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
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
|
// Copyright 2014 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 "chrome/browser/safe_browsing/incident_reporting/delayed_callback_runner.h"
#include <map>
#include <string>
#include "base/callback.h"
#include "base/macros.h"
#include "base/test/test_simple_task_runner.h"
#include "base/thread_task_runner_handle.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace {
// A class of objects that invoke a callback upon destruction. This is used as
// an owned argument on callbacks given to a DelayedCallbackRunner under test.
class CallbackArgument {
public:
explicit CallbackArgument(const base::Closure& on_delete)
: on_delete_(on_delete) {}
~CallbackArgument() { on_delete_.Run(); }
private:
base::Closure on_delete_;
DISALLOW_COPY_AND_ASSIGN(CallbackArgument);
};
} // namespace
// A test fixture that prepares a DelayedCallbackRunner instance for use and
// tracks the lifecycle of callbacks sent to it.
class DelayedCallbackRunnerTest : public testing::Test {
public:
// Registers a callback that will record its running and destruction to the
// test fixture under the given name.
void RegisterTestCallback(const std::string& name) {
callbacks_[name] = CallbackState();
instance_->RegisterCallback(MakeCallback(name));
}
protected:
DelayedCallbackRunnerTest()
: task_runner_(new base::TestSimpleTaskRunner),
thread_task_runner_handle_(task_runner_) {}
void SetUp() override {
instance_.reset(new safe_browsing::DelayedCallbackRunner(
base::TimeDelta::FromMilliseconds(1), // ignored by simple runner.
task_runner_));
}
void TearDown() override { instance_.reset(); }
void OnRun(const std::string& name, CallbackArgument* arg) {
EXPECT_FALSE(callbacks_[name].run);
callbacks_[name].run = true;
}
void OnDelete(const std::string& name) {
EXPECT_FALSE(callbacks_[name].deleted);
callbacks_[name].deleted = true;
}
// Returns a callback argument that calls the test fixture's OnDelete method
// on behalf of the given callback name.
scoped_ptr<CallbackArgument> MakeCallbackArgument(const std::string& name) {
return make_scoped_ptr(new CallbackArgument(base::Bind(
&DelayedCallbackRunnerTest::OnDelete, base::Unretained(this), name)));
}
// Returns a closure that calls |OnRun| when run and |OnDelete| when deleted
// on behalf of the given callback name.
base::Closure MakeCallback(const std::string& name) {
return base::Bind(&DelayedCallbackRunnerTest::OnRun,
base::Unretained(this),
name,
base::Owned(MakeCallbackArgument(name).release()));
}
bool CallbackWasRun(const std::string& name) { return callbacks_[name].run; }
bool CallbackWasDeleted(const std::string& name) {
return callbacks_[name].deleted;
}
scoped_refptr<base::TestSimpleTaskRunner> task_runner_;
base::ThreadTaskRunnerHandle thread_task_runner_handle_;
scoped_ptr<safe_browsing::DelayedCallbackRunner> instance_;
private:
struct CallbackState {
CallbackState() : run(), deleted() {}
bool run;
bool deleted;
};
std::map<std::string, CallbackState> callbacks_;
};
// Tests that a callback is deleted when not run before the runner is destroyed.
TEST_F(DelayedCallbackRunnerTest, NotRunDeleted) {
const std::string name("one");
RegisterTestCallback(name);
instance_.reset();
EXPECT_FALSE(CallbackWasRun(name));
EXPECT_TRUE(CallbackWasDeleted(name));
}
// Tests that a callback is run and deleted while the runner is alive.
TEST_F(DelayedCallbackRunnerTest, RunDeleted) {
const std::string name("one");
RegisterTestCallback(name);
instance_->Start();
task_runner_->RunUntilIdle();
EXPECT_TRUE(CallbackWasRun(name));
EXPECT_TRUE(CallbackWasDeleted(name));
}
// Tests that a callback registered after Start() is called is also run and
// deleted.
TEST_F(DelayedCallbackRunnerTest, AddWhileRunningRun) {
const std::string name("one");
const std::string name2("two");
// Post a task to register a new callback after Start() is called.
task_runner_->PostTask(
FROM_HERE,
base::Bind(&DelayedCallbackRunnerTest::RegisterTestCallback,
base::Unretained(this),
name2));
RegisterTestCallback(name);
instance_->Start();
task_runner_->RunUntilIdle();
EXPECT_TRUE(CallbackWasRun(name));
EXPECT_TRUE(CallbackWasDeleted(name));
EXPECT_TRUE(CallbackWasRun(name2));
EXPECT_TRUE(CallbackWasDeleted(name2));
}
TEST_F(DelayedCallbackRunnerTest, MultipleRuns) {
const std::string name("one");
const std::string name2("two");
RegisterTestCallback(name);
instance_->Start();
task_runner_->RunUntilIdle();
EXPECT_TRUE(CallbackWasRun(name));
EXPECT_TRUE(CallbackWasDeleted(name));
RegisterTestCallback(name2);
instance_->Start();
task_runner_->RunUntilIdle();
EXPECT_TRUE(CallbackWasRun(name2));
EXPECT_TRUE(CallbackWasDeleted(name2));
}
|