Add support for multipage distillation.

components/dom_distiller/core/task_tracker_unittest.cc

 // Copyright 2013 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "components/dom_distiller/core/task_tracker.h"
 
 #include "base/run_loop.h"
 #include "components/dom_distiller/core/article_entry.h"
 #include "components/dom_distiller/core/fake_distiller.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 using testing::Return;
 using testing::_;
 
 namespace dom_distiller {
 namespace test {
 
 class FakeViewRequestDelegate : public ViewRequestDelegate {
  public:
   virtual ~FakeViewRequestDelegate() {}
-  MOCK_METHOD1(OnArticleReady, void(DistilledPageProto* proto));
+  MOCK_METHOD1(OnArticleReady,
+               void(const DistilledArticleProto* article_proto));
 };
 
 class TestCancelCallback {
  public:
   TestCancelCallback() : cancelled_(false) {}
   TaskTracker::CancelCallback GetCallback() {
     return base::Bind(&TestCancelCallback::Cancel, base::Unretained(this));
   }
   void Cancel(TaskTracker*) { cancelled_ = true; }
   bool Cancelled() { return cancelled_; }
 
  private:
   bool cancelled_;
 };
 
 class MockSaveCallback {
  public:
-  MOCK_METHOD3(Save, void(const ArticleEntry&, DistilledPageProto*, bool));
+  MOCK_METHOD3(Save,
+               void(const ArticleEntry&, const DistilledArticleProto*, bool));
 };
 
 class DomDistillerTaskTrackerTest : public testing::Test {
  public:
   virtual void SetUp() OVERRIDE {
     message_loop_.reset(new base::MessageLoop());
     entry_id_ = "id0";
     page_0_url_ = GURL("http://www.example.com/1");
     page_1_url_ = GURL("http://www.example.com/2");
   }
@@ skipping 104 matching lines @@
   EXPECT_CALL(save_callback, Save(_, _, _));
 
   task_tracker.StartDistiller(&distiller_factory);
   base::RunLoop().RunUntilIdle();
 
   EXPECT_TRUE(cancel_callback.Cancelled());
 }
 
 }  // namespace test
 }  // namespace dom_distiller