Index: components/dom_distiller/content/dom_distiller_viewer_source_unittest.cc |
diff --git a/components/dom_distiller/content/dom_distiller_viewer_source_unittest.cc b/components/dom_distiller/content/dom_distiller_viewer_source_unittest.cc |
index 3c4ee7b8c7f8f52507337cb1d189e0692b4a9598..7e5db1978cb7afc0a006fced826a487639d3c2e8 100644 |
--- a/components/dom_distiller/content/dom_distiller_viewer_source_unittest.cc |
+++ b/components/dom_distiller/content/dom_distiller_viewer_source_unittest.cc |
@@ -4,7 +4,10 @@ |
#include "components/dom_distiller/content/dom_distiller_viewer_source.h" |
+#include <vector> |
+ |
#include "base/callback.h" |
+#include "components/dom_distiller/core/article_distillation_update.h" |
#include "components/dom_distiller/core/article_entry.h" |
#include "components/dom_distiller/core/dom_distiller_service.h" |
#include "components/dom_distiller/core/dom_distiller_store.h" |
@@ -23,6 +26,8 @@ class FakeViewRequestDelegate : public ViewRequestDelegate { |
public: |
virtual ~FakeViewRequestDelegate() {} |
MOCK_METHOD1(OnArticleReady, void(const DistilledArticleProto* proto)); |
+ MOCK_METHOD1(OnArticleUpdated, |
+ void(ArticleDistillationUpdate article_update)); |
}; |
class TestDomDistillerService : public DomDistillerServiceInterface { |