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..048d9aac2cf030cc6aec8022f7306ad904945c96 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,23 @@ class FakeDistiller : public Distiller { |
MOCK_METHOD0(Die, void()); |
virtual void DistillPage(const GURL& url, |
- const DistillerCallback& callback) OVERRIDE; |
+ const ArticleDistillationCallback& article_callback, |
+ const PageDistillationCallback& page_callback) |
+ OVERRIDE; |
void RunDistillerCallback(scoped_ptr<DistilledArticleProto> proto); |
GURL GetUrl() { return url_; } |
- DistillerCallback GetCallback() { return callback_; } |
+ ArticleDistillationCallback GetArticleCallback() { return article_callback_; } |
private: |
void RunDistillerCallbackInternal(scoped_ptr<DistilledArticleProto> proto); |
bool execute_callback_; |
GURL url_; |
- DistillerCallback callback_; |
+ ArticleDistillationCallback article_callback_; |
+ PageDistillationCallback page_callback_; |
}; |
} // namespace test |