src/components/dom_distiller/core/viewer_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/core/viewer.h"
   6
   7 #include "components/dom_distiller/core/dom_distiller_service.h"
   8 #include "components/dom_distiller/core/dom_distiller_test_util.h"
   9 #include "components/dom_distiller/core/task_tracker.h"
  10 #include "components/dom_distiller/core/url_constants.h"
  11 #include "testing/gtest/include/gtest/gtest.h"
  12
  13 namespace dom_distiller {
  14
  15 const char kTestScheme[] = "myscheme";
  16
  17 class FakeViewRequestDelegate : public ViewRequestDelegate {
  18  public:
  19   virtual ~FakeViewRequestDelegate() {}
  20   MOCK_METHOD1(OnArticleReady, void(const DistilledArticleProto* proto));
  21   MOCK_METHOD1(OnArticleUpdated,
  22                void(ArticleDistillationUpdate article_update));
  23 };
  24
  25 class TestDomDistillerService : public DomDistillerServiceInterface {
  26  public:
  27   TestDomDistillerService() {}
  28   virtual ~TestDomDistillerService() {}
  29
  30   MOCK_CONST_METHOD0(GetSyncableService, syncer::SyncableService*());
  31   MOCK_METHOD3(AddToList,
  32                const std::string(const GURL&,
  33                                  DistillerPage*,
  34                                  const ArticleAvailableCallback&));
  35   virtual const std::string AddToList(
  36       const GURL& url,
  37       scoped_ptr<DistillerPage> distiller_page,
  38       const ArticleAvailableCallback& article_cb) {
  39     return AddToList(url, distiller_page.get(), article_cb);
  40   }
  41   MOCK_CONST_METHOD0(GetEntries, std::vector<ArticleEntry>());
  42   MOCK_METHOD1(AddObserver, void(DomDistillerObserver*));
  43   MOCK_METHOD1(RemoveObserver, void(DomDistillerObserver*));
  44   MOCK_METHOD0(ViewUrlImpl, ViewerHandle*());
  45   virtual scoped_ptr<ViewerHandle> ViewUrl(
  46       ViewRequestDelegate*,
  47       scoped_ptr<DistillerPage> distiller_page,
  48       const GURL&) {
  49     return scoped_ptr<ViewerHandle>(ViewUrlImpl());
  50   }
  51   MOCK_METHOD0(ViewEntryImpl, ViewerHandle*());
  52   virtual scoped_ptr<ViewerHandle> ViewEntry(
  53       ViewRequestDelegate*,
  54       scoped_ptr<DistillerPage> distiller_page,
  55       const std::string&) {
  56     return scoped_ptr<ViewerHandle>(ViewEntryImpl());
  57   }
  58   MOCK_METHOD0(RemoveEntryImpl, ArticleEntry*());
  59   virtual scoped_ptr<ArticleEntry> RemoveEntry(const std::string&) {
  60     return scoped_ptr<ArticleEntry>(RemoveEntryImpl());
  61   }
  62   virtual scoped_ptr<DistillerPage> CreateDefaultDistillerPage() {
  63     return scoped_ptr<DistillerPage>();
  64   }
  65 };
  66
  67 class DomDistillerViewerTest : public testing::Test {
  68  public:
  69   virtual void SetUp() OVERRIDE {
  70     service_.reset(new TestDomDistillerService());
  71   }
  72
  73  protected:
  74   scoped_ptr<ViewerHandle> CreateViewRequest(
  75       const std::string& path,
  76       ViewRequestDelegate* view_request_delegate) {
  77     return viewer::CreateViewRequest(
  78         service_.get(), path, view_request_delegate);
  79   }
  80
  81   scoped_ptr<TestDomDistillerService> service_;
  82 };
  83
  84 TEST_F(DomDistillerViewerTest, TestCreatingViewUrlRequest) {
  85   scoped_ptr<FakeViewRequestDelegate> view_request_delegate(
  86       new FakeViewRequestDelegate());
  87   ViewerHandle* viewer_handle(new ViewerHandle(ViewerHandle::CancelCallback()));
  88   EXPECT_CALL(*service_.get(), ViewUrlImpl())
  89       .WillOnce(testing::Return(viewer_handle));
  90   EXPECT_CALL(*service_.get(), ViewEntryImpl()).Times(0);
  91   CreateViewRequest(
  92       std::string("?") + kUrlKey + "=http%3A%2F%2Fwww.example.com%2F",
  93       view_request_delegate.get());
  94 }
  95
  96 TEST_F(DomDistillerViewerTest, TestCreatingViewEntryRequest) {
  97   scoped_ptr<FakeViewRequestDelegate> view_request_delegate(
  98       new FakeViewRequestDelegate());
  99   ViewerHandle* viewer_handle(new ViewerHandle(ViewerHandle::CancelCallback()));
 100   EXPECT_CALL(*service_.get(), ViewEntryImpl())
 101       .WillOnce(testing::Return(viewer_handle));
 102   EXPECT_CALL(*service_.get(), ViewUrlImpl()).Times(0);
 103   CreateViewRequest(std::string("?") + kEntryIdKey + "=abc-def",
 104                     view_request_delegate.get());
 105 }
 106
 107 TEST_F(DomDistillerViewerTest, TestCreatingInvalidViewRequest) {
 108   scoped_ptr<FakeViewRequestDelegate> view_request_delegate(
 109       new FakeViewRequestDelegate());
 110   EXPECT_CALL(*service_.get(), ViewEntryImpl()).Times(0);
 111   EXPECT_CALL(*service_.get(), ViewUrlImpl()).Times(0);
 112   // Specify none of the required query parameters.
 113   CreateViewRequest("?foo=bar", view_request_delegate.get());
 114   // Specify both of the required query parameters.
 115   CreateViewRequest("?" + std::string(kUrlKey) +
 116                         "=http%3A%2F%2Fwww.example.com%2F&" +
 117                         std::string(kEntryIdKey) + "=abc-def",
 118                     view_request_delegate.get());
 119   // Specify an internal Chrome page.
 120   CreateViewRequest("?" + std::string(kUrlKey) + "=chrome%3A%2F%2Fsettings%2F",
 121                     view_request_delegate.get());
 122   // Specify a recursive URL.
 123   CreateViewRequest("?" + std::string(kUrlKey) + "=" +
 124                         std::string(kTestScheme) + "%3A%2F%2Fabc-def%2F",
 125                     view_request_delegate.get());
 126 }
 127
 128 }  // namespace dom_distiller