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#ifndef COMPONENTS_DOM_DISTILLER_CORE_FAKE_DISTILLER_PAGE_H_
6#define COMPONENTS_DOM_DISTILLER_CORE_FAKE_DISTILLER_PAGE_H_
7
8#include "components/dom_distiller/core/distiller_page.h"
9#include "testing/gmock/include/gmock/gmock.h"
10
11namespace dom_distiller {
12namespace test {
13
14class MockDistillerPageFactory : public DistillerPageFactory {
15 public:
16  MockDistillerPageFactory();
17  virtual ~MockDistillerPageFactory();
18  MOCK_CONST_METHOD0(CreateDistillerPageImpl, DistillerPage*());
19  virtual scoped_ptr<DistillerPage> CreateDistillerPage(
20      const gfx::Size& render_view_size) const OVERRIDE {
21    return scoped_ptr<DistillerPage>(CreateDistillerPageImpl());
22  }
23  virtual scoped_ptr<DistillerPage> CreateDistillerPageWithHandle(
24      scoped_ptr<SourcePageHandle> handle) const OVERRIDE {
25    return scoped_ptr<DistillerPage>(CreateDistillerPageImpl());
26  }
27};
28
29class MockDistillerPage : public DistillerPage {
30 public:
31  MockDistillerPage();
32  virtual ~MockDistillerPage();
33  MOCK_METHOD2(DistillPageImpl,
34               void(const GURL& gurl, const std::string& script));
35};
36
37}  // namespace test
38}  // namespace dom_distiller
39
40#endif  // COMPONENTS_DOM_DISTILLER_CORE_FAKE_DISTILLER_PAGE_H_
41