Index: client/bin/job_unittest.py |
diff --git a/client/bin/job_unittest.py b/client/bin/job_unittest.py |
index 24c0501241bd6a389e933381063a17047c4d985e..0e5aedeebc74491d8700ad4e243c65e31650b0c2 100755 |
--- a/client/bin/job_unittest.py |
+++ b/client/bin/job_unittest.py |
@@ -18,7 +18,7 @@ class job_test_case(unittest.TestCase): |
job_class = job.base_client_job |
def setUp(self): |
- self.god = mock.mock_god() |
+ self.god = mock.mock_god(ut=self) |
self.god.stub_with(job.base_client_job, '_get_environ_autodir', |
classmethod(lambda cls: '/adir')) |
self.job = self.job_class.__new__(self.job_class) |
@@ -86,6 +86,7 @@ class test_init_minimal_options(abstract_test_init, job_test_case): |
user = None |
log = False |
args = '' |
+ tap_report = None |
self.god.stub_function_to_return(job.utils, 'drop_caches', None) |
self.job._job_state = base_job_unittest.stub_job_state |
@@ -109,7 +110,7 @@ class first_line_comparator(mock.argument_comparator): |
class test_base_job(unittest.TestCase): |
def setUp(self): |
# make god |
- self.god = mock.mock_god() |
+ self.god = mock.mock_god(ut=self) |
# need to set some environ variables |
self.autodir = "autodir" |
@@ -241,6 +242,7 @@ class test_base_job(unittest.TestCase): |
options.hostname = 'localhost' |
options.user = 'my_user' |
options.args = '' |
+ options.tap_report = None |
self.job.__init__(self.control, options, |
extra_copy_cmdline=['more-blah']) |
@@ -280,6 +282,7 @@ class test_base_job(unittest.TestCase): |
options.hostname = 'localhost' |
options.user = 'my_user' |
options.args = '' |
+ options.tap_report = None |
error = Exception('fail') |
self.god.stub_function(self.job, '_post_record_init') |
@@ -691,5 +694,18 @@ class test_base_job(unittest.TestCase): |
self.god.check_playback() |
+ def test_parse_args(self): |
+ test_set = {"a='foo bar baz' b='moo apt'": |
+ ["a='foo bar baz'", "b='moo apt'"], |
+ "a='foo bar baz' only=gah": |
+ ["a='foo bar baz'", "only=gah"], |
+ "a='b c d' no=argh": |
+ ["a='b c d'", "no=argh"]} |
+ for t in test_set: |
+ parsed_args = job.base_client_job._parse_args(t) |
+ expected_args = test_set[t] |
+ self.assertEqual(parsed_args, expected_args) |
+ |
+ |
if __name__ == "__main__": |
unittest.main() |