[Doodle] Cache the DoodleConfig in prefs

components/doodle/doodle_service_unittest.cc

 // Copyright 2017 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "components/doodle/doodle_service.h"
 
 #include <memory>
 #include <utility>
 #include <vector>
 
 #include "base/bind.h"
 #include "base/memory/ptr_util.h"
 #include "base/memory/ref_counted.h"
 #include "base/test/simple_test_tick_clock.h"
 #include "base/test/test_mock_time_task_runner.h"
 #include "base/threading/thread_task_runner_handle.h"
 #include "base/time/time.h"
+#include "components/prefs/testing_pref_service.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 using testing::Eq;
 using testing::StrictMock;
 
 namespace doodle {
 
 namespace {
 
@@ skipping 29 matching lines @@
 
 DoodleConfig CreateConfig(DoodleType type) {
   return DoodleConfig(type, DoodleImage(GURL("https://doodle.com/image.jpg")));
 }
 
 }  // namespace
 
 class DoodleServiceTest : public testing::Test {
  public:
   DoodleServiceTest()
-      : fetcher_(nullptr),
-        task_runner_(new base::TestMockTimeTaskRunner()),
+      : task_runner_(new base::TestMockTimeTaskRunner()),
         task_runner_handle_(task_runner_),
         tick_clock_(task_runner_->GetMockTickClock()),
+        fetcher_(nullptr),
         expiry_timer_(nullptr) {
+    DoodleService::RegisterProfilePrefs(pref_service_.registry());
+
     task_runner_->FastForwardBy(base::TimeDelta::FromHours(12345));
 
+    RecreateService();
+  }
+
+  void DestroyService() { service_ = nullptr; }
+
+  void RecreateService() {
     auto expiry_timer = base::MakeUnique<base::OneShotTimer>(tick_clock_.get());
     expiry_timer->SetTaskRunner(task_runner_);
     expiry_timer_ = expiry_timer.get();
 
     auto fetcher = base::MakeUnique<FakeDoodleFetcher>();
     fetcher_ = fetcher.get();
 
-    service_ = base::MakeUnique<DoodleService>(std::move(fetcher),
-                                               std::move(expiry_timer));
+    service_ = base::MakeUnique<DoodleService>(
+        &pref_service_, std::move(fetcher), std::move(expiry_timer),
+        task_runner_->GetMockClock());
   }
 
   DoodleService* service() { return service_.get(); }
   FakeDoodleFetcher* fetcher() { return fetcher_; }
 
   base::TestMockTimeTaskRunner* task_runner() { return task_runner_.get(); }
 
  private:
-  // Weak, owned by the service.
-  FakeDoodleFetcher* fetcher_;
+  TestingPrefServiceSimple pref_service_;
 
   scoped_refptr<base::TestMockTimeTaskRunner> task_runner_;
   base::ThreadTaskRunnerHandle task_runner_handle_;
   std::unique_ptr<base::TickClock> tick_clock_;
-  // Weak, owned by the service.
-  base::OneShotTimer* expiry_timer_;
 
   std::unique_ptr<DoodleService> service_;
+
+  // Weak, owned by the service.
+  FakeDoodleFetcher* fetcher_;
+  base::OneShotTimer* expiry_timer_;
 };
 
 TEST_F(DoodleServiceTest, FetchesConfigOnRefresh) {
   ASSERT_THAT(service()->config(), Eq(base::nullopt));
 
   // Request a refresh of the doodle config.
   service()->Refresh();
   // The request should have arrived at the fetcher.
   EXPECT_THAT(fetcher()->num_pending_callbacks(), Eq(1u));
 
@@ skipping 13 matching lines @@
   // Serve it with a different config.
   DoodleConfig other_config = CreateConfig(DoodleType::SLIDESHOW);
   DCHECK(config != other_config);
   fetcher()->ServeAllCallbacks(DoodleState::AVAILABLE,
                                base::TimeDelta::FromHours(1), other_config);
 
   // The config should have been updated.
   EXPECT_THAT(service()->config(), Eq(other_config));
 }
 
+TEST_F(DoodleServiceTest, PersistsConfig) {
+  // Load some doodle config.
+  service()->Refresh();
+  DoodleConfig config = CreateConfig(DoodleType::SIMPLE);
+  config.large_image.url = GURL("https://doodle.com/doodle.jpg");
+  fetcher()->ServeAllCallbacks(DoodleState::AVAILABLE,
+                               base::TimeDelta::FromHours(1), config);
+  ASSERT_THAT(service()->config(), Eq(config));
+
+  // Re-create the service. It should have persisted the config, and load it
+  // again automatically.
+  RecreateService();
+  EXPECT_THAT(service()->config(), Eq(config));
+}
+
+TEST_F(DoodleServiceTest, PersistsExpiryDate) {
+  // Load some doodle config.
+  service()->Refresh();
+  DoodleConfig config = CreateConfig(DoodleType::SIMPLE);
+  config.large_image.url = GURL("https://doodle.com/doodle.jpg");
+  fetcher()->ServeAllCallbacks(DoodleState::AVAILABLE,
+                               base::TimeDelta::FromHours(1), config);
+  ASSERT_THAT(service()->config(), Eq(config));
+
+  // Destroy the service, and let some time pass.
+  DestroyService();
+  task_runner()->FastForwardBy(base::TimeDelta::FromMinutes(15));
+
+  // Remove the abandoned expiry task from the task runner.
+  ASSERT_THAT(task_runner()->GetPendingTaskCount(), Eq(1u));
+  task_runner()->ClearPendingTasks();
+
+  // Re-create the service. The persisted config should have been loaded again.
+  RecreateService();
+  EXPECT_THAT(service()->config(), Eq(config));
+
+  // Its time-to-live should have been updated.
+  EXPECT_THAT(task_runner()->GetPendingTaskCount(), Eq(1u));
+  EXPECT_THAT(task_runner()->NextPendingTaskDelay(),
+              Eq(base::TimeDelta::FromMinutes(45)));
+}
+
+TEST_F(DoodleServiceTest, PersistedConfigExpires) {
+  // Load some doodle config.
+  service()->Refresh();
+  DoodleConfig config = CreateConfig(DoodleType::SIMPLE);
+  config.large_image.url = GURL("https://doodle.com/doodle.jpg");
+  fetcher()->ServeAllCallbacks(DoodleState::AVAILABLE,
+                               base::TimeDelta::FromHours(1), config);
+  ASSERT_THAT(service()->config(), Eq(config));
+
+  // Destroy the service, and let enough time pass so that the config expires.
+  DestroyService();
+  task_runner()->FastForwardBy(base::TimeDelta::FromHours(1));
+
+  // Re-create the service. The persisted config should have been discarded
+  // because it is expired.
+  RecreateService();
+  EXPECT_THAT(service()->config(), Eq(base::nullopt));
+}
+
 TEST_F(DoodleServiceTest, CallsObserverOnConfigReceived) {
   StrictMock<MockDoodleObserver> observer;
   service()->AddObserver(&observer);
 
   ASSERT_THAT(service()->config(), Eq(base::nullopt));
 
   // Request a refresh of the doodle config.
   service()->Refresh();
   // The request should have arrived at the fetcher.
   ASSERT_THAT(fetcher()->num_pending_callbacks(), Eq(1u));
@@ skipping 85 matching lines @@
 
 TEST_F(DoodleServiceTest, CallsObserverWhenConfigExpires) {
   // Load some doodle config.
   service()->Refresh();
   DoodleConfig config = CreateConfig(DoodleType::SIMPLE);
   fetcher()->ServeAllCallbacks(DoodleState::AVAILABLE,
                                base::TimeDelta::FromHours(1), config);
   ASSERT_THAT(service()->config(), Eq(config));
 
   // Make sure the task arrived at the timer's task runner.
-  ASSERT_THAT(task_runner()->GetPendingTaskCount(), Eq(1u));
+  EXPECT_THAT(task_runner()->GetPendingTaskCount(), Eq(1u));
   EXPECT_THAT(task_runner()->NextPendingTaskDelay(),
               Eq(base::TimeDelta::FromHours(1)));
 
   // Register an observer.
   StrictMock<MockDoodleObserver> observer;
   service()->AddObserver(&observer);
 
   // Fast-forward time so that the expiry task will run. The observer should get
   // notified that there's no config anymore.
   EXPECT_CALL(observer, OnDoodleConfigUpdated(Eq(base::nullopt)));
@@ skipping 28 matching lines @@
   service()->Refresh();
   EXPECT_CALL(observer, OnDoodleConfigUpdated(Eq(base::nullopt)));
   fetcher()->ServeAllCallbacks(DoodleState::AVAILABLE,
                                base::TimeDelta::FromSeconds(0), config);
 
   // Remove the observer before the service gets destroyed.
   service()->RemoveObserver(&observer);
 }
 
 }  // namespace doodle