Index: components/dom_distiller/core/dom_distiller_service.cc |
diff --git a/components/dom_distiller/core/dom_distiller_service.cc b/components/dom_distiller/core/dom_distiller_service.cc |
index 797827ba0b4696c9d1006e33088c81ed851d4845..6ee2f2fe3a22078fc35ebbf2b3142232e0d75c29 100644 |
--- a/components/dom_distiller/core/dom_distiller_service.cc |
+++ b/components/dom_distiller/core/dom_distiller_service.cc |
@@ -7,6 +7,7 @@ |
#include "base/guid.h" |
#include "base/message_loop/message_loop.h" |
#include "components/dom_distiller/core/dom_distiller_store.h" |
+#include "components/dom_distiller/core/proto/distilled_article.pb.h" |
#include "components/dom_distiller/core/task_tracker.h" |
#include "url/gurl.h" |
@@ -27,7 +28,7 @@ ArticleEntry CreateSkeletonEntryForUrl(const GURL& url) { |
void RunArticleAvailableCallback( |
const DomDistillerService::ArticleAvailableCallback& article_cb, |
const ArticleEntry& entry, |
- DistilledPageProto* proto, |
+ const DistilledArticleProto* article_proto, |
bool distillation_succeeded) { |
article_cb.Run(distillation_succeeded); |
} |
@@ -189,13 +190,16 @@ void DomDistillerService::CancelTask(TaskTracker* task) { |
} |
} |
-void DomDistillerService::AddDistilledPageToList(const ArticleEntry& entry, |
- DistilledPageProto* proto, |
- bool distillation_succeeded) { |
+void DomDistillerService::AddDistilledPageToList( |
+ const ArticleEntry& entry, |
+ const DistilledArticleProto* article_proto, |
+ bool distillation_succeeded) { |
DCHECK(IsEntryValid(entry)); |
if (distillation_succeeded) { |
- DCHECK(proto); |
+ DCHECK(article_proto); |
+ DCHECK_GT(article_proto->pages_size(), 0); |
store_->UpdateEntry(entry); |
+ DCHECK_EQ(article_proto->pages_size(), entry.pages_size()); |
} |
} |