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 | |
| 15 | class TestBaseAutotest(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 |
mbligh | 0910844 | 2008-10-15 16:27:38 +0000 | [diff] [blame] | 26 | self.host.job.last_boot_tag = 'Autotest' |
jadmanski | c09fc15 | 2008-10-15 17:56:59 +0000 | [diff] [blame] | 27 | self.host.job.sysinfo = self.god.create_mock_class( |
| 28 | sysinfo.sysinfo, "sysinfo") |
jadmanski | cb0e161 | 2009-02-27 18:03:10 +0000 | [diff] [blame] | 29 | self.host.job.profilers = self.god.create_mock_class( |
| 30 | profilers.profilers, "profilers") |
| 31 | self.host.job.profilers.add_log = {} |
jadmanski | c09fc15 | 2008-10-15 17:56:59 +0000 | [diff] [blame] | 32 | self.host.job.tmpdir = "/job/tmp" |
showard | a6082ef | 2009-10-12 20:25:44 +0000 | [diff] [blame] | 33 | self.host.job.default_profile_only = False |
jadmanski | 808f4b1 | 2010-04-09 22:30:31 +0000 | [diff] [blame] | 34 | self.host.job.args = [] |
| 35 | self.host.job.record = 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 |
| 73 | self.base_autotest = autotest.BaseAutotest(self.host) |
jadmanski | 808f4b1 | 2010-04-09 22:30:31 +0000 | [diff] [blame] | 74 | self.base_autotest.job = self.host.job |
mbligh | b8aa75b | 2009-09-18 16:50:37 +0000 | [diff] [blame] | 75 | self.god.stub_function(self.base_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") |
| 89 | self.base_autotest.got = False |
| 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 | os.getcwd.expect_call().and_return('cwd') |
| 95 | os.chdir.expect_call(os.path.join(self.serverdir, '../client')) |
showard | 4b97607 | 2009-10-20 23:50:08 +0000 | [diff] [blame] | 96 | utils.system.expect_call('tools/make_clean', ignore_status=True) |
mbligh | 3c7a150 | 2008-07-24 18:08:47 +0000 | [diff] [blame] | 97 | os.chdir.expect_call('cwd') |
| 98 | utils.get.expect_call(os.path.join(self.serverdir, |
| 99 | '../client')).and_return('source_material') |
| 100 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 101 | self.host.wait_up.expect_call(timeout=30) |
| 102 | self.host.setup.expect_call() |
mbligh | 24f8f5b | 2008-06-12 19:56:22 +0000 | [diff] [blame] | 103 | self.host.get_autodir.expect_call().and_return("autodir") |
mbligh | 0562e65 | 2008-08-20 20:11:45 +0000 | [diff] [blame] | 104 | self.host.set_autodir.expect_call("autodir") |
jadmanski | 3c23694 | 2009-03-04 17:51:26 +0000 | [diff] [blame] | 105 | self.host.run.expect_call('mkdir -p autodir') |
jadmanski | 3c23694 | 2009-03-04 17:51:26 +0000 | [diff] [blame] | 106 | self.host.run.expect_call('rm -rf autodir/results/*', |
jadmanski | 1c3c07b | 2009-03-03 23:29:36 +0000 | [diff] [blame] | 107 | ignore_status=True) |
mbligh | b8aa75b | 2009-09-18 16:50:37 +0000 | [diff] [blame] | 108 | |
| 109 | |
| 110 | def test_constructor(self): |
| 111 | self.construct() |
| 112 | |
| 113 | # we should check the calls |
| 114 | self.god.check_playback() |
| 115 | |
| 116 | |
| 117 | def test_full_client_install(self): |
| 118 | self.record_install_prologue() |
| 119 | |
mbligh | 3c7a150 | 2008-07-24 18:08:47 +0000 | [diff] [blame] | 120 | c = autotest.global_config.global_config |
mbligh | b8aa75b | 2009-09-18 16:50:37 +0000 | [diff] [blame] | 121 | c.get_config_value.expect_call('PACKAGES', |
| 122 | 'serve_packages_from_autoserv', |
| 123 | type=bool).and_return(False) |
| 124 | self.host.send_file.expect_call('source_material', 'autodir', |
| 125 | delete_dest=True) |
| 126 | |
| 127 | # run and check |
| 128 | self.base_autotest.install_full_client() |
| 129 | self.god.check_playback() |
| 130 | |
| 131 | |
| 132 | def test_autoserv_install(self): |
| 133 | self.record_install_prologue() |
| 134 | |
| 135 | c = autotest.global_config.global_config |
| 136 | c.get_config_value.expect_call('PACKAGES', |
jadmanski | 2315a7e | 2009-09-18 18:39:37 +0000 | [diff] [blame] | 137 | 'fetch_location', type=list, default=[]).and_return([]) |
mbligh | b8aa75b | 2009-09-18 16:50:37 +0000 | [diff] [blame] | 138 | |
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) |
| 142 | self.base_autotest._install_using_send_file.expect_call(self.host, |
| 143 | 'autodir') |
mbligh | b8aa75b | 2009-09-18 16:50:37 +0000 | [diff] [blame] | 144 | # run and check |
| 145 | self.base_autotest.install() |
| 146 | self.god.check_playback() |
| 147 | |
| 148 | |
| 149 | def test_packaging_install(self): |
| 150 | self.record_install_prologue() |
| 151 | |
| 152 | c = autotest.global_config.global_config |
| 153 | c.get_config_value.expect_call('PACKAGES', |
jadmanski | 2315a7e | 2009-09-18 18:39:37 +0000 | [diff] [blame] | 154 | 'fetch_location', type=list, default=[]).and_return(['repo']) |
mbligh | 3c7a150 | 2008-07-24 18:08:47 +0000 | [diff] [blame] | 155 | pkgmgr = packages.PackageManager.expect_new('autodir', |
jadmanski | ede7e24 | 2009-08-10 15:43:33 +0000 | [diff] [blame] | 156 | repo_urls=['repo'], hostname='hostname', do_locking=False, |
mbligh | 76d19f7 | 2008-10-15 16:24:43 +0000 | [diff] [blame] | 157 | run_function=self.host.run, run_function_dargs=dict(timeout=600)) |
mbligh | 3c7a150 | 2008-07-24 18:08:47 +0000 | [diff] [blame] | 158 | pkg_dir = os.path.join('autodir', 'packages') |
mbligh | c5ddfd1 | 2008-08-04 17:15:00 +0000 | [diff] [blame] | 159 | cmd = ('cd autodir && ls | grep -v "^packages$"' |
| 160 | ' | xargs rm -rf && rm -rf .[^.]*') |
| 161 | self.host.run.expect_call(cmd) |
mbligh | 3c7a150 | 2008-07-24 18:08:47 +0000 | [diff] [blame] | 162 | pkgmgr.install_pkg.expect_call('autotest', 'client', pkg_dir, |
| 163 | 'autodir', preserve_install_dir=True) |
mbligh | 5fe5c95 | 2008-05-27 20:14:43 +0000 | [diff] [blame] | 164 | |
mbligh | 3c7a150 | 2008-07-24 18:08:47 +0000 | [diff] [blame] | 165 | # run and check |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 166 | self.base_autotest.install() |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 167 | self.god.check_playback() |
mbligh | 5fe5c95 | 2008-05-27 20:14:43 +0000 | [diff] [blame] | 168 | |
mbligh | 5fe5c95 | 2008-05-27 20:14:43 +0000 | [diff] [blame] | 169 | |
mbligh | 24f8f5b | 2008-06-12 19:56:22 +0000 | [diff] [blame] | 170 | def test_run(self): |
| 171 | self.construct() |
mbligh | 5fe5c95 | 2008-05-27 20:14:43 +0000 | [diff] [blame] | 172 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 173 | # setup |
mbligh | 24f8f5b | 2008-06-12 19:56:22 +0000 | [diff] [blame] | 174 | control = "control" |
mbligh | 5fe5c95 | 2008-05-27 20:14:43 +0000 | [diff] [blame] | 175 | |
mbligh | 24f8f5b | 2008-06-12 19:56:22 +0000 | [diff] [blame] | 176 | # stub out install |
| 177 | self.god.stub_function(self.base_autotest, "install") |
mbligh | 5fe5c95 | 2008-05-27 20:14:43 +0000 | [diff] [blame] | 178 | |
mbligh | 24f8f5b | 2008-06-12 19:56:22 +0000 | [diff] [blame] | 179 | # record |
Dan Shi | b669cbd | 2013-09-13 11:17:17 -0700 | [diff] [blame^] | 180 | self.base_autotest.install.expect_call(self.host, use_packaging=True) |
mbligh | 24f8f5b | 2008-06-12 19:56:22 +0000 | [diff] [blame] | 181 | self.host.wait_up.expect_call(timeout=30) |
| 182 | os.path.abspath.expect_call('.').and_return('.') |
mbligh | b3c0c91 | 2008-11-27 00:32:45 +0000 | [diff] [blame] | 183 | run_obj = autotest._Run.expect_new(self.host, '.', None, False, False) |
mbligh | 24f8f5b | 2008-06-12 19:56:22 +0000 | [diff] [blame] | 184 | tag = None |
| 185 | run_obj.manual_control_file = os.path.join('autodir', |
| 186 | 'control.%s' % tag) |
| 187 | run_obj.remote_control_file = os.path.join('autodir', |
| 188 | 'control.%s.autoserv' % tag) |
| 189 | run_obj.tag = tag |
| 190 | run_obj.autodir = 'autodir' |
| 191 | run_obj.verify_machine.expect_call() |
mbligh | b3c0c91 | 2008-11-27 00:32:45 +0000 | [diff] [blame] | 192 | run_obj.background = False |
mbligh | 24f8f5b | 2008-06-12 19:56:22 +0000 | [diff] [blame] | 193 | debug = os.path.join('.', 'debug') |
| 194 | os.makedirs.expect_call(debug) |
mbligh | 0910844 | 2008-10-15 16:27:38 +0000 | [diff] [blame] | 195 | delete_file_list = [run_obj.remote_control_file, |
| 196 | run_obj.remote_control_file + '.state', |
| 197 | run_obj.manual_control_file, |
| 198 | run_obj.manual_control_file + '.state'] |
| 199 | cmd = ';'.join('rm -f ' + control for control in delete_file_list) |
| 200 | self.host.run.expect_call(cmd, ignore_status=True) |
mbligh | 24f8f5b | 2008-06-12 19:56:22 +0000 | [diff] [blame] | 201 | |
Aviv Keshet | e9bc833 | 2013-02-13 14:30:36 -0800 | [diff] [blame] | 202 | utils.get.expect_call(control, local_copy=True).and_return("temp") |
mbligh | 24f8f5b | 2008-06-12 19:56:22 +0000 | [diff] [blame] | 203 | |
mbligh | 3c7a150 | 2008-07-24 18:08:47 +0000 | [diff] [blame] | 204 | c = autotest.global_config.global_config |
| 205 | c.get_config_value.expect_call("PACKAGES", |
Dale Curtis | 74a314b | 2011-06-23 14:55:46 -0700 | [diff] [blame] | 206 | 'fetch_location', type=list, default=[]).and_return(['repo']) |
mbligh | 0910844 | 2008-10-15 16:27:38 +0000 | [diff] [blame] | 207 | |
| 208 | cfile = self.god.create_mock_class(file, "file") |
| 209 | cfile_orig = "original control file" |
jadmanski | 808f4b1 | 2010-04-09 22:30:31 +0000 | [diff] [blame] | 210 | cfile_new = "args = []\njob.add_repository(['repo'])\n" |
mbligh | 0910844 | 2008-10-15 16:27:38 +0000 | [diff] [blame] | 211 | cfile_new += cfile_orig |
jadmanski | c09fc15 | 2008-10-15 17:56:59 +0000 | [diff] [blame] | 212 | |
mbligh | 0910844 | 2008-10-15 16:27:38 +0000 | [diff] [blame] | 213 | autotest.open.expect_call("temp").and_return(cfile) |
| 214 | cfile.read.expect_call().and_return(cfile_orig) |
mbligh | 3c7a150 | 2008-07-24 18:08:47 +0000 | [diff] [blame] | 215 | autotest.open.expect_call("temp", 'w').and_return(cfile) |
mbligh | 0910844 | 2008-10-15 16:27:38 +0000 | [diff] [blame] | 216 | cfile.write.expect_call(cfile_new) |
mbligh | 3c7a150 | 2008-07-24 18:08:47 +0000 | [diff] [blame] | 217 | |
mbligh | fc3da5b | 2010-01-06 18:37:22 +0000 | [diff] [blame] | 218 | self.host.job.preprocess_client_state.expect_call().and_return( |
| 219 | '/job/tmp/file1') |
mbligh | fbf73ae | 2009-12-19 05:22:42 +0000 | [diff] [blame] | 220 | self.host.send_file.expect_call( |
| 221 | "/job/tmp/file1", "autodir/control.None.autoserv.init.state") |
jadmanski | c09fc15 | 2008-10-15 17:56:59 +0000 | [diff] [blame] | 222 | os.remove.expect_call("/job/tmp/file1") |
| 223 | |
mbligh | 3c7a150 | 2008-07-24 18:08:47 +0000 | [diff] [blame] | 224 | self.host.send_file.expect_call("temp", run_obj.remote_control_file) |
| 225 | os.path.abspath.expect_call('temp').and_return('control_file') |
mbligh | 0910844 | 2008-10-15 16:27:38 +0000 | [diff] [blame] | 226 | os.path.abspath.expect_call('control').and_return('control') |
mbligh | 3c7a150 | 2008-07-24 18:08:47 +0000 | [diff] [blame] | 227 | os.remove.expect_call("temp") |
jadmanski | 6bb32d7 | 2009-03-19 20:25:24 +0000 | [diff] [blame] | 228 | |
J. Richard Barnette | f44ff96 | 2011-01-05 10:14:28 -0800 | [diff] [blame] | 229 | run_obj.execute_control.expect_call(timeout=30, |
Chris Sosa | b9ada9b | 2013-06-12 12:47:47 -0700 | [diff] [blame] | 230 | client_disconnect_timeout=240) |
jadmanski | 23afbec | 2008-09-17 18:12:07 +0000 | [diff] [blame] | 231 | |
mbligh | 24f8f5b | 2008-06-12 19:56:22 +0000 | [diff] [blame] | 232 | # run and check output |
mbligh | 3c7a150 | 2008-07-24 18:08:47 +0000 | [diff] [blame] | 233 | self.base_autotest.run(control, timeout=30) |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 234 | self.god.check_playback() |
mbligh | 5fe5c95 | 2008-05-27 20:14:43 +0000 | [diff] [blame] | 235 | |
mbligh | 3c7a150 | 2008-07-24 18:08:47 +0000 | [diff] [blame] | 236 | |
showard | ad812bf | 2009-10-20 23:49:56 +0000 | [diff] [blame] | 237 | def _stub_get_client_autodir_paths(self): |
| 238 | def mock_get_client_autodir_paths(cls, host): |
| 239 | return ['/some/path', '/another/path'] |
| 240 | self.god.stub_with(autotest.Autotest, 'get_client_autodir_paths', |
| 241 | classmethod(mock_get_client_autodir_paths)) |
| 242 | |
| 243 | |
| 244 | def _expect_failed_run(self, command): |
| 245 | (self.host.run.expect_call(command) |
| 246 | .and_raises(error.AutoservRunError('dummy', object()))) |
| 247 | |
| 248 | |
| 249 | def test_get_installed_autodir(self): |
| 250 | self._stub_get_client_autodir_paths() |
| 251 | self.host.get_autodir.expect_call().and_return(None) |
| 252 | self._expect_failed_run('test -x /some/path/bin/autotest') |
| 253 | self.host.run.expect_call('test -x /another/path/bin/autotest') |
Eric Li | 861b2d5 | 2011-02-04 14:50:35 -0800 | [diff] [blame] | 254 | self.host.run.expect_call('test -w /another/path') |
showard | ad812bf | 2009-10-20 23:49:56 +0000 | [diff] [blame] | 255 | |
| 256 | autodir = autotest.Autotest.get_installed_autodir(self.host) |
| 257 | self.assertEquals(autodir, '/another/path') |
| 258 | |
| 259 | |
| 260 | def test_get_install_dir(self): |
| 261 | self._stub_get_client_autodir_paths() |
| 262 | self.host.get_autodir.expect_call().and_return(None) |
| 263 | self._expect_failed_run('test -x /some/path/bin/autotest') |
| 264 | self._expect_failed_run('test -x /another/path/bin/autotest') |
J. Richard Barnette | f44ff96 | 2011-01-05 10:14:28 -0800 | [diff] [blame] | 265 | self._expect_failed_run('mkdir -p /some/path') |
| 266 | self.host.run.expect_call('mkdir -p /another/path') |
Eric Li | 861b2d5 | 2011-02-04 14:50:35 -0800 | [diff] [blame] | 267 | self.host.run.expect_call('test -w /another/path') |
showard | ad812bf | 2009-10-20 23:49:56 +0000 | [diff] [blame] | 268 | |
| 269 | install_dir = autotest.Autotest.get_install_dir(self.host) |
J. Richard Barnette | f44ff96 | 2011-01-05 10:14:28 -0800 | [diff] [blame] | 270 | self.assertEquals(install_dir, '/another/path') |
showard | ad812bf | 2009-10-20 23:49:56 +0000 | [diff] [blame] | 271 | |
| 272 | |
mbligh | 060c471 | 2009-12-29 02:43:35 +0000 | [diff] [blame] | 273 | def test_client_logger_process_line_log_copy_collection_failure(self): |
| 274 | collector = autotest.log_collector.expect_new(self.host, '', '') |
| 275 | logger = autotest.client_logger(self.host, '', '') |
| 276 | collector.collect_client_job_results.expect_call().and_raises( |
| 277 | Exception('log copy failure')) |
| 278 | logging.exception.expect_call(mock.is_string_comparator()) |
| 279 | logger._process_line('AUTOTEST_TEST_COMPLETE:/autotest/fifo1') |
| 280 | |
| 281 | |
| 282 | def test_client_logger_process_line_log_copy_fifo_failure(self): |
| 283 | collector = autotest.log_collector.expect_new(self.host, '', '') |
| 284 | logger = autotest.client_logger(self.host, '', '') |
| 285 | collector.collect_client_job_results.expect_call() |
| 286 | self.host.run.expect_call('echo A > /autotest/fifo2').and_raises( |
| 287 | Exception('fifo failure')) |
| 288 | logging.exception.expect_call(mock.is_string_comparator()) |
| 289 | logger._process_line('AUTOTEST_TEST_COMPLETE:/autotest/fifo2') |
| 290 | |
| 291 | |
| 292 | def test_client_logger_process_line_package_install_fifo_failure(self): |
| 293 | collector = autotest.log_collector.expect_new(self.host, '', '') |
| 294 | logger = autotest.client_logger(self.host, '', '') |
| 295 | self.god.stub_function(logger, '_send_tarball') |
| 296 | |
| 297 | c = autotest.global_config.global_config |
| 298 | c.get_config_value.expect_call('PACKAGES', |
| 299 | 'serve_packages_from_autoserv', |
| 300 | type=bool).and_return(True) |
Aviv Keshet | e9bc833 | 2013-02-13 14:30:36 -0800 | [diff] [blame] | 301 | c.get_config_value.expect_call('PACKAGES', |
| 302 | 'serve_packages_from_autoserv', |
| 303 | type=bool).and_return(True) |
mbligh | 060c471 | 2009-12-29 02:43:35 +0000 | [diff] [blame] | 304 | logger._send_tarball.expect_call('pkgname.tar.bz2', '/autotest/dest/') |
lmr | 12b4558 | 2010-01-11 21:22:02 +0000 | [diff] [blame] | 305 | |
mbligh | 060c471 | 2009-12-29 02:43:35 +0000 | [diff] [blame] | 306 | self.host.run.expect_call('echo B > /autotest/fifo3').and_raises( |
| 307 | Exception('fifo failure')) |
| 308 | logging.exception.expect_call(mock.is_string_comparator()) |
| 309 | logger._process_line('AUTOTEST_FETCH_PACKAGE:pkgname.tar.bz2:' |
| 310 | '/autotest/dest/:/autotest/fifo3') |
| 311 | |
jadmanski | 69bdaac | 2010-07-28 16:27:20 +0000 | [diff] [blame] | 312 | |
jadmanski | c6136e9 | 2010-07-19 16:41:49 +0000 | [diff] [blame] | 313 | class test_autotest_mixin(unittest.TestCase): |
| 314 | def setUp(self): |
| 315 | # a dummy Autotest and job class for use in the mixin |
| 316 | class stub_autotest(object): |
| 317 | class job(object): |
| 318 | state_dict = {} |
| 319 | def get_state(self, var, default): |
| 320 | return self.state_dict.get(var, default) |
| 321 | job = job() |
| 322 | |
| 323 | @staticmethod |
| 324 | def run(control_file, host=None): |
| 325 | self.control_file = control_file |
| 326 | self.host = host |
| 327 | |
| 328 | self.mixin = autotest.AutotestHostMixin() |
| 329 | self.mixin._Autotest = stub_autotest |
| 330 | self.job = self.mixin._Autotest.job |
| 331 | |
| 332 | |
| 333 | def test_passes(self): |
| 334 | self.job.state_dict['test_result'] = True |
| 335 | self.assertEqual(True, self.mixin.run_test('sleeptest', seconds=1)) |
| 336 | self.assert_("job.run_test('sleeptest', seconds=1)\n" |
| 337 | in self.control_file) |
| 338 | self.assertEqual(self.mixin, self.host) |
| 339 | |
| 340 | |
| 341 | def test_fails_clean(self): |
| 342 | self.job.state_dict['test_result'] = False |
| 343 | self.assertEqual(False, self.mixin.run_test('sleeptest', seconds='2')) |
| 344 | self.assert_("job.run_test('sleeptest', seconds='2')\n" |
| 345 | in self.control_file) |
| 346 | self.assertEqual(self.mixin, self.host) |
| 347 | |
| 348 | |
| 349 | def test_fails_with_exception(self): |
| 350 | self.assertEqual(False, self.mixin.run_test('sleeptest')) |
| 351 | self.assert_("job.run_test('sleeptest')\n" in self.control_file) |
| 352 | self.assertEqual(self.mixin, self.host) |
| 353 | |
mbligh | 060c471 | 2009-12-29 02:43:35 +0000 | [diff] [blame] | 354 | |
mbligh | 5fe5c95 | 2008-05-27 20:14:43 +0000 | [diff] [blame] | 355 | if __name__ == "__main__": |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 356 | unittest.main() |