1 // Copyright 2013 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.
5 #ifndef COMPONENTS_DOM_DISTILLER_CORE_FAKE_DISTILLER_H_
6 #define COMPONENTS_DOM_DISTILLER_CORE_FAKE_DISTILLER_H_
8 #include "base/callback.h"
9 #include "components/dom_distiller/core/article_distillation_update.h"
10 #include "components/dom_distiller/core/article_entry.h"
11 #include "components/dom_distiller/core/distiller.h"
12 #include "testing/gmock/include/gmock/gmock.h"
17 namespace dom_distiller {
20 class MockDistillerFactory : public DistillerFactory {
22 MockDistillerFactory();
23 virtual ~MockDistillerFactory();
24 MOCK_METHOD0(CreateDistillerImpl, Distiller*());
25 virtual scoped_ptr<Distiller> CreateDistillerForUrl(
26 const GURL& unused) override {
27 return scoped_ptr<Distiller>(CreateDistillerImpl());
31 class FakeDistiller : public Distiller {
33 // If execute_callback is true, when DistillPage is called, a task will
34 // immediately be posted to execute the callback with a simple
35 // DistilledArticleProto.
36 explicit FakeDistiller(bool execute_callback);
37 // TODO(yfriedman): Drop execute_callback from this and give the option of
38 // "auto-distilling" or calling the provided closure.
39 explicit FakeDistiller(bool execute_callback,
40 const base::Closure& distillation_initiated_callback);
41 virtual ~FakeDistiller();
42 MOCK_METHOD0(Die, void());
44 virtual void DistillPage(
46 scoped_ptr<DistillerPage> distiller_page,
47 const DistillationFinishedCallback& article_callback,
48 const DistillationUpdateCallback& page_callback) override;
50 void RunDistillerCallback(scoped_ptr<DistilledArticleProto> proto);
51 void RunDistillerUpdateCallback(const ArticleDistillationUpdate& update);
53 GURL GetUrl() { return url_; }
55 DistillationFinishedCallback GetArticleCallback() {
56 return article_callback_;
60 void PostDistillerCallback(scoped_ptr<DistilledArticleProto> proto);
61 void RunDistillerCallbackInternal(scoped_ptr<DistilledArticleProto> proto);
63 bool execute_callback_;
65 DistillationFinishedCallback article_callback_;
66 DistillationUpdateCallback page_callback_;
67 bool destruction_allowed_;
68 // Used to notify when distillation is complete.
69 base::Closure distillation_initiated_callback_;
73 } // namespace dom_distiller
75 #endif // COMPONENTS_DOM_DISTILLER_CORE_FAKE_DISTILLER_H_