OLD | NEW |
1 #!/usr/bin/python | 1 #!/usr/bin/python |
2 # Copyright (c) 2010 The Chromium OS Authors. All rights reserved. | 2 # Copyright (c) 2010 The Chromium OS Authors. All rights reserved. |
3 # Use of this source code is governed by a BSD-style license that can be | 3 # Use of this source code is governed by a BSD-style license that can be |
4 # found in the LICENSE file. | 4 # found in the LICENSE file. |
5 | 5 |
6 import copy | 6 import copy |
7 import mox | 7 import mox |
8 import os | 8 import os |
9 import multiprocessing | 9 import multiprocessing |
10 import shutil | 10 import shutil |
(...skipping 380 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
391 self.pkgindex.WriteToNamedTemporaryFile().AndReturn(fake_pkgs_file) | 391 self.pkgindex.WriteToNamedTemporaryFile().AndReturn(fake_pkgs_file) |
392 | 392 |
393 def tearDown(self): | 393 def tearDown(self): |
394 self.mox.UnsetStubs() | 394 self.mox.UnsetStubs() |
395 self.mox.VerifyAll() | 395 self.mox.VerifyAll() |
396 | 396 |
397 def doRsyncUpload(self, suffix): | 397 def doRsyncUpload(self, suffix): |
398 self.mox.StubOutWithMock(prebuilt, '_RetryRun') | 398 self.mox.StubOutWithMock(prebuilt, '_RetryRun') |
399 remote_path = '/dir/%s' % suffix.rstrip('/') | 399 remote_path = '/dir/%s' % suffix.rstrip('/') |
400 full_remote_path = 'chromeos-prebuilt:%s' % remote_path | 400 full_remote_path = 'chromeos-prebuilt:%s' % remote_path |
401 cmds = ['ssh chromeos-prebuilt mkdir -p %s' % remote_path, | 401 cmds = [['ssh', 'chromeos-prebuilt', 'mkdir', '-p',remote_path], |
402 'rsync -av --chmod=a+r fake %s/Packages' % full_remote_path, | 402 ['rsync', '-av', '--chmod=a+r', 'fake', |
403 'rsync -Rav private.tbz2 %s/' % full_remote_path] | 403 full_remote_path + '/Packages'], |
| 404 ['rsync', '-Rav', 'private.tbz2', full_remote_path + '/']] |
404 for cmd in cmds: | 405 for cmd in cmds: |
405 prebuilt._RetryRun(cmd, shell=True, cwd='/packages').AndReturn(True) | 406 prebuilt._RetryRun(cmd, cwd='/packages').AndReturn(True) |
406 self.mox.ReplayAll() | 407 self.mox.ReplayAll() |
407 uri = self.pkgindex.header['URI'] | 408 uri = self.pkgindex.header['URI'] |
408 uploader = prebuilt.PrebuiltUploader('chromeos-prebuilt:/dir', | 409 uploader = prebuilt.PrebuiltUploader('chromeos-prebuilt:/dir', |
409 'public-read', uri, []) | 410 'public-read', uri, []) |
410 uploader._UploadPrebuilt('/packages', suffix) | 411 uploader._UploadPrebuilt('/packages', suffix) |
411 | 412 |
412 def testSuccessfulGsUpload(self): | 413 def testSuccessfulGsUpload(self): |
413 uploads = {'/packages/private.tbz2': 'gs://foo/private.tbz2'} | 414 uploads = {'/packages/private.tbz2': 'gs://foo/private.tbz2'} |
414 self.mox.StubOutWithMock(prebuilt, 'GenerateUploadDict') | 415 self.mox.StubOutWithMock(prebuilt, 'GenerateUploadDict') |
415 prebuilt.GenerateUploadDict('/packages', 'gs://foo/suffix', | 416 prebuilt.GenerateUploadDict('/packages', 'gs://foo/suffix', |
(...skipping 19 matching lines...) Expand all Loading... |
435 def setUp(self): | 436 def setUp(self): |
436 self.mox = mox.Mox() | 437 self.mox = mox.Mox() |
437 self.mox.StubOutWithMock(prebuilt, 'DeterminePrebuiltConfFile') | 438 self.mox.StubOutWithMock(prebuilt, 'DeterminePrebuiltConfFile') |
438 self.mox.StubOutWithMock(prebuilt, 'RevGitFile') | 439 self.mox.StubOutWithMock(prebuilt, 'RevGitFile') |
439 self.mox.StubOutWithMock(prebuilt, 'UpdateBinhostConfFile') | 440 self.mox.StubOutWithMock(prebuilt, 'UpdateBinhostConfFile') |
440 self.build_path = '/trunk' | 441 self.build_path = '/trunk' |
441 self.upload_location = 'gs://upload/' | 442 self.upload_location = 'gs://upload/' |
442 self.version = '1' | 443 self.version = '1' |
443 self.binhost = 'http://prebuilt/' | 444 self.binhost = 'http://prebuilt/' |
444 self.key = 'PORTAGE_BINHOST' | 445 self.key = 'PORTAGE_BINHOST' |
445 self.uploader = prebuilt.PrebuiltUploader(self.upload_location, | 446 self.uploader = prebuilt.PrebuiltUploader( |
446 'public-read', self.binhost, []) | 447 self.upload_location, 'public-read', self.binhost, []) |
447 self.mox.StubOutWithMock(self.uploader, '_UploadPrebuilt') | 448 self.mox.StubOutWithMock(self.uploader, '_UploadPrebuilt') |
448 | 449 |
449 def tearDown(self): | 450 def tearDown(self): |
450 self.mox.UnsetStubs() | 451 self.mox.UnsetStubs() |
451 self.mox.VerifyAll() | 452 self.mox.VerifyAll() |
452 | 453 |
453 def testSyncHostPrebuilts(self): | 454 def testSyncHostPrebuilts(self): |
454 package_path = os.path.join(self.build_path, | 455 package_path = os.path.join(self.build_path, |
455 prebuilt._HOST_PACKAGES_PATH) | 456 prebuilt._HOST_PACKAGES_PATH) |
456 url_suffix = prebuilt._REL_HOST_PATH % {'version': self.version, | 457 url_suffix = prebuilt._REL_HOST_PATH % {'version': self.version, |
(...skipping 84 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
541 options.sync_binhost_conf) | 542 options.sync_binhost_conf) |
542 self.mox.StubOutWithMock(prebuilt.PrebuiltUploader, '_SyncBoardPrebuilts') | 543 self.mox.StubOutWithMock(prebuilt.PrebuiltUploader, '_SyncBoardPrebuilts') |
543 prebuilt.PrebuiltUploader._SyncBoardPrebuilts(options.board, | 544 prebuilt.PrebuiltUploader._SyncBoardPrebuilts(options.board, |
544 options.build_path, mox.IgnoreArg(), options.key, options.git_sync, | 545 options.build_path, mox.IgnoreArg(), options.key, options.git_sync, |
545 options.sync_binhost_conf, options.upload_board_tarball) | 546 options.sync_binhost_conf, options.upload_board_tarball) |
546 self.mox.ReplayAll() | 547 self.mox.ReplayAll() |
547 prebuilt.main() | 548 prebuilt.main() |
548 | 549 |
549 if __name__ == '__main__': | 550 if __name__ == '__main__': |
550 unittest.main() | 551 unittest.main() |
OLD | NEW |