OLD | NEW |
| (Empty) |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | |
2 // Use of this source code is governed by a BSD-style license that can be | |
3 // found in the LICENSE file. | |
4 | |
5 #include "chrome/browser/chromeos/gdata/gdata_operation_registry.h" | |
6 | |
7 #include "base/compiler_specific.h" | |
8 #include "base/memory/weak_ptr.h" | |
9 #include "base/message_loop.h" | |
10 #include "content/public/test/test_browser_thread.h" | |
11 #include "testing/gmock/include/gmock/gmock.h" | |
12 #include "testing/gtest/include/gtest/gtest.h" | |
13 | |
14 using testing::ElementsAre; | |
15 | |
16 namespace gdata { | |
17 | |
18 namespace { | |
19 | |
20 class MockOperation : public GDataOperationRegistry::Operation, | |
21 public base::SupportsWeakPtr<MockOperation> { | |
22 public: | |
23 MockOperation(GDataOperationRegistry* registry, | |
24 GDataOperationRegistry::OperationType type, | |
25 const FilePath& path) | |
26 : GDataOperationRegistry::Operation(registry, type, path) {} | |
27 | |
28 MOCK_METHOD0(DoCancel, void()); | |
29 | |
30 // Make them public so that they can be called from test cases. | |
31 using GDataOperationRegistry::Operation::NotifyStart; | |
32 using GDataOperationRegistry::Operation::NotifyProgress; | |
33 using GDataOperationRegistry::Operation::NotifyFinish; | |
34 }; | |
35 | |
36 class MockUploadOperation : public MockOperation { | |
37 public: | |
38 explicit MockUploadOperation(GDataOperationRegistry* registry) | |
39 : MockOperation(registry, | |
40 GDataOperationRegistry::OPERATION_UPLOAD, | |
41 FilePath("/dummy/upload")) {} | |
42 }; | |
43 | |
44 class MockDownloadOperation : public MockOperation { | |
45 public: | |
46 explicit MockDownloadOperation(GDataOperationRegistry* registry) | |
47 : MockOperation(registry, | |
48 GDataOperationRegistry::OPERATION_DOWNLOAD, | |
49 FilePath("/dummy/download")) {} | |
50 }; | |
51 | |
52 class MockOtherOperation : public MockOperation { | |
53 public: | |
54 explicit MockOtherOperation(GDataOperationRegistry* registry) | |
55 : MockOperation(registry, | |
56 GDataOperationRegistry::OPERATION_OTHER, | |
57 FilePath("/dummy/other")) {} | |
58 }; | |
59 | |
60 class TestObserver : public GDataOperationRegistry::Observer { | |
61 public: | |
62 virtual void OnProgressUpdate( | |
63 const std::vector<GDataOperationRegistry::ProgressStatus>& list) | |
64 OVERRIDE { | |
65 status_ = list; | |
66 } | |
67 | |
68 const std::vector<GDataOperationRegistry::ProgressStatus>& status() const { | |
69 return status_; | |
70 } | |
71 | |
72 private: | |
73 std::vector<GDataOperationRegistry::ProgressStatus> status_; | |
74 }; | |
75 | |
76 class ProgressMatcher | |
77 : public ::testing::MatcherInterface< | |
78 const GDataOperationRegistry::ProgressStatus&> { | |
79 public: | |
80 ProgressMatcher(int64 expected_current, int64 expected_total) | |
81 : expected_current_(expected_current), | |
82 expected_total_(expected_total) {} | |
83 | |
84 virtual bool MatchAndExplain( | |
85 const GDataOperationRegistry::ProgressStatus& status, | |
86 testing::MatchResultListener* /* listener */) const OVERRIDE { | |
87 return status.progress_current == expected_current_ && | |
88 status.progress_total == expected_total_; | |
89 } | |
90 | |
91 virtual void DescribeTo(::std::ostream* os) const OVERRIDE { | |
92 *os << "current / total equals " << expected_current_ << " / " << | |
93 expected_total_; | |
94 } | |
95 | |
96 virtual void DescribeNegationTo(::std::ostream* os) const OVERRIDE { | |
97 *os << "current / total does not equal " << expected_current_ << " / " << | |
98 expected_total_; | |
99 } | |
100 | |
101 private: | |
102 const int64 expected_current_; | |
103 const int64 expected_total_; | |
104 }; | |
105 | |
106 testing::Matcher<const GDataOperationRegistry::ProgressStatus&> Progress( | |
107 int64 current, int64 total) { | |
108 return testing::MakeMatcher(new ProgressMatcher(current, total)); | |
109 } | |
110 | |
111 } // namespace | |
112 | |
113 // Pretty-prints ProgressStatus for testing purpose. | |
114 std::ostream& operator<<(std::ostream& os, | |
115 const GDataOperationRegistry::ProgressStatus& status) { | |
116 return os << status.DebugString(); | |
117 } | |
118 | |
119 class GDataOperationRegistryTest : public testing::Test { | |
120 protected: | |
121 GDataOperationRegistryTest() | |
122 : ui_thread_(content::BrowserThread::UI, &message_loop_) { | |
123 } | |
124 MessageLoopForUI message_loop_; | |
125 content::TestBrowserThread ui_thread_; | |
126 }; | |
127 | |
128 TEST_F(GDataOperationRegistryTest, OneSuccess) { | |
129 TestObserver observer; | |
130 GDataOperationRegistry registry; | |
131 registry.DisableNotificationFrequencyControlForTest(); | |
132 registry.AddObserver(&observer); | |
133 | |
134 base::WeakPtr<MockOperation> op1 = | |
135 (new MockUploadOperation(®istry))->AsWeakPtr(); | |
136 EXPECT_CALL(*op1, DoCancel()).Times(0); | |
137 | |
138 EXPECT_EQ(0U, observer.status().size()); | |
139 op1->NotifyStart(); | |
140 EXPECT_THAT(observer.status(), ElementsAre(Progress(0, -1))); | |
141 op1->NotifyProgress(0, 100); | |
142 EXPECT_THAT(observer.status(), ElementsAre(Progress(0, 100))); | |
143 op1->NotifyProgress(100, 100); | |
144 EXPECT_THAT(observer.status(), ElementsAre(Progress(100, 100))); | |
145 op1->NotifyFinish(GDataOperationRegistry::OPERATION_COMPLETED); | |
146 // Contains one "COMPLETED" notification. | |
147 EXPECT_THAT(observer.status(), ElementsAre(Progress(100, 100))); | |
148 // Then it is removed. | |
149 EXPECT_EQ(0U, registry.GetProgressStatusList().size()); | |
150 EXPECT_EQ(NULL, op1.get()); // deleted | |
151 } | |
152 | |
153 TEST_F(GDataOperationRegistryTest, OneCancel) { | |
154 TestObserver observer; | |
155 GDataOperationRegistry registry; | |
156 registry.DisableNotificationFrequencyControlForTest(); | |
157 registry.AddObserver(&observer); | |
158 | |
159 base::WeakPtr<MockOperation> op1 = | |
160 (new MockUploadOperation(®istry))->AsWeakPtr(); | |
161 EXPECT_CALL(*op1, DoCancel()); | |
162 | |
163 EXPECT_EQ(0U, observer.status().size()); | |
164 op1->NotifyStart(); | |
165 EXPECT_THAT(observer.status(), ElementsAre(Progress(0, -1))); | |
166 op1->NotifyProgress(0, 100); | |
167 EXPECT_THAT(observer.status(), ElementsAre(Progress(0, 100))); | |
168 registry.CancelAll(); | |
169 EXPECT_THAT(observer.status(), ElementsAre(Progress(0, 100))); | |
170 EXPECT_EQ(0U, registry.GetProgressStatusList().size()); | |
171 EXPECT_EQ(NULL, op1.get()); // deleted | |
172 } | |
173 | |
174 TEST_F(GDataOperationRegistryTest, TwoSuccess) { | |
175 TestObserver observer; | |
176 GDataOperationRegistry registry; | |
177 registry.DisableNotificationFrequencyControlForTest(); | |
178 registry.AddObserver(&observer); | |
179 | |
180 base::WeakPtr<MockOperation> op1 = | |
181 (new MockUploadOperation(®istry))->AsWeakPtr(); | |
182 base::WeakPtr<MockOperation> op2 = | |
183 (new MockDownloadOperation(®istry))->AsWeakPtr(); | |
184 EXPECT_CALL(*op1, DoCancel()).Times(0); | |
185 EXPECT_CALL(*op2, DoCancel()).Times(0); | |
186 | |
187 EXPECT_EQ(0U, observer.status().size()); | |
188 op1->NotifyStart(); | |
189 op1->NotifyProgress(0, 100); | |
190 EXPECT_THAT(observer.status(), ElementsAre(Progress(0, 100))); | |
191 op2->NotifyStart(); | |
192 op2->NotifyProgress(0, 200); | |
193 op1->NotifyProgress(50, 100); | |
194 EXPECT_THAT(observer.status(), ElementsAre(Progress(50, 100), | |
195 Progress(0, 200))); | |
196 op1->NotifyFinish(GDataOperationRegistry::OPERATION_COMPLETED); | |
197 EXPECT_THAT(observer.status(), ElementsAre(Progress(50, 100), | |
198 Progress(0, 200))); | |
199 EXPECT_EQ(1U, registry.GetProgressStatusList().size()); | |
200 op2->NotifyFinish(GDataOperationRegistry::OPERATION_COMPLETED); | |
201 EXPECT_THAT(observer.status(), ElementsAre(Progress(0, 200))); | |
202 EXPECT_EQ(0U, registry.GetProgressStatusList().size()); | |
203 EXPECT_EQ(NULL, op1.get()); // deleted | |
204 EXPECT_EQ(NULL, op2.get()); // deleted | |
205 } | |
206 | |
207 TEST_F(GDataOperationRegistryTest, ThreeCancel) { | |
208 TestObserver observer; | |
209 GDataOperationRegistry registry; | |
210 registry.DisableNotificationFrequencyControlForTest(); | |
211 registry.AddObserver(&observer); | |
212 | |
213 base::WeakPtr<MockOperation> op1 = | |
214 (new MockUploadOperation(®istry))->AsWeakPtr(); | |
215 base::WeakPtr<MockOperation> op2 = | |
216 (new MockDownloadOperation(®istry))->AsWeakPtr(); | |
217 base::WeakPtr<MockOperation> op3 = | |
218 (new MockOtherOperation(®istry))->AsWeakPtr(); | |
219 EXPECT_CALL(*op1, DoCancel()); | |
220 EXPECT_CALL(*op2, DoCancel()); | |
221 EXPECT_CALL(*op3, DoCancel()); | |
222 | |
223 EXPECT_EQ(0U, observer.status().size()); | |
224 op1->NotifyStart(); | |
225 EXPECT_EQ(1U, observer.status().size()); | |
226 op2->NotifyStart(); | |
227 EXPECT_EQ(2U, observer.status().size()); | |
228 op3->NotifyStart(); | |
229 EXPECT_EQ(2U, observer.status().size()); // only upload/download is reported. | |
230 registry.CancelAll(); | |
231 EXPECT_EQ(1U, observer.status().size()); // holds the last one "COMPLETED" | |
232 EXPECT_EQ(0U, registry.GetProgressStatusList().size()); | |
233 EXPECT_EQ(NULL, op1.get()); // deleted | |
234 EXPECT_EQ(NULL, op2.get()); // deleted | |
235 EXPECT_EQ(NULL, op3.get()); // deleted. CancelAll cares all operations. | |
236 } | |
237 | |
238 TEST_F(GDataOperationRegistryTest, RestartOperation) { | |
239 TestObserver observer; | |
240 GDataOperationRegistry registry; | |
241 registry.DisableNotificationFrequencyControlForTest(); | |
242 registry.AddObserver(&observer); | |
243 | |
244 base::WeakPtr<MockOperation> op1 = | |
245 (new MockUploadOperation(®istry))->AsWeakPtr(); | |
246 EXPECT_CALL(*op1, DoCancel()).Times(0); | |
247 | |
248 op1->NotifyStart(); | |
249 EXPECT_EQ(1U, registry.GetProgressStatusList().size()); | |
250 op1->NotifyStart(); // restart | |
251 EXPECT_EQ(1U, registry.GetProgressStatusList().size()); | |
252 op1->NotifyProgress(0, 200); | |
253 op1->NotifyFinish(GDataOperationRegistry::OPERATION_COMPLETED); | |
254 EXPECT_EQ(0U, registry.GetProgressStatusList().size()); | |
255 EXPECT_EQ(NULL, op1.get()); // deleted | |
256 } | |
257 | |
258 } // namespace gdata | |
OLD | NEW |