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