| Index: components/dom_distiller/core/fake_distiller.h
|
| diff --git a/components/dom_distiller/core/fake_distiller.h b/components/dom_distiller/core/fake_distiller.h
|
| index 98ab7096dc942be44fb1a0841fc913e540b95701..eeadb62a8083a8828c7207d30ff93923a20b42f4 100644
|
| --- a/components/dom_distiller/core/fake_distiller.h
|
| +++ b/components/dom_distiller/core/fake_distiller.h
|
| @@ -5,6 +5,7 @@
|
| #ifndef COMPONENTS_DOM_DISTILLER_CORE_FAKE_DISTILLER_H_
|
| #define COMPONENTS_DOM_DISTILLER_CORE_FAKE_DISTILLER_H_
|
|
|
| +#include "components/dom_distiller/core/article_distillation_update.h"
|
| #include "components/dom_distiller/core/article_entry.h"
|
| #include "components/dom_distiller/core/distiller.h"
|
| #include "testing/gmock/include/gmock/gmock.h"
|
| @@ -32,20 +33,25 @@ class FakeDistiller : public Distiller {
|
| MOCK_METHOD0(Die, void());
|
|
|
| virtual void DistillPage(const GURL& url,
|
| - const DistillerCallback& callback) OVERRIDE;
|
| + const DistillationFinishedCallback& article_callback,
|
| + const DistillationUpdateCallback& page_callback)
|
| + OVERRIDE;
|
|
|
| void RunDistillerCallback(scoped_ptr<DistilledArticleProto> proto);
|
|
|
| GURL GetUrl() { return url_; }
|
|
|
| - DistillerCallback GetCallback() { return callback_; }
|
| + DistillationFinishedCallback GetArticleCallback() {
|
| + return article_callback_;
|
| + }
|
|
|
| private:
|
| void RunDistillerCallbackInternal(scoped_ptr<DistilledArticleProto> proto);
|
|
|
| bool execute_callback_;
|
| GURL url_;
|
| - DistillerCallback callback_;
|
| + DistillationFinishedCallback article_callback_;
|
| + DistillationUpdateCallback page_callback_;
|
| };
|
|
|
| } // namespace test
|
|
|