harness_unittest.py revision 6631273af8b88842cbd6202cc4615daf050cc957
1#!/usr/bin/python 2import unittest 3import common 4from autotest_lib.client.common_lib.test_utils import mock 5from autotest_lib.client.bin import harness, harness_standalone, harness_ABAT 6 7 8class harness_unittest(unittest.TestCase): 9 def setUp(self): 10 self.god = mock.mock_god() 11 12 13 def tearDown(self): 14 self.god.unstub_all() 15 16 17 def test_select_none(self): 18 job = object() 19 self.god.stub_class(harness_standalone, "harness_standalone") 20 21 harness_standalone.harness_standalone.expect_new(job) 22 harness.select(None, job) 23 self.god.check_playback() 24 25 26 def test_select_standalone(self): 27 job = object() 28 self.god.stub_class(harness_standalone, "harness_standalone") 29 30 harness_standalone.harness_standalone.expect_new(job) 31 harness.select('standalone', job) 32 self.god.check_playback() 33 34 35 def test_select_ABAT(self): 36 job = object() 37 self.god.stub_class(harness_ABAT, "harness_ABAT") 38 39 harness_ABAT.harness_ABAT.expect_new(job) 40 harness.select('ABAT', job) 41 self.god.check_playback() 42 43 44if __name__ == "__main__": 45 unittest.main() 46