blob: 44c2847ffcd4bbb07489d69a56e5f6bbc7f48ab0 [file] [log] [blame]
mbligh4046d5c2008-06-11 22:16:31 +00001#!/usr/bin/python
2import unittest
3import common
4import harness, harness_standalone, harness_ABAT
5from autotest_lib.client.common_lib.test_utils import mock
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()