client/bin/harness_unittest.py

#!/usr/bin/python
import unittest
import common
from autotest_lib.client.common_lib.test_utils import mock
from autotest_lib.client.bin import harness, harness_standalone, harness_ABAT


class harness_unittest(unittest.TestCase):
    def setUp(self):
        self.god = mock.mock_god()


    def tearDown(self):
        self.god.unstub_all()


    def test_select_none(self):
        job = object()
        self.god.stub_class(harness_standalone, "harness_standalone")

        harness_standalone.harness_standalone.expect_new(job)
        harness.select(None, job)
        self.god.check_playback()


    def test_select_standalone(self):
        job = object()
        self.god.stub_class(harness_standalone, "harness_standalone")

        harness_standalone.harness_standalone.expect_new(job)
        harness.select('standalone', job)
        self.god.check_playback()


    def test_select_ABAT(self):
        job = object()
        self.god.stub_class(harness_ABAT, "harness_ABAT")

        harness_ABAT.harness_ABAT.expect_new(job)
        harness.select('ABAT', job)
        self.god.check_playback()


if  __name__ == "__main__":
    unittest.main()