blob: 0398ffd73a24632f0cae1987ffe1e74726d12ef3 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
|
// Copyright 2016 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "components/previews/core/previews_ui_service.h"
#include <memory>
#include "base/memory/ptr_util.h"
#include "base/memory/ref_counted.h"
#include "base/message_loop/message_loop.h"
#include "base/run_loop.h"
#include "base/single_thread_task_runner.h"
#include "components/previews/core/previews_io_data.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace previews {
namespace {
class TestPreviewsUIService : public PreviewsUIService {
public:
TestPreviewsUIService(
PreviewsIOData* previews_io_data,
const scoped_refptr<base::SingleThreadTaskRunner>& io_task_runner,
std::unique_ptr<PreviewsOptOutStore> previews_opt_out_store)
: PreviewsUIService(previews_io_data,
io_task_runner,
std::move(previews_opt_out_store),
PreviewsIsEnabledCallback()),
io_data_set_(false) {}
~TestPreviewsUIService() override {}
// Set |io_data_set_| to true and use base class functionality.
void SetIOData(base::WeakPtr<PreviewsIOData> previews_io_data) override {
io_data_set_ = true;
PreviewsUIService::SetIOData(previews_io_data);
}
// Whether SetIOData was called.
bool io_data_set() { return io_data_set_; }
private:
// Whether SetIOData was called.
bool io_data_set_;
};
class PreviewsUIServiceTest : public testing::Test {
public:
PreviewsUIServiceTest() {}
~PreviewsUIServiceTest() override {}
void set_io_data(std::unique_ptr<PreviewsIOData> io_data) {
io_data_ = std::move(io_data);
}
PreviewsIOData* io_data() { return io_data_.get(); }
void set_ui_service(std::unique_ptr<TestPreviewsUIService> ui_service) {
ui_service_ = std::move(ui_service);
}
TestPreviewsUIService* ui_service() { return ui_service_.get(); }
protected:
// Run this test on a single thread.
base::MessageLoopForIO loop_;
private:
std::unique_ptr<PreviewsIOData> io_data_;
std::unique_ptr<TestPreviewsUIService> ui_service_;
};
} // namespace
TEST_F(PreviewsUIServiceTest, TestInitialization) {
set_io_data(base::WrapUnique(
new PreviewsIOData(loop_.task_runner(), loop_.task_runner())));
set_ui_service(base::WrapUnique(
new TestPreviewsUIService(io_data(), loop_.task_runner(), nullptr)));
base::RunLoop().RunUntilIdle();
// After the outstanding posted tasks have run, SetIOData should have been
// called for |ui_service_|.
EXPECT_TRUE(ui_service()->io_data_set());
}
} // namespace previews
|
