[Hotword] Sync Audio History pref every 24 hours, when opening chrome://settings and . . .

chrome/browser/search/hotword_service_unittest.cc

 // Copyright 2014 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 "base/command_line.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/metrics/field_trial.h"
 #include "base/prefs/pref_service.h"
+#include "base/test/test_simple_task_runner.h"
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/extensions/extension_service_test_base.h"
 #include "chrome/browser/extensions/test_extension_service.h"
+#include "chrome/browser/history/web_history_service.h"
+#include "chrome/browser/search/hotword_audio_history_handler.h"
 #include "chrome/browser/search/hotword_service.h"
 #include "chrome/browser/search/hotword_service_factory.h"
 #include "chrome/common/chrome_switches.h"
 #include "chrome/common/extensions/extension_constants.h"
 #include "chrome/common/pref_names.h"
 #include "chrome/test/base/testing_profile.h"
 #include "content/public/test/test_browser_thread_bundle.h"
 #include "extensions/browser/extension_system.h"
 #include "extensions/common/extension.h"
 #include "extensions/common/extension_builder.h"
 #include "extensions/common/manifest.h"
 #include "extensions/common/one_shot_event.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace {
 
+class MockAudioHistoryHandler : public HotwordAudioHistoryHandler {
+ public:
+  MockAudioHistoryHandler(
+      content::BrowserContext* context,
+      const scoped_refptr<base::SingleThreadTaskRunner>& task_runner,
+      history::WebHistoryService* web_history)
+      : HotwordAudioHistoryHandler(context, task_runner),
+        get_audio_history_calls_(0),
+        web_history_(web_history) {
+  }
+  ~MockAudioHistoryHandler() override {}
+
+  void GetAudioHistoryEnabled(
+      const HotwordAudioHistoryCallback& callback) override {
+    get_audio_history_calls_++;
+    callback.Run(true, true);
+  }
+
+  history::WebHistoryService* GetWebHistory() override {
+    return web_history_.get();
+  }
+
+  int GetAudioHistoryCalls() {
+    return get_audio_history_calls_;
+  }
+
+ private:
+  int get_audio_history_calls_;
+  scoped_ptr<history::WebHistoryService> web_history_;
+};
+
 class MockHotwordService : public HotwordService {
  public:
   explicit MockHotwordService(Profile* profile)
       : HotwordService(profile),
         uninstall_count_(0) {
   }
 
   bool UninstallHotwordExtension(ExtensionService* extension_service) override {
     uninstall_count_++;
     return HotwordService::UninstallHotwordExtension(extension_service);
@@ skipping 455 matching lines @@
 
   // Set pref.
   profile()->GetPrefs()->SetBoolean(prefs::kHotwordSearchEnabled, true);
   EXPECT_FALSE(hotword_service->IsSometimesOnEnabled());
 
   // Disable always-on.
   profile()->GetPrefs()->SetBoolean(prefs::kHotwordAlwaysOnSearchEnabled,
                                     false);
   EXPECT_FALSE(hotword_service->IsSometimesOnEnabled());
 }
+
+TEST_P(HotwordServiceTest, AudioHistorySyncOccurs) {
+  // Set the field trial to a valid one.
+  ASSERT_TRUE(base::FieldTrialList::CreateFieldTrial(
+    hotword_internal::kHotwordFieldTrialName, "Install"));
+
+  InitializeEmptyExtensionService();
+  service_->Init();
+
+  HotwordServiceFactory* hotword_service_factory =
+    HotwordServiceFactory::GetInstance();
+
+  MockHotwordService* hotword_service = static_cast<MockHotwordService*>(
+    hotword_service_factory->SetTestingFactoryAndUse(
+    profile(), BuildMockHotwordService));
+  EXPECT_TRUE(hotword_service != NULL);
+  hotword_service->SetExtensionService(service());
+  hotword_service->SetExtensionId(extension_id_);
+
+  scoped_refptr<base::TestSimpleTaskRunner> test_task_runner(
+      new base::TestSimpleTaskRunner());
+  MockAudioHistoryHandler* audio_history_handler =
+      new MockAudioHistoryHandler(profile(), test_task_runner, nullptr);
+  hotword_service->SetAudioHistoryHandler(audio_history_handler);
+  EXPECT_EQ(1, audio_history_handler->GetAudioHistoryCalls());
+  // We expect the next check for audio history to be in the queue.
+  EXPECT_EQ(base::TimeDelta::FromDays(1),
+            test_task_runner->NextPendingTaskDelay());
+  EXPECT_TRUE(test_task_runner->HasPendingTask());
+  test_task_runner->RunPendingTasks();
+  EXPECT_EQ(2, audio_history_handler->GetAudioHistoryCalls());
+  EXPECT_TRUE(test_task_runner->HasPendingTask());
+  test_task_runner->ClearPendingTasks();
+}