functest/tests/unit/features/test_barometer.py

   1 #!/usr/bin/env python
   2
   3 # Copyright (c) 2017 Orange and others.
   4 #
   5 # All rights reserved. This program and the accompanying materials
   6 # are made available under the terms of the Apache License, Version 2.0
   7 # which accompanies this distribution, and is available at
   8 # http://www.apache.org/licenses/LICENSE-2.0
   9
  10 # pylint: disable=missing-docstring
  11
  12 import logging
  13 import sys
  14 import unittest
  15
  16 import mock
  17
  18 from functest.core import testcase
  19 sys.modules['baro_tests'] = mock.Mock()  # noqa
  20 # pylint: disable=wrong-import-position
  21 from functest.opnfv_tests.features import barometer
  22
  23
  24 class BarometerTesting(unittest.TestCase):
  25
  26     _case_name = "barometercollectd"
  27     _project_name = "barometer"
  28
  29     def setUp(self):
  30         self.barometer = barometer.BarometerCollectd(
  31             case_name=self._case_name, project_name=self._project_name)
  32
  33     def test_init(self):
  34         self.assertEqual(self.barometer.project_name, self._project_name)
  35         self.assertEqual(self.barometer.case_name, self._case_name)
  36
  37     def test_run_ko(self):
  38         sys.modules['baro_tests'].collectd.main = mock.Mock(return_value=1)
  39         self.assertEqual(self.barometer.run(),
  40                          testcase.TestCase.EX_RUN_ERROR)
  41
  42     def test_run(self):
  43         sys.modules['baro_tests'].collectd.main = mock.Mock(return_value=0)
  44         self.assertEqual(self.barometer.run(), testcase.TestCase.EX_OK)
  45
  46
  47 if __name__ == "__main__":
  48     logging.disable(logging.CRITICAL)
  49     unittest.main(verbosity=2)