client/bin/harness_unittest.py

   1 #!/usr/bin/python
   2 import unittest
   3 import common
   4 from autotest_lib.client.common_lib.test_utils import mock
   5 from autotest_lib.client.bin import harness, harness_standalone, harness_ABAT
   6
   7
   8 class 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_args = ''
  22         harness_standalone.harness_standalone.expect_new(job, harness_args)
  23         harness.select(None, job, harness_args)
  24         self.god.check_playback()
  25
  26
  27     def test_select_standalone(self):
  28         job = object()
  29         self.god.stub_class(harness_standalone, "harness_standalone")
  30
  31         harness_args = ''
  32         harness_standalone.harness_standalone.expect_new(job, harness_args)
  33         harness.select('standalone', job, harness_args)
  34         self.god.check_playback()
  35
  36
  37     def test_select_ABAT(self):
  38         job = object()
  39         self.god.stub_class(harness_ABAT, "harness_ABAT")
  40
  41         harness_args = ''
  42         harness_ABAT.harness_ABAT.expect_new(job, harness_args)
  43         harness.select('ABAT', job, harness_args)
  44         self.god.check_playback()
  45
  46
  47 if  __name__ == "__main__":
  48     unittest.main()