OLD | NEW |
1 // Copyright 2016 The Chromium Authors. All rights reserved. | 1 // Copyright 2016 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/previews/core/previews_ui_service.h" | 5 #include "components/previews/core/previews_ui_service.h" |
6 | 6 |
7 #include <memory> | 7 #include <memory> |
8 | 8 |
9 #include "base/memory/ptr_util.h" | 9 #include "base/memory/ptr_util.h" |
10 #include "base/memory/ref_counted.h" | 10 #include "base/memory/ref_counted.h" |
11 #include "base/message_loop/message_loop.h" | 11 #include "base/message_loop/message_loop.h" |
12 #include "base/run_loop.h" | |
13 #include "base/single_thread_task_runner.h" | 12 #include "base/single_thread_task_runner.h" |
14 #include "components/previews/core/previews_io_data.h" | 13 #include "components/previews/core/previews_io_data.h" |
15 #include "testing/gtest/include/gtest/gtest.h" | 14 #include "testing/gtest/include/gtest/gtest.h" |
16 | 15 |
17 namespace previews { | 16 namespace previews { |
18 | 17 |
19 namespace { | 18 namespace { |
20 | 19 |
21 class TestPreviewsUIService : public PreviewsUIService { | 20 class TestPreviewsUIService : public PreviewsUIService { |
22 public: | 21 public: |
23 TestPreviewsUIService( | 22 TestPreviewsUIService( |
24 PreviewsIOData* previews_io_data, | 23 PreviewsIOData* previews_io_data, |
25 const scoped_refptr<base::SingleThreadTaskRunner>& io_task_runner) | 24 const scoped_refptr<base::SingleThreadTaskRunner>& io_task_runner, |
26 : PreviewsUIService(previews_io_data, io_task_runner), | 25 std::unique_ptr<PreviewsOptOutStore> previews_opt_out_store) |
| 26 : PreviewsUIService(previews_io_data, |
| 27 io_task_runner, |
| 28 std::move(previews_opt_out_store)), |
27 io_data_set_(false) {} | 29 io_data_set_(false) {} |
28 ~TestPreviewsUIService() override {} | 30 ~TestPreviewsUIService() override {} |
29 | 31 |
30 // Set |io_data_set_| to true and use base class functionality. | 32 // Set |io_data_set_| to true and use base class functionality. |
31 void SetIOData(base::WeakPtr<PreviewsIOData> previews_io_data) override { | 33 void SetIOData(base::WeakPtr<PreviewsIOData> previews_io_data) override { |
32 io_data_set_ = true; | 34 io_data_set_ = true; |
33 PreviewsUIService::SetIOData(previews_io_data); | 35 PreviewsUIService::SetIOData(previews_io_data); |
34 } | 36 } |
35 | 37 |
36 // Whether SetIOData was called. | 38 // Whether SetIOData was called. |
(...skipping 30 matching lines...) Expand all Loading... |
67 std::unique_ptr<PreviewsIOData> io_data_; | 69 std::unique_ptr<PreviewsIOData> io_data_; |
68 std::unique_ptr<TestPreviewsUIService> ui_service_; | 70 std::unique_ptr<TestPreviewsUIService> ui_service_; |
69 }; | 71 }; |
70 | 72 |
71 } // namespace | 73 } // namespace |
72 | 74 |
73 TEST_F(PreviewsUIServiceTest, TestInitialization) { | 75 TEST_F(PreviewsUIServiceTest, TestInitialization) { |
74 set_io_data(base::WrapUnique( | 76 set_io_data(base::WrapUnique( |
75 new PreviewsIOData(loop_.task_runner(), loop_.task_runner()))); | 77 new PreviewsIOData(loop_.task_runner(), loop_.task_runner()))); |
76 set_ui_service(base::WrapUnique( | 78 set_ui_service(base::WrapUnique( |
77 new TestPreviewsUIService(io_data(), loop_.task_runner()))); | 79 new TestPreviewsUIService(io_data(), loop_.task_runner(), nullptr))); |
78 base::RunLoop().RunUntilIdle(); | 80 loop_.RunUntilIdle(); |
79 // After the outstanding posted tasks have run, SetIOData should have been | 81 // After the outstanding posted tasks have run, SetIOData should have been |
80 // called for |ui_service_|. | 82 // called for |ui_service_|. |
81 EXPECT_TRUE(ui_service()->io_data_set()); | 83 EXPECT_TRUE(ui_service()->io_data_set()); |
82 } | 84 } |
83 | 85 |
84 } // namespace previews | 86 } // namespace previews |
OLD | NEW |