| OLD | NEW |
| 1 // Copyright 2013 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "components/dom_distiller/core/fake_distiller.h" | 5 #include "components/dom_distiller/core/fake_distiller.h" |
| 6 | 6 |
| 7 #include "base/bind.h" | 7 #include "base/bind.h" |
| 8 #include "base/message_loop/message_loop.h" | 8 #include "base/message_loop/message_loop.h" |
| 9 #include "testing/gtest/include/gtest/gtest.h" | 9 #include "testing/gtest/include/gtest/gtest.h" |
| 10 | 10 |
| 11 namespace dom_distiller { | 11 namespace dom_distiller { |
| 12 namespace test { | 12 namespace test { |
| 13 | 13 |
| 14 MockDistillerFactory::MockDistillerFactory() {} | 14 MockDistillerFactory::MockDistillerFactory() {} |
| 15 MockDistillerFactory::~MockDistillerFactory() {} | 15 MockDistillerFactory::~MockDistillerFactory() {} |
| 16 | 16 |
| 17 FakeDistiller::FakeDistiller(bool execute_callback) | 17 FakeDistiller::FakeDistiller(bool execute_callback) |
| 18 : execute_callback_(execute_callback) { | 18 : execute_callback_(execute_callback) { |
| 19 EXPECT_CALL(*this, Die()).Times(testing::AnyNumber()); | 19 EXPECT_CALL(*this, Die()).Times(testing::AnyNumber()); |
| 20 } | 20 } |
| 21 | 21 |
| 22 FakeDistiller::~FakeDistiller() { Die(); } | 22 FakeDistiller::~FakeDistiller() { Die(); } |
| 23 | 23 |
| 24 void FakeDistiller::DistillPage(const GURL& url, | 24 void FakeDistiller::DistillPage( |
| 25 const DistillerCallback& callback) { | 25 const GURL& url, |
| 26 const DistillationFinishedCallback& article_callback, |
| 27 const DistillationUpdateCallback& page_callback) { |
| 26 url_ = url; | 28 url_ = url; |
| 27 callback_ = callback; | 29 article_callback_ = article_callback; |
| 30 page_callback_ = page_callback; |
| 28 if (execute_callback_) { | 31 if (execute_callback_) { |
| 29 scoped_ptr<DistilledArticleProto> proto(new DistilledArticleProto); | 32 scoped_ptr<DistilledArticleProto> proto(new DistilledArticleProto); |
| 30 proto->add_pages()->set_url(url_.spec()); | 33 proto->add_pages()->set_url(url_.spec()); |
| 31 RunDistillerCallback(proto.Pass()); | 34 RunDistillerCallback(proto.Pass()); |
| 32 } | 35 } |
| 33 } | 36 } |
| 34 | 37 |
| 35 void FakeDistiller::RunDistillerCallback( | 38 void FakeDistiller::RunDistillerCallback( |
| 36 scoped_ptr<DistilledArticleProto> proto) { | 39 scoped_ptr<DistilledArticleProto> proto) { |
| 37 base::MessageLoop::current()->PostTask( | 40 base::MessageLoop::current()->PostTask( |
| 38 FROM_HERE, | 41 FROM_HERE, |
| 39 base::Bind(&FakeDistiller::RunDistillerCallbackInternal, | 42 base::Bind(&FakeDistiller::RunDistillerCallbackInternal, |
| 40 base::Unretained(this), | 43 base::Unretained(this), |
| 41 base::Passed(&proto))); | 44 base::Passed(&proto))); |
| 42 } | 45 } |
| 43 | 46 |
| 44 void FakeDistiller::RunDistillerCallbackInternal( | 47 void FakeDistiller::RunDistillerCallbackInternal( |
| 45 scoped_ptr<DistilledArticleProto> proto) { | 48 scoped_ptr<DistilledArticleProto> proto) { |
| 46 EXPECT_FALSE(callback_.is_null()); | 49 EXPECT_FALSE(article_callback_.is_null()); |
| 47 callback_.Run(proto.Pass()); | 50 article_callback_.Run(proto.Pass()); |
| 48 callback_.Reset(); | 51 article_callback_.Reset(); |
| 49 } | 52 } |
| 50 | 53 |
| 51 } // namespace test | 54 } // namespace test |
| 52 } // namespace dom_distiller | 55 } // namespace dom_distiller |
| OLD | NEW |