components/dom_distiller/content/browser/dom_distiller_viewer_source_unittest.cc

   1 // Copyright 2014 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "components/dom_distiller/content/browser/dom_distiller_viewer_source.h"
   6
   7 #include "components/dom_distiller/core/url_constants.h"
   8 #include "testing/gtest/include/gtest/gtest.h"
   9
  10 namespace dom_distiller {
  11
  12 const char kTestScheme[] = "myscheme";
  13
  14 class DomDistillerViewerSourceTest : public testing::Test {
  15  public:
  16   void SetUp() override {
  17     source_.reset(new DomDistillerViewerSource(nullptr, kTestScheme, nullptr));
  18   }
  19
  20  protected:
  21   scoped_ptr<DomDistillerViewerSource> source_;
  22 };
  23
  24 TEST_F(DomDistillerViewerSourceTest, TestMimeType) {
  25   EXPECT_EQ("text/css", source_.get()->GetMimeType(kViewerCssPath));
  26   EXPECT_EQ("text/html", source_.get()->GetMimeType("anythingelse"));
  27   EXPECT_EQ("text/javascript", source_.get()->GetMimeType(kViewerJsPath));
  28
  29 }
  30
  31 }  // namespace dom_distiller