mbligh | 24f8f5b | 2008-06-12 19:56:22 +0000 | [diff] [blame] | 1 | #!/usr/bin/python |
Aviv Keshet | e9bc833 | 2013-02-13 14:30:36 -0800 | [diff] [blame] | 2 | #pylint: disable-msg=C0111 |
mbligh | 5fe5c95 | 2008-05-27 20:14:43 +0000 | [diff] [blame] | 3 | __author__ = "raphtee@google.com (Travis Miller)" |
| 4 | |
mbligh | 060c471 | 2009-12-29 02:43:35 +0000 | [diff] [blame] | 5 | import unittest, os, tempfile, logging |
jadmanski | 69bdaac | 2010-07-28 16:27:20 +0000 | [diff] [blame] | 6 | |
mbligh | 5fe5c95 | 2008-05-27 20:14:43 +0000 | [diff] [blame] | 7 | import common |
jadmanski | cb0e161 | 2009-02-27 18:03:10 +0000 | [diff] [blame] | 8 | from autotest_lib.server import autotest, utils, hosts, server_job, profilers |
jadmanski | c09fc15 | 2008-10-15 17:56:59 +0000 | [diff] [blame] | 9 | from autotest_lib.client.bin import sysinfo |
Aviv Keshet | e9bc833 | 2013-02-13 14:30:36 -0800 | [diff] [blame] | 10 | from autotest_lib.client.common_lib import packages |
showard | ad812bf | 2009-10-20 23:49:56 +0000 | [diff] [blame] | 11 | from autotest_lib.client.common_lib import error |
jadmanski | 3d161b0 | 2008-06-06 15:43:36 +0000 | [diff] [blame] | 12 | from autotest_lib.client.common_lib.test_utils import mock |
jadmanski | 8d631c9 | 2008-08-18 21:12:40 +0000 | [diff] [blame] | 13 | |
mbligh | 5fe5c95 | 2008-05-27 20:14:43 +0000 | [diff] [blame] | 14 | |
Allen Li | d5abdab | 2017-02-07 16:03:43 -0800 | [diff] [blame] | 15 | class TestAutotest(unittest.TestCase): |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 16 | def setUp(self): |
| 17 | # create god |
mbligh | 0910844 | 2008-10-15 16:27:38 +0000 | [diff] [blame] | 18 | self.god = mock.mock_god() |
mbligh | 5fe5c95 | 2008-05-27 20:14:43 +0000 | [diff] [blame] | 19 | |
mbligh | 24f8f5b | 2008-06-12 19:56:22 +0000 | [diff] [blame] | 20 | # create mock host object |
jadmanski | 8d631c9 | 2008-08-18 21:12:40 +0000 | [diff] [blame] | 21 | self.host = self.god.create_mock_class(hosts.RemoteHost, "host") |
mbligh | 24f8f5b | 2008-06-12 19:56:22 +0000 | [diff] [blame] | 22 | self.host.hostname = "hostname" |
jadmanski | b6eb2f1 | 2008-09-12 16:39:36 +0000 | [diff] [blame] | 23 | self.host.job = self.god.create_mock_class(server_job.server_job, |
| 24 | "job") |
jadmanski | 23afbec | 2008-09-17 18:12:07 +0000 | [diff] [blame] | 25 | self.host.job.run_test_cleanup = True |
jadmanski | c09fc15 | 2008-10-15 17:56:59 +0000 | [diff] [blame] | 26 | self.host.job.sysinfo = self.god.create_mock_class( |
| 27 | sysinfo.sysinfo, "sysinfo") |
jadmanski | cb0e161 | 2009-02-27 18:03:10 +0000 | [diff] [blame] | 28 | self.host.job.profilers = self.god.create_mock_class( |
| 29 | profilers.profilers, "profilers") |
| 30 | self.host.job.profilers.add_log = {} |
jadmanski | c09fc15 | 2008-10-15 17:56:59 +0000 | [diff] [blame] | 31 | self.host.job.tmpdir = "/job/tmp" |
showard | a6082ef | 2009-10-12 20:25:44 +0000 | [diff] [blame] | 32 | self.host.job.default_profile_only = False |
jadmanski | 808f4b1 | 2010-04-09 22:30:31 +0000 | [diff] [blame] | 33 | self.host.job.args = [] |
| 34 | self.host.job.record = lambda *args: None |
Matt Mallett | 6da5d6b | 2018-09-27 12:57:47 -0700 | [diff] [blame] | 35 | self.host.verify_job_repo_url = lambda *args: None |
mbligh | 5fe5c95 | 2008-05-27 20:14:43 +0000 | [diff] [blame] | 36 | |
mbligh | 24f8f5b | 2008-06-12 19:56:22 +0000 | [diff] [blame] | 37 | # stubs |
mbligh | 3c7a150 | 2008-07-24 18:08:47 +0000 | [diff] [blame] | 38 | self.god.stub_function(utils, "get_server_dir") |
| 39 | self.god.stub_function(utils, "run") |
| 40 | self.god.stub_function(utils, "get") |
| 41 | self.god.stub_function(utils, "read_keyval") |
| 42 | self.god.stub_function(utils, "write_keyval") |
showard | 4b97607 | 2009-10-20 23:50:08 +0000 | [diff] [blame] | 43 | self.god.stub_function(utils, "system") |
mbligh | 24f8f5b | 2008-06-12 19:56:22 +0000 | [diff] [blame] | 44 | self.god.stub_function(tempfile, "mkstemp") |
| 45 | self.god.stub_function(tempfile, "mktemp") |
| 46 | self.god.stub_function(os, "getcwd") |
| 47 | self.god.stub_function(os, "system") |
| 48 | self.god.stub_function(os, "chdir") |
| 49 | self.god.stub_function(os, "makedirs") |
| 50 | self.god.stub_function(os, "remove") |
jadmanski | c09fc15 | 2008-10-15 17:56:59 +0000 | [diff] [blame] | 51 | self.god.stub_function(os, "fdopen") |
mbligh | 24f8f5b | 2008-06-12 19:56:22 +0000 | [diff] [blame] | 52 | self.god.stub_function(os.path, "exists") |
mbligh | 3c7a150 | 2008-07-24 18:08:47 +0000 | [diff] [blame] | 53 | self.god.stub_function(autotest, "open") |
| 54 | self.god.stub_function(autotest.global_config.global_config, |
| 55 | "get_config_value") |
mbligh | 060c471 | 2009-12-29 02:43:35 +0000 | [diff] [blame] | 56 | self.god.stub_function(logging, "exception") |
mbligh | 24f8f5b | 2008-06-12 19:56:22 +0000 | [diff] [blame] | 57 | self.god.stub_class(autotest, "_Run") |
jadmanski | 043e113 | 2008-11-19 17:10:32 +0000 | [diff] [blame] | 58 | self.god.stub_class(autotest, "log_collector") |
mbligh | 5fe5c95 | 2008-05-27 20:14:43 +0000 | [diff] [blame] | 59 | |
mbligh | 5fe5c95 | 2008-05-27 20:14:43 +0000 | [diff] [blame] | 60 | |
mbligh | 24f8f5b | 2008-06-12 19:56:22 +0000 | [diff] [blame] | 61 | def tearDown(self): |
mbligh | 1ef218d | 2009-08-03 16:57:56 +0000 | [diff] [blame] | 62 | self.god.unstub_all() |
mbligh | 5fe5c95 | 2008-05-27 20:14:43 +0000 | [diff] [blame] | 63 | |
mbligh | 24f8f5b | 2008-06-12 19:56:22 +0000 | [diff] [blame] | 64 | |
| 65 | def construct(self): |
| 66 | # setup |
| 67 | self.serverdir = "serverdir" |
| 68 | |
| 69 | # record |
mbligh | 3c7a150 | 2008-07-24 18:08:47 +0000 | [diff] [blame] | 70 | utils.get_server_dir.expect_call().and_return(self.serverdir) |
mbligh | 5fe5c95 | 2008-05-27 20:14:43 +0000 | [diff] [blame] | 71 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 72 | # create the autotest object |
Allen Li | d5abdab | 2017-02-07 16:03:43 -0800 | [diff] [blame] | 73 | self.autotest = autotest.Autotest(self.host) |
| 74 | self.autotest.job = self.host.job |
| 75 | self.god.stub_function(self.autotest, "_install_using_send_file") |
mbligh | 5fe5c95 | 2008-05-27 20:14:43 +0000 | [diff] [blame] | 76 | |
jadmanski | a49e9c4 | 2008-10-09 22:30:49 +0000 | [diff] [blame] | 77 | # stub out abspath |
| 78 | self.god.stub_function(os.path, "abspath") |
| 79 | |
mbligh | 24f8f5b | 2008-06-12 19:56:22 +0000 | [diff] [blame] | 80 | # check |
| 81 | self.god.check_playback() |
mbligh | 5fe5c95 | 2008-05-27 20:14:43 +0000 | [diff] [blame] | 82 | |
| 83 | |
mbligh | b8aa75b | 2009-09-18 16:50:37 +0000 | [diff] [blame] | 84 | def record_install_prologue(self): |
mbligh | 24f8f5b | 2008-06-12 19:56:22 +0000 | [diff] [blame] | 85 | self.construct() |
| 86 | |
| 87 | # setup |
mbligh | 3c7a150 | 2008-07-24 18:08:47 +0000 | [diff] [blame] | 88 | self.god.stub_class(packages, "PackageManager") |
Allen Li | d5abdab | 2017-02-07 16:03:43 -0800 | [diff] [blame] | 89 | self.autotest.got = False |
mbligh | 3c7a150 | 2008-07-24 18:08:47 +0000 | [diff] [blame] | 90 | location = os.path.join(self.serverdir, '../client') |
| 91 | location = os.path.abspath.expect_call(location).and_return(location) |
jadmanski | 3d161b0 | 2008-06-06 15:43:36 +0000 | [diff] [blame] | 92 | |
mbligh | 24f8f5b | 2008-06-12 19:56:22 +0000 | [diff] [blame] | 93 | # record |
mbligh | 3c7a150 | 2008-07-24 18:08:47 +0000 | [diff] [blame] | 94 | utils.get.expect_call(os.path.join(self.serverdir, |
| 95 | '../client')).and_return('source_material') |
| 96 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 97 | self.host.wait_up.expect_call(timeout=30) |
| 98 | self.host.setup.expect_call() |
mbligh | 24f8f5b | 2008-06-12 19:56:22 +0000 | [diff] [blame] | 99 | self.host.get_autodir.expect_call().and_return("autodir") |
mbligh | 0562e65 | 2008-08-20 20:11:45 +0000 | [diff] [blame] | 100 | self.host.set_autodir.expect_call("autodir") |
jadmanski | 3c23694 | 2009-03-04 17:51:26 +0000 | [diff] [blame] | 101 | self.host.run.expect_call('mkdir -p autodir') |
jadmanski | 3c23694 | 2009-03-04 17:51:26 +0000 | [diff] [blame] | 102 | self.host.run.expect_call('rm -rf autodir/results/*', |
jadmanski | 1c3c07b | 2009-03-03 23:29:36 +0000 | [diff] [blame] | 103 | ignore_status=True) |
mbligh | b8aa75b | 2009-09-18 16:50:37 +0000 | [diff] [blame] | 104 | |
| 105 | |
| 106 | def test_constructor(self): |
| 107 | self.construct() |
| 108 | |
| 109 | # we should check the calls |
| 110 | self.god.check_playback() |
| 111 | |
| 112 | |
| 113 | def test_full_client_install(self): |
| 114 | self.record_install_prologue() |
| 115 | |
Dan Shi | 1889ca1 | 2015-04-15 09:36:06 -0700 | [diff] [blame] | 116 | self.host.run.expect_call('rm -f "autodir/packages.checksum"') |
mbligh | 3c7a150 | 2008-07-24 18:08:47 +0000 | [diff] [blame] | 117 | c = autotest.global_config.global_config |
mbligh | b8aa75b | 2009-09-18 16:50:37 +0000 | [diff] [blame] | 118 | c.get_config_value.expect_call('PACKAGES', |
| 119 | 'serve_packages_from_autoserv', |
| 120 | type=bool).and_return(False) |
| 121 | self.host.send_file.expect_call('source_material', 'autodir', |
| 122 | delete_dest=True) |
Allen Li | 7962369 | 2017-08-17 16:27:11 -0700 | [diff] [blame] | 123 | self.god.stub_function(autotest.Autotest, "_send_shadow_config") |
| 124 | autotest.Autotest._send_shadow_config.expect_call() |
mbligh | b8aa75b | 2009-09-18 16:50:37 +0000 | [diff] [blame] | 125 | |
| 126 | # run and check |
Allen Li | d5abdab | 2017-02-07 16:03:43 -0800 | [diff] [blame] | 127 | self.autotest.install_full_client() |
mbligh | b8aa75b | 2009-09-18 16:50:37 +0000 | [diff] [blame] | 128 | self.god.check_playback() |
| 129 | |
| 130 | |
| 131 | def test_autoserv_install(self): |
| 132 | self.record_install_prologue() |
| 133 | |
| 134 | c = autotest.global_config.global_config |
| 135 | c.get_config_value.expect_call('PACKAGES', |
jadmanski | 2315a7e | 2009-09-18 18:39:37 +0000 | [diff] [blame] | 136 | 'fetch_location', type=list, default=[]).and_return([]) |
mbligh | b8aa75b | 2009-09-18 16:50:37 +0000 | [diff] [blame] | 137 | |
Allen Li | d5abdab | 2017-02-07 16:03:43 -0800 | [diff] [blame] | 138 | os.path.exists.expect_call('/etc/cros_chroot_version').and_return(True) |
mbligh | b8aa75b | 2009-09-18 16:50:37 +0000 | [diff] [blame] | 139 | c.get_config_value.expect_call('PACKAGES', |
| 140 | 'serve_packages_from_autoserv', |
| 141 | type=bool).and_return(True) |
Allen Li | d5abdab | 2017-02-07 16:03:43 -0800 | [diff] [blame] | 142 | self.autotest._install_using_send_file.expect_call(self.host, |
Allen Li | 7962369 | 2017-08-17 16:27:11 -0700 | [diff] [blame] | 143 | 'autodir') |
| 144 | self.god.stub_function(autotest.Autotest, "_send_shadow_config") |
| 145 | autotest.Autotest._send_shadow_config.expect_call() |
mbligh | b8aa75b | 2009-09-18 16:50:37 +0000 | [diff] [blame] | 146 | # run and check |
Allen Li | d5abdab | 2017-02-07 16:03:43 -0800 | [diff] [blame] | 147 | self.autotest.install() |
mbligh | b8aa75b | 2009-09-18 16:50:37 +0000 | [diff] [blame] | 148 | self.god.check_playback() |
| 149 | |
| 150 | |
| 151 | def test_packaging_install(self): |
| 152 | self.record_install_prologue() |
| 153 | |
| 154 | c = autotest.global_config.global_config |
| 155 | c.get_config_value.expect_call('PACKAGES', |
jadmanski | 2315a7e | 2009-09-18 18:39:37 +0000 | [diff] [blame] | 156 | 'fetch_location', type=list, default=[]).and_return(['repo']) |
Allen Li | d5abdab | 2017-02-07 16:03:43 -0800 | [diff] [blame] | 157 | os.path.exists.expect_call('/etc/cros_chroot_version').and_return(True) |
mbligh | 3c7a150 | 2008-07-24 18:08:47 +0000 | [diff] [blame] | 158 | pkgmgr = packages.PackageManager.expect_new('autodir', |
jadmanski | ede7e24 | 2009-08-10 15:43:33 +0000 | [diff] [blame] | 159 | repo_urls=['repo'], hostname='hostname', do_locking=False, |
mbligh | 76d19f7 | 2008-10-15 16:24:43 +0000 | [diff] [blame] | 160 | run_function=self.host.run, run_function_dargs=dict(timeout=600)) |
mbligh | 3c7a150 | 2008-07-24 18:08:47 +0000 | [diff] [blame] | 161 | pkg_dir = os.path.join('autodir', 'packages') |
Dan Shi | 86de094 | 2017-08-24 21:37:24 -0700 | [diff] [blame] | 162 | cmd = ('cd autodir && ls | grep -v "^packages$" | ' |
| 163 | 'grep -v "^result_tools$" | ' |
| 164 | 'xargs rm -rf && rm -rf .[!.]*') |
mbligh | c5ddfd1 | 2008-08-04 17:15:00 +0000 | [diff] [blame] | 165 | self.host.run.expect_call(cmd) |
mbligh | 3c7a150 | 2008-07-24 18:08:47 +0000 | [diff] [blame] | 166 | pkgmgr.install_pkg.expect_call('autotest', 'client', pkg_dir, |
| 167 | 'autodir', preserve_install_dir=True) |
mbligh | 5fe5c95 | 2008-05-27 20:14:43 +0000 | [diff] [blame] | 168 | |
mbligh | 3c7a150 | 2008-07-24 18:08:47 +0000 | [diff] [blame] | 169 | # run and check |
Allen Li | d5abdab | 2017-02-07 16:03:43 -0800 | [diff] [blame] | 170 | self.autotest.install() |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 171 | self.god.check_playback() |
mbligh | 5fe5c95 | 2008-05-27 20:14:43 +0000 | [diff] [blame] | 172 | |
mbligh | 5fe5c95 | 2008-05-27 20:14:43 +0000 | [diff] [blame] | 173 | |
mbligh | 24f8f5b | 2008-06-12 19:56:22 +0000 | [diff] [blame] | 174 | def test_run(self): |
| 175 | self.construct() |
mbligh | 5fe5c95 | 2008-05-27 20:14:43 +0000 | [diff] [blame] | 176 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 177 | # setup |
mbligh | 24f8f5b | 2008-06-12 19:56:22 +0000 | [diff] [blame] | 178 | control = "control" |
mbligh | 5fe5c95 | 2008-05-27 20:14:43 +0000 | [diff] [blame] | 179 | |
mbligh | 24f8f5b | 2008-06-12 19:56:22 +0000 | [diff] [blame] | 180 | # stub out install |
Allen Li | d5abdab | 2017-02-07 16:03:43 -0800 | [diff] [blame] | 181 | self.god.stub_function(self.autotest, "install") |
mbligh | 5fe5c95 | 2008-05-27 20:14:43 +0000 | [diff] [blame] | 182 | |
mbligh | 24f8f5b | 2008-06-12 19:56:22 +0000 | [diff] [blame] | 183 | # record |
Allen Li | d5abdab | 2017-02-07 16:03:43 -0800 | [diff] [blame] | 184 | self.autotest.install.expect_call(self.host, use_packaging=True) |
mbligh | 24f8f5b | 2008-06-12 19:56:22 +0000 | [diff] [blame] | 185 | self.host.wait_up.expect_call(timeout=30) |
| 186 | os.path.abspath.expect_call('.').and_return('.') |
Prathmesh Prabhu | beaee90 | 2017-07-26 11:58:44 -0700 | [diff] [blame] | 187 | run_obj = autotest._Run.expect_new(self.host, '.', None, False, False) |
mbligh | 24f8f5b | 2008-06-12 19:56:22 +0000 | [diff] [blame] | 188 | tag = None |
| 189 | run_obj.manual_control_file = os.path.join('autodir', |
| 190 | 'control.%s' % tag) |
| 191 | run_obj.remote_control_file = os.path.join('autodir', |
| 192 | 'control.%s.autoserv' % tag) |
| 193 | run_obj.tag = tag |
| 194 | run_obj.autodir = 'autodir' |
| 195 | run_obj.verify_machine.expect_call() |
Prathmesh Prabhu | beaee90 | 2017-07-26 11:58:44 -0700 | [diff] [blame] | 196 | run_obj.background = False |
mbligh | 24f8f5b | 2008-06-12 19:56:22 +0000 | [diff] [blame] | 197 | debug = os.path.join('.', 'debug') |
| 198 | os.makedirs.expect_call(debug) |
mbligh | 0910844 | 2008-10-15 16:27:38 +0000 | [diff] [blame] | 199 | delete_file_list = [run_obj.remote_control_file, |
| 200 | run_obj.remote_control_file + '.state', |
| 201 | run_obj.manual_control_file, |
| 202 | run_obj.manual_control_file + '.state'] |
| 203 | cmd = ';'.join('rm -f ' + control for control in delete_file_list) |
| 204 | self.host.run.expect_call(cmd, ignore_status=True) |
mbligh | 24f8f5b | 2008-06-12 19:56:22 +0000 | [diff] [blame] | 205 | |
Aviv Keshet | e9bc833 | 2013-02-13 14:30:36 -0800 | [diff] [blame] | 206 | utils.get.expect_call(control, local_copy=True).and_return("temp") |
mbligh | 24f8f5b | 2008-06-12 19:56:22 +0000 | [diff] [blame] | 207 | |
mbligh | 3c7a150 | 2008-07-24 18:08:47 +0000 | [diff] [blame] | 208 | c = autotest.global_config.global_config |
| 209 | c.get_config_value.expect_call("PACKAGES", |
Dale Curtis | 74a314b | 2011-06-23 14:55:46 -0700 | [diff] [blame] | 210 | 'fetch_location', type=list, default=[]).and_return(['repo']) |
mbligh | 0910844 | 2008-10-15 16:27:38 +0000 | [diff] [blame] | 211 | |
| 212 | cfile = self.god.create_mock_class(file, "file") |
| 213 | cfile_orig = "original control file" |
jadmanski | 808f4b1 | 2010-04-09 22:30:31 +0000 | [diff] [blame] | 214 | cfile_new = "args = []\njob.add_repository(['repo'])\n" |
mbligh | 0910844 | 2008-10-15 16:27:38 +0000 | [diff] [blame] | 215 | cfile_new += cfile_orig |
jadmanski | c09fc15 | 2008-10-15 17:56:59 +0000 | [diff] [blame] | 216 | |
Allen Li | d5abdab | 2017-02-07 16:03:43 -0800 | [diff] [blame] | 217 | os.path.exists.expect_call('/etc/cros_chroot_version').and_return(True) |
mbligh | 0910844 | 2008-10-15 16:27:38 +0000 | [diff] [blame] | 218 | autotest.open.expect_call("temp").and_return(cfile) |
| 219 | cfile.read.expect_call().and_return(cfile_orig) |
mbligh | 3c7a150 | 2008-07-24 18:08:47 +0000 | [diff] [blame] | 220 | autotest.open.expect_call("temp", 'w').and_return(cfile) |
mbligh | 0910844 | 2008-10-15 16:27:38 +0000 | [diff] [blame] | 221 | cfile.write.expect_call(cfile_new) |
mbligh | 3c7a150 | 2008-07-24 18:08:47 +0000 | [diff] [blame] | 222 | |
mbligh | fc3da5b | 2010-01-06 18:37:22 +0000 | [diff] [blame] | 223 | self.host.job.preprocess_client_state.expect_call().and_return( |
| 224 | '/job/tmp/file1') |
mbligh | fbf73ae | 2009-12-19 05:22:42 +0000 | [diff] [blame] | 225 | self.host.send_file.expect_call( |
| 226 | "/job/tmp/file1", "autodir/control.None.autoserv.init.state") |
jadmanski | c09fc15 | 2008-10-15 17:56:59 +0000 | [diff] [blame] | 227 | os.remove.expect_call("/job/tmp/file1") |
| 228 | |
mbligh | 3c7a150 | 2008-07-24 18:08:47 +0000 | [diff] [blame] | 229 | self.host.send_file.expect_call("temp", run_obj.remote_control_file) |
| 230 | os.path.abspath.expect_call('temp').and_return('control_file') |
mbligh | 0910844 | 2008-10-15 16:27:38 +0000 | [diff] [blame] | 231 | os.path.abspath.expect_call('control').and_return('control') |
mbligh | 3c7a150 | 2008-07-24 18:08:47 +0000 | [diff] [blame] | 232 | os.remove.expect_call("temp") |
jadmanski | 6bb32d7 | 2009-03-19 20:25:24 +0000 | [diff] [blame] | 233 | |
J. Richard Barnette | f44ff96 | 2011-01-05 10:14:28 -0800 | [diff] [blame] | 234 | run_obj.execute_control.expect_call(timeout=30, |
Chris Sosa | b9ada9b | 2013-06-12 12:47:47 -0700 | [diff] [blame] | 235 | client_disconnect_timeout=240) |
jadmanski | 23afbec | 2008-09-17 18:12:07 +0000 | [diff] [blame] | 236 | |
mbligh | 24f8f5b | 2008-06-12 19:56:22 +0000 | [diff] [blame] | 237 | # run and check output |
Allen Li | d5abdab | 2017-02-07 16:03:43 -0800 | [diff] [blame] | 238 | self.autotest.run(control, timeout=30) |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 239 | self.god.check_playback() |
mbligh | 5fe5c95 | 2008-05-27 20:14:43 +0000 | [diff] [blame] | 240 | |
mbligh | 3c7a150 | 2008-07-24 18:08:47 +0000 | [diff] [blame] | 241 | |
showard | ad812bf | 2009-10-20 23:49:56 +0000 | [diff] [blame] | 242 | def _stub_get_client_autodir_paths(self): |
| 243 | def mock_get_client_autodir_paths(cls, host): |
| 244 | return ['/some/path', '/another/path'] |
| 245 | self.god.stub_with(autotest.Autotest, 'get_client_autodir_paths', |
| 246 | classmethod(mock_get_client_autodir_paths)) |
| 247 | |
| 248 | |
| 249 | def _expect_failed_run(self, command): |
| 250 | (self.host.run.expect_call(command) |
| 251 | .and_raises(error.AutoservRunError('dummy', object()))) |
| 252 | |
| 253 | |
| 254 | def test_get_installed_autodir(self): |
| 255 | self._stub_get_client_autodir_paths() |
| 256 | self.host.get_autodir.expect_call().and_return(None) |
| 257 | self._expect_failed_run('test -x /some/path/bin/autotest') |
| 258 | self.host.run.expect_call('test -x /another/path/bin/autotest') |
Eric Li | 861b2d5 | 2011-02-04 14:50:35 -0800 | [diff] [blame] | 259 | self.host.run.expect_call('test -w /another/path') |
showard | ad812bf | 2009-10-20 23:49:56 +0000 | [diff] [blame] | 260 | |
| 261 | autodir = autotest.Autotest.get_installed_autodir(self.host) |
| 262 | self.assertEquals(autodir, '/another/path') |
| 263 | |
| 264 | |
| 265 | def test_get_install_dir(self): |
| 266 | self._stub_get_client_autodir_paths() |
| 267 | self.host.get_autodir.expect_call().and_return(None) |
| 268 | self._expect_failed_run('test -x /some/path/bin/autotest') |
| 269 | self._expect_failed_run('test -x /another/path/bin/autotest') |
J. Richard Barnette | f44ff96 | 2011-01-05 10:14:28 -0800 | [diff] [blame] | 270 | self._expect_failed_run('mkdir -p /some/path') |
| 271 | self.host.run.expect_call('mkdir -p /another/path') |
Eric Li | 861b2d5 | 2011-02-04 14:50:35 -0800 | [diff] [blame] | 272 | self.host.run.expect_call('test -w /another/path') |
showard | ad812bf | 2009-10-20 23:49:56 +0000 | [diff] [blame] | 273 | |
| 274 | install_dir = autotest.Autotest.get_install_dir(self.host) |
J. Richard Barnette | f44ff96 | 2011-01-05 10:14:28 -0800 | [diff] [blame] | 275 | self.assertEquals(install_dir, '/another/path') |
showard | ad812bf | 2009-10-20 23:49:56 +0000 | [diff] [blame] | 276 | |
| 277 | |
mbligh | 060c471 | 2009-12-29 02:43:35 +0000 | [diff] [blame] | 278 | def test_client_logger_process_line_log_copy_collection_failure(self): |
| 279 | collector = autotest.log_collector.expect_new(self.host, '', '') |
| 280 | logger = autotest.client_logger(self.host, '', '') |
| 281 | collector.collect_client_job_results.expect_call().and_raises( |
| 282 | Exception('log copy failure')) |
| 283 | logging.exception.expect_call(mock.is_string_comparator()) |
| 284 | logger._process_line('AUTOTEST_TEST_COMPLETE:/autotest/fifo1') |
| 285 | |
| 286 | |
| 287 | def test_client_logger_process_line_log_copy_fifo_failure(self): |
| 288 | collector = autotest.log_collector.expect_new(self.host, '', '') |
| 289 | logger = autotest.client_logger(self.host, '', '') |
| 290 | collector.collect_client_job_results.expect_call() |
| 291 | self.host.run.expect_call('echo A > /autotest/fifo2').and_raises( |
| 292 | Exception('fifo failure')) |
| 293 | logging.exception.expect_call(mock.is_string_comparator()) |
| 294 | logger._process_line('AUTOTEST_TEST_COMPLETE:/autotest/fifo2') |
| 295 | |
| 296 | |
| 297 | def test_client_logger_process_line_package_install_fifo_failure(self): |
| 298 | collector = autotest.log_collector.expect_new(self.host, '', '') |
| 299 | logger = autotest.client_logger(self.host, '', '') |
| 300 | self.god.stub_function(logger, '_send_tarball') |
| 301 | |
| 302 | c = autotest.global_config.global_config |
| 303 | c.get_config_value.expect_call('PACKAGES', |
| 304 | 'serve_packages_from_autoserv', |
| 305 | type=bool).and_return(True) |
Aviv Keshet | e9bc833 | 2013-02-13 14:30:36 -0800 | [diff] [blame] | 306 | c.get_config_value.expect_call('PACKAGES', |
| 307 | 'serve_packages_from_autoserv', |
| 308 | type=bool).and_return(True) |
mbligh | 060c471 | 2009-12-29 02:43:35 +0000 | [diff] [blame] | 309 | logger._send_tarball.expect_call('pkgname.tar.bz2', '/autotest/dest/') |
lmr | 12b4558 | 2010-01-11 21:22:02 +0000 | [diff] [blame] | 310 | |
mbligh | 060c471 | 2009-12-29 02:43:35 +0000 | [diff] [blame] | 311 | self.host.run.expect_call('echo B > /autotest/fifo3').and_raises( |
| 312 | Exception('fifo failure')) |
| 313 | logging.exception.expect_call(mock.is_string_comparator()) |
| 314 | logger._process_line('AUTOTEST_FETCH_PACKAGE:pkgname.tar.bz2:' |
| 315 | '/autotest/dest/:/autotest/fifo3') |
| 316 | |
jadmanski | 69bdaac | 2010-07-28 16:27:20 +0000 | [diff] [blame] | 317 | |
mbligh | 5fe5c95 | 2008-05-27 20:14:43 +0000 | [diff] [blame] | 318 | if __name__ == "__main__": |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 319 | unittest.main() |