Theme Preferences for Distilled Pages

components/dom_distiller/core/distilled_page_prefs_unittests.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 "components/dom_distiller/core/distilled_page_prefs.h"
+
+#include "base/message_loop/message_loop.h"
+#include "base/run_loop.h"
+#include "components/pref_registry/testing_pref_service_syncable.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace dom_distiller {
+
+namespace {
+
+class TestingObserver : public DistilledPagePrefs::Observer {
+ public:
+  TestingObserver() : theme_(DistilledPagePrefs::Theme::kLight) {}
+
+  virtual void OnChangeTheme(DistilledPagePrefs::Theme new_theme) OVERRIDE {
+    theme_ = new_theme;
+  }
+
+  DistilledPagePrefs::Theme GetTheme() {
+    return theme_;
+  }
+
+ private:
+  DistilledPagePrefs::Theme theme_;
+};
+
+} // namespace
+
+class DistilledPagePrefsTest : public testing::Test {
+ protected:
+  virtual void SetUp() OVERRIDE {
+    user_prefs::TestingPrefServiceSyncable* pref_service =
+        new user_prefs::TestingPrefServiceSyncable();
+    DistilledPagePrefs::RegisterProfilePrefs(pref_service->registry());
+    distilled_page_prefs_ = new DistilledPagePrefs(pref_service);
+  }
+
+  DistilledPagePrefs* distilled_page_prefs_;
+
+ private:
+  base::MessageLoop message_loop_;
+};
+
+TEST_F(DistilledPagePrefsTest, TestingOnChangeThemeIsBeingCalled) {
+  TestingObserver* obs = new TestingObserver();
+  distilled_page_prefs_->AddObserver(obs);
+  distilled_page_prefs_->SetTheme(DistilledPagePrefs::Theme::kSepia);
+  EXPECT_EQ(DistilledPagePrefs::Theme::kLight, obs->GetTheme());
+  base::RunLoop().RunUntilIdle();
+  EXPECT_EQ(DistilledPagePrefs::Theme::kSepia, obs->GetTheme());
+  distilled_page_prefs_->SetTheme(DistilledPagePrefs::Theme::kDark);
+  base::RunLoop().RunUntilIdle();
+  EXPECT_EQ(DistilledPagePrefs::Theme::kDark, obs->GetTheme());
+}
+
+TEST_F(DistilledPagePrefsTest, TestingMultipleObservers) {
+  TestingObserver* obs = new TestingObserver();
+  distilled_page_prefs_->AddObserver(obs);
+  TestingObserver* obs2 = new TestingObserver();
+  distilled_page_prefs_->AddObserver(obs2);
+  distilled_page_prefs_->SetTheme(DistilledPagePrefs::Theme::kSepia);
+  base::RunLoop().RunUntilIdle();
+  EXPECT_EQ(DistilledPagePrefs::Theme::kSepia, obs->GetTheme());
+  EXPECT_EQ(DistilledPagePrefs::Theme::kSepia, obs2->GetTheme());
+  distilled_page_prefs_->RemoveObserver(obs);
+  distilled_page_prefs_->SetTheme(DistilledPagePrefs::Theme::kLight);
+  base::RunLoop().RunUntilIdle();
+  EXPECT_EQ(DistilledPagePrefs::Theme::kSepia, obs->GetTheme());
+  EXPECT_EQ(DistilledPagePrefs::Theme::kLight, obs2->GetTheme());
+}
+
+}  // namespace dom_distiller