Theme Preferences for Distilled Pages

components/dom_distiller/core/viewer_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 "components/dom_distiller/core/viewer.h"
 
+#include "components/dom_distiller/core/distilled_page_prefs.h"
 #include "components/dom_distiller/core/dom_distiller_service.h"
 #include "components/dom_distiller/core/dom_distiller_test_util.h"
 #include "components/dom_distiller/core/task_tracker.h"
 #include "components/dom_distiller/core/url_constants.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace dom_distiller {
 
 const char kTestScheme[] = "myscheme";
 
@@ skipping 42 matching lines @@
   virtual scoped_ptr<ArticleEntry> RemoveEntry(const std::string&) {
     return scoped_ptr<ArticleEntry>(RemoveEntryImpl());
   }
   virtual scoped_ptr<DistillerPage> CreateDefaultDistillerPage() {
     return scoped_ptr<DistillerPage>();
   }
   virtual scoped_ptr<DistillerPage> CreateDefaultDistillerPageWithHandle(
       scoped_ptr<SourcePageHandle> handle) {
     return scoped_ptr<DistillerPage>();
   }
+  virtual DistilledPagePrefs* GetDistilledPagePrefs() OVERRIDE;
 };
 
 class DomDistillerViewerTest : public testing::Test {
  public:
   virtual void SetUp() OVERRIDE {
     service_.reset(new TestDomDistillerService());
   }
 
  protected:
   scoped_ptr<ViewerHandle> CreateViewRequest(
@@ skipping 43 matching lines @@
                     view_request_delegate.get());
   // Specify an internal Chrome page.
   CreateViewRequest("?" + std::string(kUrlKey) + "=chrome%3A%2F%2Fsettings%2F",
                     view_request_delegate.get());
   // Specify a recursive URL.
   CreateViewRequest("?" + std::string(kUrlKey) + "=" +
                         std::string(kTestScheme) + "%3A%2F%2Fabc-def%2F",
                     view_request_delegate.get());
 }
 
+DistilledPagePrefs* TestDomDistillerService::GetDistilledPagePrefs() {
+  return NULL;
+}
+
+TEST_F(DomDistillerViewerTest, TestGetDistilledPageThemeJsOutput) {
+  std::string kDarkJs = "useTheme('dark');";
+  std::string kSepiaJs = "useTheme('sepia');";
+  std::string kLightJs = "useTheme('light');";
+  EXPECT_EQ(kDarkJs.compare(viewer::GetDistilledPageThemeJs(
+                DistilledPagePrefs::Theme::DARK)),
+            0);
+  EXPECT_EQ(kLightJs.compare(viewer::GetDistilledPageThemeJs(
+                DistilledPagePrefs::Theme::LIGHT)),
+            0);
+  EXPECT_EQ(kSepiaJs.compare(viewer::GetDistilledPageThemeJs(
+                DistilledPagePrefs::Theme::SEPIA)),
+            0);
+}
+
 }  // namespace dom_distiller