Home | History | Annotate | Download | only in core 
      1 // Copyright 2013 The Chromium Authors. All rights reserved.
      2 // Use of this source code is governed by a BSD-style license that can be
      3 // found in the LICENSE file.
      4 
      5 #include "components/dom_distiller/core/fake_distiller.h"
      6 
      7 #include "base/auto_reset.h"
      8 #include "base/bind.h"
      9 #include "base/callback_helpers.h"
     10 #include "base/message_loop/message_loop.h"
     11 #include "testing/gtest/include/gtest/gtest.h"
     12 
     13 namespace dom_distiller {
     14 namespace test {
     15 
     16 MockDistillerFactory::MockDistillerFactory() {}
     17 MockDistillerFactory::~MockDistillerFactory() {}
     18 
     19 FakeDistiller::FakeDistiller(bool execute_callback)
     20     : execute_callback_(execute_callback),
     21       destruction_allowed_(true) {
     22   EXPECT_CALL(*this, Die()).Times(testing::AnyNumber());
     23 }
     24 
     25 FakeDistiller::FakeDistiller(
     26     bool execute_callback,
     27     const base::Closure& distillation_initiated_callback)
     28     : execute_callback_(execute_callback),
     29       destruction_allowed_(true),
     30       distillation_initiated_callback_(distillation_initiated_callback) {
     31   EXPECT_CALL(*this, Die()).Times(testing::AnyNumber());
     32 }
     33 
     34 FakeDistiller::~FakeDistiller() {
     35   EXPECT_TRUE(destruction_allowed_);
     36   Die();
     37 }
     38 
     39 void FakeDistiller::DistillPage(
     40     const GURL& url,
     41     scoped_ptr<DistillerPage> distiller_page,
     42     const DistillationFinishedCallback& article_callback,
     43     const DistillationUpdateCallback& page_callback) {
     44   url_ = url;
     45   article_callback_ = article_callback;
     46   page_callback_ = page_callback;
     47   if (!distillation_initiated_callback_.is_null()) {
     48     base::ResetAndReturn(&distillation_initiated_callback_).Run();
     49   }
     50   if (execute_callback_) {
     51     scoped_ptr<DistilledArticleProto> proto(new DistilledArticleProto);
     52     proto->add_pages()->set_url(url_.spec());
     53     PostDistillerCallback(proto.Pass());
     54   }
     55 }
     56 
     57 void FakeDistiller::RunDistillerCallback(
     58     scoped_ptr<DistilledArticleProto> proto) {
     59   ASSERT_FALSE(execute_callback_) << "Cannot explicitly run the distiller "
     60                                      "callback for a fake distiller created "
     61                                      "with automatic callback execution.";
     62   PostDistillerCallback(proto.Pass());
     63 }
     64 
     65 void FakeDistiller::RunDistillerUpdateCallback(
     66     const ArticleDistillationUpdate& update) {
     67   page_callback_.Run(update);
     68 }
     69 
     70 
     71 void FakeDistiller::PostDistillerCallback(
     72     scoped_ptr<DistilledArticleProto> proto) {
     73   base::MessageLoop::current()->PostTask(
     74       FROM_HERE,
     75       base::Bind(&FakeDistiller::RunDistillerCallbackInternal,
     76                  base::Unretained(this),
     77                  base::Passed(&proto)));
     78 }
     79 
     80 void FakeDistiller::RunDistillerCallbackInternal(
     81     scoped_ptr<DistilledArticleProto> proto) {
     82   EXPECT_FALSE(article_callback_.is_null());
     83 
     84   base::AutoReset<bool> dont_delete_this_in_callback(&destruction_allowed_,
     85                                                      false);
     86   article_callback_.Run(proto.Pass());
     87   article_callback_.Reset();
     88 }
     89 
     90 }  // namespace test
     91 }  // namespace dom_distiller
     92