Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(22)

Side by Side Diff: chrome/browser/component_updater/test/component_updater_service_unittest.cc

Issue 15908002: Differential updates for components. (Closed) Base URL: svn://svn.chromium.org/chrome/trunk/src/
Patch Set: Created 7 years, 7 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch | Annotate | Revision Log
OLDNEW
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. 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 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 "chrome/browser/component_updater/component_updater_service.h"
6
7 #include <list> 5 #include <list>
8 #include <utility> 6 #include <utility>
9 7
10 #include "base/compiler_specific.h" 8 #include "base/compiler_specific.h"
11 #include "base/file_util.h" 9 #include "base/file_util.h"
12 #include "base/files/file_path.h" 10 #include "base/files/file_path.h"
13 #include "base/memory/scoped_vector.h" 11 #include "base/memory/scoped_vector.h"
14 #include "base/message_loop.h" 12 #include "base/message_loop.h"
15 #include "base/path_service.h" 13 #include "base/path_service.h"
14 #include "base/stringprintf.h"
15 #include "base/strings/string_number_conversions.h"
16 #include "base/values.h" 16 #include "base/values.h"
17 #include "chrome/browser/component_updater/component_updater_service.h"
18 #include "chrome/browser/component_updater/test/component_patcher_mock.h"
17 #include "chrome/common/chrome_notification_types.h" 19 #include "chrome/common/chrome_notification_types.h"
18 #include "chrome/common/chrome_paths.h" 20 #include "chrome/common/chrome_paths.h"
19 #include "content/public/browser/notification_observer.h" 21 #include "content/public/browser/notification_observer.h"
20 #include "content/public/browser/notification_service.h" 22 #include "content/public/browser/notification_service.h"
21 #include "content/public/test/test_browser_thread.h" 23 #include "content/public/test/test_browser_thread.h"
22 #include "content/public/test/test_notification_tracker.h" 24 #include "content/public/test/test_notification_tracker.h"
23 #include "content/test/net/url_request_prepackaged_interceptor.h" 25 #include "content/test/net/url_request_prepackaged_interceptor.h"
24 #include "googleurl/src/gurl.h" 26 #include "googleurl/src/gurl.h"
25 #include "libxml/globals.h" 27 #include "libxml/globals.h"
28 #include "net/base/upload_bytes_element_reader.h"
26 #include "net/url_request/url_fetcher.h" 29 #include "net/url_request/url_fetcher.h"
30 #include "net/url_request/url_request.h"
31 #include "net/url_request/url_request_filter.h"
32 #include "net/url_request/url_request_simple_job.h"
27 #include "net/url_request/url_request_test_util.h" 33 #include "net/url_request/url_request_test_util.h"
28 #include "testing/gtest/include/gtest/gtest.h" 34 #include "testing/gtest/include/gtest/gtest.h"
29 35
30 using content::BrowserThread; 36 using content::BrowserThread;
31 using content::TestNotificationTracker; 37 using content::TestNotificationTracker;
32 38
33 namespace { 39 namespace {
34 // Overrides some of the component updater behaviors so it is easier to test 40
35 // and loops faster. In actual usage it takes hours do to a full cycle. 41 // component 1 has extension id "jebgalgnebhfojomionfpkfelancnnkf", and
42 // the RSA public key the following hash:
43 const uint8 jebg_hash[] = {0x94, 0x16, 0x0b, 0x6d, 0x41, 0x75, 0xe9, 0xec,
44 0x8e, 0xd5, 0xfa, 0x54, 0xb0, 0xd2, 0xdd, 0xa5,
45 0x6e, 0x05, 0x6b, 0xe8, 0x73, 0x47, 0xf6, 0xc4,
46 0x11, 0x9f, 0xbc, 0xb3, 0x09, 0xb3, 0x5b, 0x40};
47 // component 2 has extension id "abagagagagagagagagagagagagagagag", and
48 // the RSA public key the following hash:
49 const uint8 abag_hash[] = {0x01, 0x06, 0x06, 0x06, 0x06, 0x06, 0x06, 0x06,
50 0x06, 0x06, 0x06, 0x06, 0x06, 0x06, 0x06, 0x06,
51 0x06, 0x06, 0x06, 0x06, 0x06, 0x06, 0x06, 0x06,
52 0x06, 0x06, 0x06, 0x06, 0x06, 0x06, 0x06, 0x01};
53 // component 3 has extension id "ihfokbkgjpifnbbojhneepfflplebdkc", and
54 // the RSA public key the following hash:
55 const uint8 ihfo_hash[] = {0x87, 0x5e, 0xa1, 0xa6, 0x9f, 0x85, 0xd1, 0x1e,
56 0x97, 0xd4, 0x4f, 0x55, 0xbf, 0xb4, 0x13, 0xa2,
57 0xe7, 0xc5, 0xc8, 0xf5, 0x60, 0x19, 0x78, 0x1b,
58 0x6d, 0xe9, 0x4c, 0xeb, 0x96, 0x05, 0x42, 0x17};
59
36 class TestConfigurator : public ComponentUpdateService::Configurator { 60 class TestConfigurator : public ComponentUpdateService::Configurator {
37 public: 61 public:
38 TestConfigurator() 62 TestConfigurator();
39 : times_(1), recheck_time_(0), ondemand_time_(0), cus_(NULL) {
40 }
41 63
42 virtual int InitialDelay() OVERRIDE { return 0; } 64 virtual int InitialDelay() OVERRIDE;
43 65
44 typedef std::pair<CrxComponent*, int> CheckAtLoopCount; 66 typedef std::pair<CrxComponent*, int> CheckAtLoopCount;
45 67
46 virtual int NextCheckDelay() OVERRIDE { 68 virtual int NextCheckDelay() OVERRIDE;
47 // This is called when a new full cycle of checking for updates is going
48 // to happen. In test we normally only test one cycle so it is a good
49 // time to break from the test messageloop Run() method so the test can
50 // finish.
51 if (--times_ <= 0) {
52 MessageLoop::current()->Quit();
53 return 0;
54 69
55 } 70 virtual int StepDelay() OVERRIDE;
56 71
57 // Look for checks to issue in the middle of the loop. 72 virtual int MinimumReCheckWait() OVERRIDE;
58 for (std::list<CheckAtLoopCount>::iterator
59 i = components_to_check_.begin();
60 i != components_to_check_.end(); ) {
61 if (i->second == times_) {
62 cus_->CheckForUpdateSoon(*i->first);
63 i = components_to_check_.erase(i);
64 } else {
65 ++i;
66 }
67 }
68 return 1;
69 }
70 73
71 virtual int StepDelay() OVERRIDE { 74 virtual int OnDemandDelay() OVERRIDE;
72 return 0;
73 }
74 75
75 virtual int MinimumReCheckWait() OVERRIDE { 76 virtual GURL UpdateUrl(CrxComponent::UrlSource source) OVERRIDE;
76 return recheck_time_;
77 }
78 77
79 virtual int OnDemandDelay() OVERRIDE { 78 virtual GURL PingUrl() OVERRIDE;
80 return ondemand_time_;
81 }
82 79
83 virtual GURL UpdateUrl(CrxComponent::UrlSource source) OVERRIDE { 80 virtual const char* ExtraRequestParams() OVERRIDE;
84 switch (source) {
85 case CrxComponent::BANDAID:
86 return GURL("http://localhost/upd");
87 case CrxComponent::CWS_PUBLIC:
88 return GURL("http://localhost/cws");
89 default:
90 return GURL("http://wronghost/bad");
91 };
92 }
93 81
94 virtual const char* ExtraRequestParams() OVERRIDE { return "extra=foo"; } 82 virtual size_t UrlSizeLimit() OVERRIDE;
95 83
96 virtual size_t UrlSizeLimit() OVERRIDE { return 256; } 84 virtual net::URLRequestContextGetter* RequestContext() OVERRIDE;
97
98 virtual net::URLRequestContextGetter* RequestContext() OVERRIDE {
99 return new net::TestURLRequestContextGetter(
100 BrowserThread::GetMessageLoopProxyForThread(BrowserThread::IO));
101 }
102 85
103 // Don't use the utility process to decode files. 86 // Don't use the utility process to decode files.
104 virtual bool InProcess() OVERRIDE { return true; } 87 virtual bool InProcess() OVERRIDE;
105 88
106 virtual void OnEvent(Events event, int extra) OVERRIDE { } 89 virtual void OnEvent(Events event, int extra) OVERRIDE;
107 90
108 // Set how many update checks are called, the default value is just once. 91 virtual ComponentPatcher* CreateComponentPatcher() OVERRIDE;
109 void SetLoopCount(int times) { times_ = times; }
110 92
111 void SetRecheckTime(int seconds) { 93 virtual bool PingsEnabled() const OVERRIDE;
112 recheck_time_ = seconds;
113 }
114 94
115 void SetOnDemandTime(int seconds) { 95 virtual bool DeltasEnabled() const OVERRIDE;
116 ondemand_time_ = seconds;
117 }
118 96
119 void AddComponentToCheck(CrxComponent* com, int at_loop_iter) { 97 void SetLoopCount(int times);
120 components_to_check_.push_back(std::make_pair(com, at_loop_iter));
121 }
122 98
123 void SetComponentUpdateService(ComponentUpdateService* cus) { 99 void SetRecheckTime(int seconds);
124 cus_ = cus; 100
125 } 101 void SetOnDemandTime(int seconds);
102
103 void AddComponentToCheck(CrxComponent* com, int at_loop_iter);
104
105 void SetComponentUpdateService(ComponentUpdateService* cus);
126 106
127 private: 107 private:
128 int times_; 108 int times_;
129 int recheck_time_; 109 int recheck_time_;
130 int ondemand_time_; 110 int ondemand_time_;
131 111
132 std::list<CheckAtLoopCount> components_to_check_; 112 std::list<CheckAtLoopCount> components_to_check_;
133 ComponentUpdateService* cus_; 113 ComponentUpdateService* cus_;
134 }; 114 };
135 115
136 class TestInstaller : public ComponentInstaller { 116 class TestInstaller : public ComponentInstaller {
137 public : 117 public :
138 explicit TestInstaller() 118 explicit TestInstaller();
139 : error_(0), install_count_(0) { 119
140 } 120 virtual void OnUpdateError(int error) OVERRIDE;
141
142 virtual void OnUpdateError(int error) OVERRIDE {
143 EXPECT_NE(0, error);
144 error_ = error;
145 }
146 121
147 virtual bool Install(const base::DictionaryValue& manifest, 122 virtual bool Install(const base::DictionaryValue& manifest,
148 const base::FilePath& unpack_path) OVERRIDE { 123 const base::FilePath& unpack_path) OVERRIDE;
149 ++install_count_; 124
150 return file_util::Delete(unpack_path, true); 125 virtual bool GetInstalledFile(const std::string& file,
151 } 126 base::FilePath* installed_file) OVERRIDE;
152 127
153 int error() const { return error_; } 128 int error() const;
154 129
155 int install_count() const { return install_count_; } 130 int install_count() const;
156 131
157 private: 132 protected:
158 int error_; 133 int error_;
159 int install_count_; 134 int install_count_;
160 }; 135 };
161 136
162 // component 1 has extension id "jebgalgnebhfojomionfpkfelancnnkf", and 137 class PersistentTestInstaller : public TestInstaller {
163 // the RSA public key the following hash: 138 public :
164 const uint8 jebg_hash[] = {0x94,0x16,0x0b,0x6d,0x41,0x75,0xe9,0xec,0x8e,0xd5, 139 explicit PersistentTestInstaller();
165 0xfa,0x54,0xb0,0xd2,0xdd,0xa5,0x6e,0x05,0x6b,0xe8, 140
166 0x73,0x47,0xf6,0xc4,0x11,0x9f,0xbc,0xb3,0x09,0xb3, 141 virtual ~PersistentTestInstaller();
167 0x5b,0x40}; 142
168 // component 2 has extension id "abagagagagagagagagagagagagagagag", and 143 virtual bool Install(const base::DictionaryValue& manifest,
169 // the RSA public key the following hash: 144 const base::FilePath& unpack_path) OVERRIDE;
170 const uint8 abag_hash[] = {0x01,0x06,0x06,0x06,0x06,0x06,0x06,0x06,0x06,0x06, 145
171 0x06,0x06,0x06,0x06,0x06,0x06,0x06,0x06,0x06,0x06, 146 virtual bool GetInstalledFile(const std::string& file,
172 0x06,0x06,0x06,0x06,0x06,0x06,0x06,0x06,0x06,0x06, 147 base::FilePath* installed_file) OVERRIDE;
173 0x06,0x01}; 148
174 149 private:
175 const char expected_crx_url[] = 150 base::FilePath install_directory_;
176 "http://localhost/download/jebgalgnebhfojomionfpkfelancnnkf.crx"; 151 Version current_version_;
177 152 };
178 } // namespace 153
179
180 // Common fixture for all the component updater tests.
181 class ComponentUpdaterTest : public testing::Test { 154 class ComponentUpdaterTest : public testing::Test {
182 public: 155 public:
183 enum TestComponents { 156 enum TestComponents {
184 kTestComponent_abag, 157 kTestComponent_abag,
185 kTestComponent_jebg 158 kTestComponent_jebg,
159 kTestComponent_ihfo,
186 }; 160 };
187 161
188 ComponentUpdaterTest() : component_updater_(NULL), test_config_(NULL) { 162 ComponentUpdaterTest();
189 // The component updater instance under test. 163
190 test_config_ = new TestConfigurator; 164 virtual ~ComponentUpdaterTest();
191 component_updater_.reset(ComponentUpdateServiceFactory(test_config_)); 165
192 test_config_->SetComponentUpdateService(component_updater_.get()); 166 virtual void TearDown();
193 // The test directory is chrome/test/data/components. 167
194 PathService::Get(chrome::DIR_TEST_DATA, &test_data_dir_); 168 ComponentUpdateService* component_updater();
195 test_data_dir_ = test_data_dir_.AppendASCII("components");
196
197 // Subscribe to all component updater notifications.
198 const int notifications[] = {
199 chrome::NOTIFICATION_COMPONENT_UPDATER_STARTED,
200 chrome::NOTIFICATION_COMPONENT_UPDATER_SLEEPING,
201 chrome::NOTIFICATION_COMPONENT_UPDATE_FOUND,
202 chrome::NOTIFICATION_COMPONENT_UPDATE_READY
203 };
204
205 for (int ix = 0; ix != arraysize(notifications); ++ix) {
206 notification_tracker_.ListenFor(
207 notifications[ix], content::NotificationService::AllSources());
208 }
209 net::URLFetcher::SetEnableInterceptionForTests(true);
210 }
211
212 virtual ~ComponentUpdaterTest() {
213 net::URLFetcher::SetEnableInterceptionForTests(false);
214 }
215
216 virtual void TearDown() {
217 xmlCleanupGlobals();
218 }
219
220 ComponentUpdateService* component_updater() {
221 return component_updater_.get();
222 }
223 169
224 // Makes the full path to a component updater test file. 170 // Makes the full path to a component updater test file.
225 const base::FilePath test_file(const char* file) { 171 const base::FilePath test_file(const char* file);
226 return test_data_dir_.AppendASCII(file); 172
227 } 173 TestNotificationTracker& notification_tracker();
228 174
229 TestNotificationTracker& notification_tracker() { 175 TestConfigurator* test_configurator();
230 return notification_tracker_;
231 }
232
233 TestConfigurator* test_configurator() {
234 return test_config_;
235 }
236 176
237 ComponentUpdateService::Status RegisterComponent(CrxComponent* com, 177 ComponentUpdateService::Status RegisterComponent(CrxComponent* com,
238 TestComponents component, 178 TestComponents component,
239 const Version& version) { 179 const Version& version,
240 if (component == kTestComponent_abag) { 180 TestInstaller* installer);
241 com->name = "test_abag";
242 com->pk_hash.assign(abag_hash, abag_hash + arraysize(abag_hash));
243 } else {
244 com->name = "test_jebg";
245 com->pk_hash.assign(jebg_hash, jebg_hash + arraysize(jebg_hash));
246 }
247 com->version = version;
248 TestInstaller* installer = new TestInstaller;
249 com->installer = installer;
250 test_installers_.push_back(installer);
251 return component_updater_->RegisterComponent(*com);
252 }
253 181
254 private: 182 private:
255 scoped_ptr<ComponentUpdateService> component_updater_; 183 scoped_ptr<ComponentUpdateService> component_updater_;
256 base::FilePath test_data_dir_; 184 base::FilePath test_data_dir_;
257 TestNotificationTracker notification_tracker_; 185 TestNotificationTracker notification_tracker_;
258 TestConfigurator* test_config_; 186 TestConfigurator* test_config_;
259 // ComponentInstaller objects to delete after each test. 187 };
260 ScopedVector<TestInstaller> test_installers_; 188
261 }; 189 const char expected_crx_url[] =
190 "http://localhost/download/jebgalgnebhfojomionfpkfelancnnkf.crx";
191
192 class PingChecker {
193 public:
194 explicit PingChecker(const std::map<std::string, std::string>& attributes);
195
196 virtual ~PingChecker();
197
198 void Trial(net::URLRequest* request);
199
200 int NumHits() const {
201 return num_hits_;
202 }
203 int NumMisses() const {
204 return num_misses_;
205 }
206
207 private:
208 int num_hits_;
209 int num_misses_;
210 const std::map<std::string, std::string> attributes_;
211 virtual bool Test(net::URLRequest* request);
212 };
213
214 PingChecker::PingChecker(const std::map<std::string, std::string>& attributes)
215 : num_hits_(0), num_misses_(0), attributes_(attributes) {
216 }
217
218 PingChecker::~PingChecker() {}
219
220 void PingChecker::Trial(net::URLRequest* request) {
221 if (Test(request)) {
222 ++num_hits_;
223 } else {
224 ++num_misses_;
225 }
226 }
227
228 bool PingChecker::Test(net::URLRequest* request) {
229 if (request->has_upload()) {
230 const net::UploadDataStream* stream = request->get_upload();
231 const net::UploadBytesElementReader* reader =
232 stream->element_readers()[0]->AsBytesReader();
233 int size = reader->length();
234 scoped_refptr <net::IOBuffer> buffer = new net::IOBuffer(size);
235 std::string data(reader->bytes());
236 // For now, we assume that there is only one ping per POST.
237 std::string::size_type start = data.find("<o:event");
238 if (start != std::string::npos) {
239 std::string::size_type end = data.find(">", start);
240 if (end != std::string::npos) {
241 std::string ping = data.substr(start, end - start);
242 std::map<std::string, std::string>::const_iterator iter;
243 for (iter = attributes_.begin(); iter != attributes_.end(); ++iter) {
244 // does the ping contain the specified attribute/value?
245 if (ping.find(std::string(" ") + (iter->first) +
246 std::string("=") + (iter->second)) == string::npos) {
247 return false;
248 }
249 }
250 return true;
251 }
252 }
253 }
254 return false;
255 }
256
257 class URLRequestPostInterceptor {
258 public:
259 explicit URLRequestPostInterceptor(PingChecker* counter);
260 virtual ~URLRequestPostInterceptor();
261
262 private:
263 class Delegate;
264
265 // After creation, |delegate_| lives on the IO thread, and a task to delete it
266 // is posted from ~URLRequestPostInterceptor().
267 Delegate* delegate_;
268
269 DISALLOW_COPY_AND_ASSIGN(URLRequestPostInterceptor);
270 };
271
272 class URLRequestPostInterceptor::Delegate
273 : public net::URLRequestJobFactory::ProtocolHandler {
274 public:
275 explicit Delegate(PingChecker* counter);
276 virtual ~Delegate() {}
277
278 void Register();
279
280 void Unregister();
281
282 private:
283 PingChecker* counter_;
284
285 virtual net::URLRequestJob* MaybeCreateJob(
286 net::URLRequest* request,
287 net::NetworkDelegate* network_delegate) const OVERRIDE;
288 };
289
290 class URLRequestPingMockJob : public net::URLRequestSimpleJob {
291 public:
292 URLRequestPingMockJob(net::URLRequest* request,
293 net::NetworkDelegate* network_delegate);
294
295 protected:
296 virtual int GetData(std::string* mime_type,
297 std::string* charset,
298 std::string* data,
299 const net::CompletionCallback& callback) const OVERRIDE;
300
301 private:
302 virtual ~URLRequestPingMockJob() {}
303
304 DISALLOW_COPY_AND_ASSIGN(URLRequestPingMockJob);
305 };
306
307 URLRequestPostInterceptor::URLRequestPostInterceptor(
308 PingChecker* counter) : delegate_(new Delegate(counter)) {
309 BrowserThread::PostTask(BrowserThread::IO, FROM_HERE,
310 base::Bind(&Delegate::Register,
311 base::Unretained(delegate_)));
312 }
313
314 URLRequestPostInterceptor::~URLRequestPostInterceptor() {
315 BrowserThread::PostTask(BrowserThread::IO, FROM_HERE,
316 base::Bind(&Delegate::Unregister,
317 base::Unretained(delegate_)));
318 }
319
320 URLRequestPostInterceptor::Delegate::Delegate(
321 PingChecker* counter) : counter_(counter) {
322 }
323
324 void URLRequestPostInterceptor::Delegate::Register() {
325 net::URLRequestFilter::GetInstance()->AddHostnameProtocolHandler(
326 "http", "localhost2",
327 scoped_ptr<net::URLRequestJobFactory::ProtocolHandler>(this));
328 }
329
330 void URLRequestPostInterceptor::Delegate::Unregister() {
331 net::URLRequestFilter::GetInstance()->
332 RemoveHostnameHandler("http", "localhost2");
333 }
334
335 net::URLRequestJob* URLRequestPostInterceptor::Delegate::MaybeCreateJob(
336 net::URLRequest* request,
337 net::NetworkDelegate* network_delegate) const {
338 CHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
339 if (request->has_upload()) {
340 counter_->Trial(request);
341 return new URLRequestPingMockJob(request, network_delegate);
342 }
343 return NULL;
344 }
345
346 URLRequestPingMockJob::URLRequestPingMockJob(
347 net::URLRequest* request,
348 net::NetworkDelegate* network_delegate)
349 : net::URLRequestSimpleJob(request, network_delegate) {
350 }
351
352 int URLRequestPingMockJob::GetData(
353 std::string* mime_type,
354 std::string* charset,
355 std::string* data,
356 const net::CompletionCallback& callback) const {
357 mime_type->assign("text/plain");
358 charset->assign("US-ASCII");
359 data->assign(""); // There is no reason to have a response body in this test.
360 return net::OK;
361 }
362
363 } // namespace
364
365 TestInstaller::TestInstaller()
366 : error_(0), install_count_(0) {
367 }
368
369 void TestInstaller::OnUpdateError(int error) {
370 EXPECT_NE(0, error);
371 error_ = error;
372 }
373
374 bool TestInstaller::Install(const base::DictionaryValue& manifest,
375 const base::FilePath& unpack_path) {
376 ++install_count_;
377 return file_util::Delete(unpack_path, true);
378 }
379
380 bool TestInstaller::GetInstalledFile(const std::string& file,
381 base::FilePath* installed_file) {
382 return false;
383 }
384
385 int TestInstaller::error() const { return error_; }
386
387 int TestInstaller::install_count() const { return install_count_; }
388
389 PersistentTestInstaller::PersistentTestInstaller() {
390 file_util::CreateNewTempDirectory(FILE_PATH_LITERAL("TEST_"),
391 &install_directory_);
392 }
393
394 PersistentTestInstaller::~PersistentTestInstaller() {
395 file_util::Delete(install_directory_, true);
396 }
397
398 bool PersistentTestInstaller::Install(const base::DictionaryValue& manifest,
399 const base::FilePath& unpack_path) {
400 std::string version_string;
401 manifest.GetStringASCII("version", &version_string);
402 Version version(version_string.c_str());
403
404 base::FilePath path;
405 path = install_directory_.AppendASCII(version.GetString());
406 file_util::CreateDirectory(path.DirName());
407 if (!file_util::Move(unpack_path, path))
408 return false;
409 current_version_ = version;
410 ++install_count_;
411 return true;
412 }
413
414 bool PersistentTestInstaller::GetInstalledFile(const std::string& file,
415 base::FilePath* installed_file) {
416 base::FilePath path;
417 path = install_directory_.AppendASCII(current_version_.GetString());
418 *installed_file = path.Append(base::FilePath::FromUTF8Unsafe(file));
419 return true;
420 }
421
422 TestConfigurator::TestConfigurator()
423 : times_(1), recheck_time_(0), ondemand_time_(0), cus_(NULL) {
424 }
425
426 int TestConfigurator::InitialDelay() { return 0; }
427
428 int TestConfigurator::NextCheckDelay() {
429 // This is called when a new full cycle of checking for updates is going
430 // to happen. In test we normally only test one cycle so it is a good
431 // time to break from the test messageloop Run() method so the test can
432 // finish.
433 if (--times_ <= 0) {
434 MessageLoop::current()->Quit();
435 return 0;
436 }
437
438 // Look for checks to issue in the middle of the loop.
439 for (std::list<CheckAtLoopCount>::iterator
440 i = components_to_check_.begin();
441 i != components_to_check_.end(); ) {
442 if (i->second == times_) {
443 cus_->CheckForUpdateSoon(*i->first);
444 i = components_to_check_.erase(i);
445 } else {
446 ++i;
447 }
448 }
449 return 1;
450 }
451
452 int TestConfigurator::StepDelay() {
453 return 0;
454 }
455
456 int TestConfigurator::MinimumReCheckWait() {
457 return recheck_time_;
458 }
459
460 int TestConfigurator::OnDemandDelay() {
461 return ondemand_time_;
462 }
463
464 GURL TestConfigurator::UpdateUrl(CrxComponent::UrlSource source) {
465 switch (source) {
466 case CrxComponent::BANDAID:
467 return GURL("http://localhost/upd");
468 case CrxComponent::CWS_PUBLIC:
469 return GURL("http://localhost/cws");
470 default:
471 return GURL("http://wronghost/bad");
472 };
473 }
474
475 GURL TestConfigurator::PingUrl() {
476 return GURL("http://localhost2/ping");
477 }
478
479 const char* TestConfigurator::ExtraRequestParams() { return "extra=foo"; }
480
481 size_t TestConfigurator::UrlSizeLimit() { return 256; }
482
483 net::URLRequestContextGetter* TestConfigurator::RequestContext() {
484 return new net::TestURLRequestContextGetter(
485 BrowserThread::GetMessageLoopProxyForThread(BrowserThread::IO));
486 }
487
488 // Don't use the utility process to decode files.
489 bool TestConfigurator::InProcess() { return true; }
490
491 void TestConfigurator::OnEvent(Events event, int extra) { }
492
493 ComponentPatcher* TestConfigurator::CreateComponentPatcher() {
494 return new MockComponentPatcher();
495 }
496
497 bool TestConfigurator::PingsEnabled() const {
498 return true;
499 }
500
501 bool TestConfigurator::DeltasEnabled() const {
502 return true;
503 }
504
505 // Set how many update checks are called, the default value is just once.
506 void TestConfigurator::SetLoopCount(int times) { times_ = times; }
507
508 void TestConfigurator::SetRecheckTime(int seconds) {
509 recheck_time_ = seconds;
510 }
511
512 void TestConfigurator::SetOnDemandTime(int seconds) {
513 ondemand_time_ = seconds;
514 }
515
516 void TestConfigurator::AddComponentToCheck(CrxComponent* com,
517 int at_loop_iter) {
518 components_to_check_.push_back(std::make_pair(com, at_loop_iter));
519 }
520
521 void TestConfigurator::SetComponentUpdateService(ComponentUpdateService* cus) {
522 cus_ = cus;
523 }
524
525 ComponentUpdaterTest::ComponentUpdaterTest()
526 : component_updater_(NULL), test_config_(NULL) {
527 // The component updater instance under test.
528 test_config_ = new TestConfigurator;
529 component_updater_.reset(ComponentUpdateServiceFactory(test_config_));
530 test_config_->SetComponentUpdateService(component_updater_.get());
531 // The test directory is chrome/test/data/components.
532 PathService::Get(chrome::DIR_TEST_DATA, &test_data_dir_);
533 test_data_dir_ = test_data_dir_.AppendASCII("components");
534
535 // Subscribe to all component updater notifications.
536 const int notifications[] = {
537 chrome::NOTIFICATION_COMPONENT_UPDATER_STARTED,
538 chrome::NOTIFICATION_COMPONENT_UPDATER_SLEEPING,
539 chrome::NOTIFICATION_COMPONENT_UPDATE_FOUND,
540 chrome::NOTIFICATION_COMPONENT_UPDATE_READY
541 };
542
543 for (int ix = 0; ix != arraysize(notifications); ++ix) {
544 notification_tracker_.ListenFor(
545 notifications[ix], content::NotificationService::AllSources());
546 }
547 net::URLFetcher::SetEnableInterceptionForTests(true);
548 }
549
550 ComponentUpdaterTest::~ComponentUpdaterTest() {
551 net::URLFetcher::SetEnableInterceptionForTests(false);
552 }
553
554 void ComponentUpdaterTest::TearDown() {
555 xmlCleanupGlobals();
556 }
557
558 ComponentUpdateService* ComponentUpdaterTest::component_updater() {
559 return component_updater_.get();
560 }
561
562 // Makes the full path to a component updater test file.
563 const base::FilePath ComponentUpdaterTest::test_file(const char* file) {
564 return test_data_dir_.AppendASCII(file);
565 }
566
567 TestNotificationTracker& ComponentUpdaterTest::notification_tracker() {
568 return notification_tracker_;
569 }
570
571 TestConfigurator* ComponentUpdaterTest::test_configurator() {
572 return test_config_;
573 }
574
575 ComponentUpdateService::Status ComponentUpdaterTest::RegisterComponent(
576 CrxComponent* com,
577 TestComponents component,
578 const Version& version,
579 TestInstaller* installer) {
580 if (component == kTestComponent_abag) {
581 com->name = "test_abag";
582 com->pk_hash.assign(abag_hash, abag_hash + arraysize(abag_hash));
583 } else if (component == kTestComponent_jebg) {
584 com->name = "test_jebg";
585 com->pk_hash.assign(jebg_hash, jebg_hash + arraysize(jebg_hash));
586 } else {
587 com->name = "test_ihfo";
588 com->pk_hash.assign(ihfo_hash, ihfo_hash + arraysize(ihfo_hash));
589 }
590 com->version = version;
591 com->installer = installer;
592 return component_updater_->RegisterComponent(*com);
593 }
262 594
263 // Verify that our test fixture work and the component updater can 595 // Verify that our test fixture work and the component updater can
264 // be created and destroyed with no side effects. 596 // be created and destroyed with no side effects.
265 TEST_F(ComponentUpdaterTest, VerifyFixture) { 597 TEST_F(ComponentUpdaterTest, VerifyFixture) {
266 EXPECT_TRUE(component_updater() != NULL); 598 EXPECT_TRUE(component_updater() != NULL);
267 EXPECT_EQ(0ul, notification_tracker().size()); 599 EXPECT_EQ(0ul, notification_tracker().size());
268 } 600 }
269 601
270 // Verify that the component updater can be caught in a quick 602 // Verify that the component updater can be caught in a quick
271 // start-shutdown situation. Failure of this test will be a crash. Also 603 // start-shutdown situation. Failure of this test will be a crash. Also
(...skipping 16 matching lines...) Expand all
288 MessageLoop message_loop; 620 MessageLoop message_loop;
289 content::TestBrowserThread ui_thread(BrowserThread::UI, &message_loop); 621 content::TestBrowserThread ui_thread(BrowserThread::UI, &message_loop);
290 content::TestBrowserThread file_thread(BrowserThread::FILE); 622 content::TestBrowserThread file_thread(BrowserThread::FILE);
291 content::TestBrowserThread io_thread(BrowserThread::IO); 623 content::TestBrowserThread io_thread(BrowserThread::IO);
292 624
293 io_thread.StartIOThread(); 625 io_thread.StartIOThread();
294 file_thread.Start(); 626 file_thread.Start();
295 627
296 content::URLLocalHostRequestPrepackagedInterceptor interceptor; 628 content::URLLocalHostRequestPrepackagedInterceptor interceptor;
297 629
630 TestInstaller installer;
298 CrxComponent com; 631 CrxComponent com;
299 EXPECT_EQ(ComponentUpdateService::kOk, 632 EXPECT_EQ(ComponentUpdateService::kOk,
300 RegisterComponent(&com, kTestComponent_abag, Version("1.1"))); 633 RegisterComponent(&com,
634 kTestComponent_abag,
635 Version("1.1"),
636 &installer));
301 637
302 const GURL expected_update_url( 638 const GURL expected_update_url(
303 "http://localhost/upd?extra=foo&x=id%3D" 639 "http://localhost/upd?extra=foo"
304 "abagagagagagagagagagagagagagagag%26v%3D1.1%26uc"); 640 "&x=id%3Dabagagagagagagagagagagagagagagag%26v%3D1.1%26fp%3D%26uc");
305 641
306 interceptor.SetResponse(expected_update_url, 642 interceptor.SetResponse(expected_update_url,
307 test_file("updatecheck_reply_1.xml")); 643 test_file("updatecheck_reply_1.xml"));
308 644
309 // We loop twice, but there are no updates so we expect two sleep messages. 645 // We loop twice, but there are no updates so we expect two sleep messages.
310 test_configurator()->SetLoopCount(2); 646 test_configurator()->SetLoopCount(2);
311 component_updater()->Start(); 647 component_updater()->Start();
312 648
313 ASSERT_EQ(1ul, notification_tracker().size()); 649 ASSERT_EQ(1ul, notification_tracker().size());
314 TestNotificationTracker::Event ev1 = notification_tracker().at(0); 650 TestNotificationTracker::Event ev1 = notification_tracker().at(0);
(...skipping 37 matching lines...) Expand 10 before | Expand all | Expand 10 after
352 EXPECT_EQ(0, static_cast<TestInstaller*>(com.installer)->error()); 688 EXPECT_EQ(0, static_cast<TestInstaller*>(com.installer)->error());
353 EXPECT_EQ(0, static_cast<TestInstaller*>(com.installer)->install_count()); 689 EXPECT_EQ(0, static_cast<TestInstaller*>(com.installer)->install_count());
354 690
355 component_updater()->Stop(); 691 component_updater()->Stop();
356 } 692 }
357 693
358 // Verify that we can check for updates and install one component. Besides 694 // Verify that we can check for updates and install one component. Besides
359 // the notifications above NOTIFICATION_COMPONENT_UPDATE_FOUND and 695 // the notifications above NOTIFICATION_COMPONENT_UPDATE_FOUND and
360 // NOTIFICATION_COMPONENT_UPDATE_READY should have been fired. We do two loops 696 // NOTIFICATION_COMPONENT_UPDATE_READY should have been fired. We do two loops
361 // so the second time around there should be nothing left to do. 697 // so the second time around there should be nothing left to do.
362 // We also check that only 3 network requests are issued: 698 // We also check that only 3 non-ping network requests are issued:
363 // 1- manifest check 699 // 1- manifest check
364 // 2- download crx 700 // 2- download crx
365 // 3- second manifest check. 701 // 3- second manifest check.
366 TEST_F(ComponentUpdaterTest, InstallCrx) { 702 TEST_F(ComponentUpdaterTest, InstallCrx) {
367 MessageLoop message_loop; 703 MessageLoop message_loop;
368 content::TestBrowserThread ui_thread(BrowserThread::UI, &message_loop); 704 content::TestBrowserThread ui_thread(BrowserThread::UI, &message_loop);
369 content::TestBrowserThread file_thread(BrowserThread::FILE); 705 content::TestBrowserThread file_thread(BrowserThread::FILE);
370 content::TestBrowserThread io_thread(BrowserThread::IO); 706 content::TestBrowserThread io_thread(BrowserThread::IO);
371 707
372 io_thread.StartIOThread(); 708 io_thread.StartIOThread();
373 file_thread.Start(); 709 file_thread.Start();
374 710
711 std::map<std::string, std::string> map;
712 map.insert(std::pair<std::string, std::string>("eventtype", "\"3\""));
713 map.insert(std::pair<std::string, std::string>("eventresult", "\"1\""));
714 map.insert(std::pair<std::string, std::string>("previousversion",
715 "\"0.9\""));
716 map.insert(std::pair<std::string, std::string>("nextversion", "\"1.0\""));
717 PingChecker ping_checker(map);
718 URLRequestPostInterceptor post_interceptor(&ping_checker);
375 content::URLLocalHostRequestPrepackagedInterceptor interceptor; 719 content::URLLocalHostRequestPrepackagedInterceptor interceptor;
376 720
721 TestInstaller installer1;
377 CrxComponent com1; 722 CrxComponent com1;
378 RegisterComponent(&com1, kTestComponent_jebg, Version("0.9")); 723 RegisterComponent(&com1, kTestComponent_jebg, Version("0.9"), &installer1);
724 TestInstaller installer2;
379 CrxComponent com2; 725 CrxComponent com2;
380 RegisterComponent(&com2, kTestComponent_abag, Version("2.2")); 726 RegisterComponent(&com2, kTestComponent_abag, Version("2.2"), &installer2);
381 727
382 const GURL expected_update_url_1( 728 const GURL expected_update_url_1(
383 "http://localhost/upd?extra=foo&x=id%3D" 729 "http://localhost/upd?extra=foo"
384 "jebgalgnebhfojomionfpkfelancnnkf%26v%3D0.9%26uc&x=id%3D" 730 "&x=id%3Djebgalgnebhfojomionfpkfelancnnkf%26v%3D0.9%26fp%3D%26uc"
385 "abagagagagagagagagagagagagagagag%26v%3D2.2%26uc"); 731 "&x=id%3Dabagagagagagagagagagagagagagagag%26v%3D2.2%26fp%3D%26uc");
386 732
387 const GURL expected_update_url_2( 733 const GURL expected_update_url_2(
388 "http://localhost/upd?extra=foo&x=id%3D" 734 "http://localhost/upd?extra=foo"
389 "abagagagagagagagagagagagagagagag%26v%3D2.2%26uc&x=id%3D" 735 "&x=id%3Dabagagagagagagagagagagagagagagag%26v%3D2.2%26fp%3D%26uc"
390 "jebgalgnebhfojomionfpkfelancnnkf%26v%3D1.0%26uc"); 736 "&x=id%3Djebgalgnebhfojomionfpkfelancnnkf%26v%3D1.0%26fp%3D%26uc");
391 737
392 interceptor.SetResponse(expected_update_url_1, 738 interceptor.SetResponse(expected_update_url_1,
393 test_file("updatecheck_reply_1.xml")); 739 test_file("updatecheck_reply_1.xml"));
394 interceptor.SetResponse(expected_update_url_2, 740 interceptor.SetResponse(expected_update_url_2,
395 test_file("updatecheck_reply_1.xml")); 741 test_file("updatecheck_reply_1.xml"));
396 interceptor.SetResponse(GURL(expected_crx_url), 742 interceptor.SetResponse(GURL(expected_crx_url),
397 test_file("jebgalgnebhfojomionfpkfelancnnkf.crx")); 743 test_file("jebgalgnebhfojomionfpkfelancnnkf.crx"));
398 744
399 test_configurator()->SetLoopCount(2); 745 test_configurator()->SetLoopCount(2);
400 746
401 component_updater()->Start(); 747 component_updater()->Start();
402 message_loop.Run(); 748 message_loop.Run();
403 749
404 EXPECT_EQ(0, static_cast<TestInstaller*>(com1.installer)->error()); 750 EXPECT_EQ(0, static_cast<TestInstaller*>(com1.installer)->error());
405 EXPECT_EQ(1, static_cast<TestInstaller*>(com1.installer)->install_count()); 751 EXPECT_EQ(1, static_cast<TestInstaller*>(com1.installer)->install_count());
406 EXPECT_EQ(0, static_cast<TestInstaller*>(com2.installer)->error()); 752 EXPECT_EQ(0, static_cast<TestInstaller*>(com2.installer)->error());
407 EXPECT_EQ(0, static_cast<TestInstaller*>(com2.installer)->install_count()); 753 EXPECT_EQ(0, static_cast<TestInstaller*>(com2.installer)->install_count());
408 754
409 EXPECT_EQ(3, interceptor.GetHitCount()); 755 EXPECT_EQ(3, interceptor.GetHitCount());
756 EXPECT_EQ(1, ping_checker.NumHits());
757 EXPECT_EQ(0, ping_checker.NumMisses());
410 758
411 ASSERT_EQ(5ul, notification_tracker().size()); 759 ASSERT_EQ(5ul, notification_tracker().size());
412 760
413 TestNotificationTracker::Event ev1 = notification_tracker().at(1); 761 TestNotificationTracker::Event ev1 = notification_tracker().at(1);
414 EXPECT_EQ(chrome::NOTIFICATION_COMPONENT_UPDATE_FOUND, ev1.type); 762 EXPECT_EQ(chrome::NOTIFICATION_COMPONENT_UPDATE_FOUND, ev1.type);
415 763
416 TestNotificationTracker::Event ev2 = notification_tracker().at(2); 764 TestNotificationTracker::Event ev2 = notification_tracker().at(2);
417 EXPECT_EQ(chrome::NOTIFICATION_COMPONENT_UPDATE_READY, ev2.type); 765 EXPECT_EQ(chrome::NOTIFICATION_COMPONENT_UPDATE_READY, ev2.type);
418 766
419 TestNotificationTracker::Event ev3 = notification_tracker().at(3); 767 TestNotificationTracker::Event ev3 = notification_tracker().at(3);
(...skipping 10 matching lines...) Expand all
430 // checks to different urls, each only containing one component. 778 // checks to different urls, each only containing one component.
431 TEST_F(ComponentUpdaterTest, InstallCrxTwoSources) { 779 TEST_F(ComponentUpdaterTest, InstallCrxTwoSources) {
432 MessageLoop message_loop; 780 MessageLoop message_loop;
433 content::TestBrowserThread ui_thread(BrowserThread::UI, &message_loop); 781 content::TestBrowserThread ui_thread(BrowserThread::UI, &message_loop);
434 content::TestBrowserThread file_thread(BrowserThread::FILE); 782 content::TestBrowserThread file_thread(BrowserThread::FILE);
435 content::TestBrowserThread io_thread(BrowserThread::IO); 783 content::TestBrowserThread io_thread(BrowserThread::IO);
436 784
437 io_thread.StartIOThread(); 785 io_thread.StartIOThread();
438 file_thread.Start(); 786 file_thread.Start();
439 787
788 std::map<std::string, std::string> map;
789 map.insert(std::pair<std::string, std::string>("eventtype", "\"3\""));
790 map.insert(std::pair<std::string, std::string>("eventresult", "\"1\""));
791 map.insert(std::pair<std::string, std::string>("previousversion",
792 "\"0.9\""));
793 map.insert(std::pair<std::string, std::string>("nextversion", "\"1.0\""));
794 PingChecker ping_checker(map);
795 URLRequestPostInterceptor post_interceptor(&ping_checker);
440 content::URLLocalHostRequestPrepackagedInterceptor interceptor; 796 content::URLLocalHostRequestPrepackagedInterceptor interceptor;
441 797
798 TestInstaller installer1;
442 CrxComponent com1; 799 CrxComponent com1;
443 RegisterComponent(&com1, kTestComponent_abag, Version("2.2")); 800 RegisterComponent(&com1, kTestComponent_abag, Version("2.2"), &installer1);
801 TestInstaller installer2;
444 CrxComponent com2; 802 CrxComponent com2;
445 com2.source = CrxComponent::CWS_PUBLIC; 803 com2.source = CrxComponent::CWS_PUBLIC;
446 RegisterComponent(&com2, kTestComponent_jebg, Version("0.9")); 804 RegisterComponent(&com2, kTestComponent_jebg, Version("0.9"), &installer2);
447 805
448 const GURL expected_update_url_1( 806 const GURL expected_update_url_1(
449 "http://localhost/upd?extra=foo&x=id%3D" 807 "http://localhost/upd?extra=foo&x=id%3D"
450 "abagagagagagagagagagagagagagagag%26v%3D2.2%26uc"); 808 "abagagagagagagagagagagagagagagag%26v%3D2.2%26fp%3D%26uc");
451 809
452 const GURL expected_update_url_2( 810 const GURL expected_update_url_2(
453 "http://localhost/cws?extra=foo&x=id%3D" 811 "http://localhost/cws?extra=foo&x=id%3D"
454 "jebgalgnebhfojomionfpkfelancnnkf%26v%3D0.9%26uc"); 812 "jebgalgnebhfojomionfpkfelancnnkf%26v%3D0.9%26fp%3D%26uc");
455 813
456 interceptor.SetResponse(expected_update_url_1, 814 interceptor.SetResponse(expected_update_url_1,
457 test_file("updatecheck_reply_3.xml")); 815 test_file("updatecheck_reply_3.xml"));
458 interceptor.SetResponse(expected_update_url_2, 816 interceptor.SetResponse(expected_update_url_2,
459 test_file("updatecheck_reply_1.xml")); 817 test_file("updatecheck_reply_1.xml"));
460 interceptor.SetResponse(GURL(expected_crx_url), 818 interceptor.SetResponse(GURL(expected_crx_url),
461 test_file("jebgalgnebhfojomionfpkfelancnnkf.crx")); 819 test_file("jebgalgnebhfojomionfpkfelancnnkf.crx"));
462 820
463 test_configurator()->SetLoopCount(3); 821 test_configurator()->SetLoopCount(3);
464 822
465 // We have to set SetRecheckTime to something bigger than 0 or else the 823 // We have to set SetRecheckTime to something bigger than 0 or else the
466 // component updater will keep re-checking the 'abag' component because 824 // component updater will keep re-checking the 'abag' component because
467 // the default source pre-empts the other sources. 825 // the default source pre-empts the other sources.
468 test_configurator()->SetRecheckTime(60*60); 826 test_configurator()->SetRecheckTime(60*60);
469 827
470 component_updater()->Start(); 828 component_updater()->Start();
471 message_loop.Run(); 829 message_loop.Run();
472 830
473 EXPECT_EQ(0, static_cast<TestInstaller*>(com1.installer)->error()); 831 EXPECT_EQ(0, static_cast<TestInstaller*>(com1.installer)->error());
474 EXPECT_EQ(0, static_cast<TestInstaller*>(com1.installer)->install_count()); 832 EXPECT_EQ(0, static_cast<TestInstaller*>(com1.installer)->install_count());
475 EXPECT_EQ(0, static_cast<TestInstaller*>(com2.installer)->error()); 833 EXPECT_EQ(0, static_cast<TestInstaller*>(com2.installer)->error());
476 EXPECT_EQ(1, static_cast<TestInstaller*>(com2.installer)->install_count()); 834 EXPECT_EQ(1, static_cast<TestInstaller*>(com2.installer)->install_count());
477 835
836 EXPECT_EQ(1, ping_checker.NumHits());
837 EXPECT_EQ(0, ping_checker.NumMisses());
478 EXPECT_EQ(3, interceptor.GetHitCount()); 838 EXPECT_EQ(3, interceptor.GetHitCount());
479 839
480 ASSERT_EQ(6ul, notification_tracker().size()); 840 ASSERT_EQ(6ul, notification_tracker().size());
481 841
482 TestNotificationTracker::Event ev0 = notification_tracker().at(1); 842 TestNotificationTracker::Event ev0 = notification_tracker().at(1);
483 EXPECT_EQ(chrome::NOTIFICATION_COMPONENT_UPDATER_SLEEPING, ev0.type); 843 EXPECT_EQ(chrome::NOTIFICATION_COMPONENT_UPDATER_SLEEPING, ev0.type);
484 844
485 TestNotificationTracker::Event ev1 = notification_tracker().at(2); 845 TestNotificationTracker::Event ev1 = notification_tracker().at(2);
486 EXPECT_EQ(chrome::NOTIFICATION_COMPONENT_UPDATE_FOUND, ev1.type); 846 EXPECT_EQ(chrome::NOTIFICATION_COMPONENT_UPDATE_FOUND, ev1.type);
487 847
(...skipping 14 matching lines...) Expand all
502 // version is much higher than of chrome. 862 // version is much higher than of chrome.
503 TEST_F(ComponentUpdaterTest, ProdVersionCheck) { 863 TEST_F(ComponentUpdaterTest, ProdVersionCheck) {
504 MessageLoop message_loop; 864 MessageLoop message_loop;
505 content::TestBrowserThread ui_thread(BrowserThread::UI, &message_loop); 865 content::TestBrowserThread ui_thread(BrowserThread::UI, &message_loop);
506 content::TestBrowserThread file_thread(BrowserThread::FILE); 866 content::TestBrowserThread file_thread(BrowserThread::FILE);
507 content::TestBrowserThread io_thread(BrowserThread::IO); 867 content::TestBrowserThread io_thread(BrowserThread::IO);
508 868
509 io_thread.StartIOThread(); 869 io_thread.StartIOThread();
510 file_thread.Start(); 870 file_thread.Start();
511 871
872 std::map<std::string, std::string> map;
873 PingChecker ping_checker(map);
874 URLRequestPostInterceptor post_interceptor(&ping_checker);
512 content::URLLocalHostRequestPrepackagedInterceptor interceptor; 875 content::URLLocalHostRequestPrepackagedInterceptor interceptor;
513 876
877 TestInstaller installer;
514 CrxComponent com; 878 CrxComponent com;
515 RegisterComponent(&com, kTestComponent_jebg, Version("0.9")); 879 RegisterComponent(&com, kTestComponent_jebg, Version("0.9"), &installer);
516 880
517 const GURL expected_update_url( 881 const GURL expected_update_url(
518 "http://localhost/upd?extra=foo&x=id%3D" 882 "http://localhost/upd?extra=foo&x=id%3D"
519 "jebgalgnebhfojomionfpkfelancnnkf%26v%3D0.9%26uc"); 883 "jebgalgnebhfojomionfpkfelancnnkf%26v%3D0.9%26fp%3D%26uc");
520 884
521 interceptor.SetResponse(expected_update_url, 885 interceptor.SetResponse(expected_update_url,
522 test_file("updatecheck_reply_2.xml")); 886 test_file("updatecheck_reply_2.xml"));
523 interceptor.SetResponse(GURL(expected_crx_url), 887 interceptor.SetResponse(GURL(expected_crx_url),
524 test_file("jebgalgnebhfojomionfpkfelancnnkf.crx")); 888 test_file("jebgalgnebhfojomionfpkfelancnnkf.crx"));
525 889
526 test_configurator()->SetLoopCount(1); 890 test_configurator()->SetLoopCount(1);
527 component_updater()->Start(); 891 component_updater()->Start();
528 message_loop.Run(); 892 message_loop.Run();
529 893
894 EXPECT_EQ(0, ping_checker.NumHits());
895 EXPECT_EQ(0, ping_checker.NumMisses());
530 EXPECT_EQ(1, interceptor.GetHitCount()); 896 EXPECT_EQ(1, interceptor.GetHitCount());
531 EXPECT_EQ(0, static_cast<TestInstaller*>(com.installer)->error()); 897 EXPECT_EQ(0, static_cast<TestInstaller*>(com.installer)->error());
532 EXPECT_EQ(0, static_cast<TestInstaller*>(com.installer)->install_count()); 898 EXPECT_EQ(0, static_cast<TestInstaller*>(com.installer)->install_count());
533 899
534 component_updater()->Stop(); 900 component_updater()->Stop();
535 } 901 }
536 902
537 // Test that a ping for an update check can cause installs. 903 // Test that a ping for an update check can cause installs.
538 // Here is the timeline: 904 // Here is the timeline:
539 // - First loop: we return a reply that indicates no update, so 905 // - First loop: we return a reply that indicates no update, so
540 // nothing happens. 906 // nothing happens.
541 // - We ping. 907 // - We ping.
542 // - This triggers a second loop, which has a reply that triggers an install. 908 // - This triggers a second loop, which has a reply that triggers an install.
543 TEST_F(ComponentUpdaterTest, CheckForUpdateSoon) { 909 TEST_F(ComponentUpdaterTest, CheckForUpdateSoon) {
544 MessageLoop message_loop; 910 MessageLoop message_loop;
545 content::TestBrowserThread ui_thread(BrowserThread::UI, &message_loop); 911 content::TestBrowserThread ui_thread(BrowserThread::UI, &message_loop);
546 content::TestBrowserThread file_thread(BrowserThread::FILE); 912 content::TestBrowserThread file_thread(BrowserThread::FILE);
547 content::TestBrowserThread io_thread(BrowserThread::IO); 913 content::TestBrowserThread io_thread(BrowserThread::IO);
548 914
549 io_thread.StartIOThread(); 915 io_thread.StartIOThread();
550 file_thread.Start(); 916 file_thread.Start();
551 917
918 std::map<std::string, std::string> map;
919 map.insert(std::pair<std::string, std::string>("eventtype", "\"3\""));
920 map.insert(std::pair<std::string, std::string>("eventresult", "\"1\""));
921 map.insert(std::pair<std::string, std::string>("previousversion",
922 "\"0.9\""));
923 map.insert(std::pair<std::string, std::string>("nextversion", "\"1.0\""));
924 PingChecker ping_checker(map);
925 URLRequestPostInterceptor post_interceptor(&ping_checker);
552 content::URLLocalHostRequestPrepackagedInterceptor interceptor; 926 content::URLLocalHostRequestPrepackagedInterceptor interceptor;
553 927
928 TestInstaller installer1;
554 CrxComponent com1; 929 CrxComponent com1;
555 RegisterComponent(&com1, kTestComponent_abag, Version("2.2")); 930 RegisterComponent(&com1, kTestComponent_abag, Version("2.2"), &installer1);
931 TestInstaller installer2;
556 CrxComponent com2; 932 CrxComponent com2;
557 RegisterComponent(&com2, kTestComponent_jebg, Version("0.9")); 933 RegisterComponent(&com2, kTestComponent_jebg, Version("0.9"), &installer2);
558 934
559 const GURL expected_update_url_1( 935 const GURL expected_update_url_1(
560 "http://localhost/upd?extra=foo&x=id%3D" 936 "http://localhost/upd?extra=foo"
561 "abagagagagagagagagagagagagagagag%26v%3D2.2%26uc&x=id%3D" 937 "&x=id%3Dabagagagagagagagagagagagagagagag%26v%3D2.2%26fp%3D%26uc"
562 "jebgalgnebhfojomionfpkfelancnnkf%26v%3D0.9%26uc"); 938 "&x=id%3Djebgalgnebhfojomionfpkfelancnnkf%26v%3D0.9%26fp%3D%26uc");
563 939
564 const GURL expected_update_url_2( 940 const GURL expected_update_url_2(
565 "http://localhost/upd?extra=foo&x=id%3D" 941 "http://localhost/upd?extra=foo"
566 "jebgalgnebhfojomionfpkfelancnnkf%26v%3D0.9%26uc&x=id%3D" 942 "&x=id%3Djebgalgnebhfojomionfpkfelancnnkf%26v%3D0.9%26fp%3D%26uc"
567 "abagagagagagagagagagagagagagagag%26v%3D2.2%26uc"); 943 "&x=id%3Dabagagagagagagagagagagagagagagag%26v%3D2.2%26fp%3D%26uc");
568 944
569 interceptor.SetResponse(expected_update_url_1, 945 interceptor.SetResponse(expected_update_url_1,
570 test_file("updatecheck_reply_empty")); 946 test_file("updatecheck_reply_empty"));
571 interceptor.SetResponse(expected_update_url_2, 947 interceptor.SetResponse(expected_update_url_2,
572 test_file("updatecheck_reply_1.xml")); 948 test_file("updatecheck_reply_1.xml"));
573 interceptor.SetResponse(GURL(expected_crx_url), 949 interceptor.SetResponse(GURL(expected_crx_url),
574 test_file("jebgalgnebhfojomionfpkfelancnnkf.crx")); 950 test_file("jebgalgnebhfojomionfpkfelancnnkf.crx"));
575 // Test success. 951 // Test success.
576 test_configurator()->SetLoopCount(2); 952 test_configurator()->SetLoopCount(2);
577 test_configurator()->AddComponentToCheck(&com2, 1); 953 test_configurator()->AddComponentToCheck(&com2, 1);
(...skipping 28 matching lines...) Expand all
606 test_configurator()->SetOnDemandTime(60 * 60); 982 test_configurator()->SetOnDemandTime(60 * 60);
607 EXPECT_EQ(ComponentUpdateService::kError, 983 EXPECT_EQ(ComponentUpdateService::kError,
608 component_updater()->CheckForUpdateSoon(com2)); 984 component_updater()->CheckForUpdateSoon(com2));
609 // Okay, now reset to 0 for the other tests. 985 // Okay, now reset to 0 for the other tests.
610 test_configurator()->SetOnDemandTime(0); 986 test_configurator()->SetOnDemandTime(0);
611 component_updater()->Stop(); 987 component_updater()->Stop();
612 988
613 // Test a few error cases. NOTE: We don't have callbacks for 989 // Test a few error cases. NOTE: We don't have callbacks for
614 // when the updates failed yet. 990 // when the updates failed yet.
615 const GURL expected_update_url_3( 991 const GURL expected_update_url_3(
616 "http://localhost/upd?extra=foo&x=id%3D" 992 "http://localhost/upd?extra=foo"
617 "jebgalgnebhfojomionfpkfelancnnkf%26v%3D1.0%26uc&x=id%3D" 993 "&x=id%3Djebgalgnebhfojomionfpkfelancnnkf%26v%3D1.0%26fp%3D%26uc"
618 "abagagagagagagagagagagagagagagag%26v%3D2.2%26uc"); 994 "&x=id%3Dabagagagagagagagagagagagagagagag%26v%3D2.2%26fp%3D%26uc");
619 995
620 // No update: error from no server response 996 // No update: error from no server response
621 interceptor.SetResponse(expected_update_url_3, 997 interceptor.SetResponse(expected_update_url_3,
622 test_file("updatecheck_reply_empty")); 998 test_file("updatecheck_reply_empty"));
623 notification_tracker().Reset(); 999 notification_tracker().Reset();
624 test_configurator()->SetLoopCount(1); 1000 test_configurator()->SetLoopCount(1);
625 component_updater()->Start(); 1001 component_updater()->Start();
626 EXPECT_EQ(ComponentUpdateService::kOk, 1002 EXPECT_EQ(ComponentUpdateService::kOk,
627 component_updater()->CheckForUpdateSoon(com2)); 1003 component_updater()->CheckForUpdateSoon(com2));
628 1004
(...skipping 10 matching lines...) Expand all
639 interceptor.SetResponse(expected_update_url_3, 1015 interceptor.SetResponse(expected_update_url_3,
640 test_file("updatecheck_reply_1.xml")); 1016 test_file("updatecheck_reply_1.xml"));
641 notification_tracker().Reset(); 1017 notification_tracker().Reset();
642 test_configurator()->SetLoopCount(1); 1018 test_configurator()->SetLoopCount(1);
643 component_updater()->Start(); 1019 component_updater()->Start();
644 EXPECT_EQ(ComponentUpdateService::kOk, 1020 EXPECT_EQ(ComponentUpdateService::kOk,
645 component_updater()->CheckForUpdateSoon(com2)); 1021 component_updater()->CheckForUpdateSoon(com2));
646 1022
647 message_loop.Run(); 1023 message_loop.Run();
648 1024
1025 EXPECT_EQ(1, ping_checker.NumHits());
1026 EXPECT_EQ(0, ping_checker.NumMisses());
649 ASSERT_EQ(2ul, notification_tracker().size()); 1027 ASSERT_EQ(2ul, notification_tracker().size());
650 ev0 = notification_tracker().at(0); 1028 ev0 = notification_tracker().at(0);
651 EXPECT_EQ(chrome::NOTIFICATION_COMPONENT_UPDATER_STARTED, ev0.type); 1029 EXPECT_EQ(chrome::NOTIFICATION_COMPONENT_UPDATER_STARTED, ev0.type);
652 ev1 = notification_tracker().at(1); 1030 ev1 = notification_tracker().at(1);
653 EXPECT_EQ(chrome::NOTIFICATION_COMPONENT_UPDATER_SLEEPING, ev1.type); 1031 EXPECT_EQ(chrome::NOTIFICATION_COMPONENT_UPDATER_SLEEPING, ev1.type);
654 component_updater()->Stop(); 1032 component_updater()->Stop();
655 } 1033 }
656 1034
657 // Verify that a previously registered component can get re-registered 1035 // Verify that a previously registered component can get re-registered
658 // with a different version. 1036 // with a different version.
659 TEST_F(ComponentUpdaterTest, CheckReRegistration) { 1037 TEST_F(ComponentUpdaterTest, CheckReRegistration) {
660 MessageLoop message_loop; 1038 MessageLoop message_loop;
661 content::TestBrowserThread ui_thread(BrowserThread::UI, &message_loop); 1039 content::TestBrowserThread ui_thread(BrowserThread::UI, &message_loop);
662 content::TestBrowserThread file_thread(BrowserThread::FILE); 1040 content::TestBrowserThread file_thread(BrowserThread::FILE);
663 content::TestBrowserThread io_thread(BrowserThread::IO); 1041 content::TestBrowserThread io_thread(BrowserThread::IO);
664 1042
665 io_thread.StartIOThread(); 1043 io_thread.StartIOThread();
666 file_thread.Start(); 1044 file_thread.Start();
667 1045
1046 std::map<std::string, std::string> map;
1047 map.insert(std::pair<std::string, std::string>("eventtype", "\"3\""));
1048 map.insert(std::pair<std::string, std::string>("eventresult", "\"1\""));
1049 map.insert(std::pair<std::string, std::string>("previousversion",
1050 "\"0.9\""));
1051 map.insert(std::pair<std::string, std::string>("nextversion", "\"1.0\""));
1052 PingChecker ping_checker(map);
1053 URLRequestPostInterceptor post_interceptor(&ping_checker);
668 content::URLLocalHostRequestPrepackagedInterceptor interceptor; 1054 content::URLLocalHostRequestPrepackagedInterceptor interceptor;
669 1055
1056 TestInstaller installer1;
670 CrxComponent com1; 1057 CrxComponent com1;
671 RegisterComponent(&com1, kTestComponent_jebg, Version("0.9")); 1058 RegisterComponent(&com1, kTestComponent_jebg, Version("0.9"), &installer1);
1059 TestInstaller installer2;
672 CrxComponent com2; 1060 CrxComponent com2;
673 RegisterComponent(&com2, kTestComponent_abag, Version("2.2")); 1061 RegisterComponent(&com2, kTestComponent_abag, Version("2.2"), &installer2);
674 1062
675 // Start with 0.9, and update to 1.0 1063 // Start with 0.9, and update to 1.0
676 const GURL expected_update_url_1( 1064 const GURL expected_update_url_1(
677 "http://localhost/upd?extra=foo&x=id%3D" 1065 "http://localhost/upd?extra=foo"
678 "jebgalgnebhfojomionfpkfelancnnkf%26v%3D0.9%26uc&x=id%3D" 1066 "&x=id%3Djebgalgnebhfojomionfpkfelancnnkf%26v%3D0.9%26fp%3D%26uc"
679 "abagagagagagagagagagagagagagagag%26v%3D2.2%26uc"); 1067 "&x=id%3Dabagagagagagagagagagagagagagagag%26v%3D2.2%26fp%3D%26uc");
680 1068
681 const GURL expected_update_url_2( 1069 const GURL expected_update_url_2(
682 "http://localhost/upd?extra=foo&x=id%3D" 1070 "http://localhost/upd?extra=foo"
683 "abagagagagagagagagagagagagagagag%26v%3D2.2%26uc&x=id%3D" 1071 "&x=id%3Dabagagagagagagagagagagagagagagag%26v%3D2.2%26fp%3D%26uc"
684 "jebgalgnebhfojomionfpkfelancnnkf%26v%3D1.0%26uc"); 1072 "&x=id%3Djebgalgnebhfojomionfpkfelancnnkf%26v%3D1.0%26fp%3D%26uc");
685 1073
686 interceptor.SetResponse(expected_update_url_1, 1074 interceptor.SetResponse(expected_update_url_1,
687 test_file("updatecheck_reply_1.xml")); 1075 test_file("updatecheck_reply_1.xml"));
688 interceptor.SetResponse(expected_update_url_2, 1076 interceptor.SetResponse(expected_update_url_2,
689 test_file("updatecheck_reply_1.xml")); 1077 test_file("updatecheck_reply_1.xml"));
690 interceptor.SetResponse(GURL(expected_crx_url), 1078 interceptor.SetResponse(GURL(expected_crx_url),
691 test_file("jebgalgnebhfojomionfpkfelancnnkf.crx")); 1079 test_file("jebgalgnebhfojomionfpkfelancnnkf.crx"));
692 1080
693 // Loop twice to issue two checks: (1) with original 0.9 version 1081 // Loop twice to issue two checks: (1) with original 0.9 version
694 // and (2) with the updated 1.0 version. 1082 // and (2) with the updated 1.0 version.
695 test_configurator()->SetLoopCount(2); 1083 test_configurator()->SetLoopCount(2);
696 1084
697 component_updater()->Start(); 1085 component_updater()->Start();
698 message_loop.Run(); 1086 message_loop.Run();
699 1087
700 EXPECT_EQ(0, static_cast<TestInstaller*>(com1.installer)->error()); 1088 EXPECT_EQ(0, static_cast<TestInstaller*>(com1.installer)->error());
701 EXPECT_EQ(1, static_cast<TestInstaller*>(com1.installer)->install_count()); 1089 EXPECT_EQ(1, static_cast<TestInstaller*>(com1.installer)->install_count());
702 EXPECT_EQ(0, static_cast<TestInstaller*>(com2.installer)->error()); 1090 EXPECT_EQ(0, static_cast<TestInstaller*>(com2.installer)->error());
703 EXPECT_EQ(0, static_cast<TestInstaller*>(com2.installer)->install_count()); 1091 EXPECT_EQ(0, static_cast<TestInstaller*>(com2.installer)->install_count());
704 1092
1093 EXPECT_EQ(1, ping_checker.NumHits());
1094 EXPECT_EQ(0, ping_checker.NumMisses());
705 EXPECT_EQ(3, interceptor.GetHitCount()); 1095 EXPECT_EQ(3, interceptor.GetHitCount());
706 1096
707 ASSERT_EQ(5ul, notification_tracker().size()); 1097 ASSERT_EQ(5ul, notification_tracker().size());
708 1098
709 TestNotificationTracker::Event ev0 = notification_tracker().at(0); 1099 TestNotificationTracker::Event ev0 = notification_tracker().at(0);
710 EXPECT_EQ(chrome::NOTIFICATION_COMPONENT_UPDATER_STARTED, ev0.type); 1100 EXPECT_EQ(chrome::NOTIFICATION_COMPONENT_UPDATER_STARTED, ev0.type);
711 1101
712 TestNotificationTracker::Event ev1 = notification_tracker().at(1); 1102 TestNotificationTracker::Event ev1 = notification_tracker().at(1);
713 EXPECT_EQ(chrome::NOTIFICATION_COMPONENT_UPDATE_FOUND, ev1.type); 1103 EXPECT_EQ(chrome::NOTIFICATION_COMPONENT_UPDATE_FOUND, ev1.type);
714 1104
715 TestNotificationTracker::Event ev2 = notification_tracker().at(2); 1105 TestNotificationTracker::Event ev2 = notification_tracker().at(2);
716 EXPECT_EQ(chrome::NOTIFICATION_COMPONENT_UPDATE_READY, ev2.type); 1106 EXPECT_EQ(chrome::NOTIFICATION_COMPONENT_UPDATE_READY, ev2.type);
717 1107
718 TestNotificationTracker::Event ev3 = notification_tracker().at(3); 1108 TestNotificationTracker::Event ev3 = notification_tracker().at(3);
719 EXPECT_EQ(chrome::NOTIFICATION_COMPONENT_UPDATER_SLEEPING, ev3.type); 1109 EXPECT_EQ(chrome::NOTIFICATION_COMPONENT_UPDATER_SLEEPING, ev3.type);
720 1110
721 TestNotificationTracker::Event ev4 = notification_tracker().at(4); 1111 TestNotificationTracker::Event ev4 = notification_tracker().at(4);
722 EXPECT_EQ(chrome::NOTIFICATION_COMPONENT_UPDATER_SLEEPING, ev4.type); 1112 EXPECT_EQ(chrome::NOTIFICATION_COMPONENT_UPDATER_SLEEPING, ev4.type);
723 1113
724 // Now re-register, pretending to be an even newer version (2.2) 1114 // Now re-register, pretending to be an even newer version (2.2)
1115 TestInstaller installer3;
725 component_updater()->Stop(); 1116 component_updater()->Stop();
726 EXPECT_EQ(ComponentUpdateService::kReplaced, 1117 EXPECT_EQ(ComponentUpdateService::kReplaced,
727 RegisterComponent(&com1, kTestComponent_jebg, Version("2.2"))); 1118 RegisterComponent(&com1,
1119 kTestComponent_jebg,
1120 Version("2.2"),
1121 &installer3));
728 1122
729 // Check that we send out 2.2 as our version. 1123 // Check that we send out 2.2 as our version.
730 // Interceptor's hit count should go up by 1. 1124 // Interceptor's hit count should go up by 1.
731 const GURL expected_update_url_3( 1125 const GURL expected_update_url_3(
732 "http://localhost/upd?extra=foo&x=id%3D" 1126 "http://localhost/upd?extra=foo"
733 "jebgalgnebhfojomionfpkfelancnnkf%26v%3D2.2%26uc&x=id%3D" 1127 "&x=id%3Djebgalgnebhfojomionfpkfelancnnkf%26v%3D2.2%26fp%3D%26uc"
734 "abagagagagagagagagagagagagagagag%26v%3D2.2%26uc"); 1128 "&x=id%3Dabagagagagagagagagagagagagagagag%26v%3D2.2%26fp%3D%26uc");
735 1129
736 interceptor.SetResponse(expected_update_url_3, 1130 interceptor.SetResponse(expected_update_url_3,
737 test_file("updatecheck_reply_1.xml")); 1131 test_file("updatecheck_reply_1.xml"));
738 1132
739 notification_tracker().Reset(); 1133 notification_tracker().Reset();
740 1134
741 // Loop once just to notice the check happening with the re-register version. 1135 // Loop once just to notice the check happening with the re-register version.
742 test_configurator()->SetLoopCount(1); 1136 test_configurator()->SetLoopCount(1);
743 component_updater()->Start(); 1137 component_updater()->Start();
744 message_loop.Run(); 1138 message_loop.Run();
745 1139
746 ASSERT_EQ(2ul, notification_tracker().size()); 1140 ASSERT_EQ(2ul, notification_tracker().size());
747 1141
748 ev0 = notification_tracker().at(0); 1142 ev0 = notification_tracker().at(0);
749 EXPECT_EQ(chrome::NOTIFICATION_COMPONENT_UPDATER_STARTED, ev0.type); 1143 EXPECT_EQ(chrome::NOTIFICATION_COMPONENT_UPDATER_STARTED, ev0.type);
750 1144
751 ev1 = notification_tracker().at(1); 1145 ev1 = notification_tracker().at(1);
752 EXPECT_EQ(chrome::NOTIFICATION_COMPONENT_UPDATER_SLEEPING, ev1.type); 1146 EXPECT_EQ(chrome::NOTIFICATION_COMPONENT_UPDATER_SLEEPING, ev1.type);
753 1147
754 EXPECT_EQ(4, interceptor.GetHitCount()); 1148 EXPECT_EQ(4, interceptor.GetHitCount());
755 1149
756 // The test harness's Register() function creates a new installer, 1150 // We created a new installer, so the counts go back to 0.
757 // so the counts go back to 0.
758 EXPECT_EQ(0, static_cast<TestInstaller*>(com1.installer)->error()); 1151 EXPECT_EQ(0, static_cast<TestInstaller*>(com1.installer)->error());
759 EXPECT_EQ(0, static_cast<TestInstaller*>(com1.installer)->install_count()); 1152 EXPECT_EQ(0, static_cast<TestInstaller*>(com1.installer)->install_count());
760 EXPECT_EQ(0, static_cast<TestInstaller*>(com2.installer)->error()); 1153 EXPECT_EQ(0, static_cast<TestInstaller*>(com2.installer)->error());
761 EXPECT_EQ(0, static_cast<TestInstaller*>(com2.installer)->install_count()); 1154 EXPECT_EQ(0, static_cast<TestInstaller*>(com2.installer)->install_count());
762 1155
763 component_updater()->Stop(); 1156 component_updater()->Stop();
764 } 1157 }
1158
1159 // Verify that we can download and install a component and a differential
1160 // update to that component. We do three loops; the final loop should do
1161 // nothing.
1162 // We also check that exactly 5 non-ping network requests are issued:
1163 // 1- update check (response: v1 available)
1164 // 2- download crx (v1)
1165 // 3- update check (response: v2 available)
1166 // 4- download differential crx (v1 to v2)
1167 // 5- update check (response: no further update available)
1168 // There should be two pings, one for each update. The second will bear a
1169 // diffresult=1, while the first will not.
1170 TEST_F(ComponentUpdaterTest, DifferentialUpdate) {
1171 MessageLoop message_loop;
1172 content::TestBrowserThread ui_thread(BrowserThread::UI, &message_loop);
1173 content::TestBrowserThread file_thread(BrowserThread::FILE);
1174 content::TestBrowserThread io_thread(BrowserThread::IO);
1175
1176 io_thread.StartIOThread();
1177 file_thread.Start();
1178
1179 std::map<std::string, std::string> map;
1180 map.insert(std::pair<std::string, std::string>("eventtype", "\"3\""));
1181 map.insert(std::pair<std::string, std::string>("eventresult", "\"1\""));
1182 map.insert(std::pair<std::string, std::string>("diffresult", "\"1\""));
1183 PingChecker ping_checker(map);
1184 URLRequestPostInterceptor post_interceptor(&ping_checker);
1185 content::URLLocalHostRequestPrepackagedInterceptor interceptor;
1186
1187 PersistentTestInstaller installer;
1188 CrxComponent com;
1189 RegisterComponent(&com, kTestComponent_ihfo, Version("0.0"), &installer);
1190
1191 const GURL expected_update_url_0(
1192 "http://localhost/upd?extra=foo"
1193 "&x=id%3Dihfokbkgjpifnbbojhneepfflplebdkc%26v%3D0.0%26fp%3D%26uc");
1194 const GURL expected_update_url_1(
1195 "http://localhost/upd?extra=foo"
1196 "&x=id%3Dihfokbkgjpifnbbojhneepfflplebdkc%26v%3D1.0%26fp%3D1%26uc");
1197 const GURL expected_update_url_2(
1198 "http://localhost/upd?extra=foo"
1199 "&x=id%3Dihfokbkgjpifnbbojhneepfflplebdkc%26v%3D2.0%26fp%3Df22%26uc");
1200 const GURL expected_crx_url_1(
1201 "http://localhost/download/ihfokbkgjpifnbbojhneepfflplebdkc_1.crx");
1202 const GURL expected_crx_url_1_diff_2(
1203 "http://localhost/download/ihfokbkgjpifnbbojhneepfflplebdkc_1to2.crx");
1204
1205 interceptor.SetResponse(expected_update_url_0,
1206 test_file("updatecheck_diff_reply_1.xml"));
1207 interceptor.SetResponse(expected_update_url_1,
1208 test_file("updatecheck_diff_reply_2.xml"));
1209 interceptor.SetResponse(expected_update_url_2,
1210 test_file("updatecheck_diff_reply_3.xml"));
1211 interceptor.SetResponse(expected_crx_url_1,
1212 test_file("ihfokbkgjpifnbbojhneepfflplebdkc_1.crx"));
1213 interceptor.SetResponse(
1214 expected_crx_url_1_diff_2,
1215 test_file("ihfokbkgjpifnbbojhneepfflplebdkc_1to2.crx"));
1216
1217 test_configurator()->SetLoopCount(3);
1218
1219 component_updater()->Start();
1220 message_loop.Run();
1221
1222 EXPECT_EQ(0, static_cast<TestInstaller*>(com.installer)->error());
1223 EXPECT_EQ(2, static_cast<TestInstaller*>(com.installer)->install_count());
1224
1225 EXPECT_EQ(1, ping_checker.NumHits());
1226 EXPECT_EQ(1, ping_checker.NumMisses());
1227 EXPECT_EQ(5, interceptor.GetHitCount());
1228
1229 component_updater()->Stop();
1230 }
1231
1232 // Verify that component installation falls back to downloading and installing
1233 // a full update if the differential update fails (in this case, because the
1234 // installer does not know about the existing files). We do two loops; the final
1235 // loop should do nothing.
1236 // We also check that exactly 4 non-ping network requests are issued:
1237 // 1- update check (loop 1)
1238 // 2- download differential crx
1239 // 3- download full crx
1240 // 4- update check (loop 2 - no update available)
1241 // There should be one ping for the first attempted update.
1242 TEST_F(ComponentUpdaterTest, DifferentialUpdateFails) {
1243 MessageLoop message_loop;
1244 content::TestBrowserThread ui_thread(BrowserThread::UI, &message_loop);
1245 content::TestBrowserThread file_thread(BrowserThread::FILE);
1246 content::TestBrowserThread io_thread(BrowserThread::IO);
1247
1248 io_thread.StartIOThread();
1249 file_thread.Start();
1250
1251 std::map<std::string, std::string> map;
1252 map.insert(std::pair<std::string, std::string>("eventtype", "\"3\""));
1253 map.insert(std::pair<std::string, std::string>("eventresult", "\"1\""));
1254 map.insert(std::pair<std::string, std::string>("diffresult", "\"0\""));
1255 map.insert(std::pair<std::string, std::string>("differrorcode", "\"12\""));
1256 PingChecker ping_checker(map);
1257 URLRequestPostInterceptor post_interceptor(&ping_checker);
1258 content::URLLocalHostRequestPrepackagedInterceptor interceptor;
1259
1260 TestInstaller installer;
1261 CrxComponent com;
1262 RegisterComponent(&com, kTestComponent_ihfo, Version("1.0"), &installer);
1263
1264 const GURL expected_update_url_1(
1265 "http://localhost/upd?extra=foo"
1266 "&x=id%3Dihfokbkgjpifnbbojhneepfflplebdkc%26v%3D1.0%26fp%3D%26uc");
1267 const GURL expected_update_url_2(
1268 "http://localhost/upd?extra=foo"
1269 "&x=id%3Dihfokbkgjpifnbbojhneepfflplebdkc%26v%3D2.0%26fp%3Df22%26uc");
1270 const GURL expected_crx_url_1(
1271 "http://localhost/download/ihfokbkgjpifnbbojhneepfflplebdkc_1.crx");
1272 const GURL expected_crx_url_1_diff_2(
1273 "http://localhost/download/ihfokbkgjpifnbbojhneepfflplebdkc_1to2.crx");
1274 const GURL expected_crx_url_2(
1275 "http://localhost/download/ihfokbkgjpifnbbojhneepfflplebdkc_2.crx");
1276
1277 interceptor.SetResponse(expected_update_url_1,
1278 test_file("updatecheck_diff_reply_2.xml"));
1279 interceptor.SetResponse(expected_update_url_2,
1280 test_file("updatecheck_diff_reply_3.xml"));
1281 interceptor.SetResponse(expected_crx_url_1,
1282 test_file("ihfokbkgjpifnbbojhneepfflplebdkc_1.crx"));
1283 interceptor.SetResponse(
1284 expected_crx_url_1_diff_2,
1285 test_file("ihfokbkgjpifnbbojhneepfflplebdkc_1to2.crx"));
1286 interceptor.SetResponse(expected_crx_url_2,
1287 test_file("ihfokbkgjpifnbbojhneepfflplebdkc_2.crx"));
1288
1289 test_configurator()->SetLoopCount(2);
1290
1291 component_updater()->Start();
1292 message_loop.Run();
1293
1294 // A failed differential update does not count as a failed install.
1295 EXPECT_EQ(0, static_cast<TestInstaller*>(com.installer)->error());
1296 EXPECT_EQ(1, static_cast<TestInstaller*>(com.installer)->install_count());
1297
1298 EXPECT_EQ(1, ping_checker.NumHits());
1299 EXPECT_EQ(0, ping_checker.NumMisses());
1300 EXPECT_EQ(4, interceptor.GetHitCount());
1301
1302 component_updater()->Stop();
1303 }
1304
1305 // Verify that a failed installation causes an install failure ping.
1306 TEST_F(ComponentUpdaterTest, CheckFailedInstallPing) {
1307 MessageLoop message_loop;
1308 content::TestBrowserThread ui_thread(BrowserThread::UI, &message_loop);
1309 content::TestBrowserThread file_thread(BrowserThread::FILE);
1310 content::TestBrowserThread io_thread(BrowserThread::IO);
1311
1312 io_thread.StartIOThread();
1313 file_thread.Start();
1314
1315 std::map<std::string, std::string> map;
1316 map.insert(std::pair<std::string, std::string>("eventtype", "\"3\""));
1317 map.insert(std::pair<std::string, std::string>("eventresult", "\"0\""));
1318 map.insert(std::pair<std::string, std::string>("errorcode", "\"9\""));
1319 map.insert(std::pair<std::string, std::string>("previousversion",
1320 "\"0.9\""));
1321 map.insert(std::pair<std::string, std::string>("nextversion", "\"1.0\""));
1322 PingChecker ping_checker(map);
1323 URLRequestPostInterceptor post_interceptor(&ping_checker);
1324 content::URLLocalHostRequestPrepackagedInterceptor interceptor;
1325
1326 // This test installer reports installation failure.
1327 class : public TestInstaller {
1328 virtual bool Install(const base::DictionaryValue& manifest,
1329 const base::FilePath& unpack_path) OVERRIDE {
1330 ++install_count_;
1331 file_util::Delete(unpack_path, true);
1332 return false;
1333 }
1334 } installer;
1335
1336 CrxComponent com;
1337 RegisterComponent(&com, kTestComponent_jebg, Version("0.9"), &installer);
1338
1339 // Start with 0.9, and attempt update to 1.0
1340 const GURL expected_update_url_1(
1341 "http://localhost/upd?extra=foo"
1342 "&x=id%3Djebgalgnebhfojomionfpkfelancnnkf%26v%3D0.9%26fp%3D%26uc");
1343
1344 interceptor.SetResponse(expected_update_url_1,
1345 test_file("updatecheck_reply_1.xml"));
1346 interceptor.SetResponse(GURL(expected_crx_url),
1347 test_file("jebgalgnebhfojomionfpkfelancnnkf.crx"));
1348
1349 // Loop twice to issue two checks: (1) with original 0.9 version
1350 // and (2), which should retry with 0.9.
1351 test_configurator()->SetLoopCount(2);
1352 component_updater()->Start();
1353 message_loop.Run();
1354
1355 // Loop once more, but expect no ping because a noupdate response is issued.
1356 // This is necessary to clear out the fire-and-forget ping from the previous
1357 // iteration.
1358 interceptor.SetResponse(expected_update_url_1,
1359 test_file("updatecheck_reply_noupdate.xml"));
1360 test_configurator()->SetLoopCount(1);
1361 component_updater()->Start();
1362 message_loop.Run();
1363
1364 EXPECT_EQ(0, static_cast<TestInstaller*>(com.installer)->error());
1365 EXPECT_EQ(2, static_cast<TestInstaller*>(com.installer)->install_count());
1366
1367 EXPECT_EQ(2, ping_checker.NumHits());
1368 EXPECT_EQ(0, ping_checker.NumMisses());
1369 EXPECT_EQ(5, interceptor.GetHitCount());
1370
1371 component_updater()->Stop();
1372 }
1373
1374 // Verify that we successfully propagate a patcher error.
1375 // ihfokbkgjpifnbbojhneepfflplebdkc_1to2_bad.crx contains an incorrect
1376 // patching instruction that should fail.
1377 TEST_F(ComponentUpdaterTest, DifferentialUpdateFailErrorcode) {
1378 MessageLoop message_loop;
1379 content::TestBrowserThread ui_thread(BrowserThread::UI, &message_loop);
1380 content::TestBrowserThread file_thread(BrowserThread::FILE);
1381 content::TestBrowserThread io_thread(BrowserThread::IO);
1382
1383 io_thread.StartIOThread();
1384 file_thread.Start();
1385
1386 std::map<std::string, std::string> map;
1387 map.insert(std::pair<std::string, std::string>("eventtype", "\"3\""));
1388 map.insert(std::pair<std::string, std::string>("eventresult", "\"1\""));
1389 map.insert(std::pair<std::string, std::string>("diffresult", "\"0\""));
1390 map.insert(std::pair<std::string, std::string>("differrorcode", "\"14\""));
1391 map.insert(std::pair<std::string, std::string>("diffextracode1", "\"305\""));
1392 PingChecker ping_checker(map);
1393 URLRequestPostInterceptor post_interceptor(&ping_checker);
1394 content::URLLocalHostRequestPrepackagedInterceptor interceptor;
1395
1396 PersistentTestInstaller installer;
1397 CrxComponent com;
1398 RegisterComponent(&com, kTestComponent_ihfo, Version("0.0"), &installer);
1399
1400 const GURL expected_update_url_0(
1401 "http://localhost/upd?extra=foo"
1402 "&x=id%3Dihfokbkgjpifnbbojhneepfflplebdkc%26v%3D0.0%26fp%3D%26uc");
1403 const GURL expected_update_url_1(
1404 "http://localhost/upd?extra=foo"
1405 "&x=id%3Dihfokbkgjpifnbbojhneepfflplebdkc%26v%3D1.0%26fp%3D1%26uc");
1406 const GURL expected_update_url_2(
1407 "http://localhost/upd?extra=foo"
1408 "&x=id%3Dihfokbkgjpifnbbojhneepfflplebdkc%26v%3D2.0%26fp%3Df22%26uc");
1409 const GURL expected_crx_url_1(
1410 "http://localhost/download/ihfokbkgjpifnbbojhneepfflplebdkc_1.crx");
1411 const GURL expected_crx_url_1_diff_2(
1412 "http://localhost/download/ihfokbkgjpifnbbojhneepfflplebdkc_1to2.crx");
1413 const GURL expected_crx_url_2(
1414 "http://localhost/download/ihfokbkgjpifnbbojhneepfflplebdkc_2.crx");
1415
1416 interceptor.SetResponse(expected_update_url_0,
1417 test_file("updatecheck_diff_reply_1.xml"));
1418 interceptor.SetResponse(expected_update_url_1,
1419 test_file("updatecheck_diff_reply_2.xml"));
1420 interceptor.SetResponse(expected_update_url_2,
1421 test_file("updatecheck_diff_reply_3.xml"));
1422 interceptor.SetResponse(expected_crx_url_1,
1423 test_file("ihfokbkgjpifnbbojhneepfflplebdkc_1.crx"));
1424 interceptor.SetResponse(
1425 expected_crx_url_1_diff_2,
1426 test_file("ihfokbkgjpifnbbojhneepfflplebdkc_1to2_bad.crx"));
1427 interceptor.SetResponse(expected_crx_url_2,
1428 test_file("ihfokbkgjpifnbbojhneepfflplebdkc_2.crx"));
1429
1430 test_configurator()->SetLoopCount(3);
1431
1432 component_updater()->Start();
1433 message_loop.Run();
1434
1435 EXPECT_EQ(0, static_cast<TestInstaller*>(com.installer)->error());
1436 EXPECT_EQ(2, static_cast<TestInstaller*>(com.installer)->install_count());
1437
1438 EXPECT_EQ(1, ping_checker.NumHits());
1439 EXPECT_EQ(1, ping_checker.NumMisses());
1440 EXPECT_EQ(6, interceptor.GetHitCount());
1441
1442 component_updater()->Stop();
1443 }
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698