Index: components/dom_distiller/core/task_tracker_unittest.cc |
diff --git a/components/dom_distiller/core/task_tracker_unittest.cc b/components/dom_distiller/core/task_tracker_unittest.cc |
index c88becbd1e9368d561842a6da9b8eef17bc9734c..c6eef4be519782dfc74428bac361d5c1e2a83e75 100644 |
--- a/components/dom_distiller/core/task_tracker_unittest.cc |
+++ b/components/dom_distiller/core/task_tracker_unittest.cc |
@@ -5,6 +5,7 @@ |
#include "components/dom_distiller/core/task_tracker.h" |
#include "base/run_loop.h" |
+#include "components/dom_distiller/core/article_distillation_update.h" |
#include "components/dom_distiller/core/article_entry.h" |
#include "components/dom_distiller/core/fake_distiller.h" |
#include "testing/gtest/include/gtest/gtest.h" |
@@ -20,6 +21,8 @@ class FakeViewRequestDelegate : public ViewRequestDelegate { |
virtual ~FakeViewRequestDelegate() {} |
MOCK_METHOD1(OnArticleReady, |
void(const DistilledArticleProto* article_proto)); |
+ MOCK_METHOD1(OnArticleUpdated, |
+ void(ArticleDistillationUpdate article_update)); |
}; |
class TestCancelCallback { |