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 |