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/dom_distiller_service.h"
 #include "components/dom_distiller/core/dom_distiller_test_util.h"
+#include "components/dom_distiller/core/reader_mode_preferences.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";
 
 class FakeViewRequestDelegate : public ViewRequestDelegate {
  public:
@@ skipping 40 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 ReaderModePrefs* GetReaderModePrefs() 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());
 }
 
+ReaderModePrefs* TestDomDistillerService::GetReaderModePrefs() {
+  return NULL;
+}
+
+TEST_F(DomDistillerViewerTest, TestGetBodyCssClassOutput) {
+  std::string kDark = "dark";
+  std::string kSepia = "sepia";
+  std::string kLight = "light";
+  ASSERT_EQ(kDark.compare(viewer::GetBodyCssClass(
+      ReaderModePrefs::Theme::kDark)), 0);
+  ASSERT_EQ(kLight.compare(viewer::GetBodyCssClass(
+      ReaderModePrefs::Theme::kLight)), 0);
+  ASSERT_EQ(kSepia.compare(viewer::GetBodyCssClass(
+      ReaderModePrefs::Theme::kSepia)), 0);
+}
+
 }  // namespace dom_distiller