chrome/browser/notifications/scheduler/public/schedule_service_utils_unittest.cc

// Copyright 2019 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "chrome/browser/notifications/scheduler/public/schedule_service_utils.h"

#include <string>
#include <vector>

#include "chrome/browser/notifications/scheduler/test/fake_clock.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace notifications {
namespace {

struct TestCaseInput {
  // A list of fake current time inputs.
  base::Time fake_now;

  // Expected output data.
  TimePair expected;

  TestCaseInput(std::string fake_now_str,
                std::string expected_window_start_str,
                std::string expected_window_end_str) {
    EXPECT_TRUE(base::Time::FromString(fake_now_str.c_str(), &this->fake_now));
    EXPECT_TRUE(base::Time::FromString(expected_window_start_str.c_str(),
                                       &this->expected.first));
    EXPECT_TRUE(base::Time::FromString(expected_window_end_str.c_str(),
                                       &this->expected.second));
  }
};

// Assume fixed suggested time window, and try different faked current time to
// verify next actual deliver time window calculation.
TEST(NotificationScheduleServiceUtilsTest, NextTimeWindow) {
  // Build test cases.
  TimeDeltaPair morning_window_input = {base::Hours(5), base::Hours(7)};
  TimeDeltaPair evening_window_input = {base::Hours(18), base::Hours(20)};

  std::vector<TestCaseInput> test_cases = {
      {"03/24/19 04:05:55 AM", "03/24/19 05:00:00 AM", "03/24/19 07:00:00 AM"},
      {"03/24/19 11:45:22 AM", "03/24/19 06:00:00 PM", "03/24/19 08:00:00 PM"},
      {"03/24/19 11:45:22 PM", "03/25/19 05:00:00 AM", "03/25/19 07:00:00 AM"},
  };

  // Run test cases.
  for (size_t i = 0; i < test_cases.size(); i++) {
    notifications::test::FakeClock clock;
    clock.SetNow(test_cases[i].fake_now);
    TimePair actual_output;
    EXPECT_TRUE(NextTimeWindow(&clock, morning_window_input,
                               evening_window_input, &actual_output));
    EXPECT_EQ(actual_output, test_cases[i].expected);
  }
}

}  // namespace

}  // namespace notifications