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