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 #ifndef COMPONENTS_DOM_DISTILLER_CORE_FAKE_DISTILLER_H_ | 5 #ifndef COMPONENTS_DOM_DISTILLER_CORE_FAKE_DISTILLER_H_ |
6 #define COMPONENTS_DOM_DISTILLER_CORE_FAKE_DISTILLER_H_ | 6 #define COMPONENTS_DOM_DISTILLER_CORE_FAKE_DISTILLER_H_ |
7 | 7 |
8 #include "base/callback.h" | 8 #include "base/callback.h" |
9 #include "components/dom_distiller/core/article_distillation_update.h" | 9 #include "components/dom_distiller/core/article_distillation_update.h" |
10 #include "components/dom_distiller/core/article_entry.h" | 10 #include "components/dom_distiller/core/article_entry.h" |
11 #include "components/dom_distiller/core/distiller.h" | 11 #include "components/dom_distiller/core/distiller.h" |
12 #include "testing/gmock/include/gmock/gmock.h" | 12 #include "testing/gmock/include/gmock/gmock.h" |
13 #include "url/gurl.h" | 13 #include "url/gurl.h" |
14 | 14 |
15 class GURL; | 15 class GURL; |
16 | 16 |
17 namespace dom_distiller { | 17 namespace dom_distiller { |
18 namespace test { | 18 namespace test { |
19 | 19 |
20 class MockDistillerFactory : public DistillerFactory { | 20 class MockDistillerFactory : public DistillerFactory { |
21 public: | 21 public: |
22 MockDistillerFactory(); | 22 MockDistillerFactory(); |
23 virtual ~MockDistillerFactory(); | 23 virtual ~MockDistillerFactory(); |
24 MOCK_METHOD0(CreateDistillerImpl, Distiller*()); | 24 MOCK_METHOD0(CreateDistillerImpl, Distiller*()); |
25 virtual scoped_ptr<Distiller> CreateDistiller() override { | 25 virtual scoped_ptr<Distiller> CreateDistillerForUrl( |
| 26 const GURL& unused) override { |
26 return scoped_ptr<Distiller>(CreateDistillerImpl()); | 27 return scoped_ptr<Distiller>(CreateDistillerImpl()); |
27 } | 28 } |
28 }; | 29 }; |
29 | 30 |
30 class FakeDistiller : public Distiller { | 31 class FakeDistiller : public Distiller { |
31 public: | 32 public: |
32 // If execute_callback is true, when DistillPage is called, a task will | 33 // If execute_callback is true, when DistillPage is called, a task will |
33 // immediately be posted to execute the callback with a simple | 34 // immediately be posted to execute the callback with a simple |
34 // DistilledArticleProto. | 35 // DistilledArticleProto. |
35 explicit FakeDistiller(bool execute_callback); | 36 explicit FakeDistiller(bool execute_callback); |
(...skipping 29 matching lines...) Expand all Loading... |
65 DistillationUpdateCallback page_callback_; | 66 DistillationUpdateCallback page_callback_; |
66 bool destruction_allowed_; | 67 bool destruction_allowed_; |
67 // Used to notify when distillation is complete. | 68 // Used to notify when distillation is complete. |
68 base::Closure distillation_initiated_callback_; | 69 base::Closure distillation_initiated_callback_; |
69 }; | 70 }; |
70 | 71 |
71 } // namespace test | 72 } // namespace test |
72 } // namespace dom_distiller | 73 } // namespace dom_distiller |
73 | 74 |
74 #endif // COMPONENTS_DOM_DISTILLER_CORE_FAKE_DISTILLER_H_ | 75 #endif // COMPONENTS_DOM_DISTILLER_CORE_FAKE_DISTILLER_H_ |
OLD | NEW |