Index: slave/skia_slave_scripts/utils/old_gs_utils_test.py |
diff --git a/slave/skia_slave_scripts/utils/old_gs_utils_test.py b/slave/skia_slave_scripts/utils/old_gs_utils_test.py |
deleted file mode 100644 |
index 8e24e9841100e9241436493007abdb8a153bd8cb..0000000000000000000000000000000000000000 |
--- a/slave/skia_slave_scripts/utils/old_gs_utils_test.py |
+++ /dev/null |
@@ -1,225 +0,0 @@ |
-#!/usr/bin/env python |
-# Copyright (c) 2012 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. |
- |
-"""Tests for module old_gs_utils.""" |
- |
-import __builtin__ |
-import os |
-import posixpath |
-import shutil |
-import sys |
-import tempfile |
-import time |
- |
-# Appending to PYTHONPATH to find common. |
-BUILDBOT_PATH = os.path.realpath(os.path.join( |
- os.path.dirname(os.path.abspath(__file__)), os.pardir, os.pardir, os.pardir) |
-) |
-sys.path.append(os.path.join(BUILDBOT_PATH, 'common')) |
-sys.path.append(os.path.join(BUILDBOT_PATH, 'third_party', 'chromium_buildbot', |
- 'scripts')) |
-sys.path.append(os.path.join(BUILDBOT_PATH, 'third_party', 'chromium_buildbot', |
- 'scripts', 'common')) |
-sys.path.append(os.path.join(BUILDBOT_PATH, 'third_party', 'chromium_buildbot', |
- 'site_config')) |
-sys.path.append(os.path.join(BUILDBOT_PATH, 'third_party', 'chromium_buildbot', |
- 'third_party', 'twisted_10_2')) |
- |
-from py.utils import shell_utils |
-from slave import slave_utils |
-import old_gs_utils as gs_utils |
-import unittest |
- |
- |
-GSUTIL_LOCATION = slave_utils.GSUtilSetup() |
- |
-TEST_TIMESTAMP = '1354128965' |
-TEST_TIMESTAMP_2 = '1354128985' |
- |
- |
-class TestGSUtils(unittest.TestCase): |
- |
- def setUp(self): |
- self._expected_commands = [] |
- self._test_temp_file = None |
- self._test_gs_base = None |
- self._test_destdir = None |
- self._test_gs_acl = None |
- self._local_tempdir = tempfile.mkdtemp() |
- |
- def _MockCommand(command): |
- self.assertEquals(self._expected_commands.pop(0), ' '.join(command)) |
- |
- def _MockGSUtilFileCopy(filename, gs_base, subdir, gs_acl): |
- self.assertEquals(self._test_temp_file, filename) |
- self.assertEquals(self._test_gs_base, gs_base) |
- self.assertEquals(self._test_destdir, subdir) |
- self.assertEquals(self._test_gs_acl, gs_acl) |
- |
- def _MockGSUtilDownloadFile(src, dst): |
- pass |
- |
- self._original_bash_run_command = shell_utils.run |
- shell_utils.run = _MockCommand |
- |
- self._original_gsutil_file_copy = slave_utils.GSUtilCopyFile |
- slave_utils.GSUtilCopyFile = _MockGSUtilFileCopy |
- |
- self._original_gsutil_download_file = slave_utils.GSUtilDownloadFile |
- slave_utils.GSUtilDownloadFile = _MockGSUtilDownloadFile |
- |
- self._original_file = __builtin__.open |
- |
- def tearDown(self): |
- self.assertEquals(len(self._expected_commands), 0) |
- shell_utils.run = self._original_bash_run_command |
- slave_utils.GSUtilCopyFile = self._original_gsutil_file_copy |
- slave_utils.GSUtilDownloadFile = self._original_gsutil_download_file |
- __builtin__.open = self._original_file |
- shutil.rmtree(self._local_tempdir) |
- |
- def test_delete_storage_object(self): |
- self._expected_commands = [('%s rm -R superman' % GSUTIL_LOCATION)] |
- gs_utils.delete_storage_object('superman') |
- |
- def test_upload_file(self): |
- self._expected_commands = [( |
- '%s cp -a public /fake/local/src/path gs://fake/remote/dest/path' % |
- GSUTIL_LOCATION)] |
- gs_utils.upload_file( |
- local_src_path='/fake/local/src/path', |
- remote_dest_path='gs://fake/remote/dest/path', |
- gs_acl='public') |
- |
- def test_upload_dir_contents_empty(self): |
- self._expected_commands = [] |
- gs_utils.upload_dir_contents( |
- local_src_dir=self._local_tempdir, remote_dest_dir='remote_dest_dir', |
- gs_acl='public') |
- |
- def test_upload_dir_contents_one_file(self): |
- """Upload src_dir containing one file, and no subdirs.""" |
- self._test_upload_dir_contents(filenames=['file1']) |
- |
- def test_upload_dir_contents_multiple_files(self): |
- """Upload src_dir containing multiple files, and no subdirs.""" |
- self._test_upload_dir_contents(filenames=['file1', 'file2']) |
- |
- def _test_upload_dir_contents(self, filenames): |
- """Helper function for upload_dir_contents() unittests. |
- |
- Args: |
- filenames: basenames of files to create within local_src_dir |
- """ |
- local_src_dir = self._local_tempdir |
- remote_dest_dir = 'remote_dest_dir' |
- for filename in filenames: |
- self._expected_commands.append('%s cp -a public %s %s' % ( |
- GSUTIL_LOCATION, |
- os.path.join(local_src_dir, filename), |
- posixpath.join(remote_dest_dir, filename))) |
- with open(os.path.join(local_src_dir, filename), 'w'): |
- pass |
- gs_utils.upload_dir_contents( |
- local_src_dir=local_src_dir, remote_dest_dir=remote_dest_dir, |
- gs_acl='public') |
- |
- def test_upload_dir_contents_one_dir(self): |
- """Upload src_dir containing a subdir, which in turn contains files.""" |
- local_src_dir = self._local_tempdir |
- remote_dest_dir = 'remote_dest_dir' |
- subdir = 'subdir' |
- os.mkdir(os.path.join(local_src_dir, subdir)) |
- for filename in ['file1', 'file2']: |
- self._expected_commands.append('%s cp -a public %s %s' % ( |
- GSUTIL_LOCATION, |
- os.path.join(local_src_dir, subdir, filename), |
- posixpath.join(remote_dest_dir, subdir, filename))) |
- with open(os.path.join(local_src_dir, subdir, filename), 'w'): |
- pass |
- gs_utils.upload_dir_contents( |
- local_src_dir=local_src_dir, remote_dest_dir=remote_dest_dir, |
- gs_acl='public') |
- |
- def test_download_dir_contents(self): |
- self._expected_commands = [( |
- '%s -m cp -R superman batman' % GSUTIL_LOCATION)] |
- gs_utils.download_dir_contents('superman', 'batman') |
- |
- def test_copy_dir_contents(self): |
- self._expected_commands = [( |
- '%s -m cp -a public -R superman batman' % GSUTIL_LOCATION)] |
- gs_utils.copy_dir_contents('superman', 'batman', 'public') |
- |
- def test_does_storage_object_exist(self): |
- self._expected_commands = [('%s ls superman' % GSUTIL_LOCATION)] |
- gs_utils.does_storage_object_exist('superman') |
- |
- def test_write_timestamp_file(self): |
- self._test_temp_file = os.path.join(tempfile.gettempdir(), 'TIMESTAMP') |
- self._test_gs_base = 'gs://test' |
- self._test_destdir = 'testdir' |
- self._test_gs_acl = 'private' |
- gs_utils.write_timestamp_file( |
- timestamp_file_name='TIMESTAMP', |
- timestamp_value=time.time(), |
- gs_base=self._test_gs_base, |
- gs_relative_dir=self._test_destdir, |
- gs_acl=self._test_gs_acl, |
- local_dir=self._local_tempdir) |
- |
- def test_AreTimeStampsEqual(self): |
- self._test_gs_base = 'gs://test' |
- self._test_destdir = 'testdir' |
- local_dir = self._local_tempdir |
- |
- class _MockFile(): |
- def __init__(self, name, attributes): |
- self._name = name |
- |
- def readlines(self): |
- return [] |
- |
- def read(self, arg1=None): |
- if self._name == os.path.join(tempfile.gettempdir(), 'TIMESTAMP'): |
- return TEST_TIMESTAMP |
- else: |
- return TEST_TIMESTAMP_2 |
- |
- def close(self): |
- pass |
- |
- def __enter__(self): |
- return self |
- |
- def __exit__(self, *args): |
- pass |
- |
- def write(self, string): |
- pass |
- |
- __builtin__.open = _MockFile |
- |
- # Will be false because the tmp directory will have no TIMESTAMP in it. |
- # pylint: disable=W0212 |
- self.assertFalse( |
- gs_utils._are_timestamps_equal( |
- local_dir=local_dir, |
- gs_base=self._test_gs_base, |
- gs_relative_dir=self._test_destdir)) |
- |
- self._test_temp_file = os.path.join(local_dir, 'TIMESTAMP') |
- |
- # Will be false because the timestamps are different. |
- # pylint: disable=W0212 |
- self.assertFalse( |
- gs_utils._are_timestamps_equal( |
- local_dir=local_dir, |
- gs_base=self._test_gs_base, |
- gs_relative_dir=self._test_destdir)) |
- |
- |
-if __name__ == '__main__': |
- unittest.main() |