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 with mock.patch('functest.utils.functest_utils.get_parameter_from_yaml'):
  20     from functest.opnfv_tests.features import barometer
  21
  22
  23 class BarometerTesting(unittest.TestCase):
  24
  25     _case_name = "barometercollectd"
  26     _project_name = "barometer"
  27
  28     def setUp(self):
  29         self.barometer = barometer.BarometerCollectd(
  30             case_name=self._case_name, project_name=self._project_name)
  31
  32     def test_init(self):
  33         self.assertEqual(self.barometer.project_name, self._project_name)
  34         self.assertEqual(self.barometer.case_name, self._case_name)
  35
  36     def test_run_ko(self):
  37         sys.modules['baro_tests'].collectd.main = mock.Mock(return_value=1)
  38         self.assertEqual(self.barometer.run(),
  39                          testcase.TestCase.EX_RUN_ERROR)
  40
  41     def test_run(self):
  42         sys.modules['baro_tests'].collectd.main = mock.Mock(return_value=0)
  43         self.assertEqual(self.barometer.run(), testcase.TestCase.EX_OK)
  44
  45
  46 if __name__ == "__main__":
  47     logging.disable(logging.CRITICAL)
  48     unittest.main(verbosity=2)