Introduced the URLBlacklistManager, and wired it to various places.

chrome/browser/policy/url_blacklist_manager_unittest.cc

+// Copyright (c) 2011 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 "chrome/browser/policy/url_blacklist_manager.h"
+
+#include "base/basictypes.h"
+#include "base/message_loop.h"
+#include "chrome/common/pref_names.h"
+#include "chrome/test/base/testing_pref_service.h"
+#include "chrome/test/base/testing_profile.h"
+#include "content/browser/browser_thread.h"
+#include "content/common/notification_service.h"
+#include "testing/gmock/include/gmock/gmock.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace {
+
+using ::testing::_;
+using ::testing::Invoke;
+using ::testing::Mock;
+
+class TestingURLBlacklistManager : public policy::URLBlacklistManager {
+ public:
+  explicit TestingURLBlacklistManager(Profile* profile)
+      : URLBlacklistManager(profile) {
+  }
+
+  virtual ~TestingURLBlacklistManager() {
+  }
+
+  // Make this method public for testing.
+  using URLBlacklistManager::ScheduleUpdate;
+
+  virtual void PostUpdateTask(Task* task) OVERRIDE {
+    // Post tasks without a delay during tests.
+    MessageLoop::current()->PostTask(FROM_HERE, task);
+  }
+
+  MOCK_METHOD0(Update, void());
+  MOCK_METHOD1(SetBlacklist, void(policy::Blacklist*));
+
+  void UpdateNotMocked() {
+    URLBlacklistManager::Update();
+  }
+
+ private:
+  DISALLOW_COPY_AND_ASSIGN(TestingURLBlacklistManager);
+};
+
+class URLBlacklistManagerTest : public testing::Test {
+ protected:
+  URLBlacklistManagerTest()
+      : ui_thread_(BrowserThread::UI, &loop_),
+        file_thread_(BrowserThread::FILE, &loop_),
+        io_thread_(BrowserThread::IO, &loop_) {
+  }
+
+  virtual void SetUp() OVERRIDE {
+    pref_service_ = profile_.GetTestingPrefService();
+    url_blacklist_manager_.reset(new TestingURLBlacklistManager(&profile_));
+    url_blacklist_manager_->InitializeOnIOThread();
+    loop_.RunAllPending();
+  }
+
+  virtual void TearDown() OVERRIDE {
+    if (url_blacklist_manager_.get())
+      url_blacklist_manager_->ShutdownOnUIThread();
+    loop_.RunAllPending();
+  }
+
+  void ExpectUpdate() {
+    EXPECT_CALL(*url_blacklist_manager_, Update())
+        .WillOnce(Invoke(url_blacklist_manager_.get(),
+                         &TestingURLBlacklistManager::UpdateNotMocked));
+  }
+
+  MessageLoop loop_;
+  TestingProfile profile_;
+  TestingPrefService* pref_service_;
+  scoped_ptr<TestingURLBlacklistManager> url_blacklist_manager_;
+
+ private:
+  BrowserThread ui_thread_;
+  BrowserThread file_thread_;
+  BrowserThread io_thread_;
+
+  DISALLOW_COPY_AND_ASSIGN(URLBlacklistManagerTest);
+};
+
+TEST_F(URLBlacklistManagerTest, SingleUpdateForTwoPrefChanges) {
+  ExpectUpdate();
+
+  ListValue* blacklist = new ListValue;
+  blacklist->Append(new StringValue("*.google.com"));
+  ListValue* whitelist = new ListValue;
+  whitelist->Append(new StringValue("mail.google.com"));
+  pref_service_->SetManagedPref(prefs::kUrlBlacklist, blacklist);
+  pref_service_->SetManagedPref(prefs::kUrlBlacklist, whitelist);
+  loop_.RunAllPending();
+
+  Mock::VerifyAndClearExpectations(url_blacklist_manager_.get());
+}
+
+TEST_F(URLBlacklistManagerTest, ShutdownWithPendingTask0) {
+  // Post an update task to the UI thread.
+  url_blacklist_manager_->ScheduleUpdate();
+  // Shutdown comes before the task is executed.
+  url_blacklist_manager_->ShutdownOnUIThread();
+  url_blacklist_manager_.reset();
+  // Run the task after shutdown and deletion.
+  loop_.RunAllPending();
+}
+
+TEST_F(URLBlacklistManagerTest, ShutdownWithPendingTask1) {
+  EXPECT_CALL(*url_blacklist_manager_, Update()).Times(0);
+  // Post an update task.
+  url_blacklist_manager_->ScheduleUpdate();
+  // Shutdown comes before the task is executed.
+  url_blacklist_manager_->ShutdownOnUIThread();
+  // Run the task after shutdown, but before deletion.
+  loop_.RunAllPending();
+  Mock::VerifyAndClearExpectations(url_blacklist_manager_.get());
+  url_blacklist_manager_.reset();
+  loop_.RunAllPending();
+}
+
+TEST_F(URLBlacklistManagerTest, ShutdownWithPendingTask2) {
+  // Update posts a BuildBlacklistTask to the FILE thread.
+  url_blacklist_manager_->UpdateNotMocked();
+  // Shutdown comes before the task is executed.
+  url_blacklist_manager_->ShutdownOnUIThread();
+  url_blacklist_manager_.reset();
+  // Run the task after shutdown and deletion.
+  loop_.RunAllPending();
+}
+
+TEST_F(URLBlacklistManagerTest, ShutdownWithPendingTask3) {
+  EXPECT_CALL(*url_blacklist_manager_, SetBlacklist(_)).Times(0);
+  // Update posts a BuildBlacklistTask to the FILE thread.
+  url_blacklist_manager_->UpdateNotMocked();
+  // Shutdown comes before the task is executed.
+  url_blacklist_manager_->ShutdownOnUIThread();
+  // Run the task after shutdown, but before deletion.
+  loop_.RunAllPending();
+  Mock::VerifyAndClearExpectations(url_blacklist_manager_.get());
+  url_blacklist_manager_.reset();
+  loop_.RunAllPending();
+}
+
+}  // namespace