| // Copyright 2014 The Chromium Authors |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include "components/sync/driver/startup_controller.h" |
| |
| #include <memory> |
| |
| #include "base/functional/bind.h" |
| #include "base/test/task_environment.h" |
| #include "testing/gmock/include/gmock/gmock.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| |
| namespace syncer { |
| |
| class StartupControllerTest : public testing::Test { |
| public: |
| StartupControllerTest() = default; |
| |
| void SetUp() override { |
| controller_ = std::make_unique<StartupController>( |
| base::BindRepeating(&StartupControllerTest::ShouldStart, |
| base::Unretained(this)), |
| base::BindRepeating(&StartupControllerTest::FakeStartBackend, |
| base::Unretained(this))); |
| } |
| |
| void SetShouldStart(bool should_start) { should_start_ = should_start; } |
| |
| void ExpectStarted() { |
| task_environment_.RunUntilIdle(); |
| EXPECT_TRUE(started()); |
| EXPECT_EQ(StartupController::State::STARTED, controller()->GetState()); |
| } |
| |
| void ExpectStartDeferred() { |
| task_environment_.RunUntilIdle(); |
| EXPECT_FALSE(started()); |
| EXPECT_EQ(StartupController::State::STARTING_DEFERRED, |
| controller()->GetState()); |
| } |
| |
| void ExpectNotStarted() { |
| task_environment_.RunUntilIdle(); |
| EXPECT_FALSE(started()); |
| EXPECT_EQ(StartupController::State::NOT_STARTED, controller()->GetState()); |
| } |
| |
| bool started() const { return started_; } |
| void clear_started() { started_ = false; } |
| StartupController* controller() { return controller_.get(); } |
| |
| void RunDeferredTasks() { task_environment_.FastForwardUntilNoTasksRemain(); } |
| |
| private: |
| bool ShouldStart() { return should_start_; } |
| void FakeStartBackend() { started_ = true; } |
| |
| base::test::SingleThreadTaskEnvironment task_environment_{ |
| base::test::TaskEnvironment::TimeSource::MOCK_TIME}; |
| bool should_start_ = false; |
| bool started_ = false; |
| std::unique_ptr<StartupController> controller_; |
| }; |
| |
| // Test that sync doesn't start if the "should sync start" callback returns |
| // false. |
| TEST_F(StartupControllerTest, ShouldNotStart) { |
| controller()->TryStart(/*force_immediate=*/false); |
| ExpectNotStarted(); |
| |
| controller()->TryStart(/*force_immediate=*/true); |
| ExpectNotStarted(); |
| } |
| |
| TEST_F(StartupControllerTest, DefersByDefault) { |
| SetShouldStart(true); |
| controller()->TryStart(/*force_immediate=*/false); |
| ExpectStartDeferred(); |
| } |
| |
| // Test that a data type triggering startup starts sync immediately. |
| TEST_F(StartupControllerTest, NoDeferralDataTypeTrigger) { |
| SetShouldStart(true); |
| controller()->OnDataTypeRequestsSyncStartup(SESSIONS); |
| ExpectStarted(); |
| } |
| |
| // Test that a data type trigger interrupts the deferral timer and starts |
| // sync immediately. |
| TEST_F(StartupControllerTest, DataTypeTriggerInterruptsDeferral) { |
| SetShouldStart(true); |
| controller()->TryStart(/*force_immediate=*/false); |
| ExpectStartDeferred(); |
| |
| controller()->OnDataTypeRequestsSyncStartup(SESSIONS); |
| ExpectStarted(); |
| |
| // The fallback timer shouldn't result in another invocation of the closure |
| // we passed to the StartupController. |
| clear_started(); |
| RunDeferredTasks(); |
| EXPECT_FALSE(started()); |
| } |
| |
| // Test that the fallback timer starts sync in the event all conditions are met |
| // and no data type requests sync. |
| TEST_F(StartupControllerTest, FallbackTimer) { |
| SetShouldStart(true); |
| controller()->TryStart(/*force_immediate=*/false); |
| ExpectStartDeferred(); |
| |
| RunDeferredTasks(); |
| ExpectStarted(); |
| } |
| |
| // Sanity check that the fallback timer doesn't fire before startup |
| // conditions are met. |
| TEST_F(StartupControllerTest, FallbackTimerWaits) { |
| controller()->TryStart(/*force_immediate=*/false); |
| ExpectNotStarted(); |
| RunDeferredTasks(); |
| ExpectNotStarted(); |
| } |
| |
| // Test that sync starts immediately when told to do so. |
| TEST_F(StartupControllerTest, NoDeferralIfForceImmediate) { |
| SetShouldStart(true); |
| ExpectNotStarted(); |
| controller()->TryStart(/*force_immediate=*/true); |
| ExpectStarted(); |
| } |
| |
| // Test that setting |force_immediate| interrupts the deferral timer and starts |
| // sync immediately. |
| TEST_F(StartupControllerTest, ForceImmediateInterruptsDeferral) { |
| SetShouldStart(true); |
| controller()->TryStart(/*force_immediate=*/false); |
| ExpectStartDeferred(); |
| |
| controller()->TryStart(/*force_immediate=*/true); |
| ExpectStarted(); |
| } |
| } // namespace syncer |