tests/unit/benchmark/scenarios/availability/test_basemonitor.py

   1 #!/usr/bin/env python
   2
   3 ##############################################################################
   4 # Copyright (c) 2015 Huawei Technologies Co.,Ltd and others.
   5 #
   6 # All rights reserved. This program and the accompanying materials
   7 # are made available under the terms of the Apache License, Version 2.0
   8 # which accompanies this distribution, and is available at
   9 # http://www.apache.org/licenses/LICENSE-2.0
  10 ##############################################################################
  11
  12 # Unittest for
  13 # yardstick.benchmark.scenarios.availability.monitor.monitor_command
  14
  15 from __future__ import absolute_import
  16 import mock
  17 import unittest
  18
  19 from yardstick.benchmark.scenarios.availability.monitor import basemonitor
  20
  21
  22 @mock.patch(
  23     'yardstick.benchmark.scenarios.availability.monitor.basemonitor'
  24     '.BaseMonitor')
  25 class MonitorMgrTestCase(unittest.TestCase):
  26
  27     def setUp(self):
  28         config = {
  29             'monitor_type': 'openstack-api',
  30             'key': 'service-status'
  31         }
  32
  33         self.monitor_configs = []
  34         self.monitor_configs.append(config)
  35
  36     def test__MonitorMgr_setup_successful(self, mock_monitor):
  37         instance = basemonitor.MonitorMgr()
  38         instance.init_monitors(self.monitor_configs, None)
  39         instance.start_monitors()
  40         instance.wait_monitors()
  41
  42         ret = instance.verify_SLA()
  43
  44     def test_MonitorMgr_getitem(self, mock_monitor):
  45         monitorMgr = basemonitor.MonitorMgr()
  46         monitorMgr.init_monitors(self.monitor_configs, None)
  47         monitorIns = monitorMgr['service-status']
  48
  49
  50 class BaseMonitorTestCase(unittest.TestCase):
  51
  52     class MonitorSimple(basemonitor.BaseMonitor):
  53         __monitor_type__ = "MonitorForTest"
  54
  55         def setup(self):
  56             self.monitor_result = False
  57
  58         def monitor_func(self):
  59             return self.monitor_result
  60
  61     def setUp(self):
  62         self.monitor_cfg = {
  63             'monitor_type': 'MonitorForTest',
  64             'command_name': 'nova image-list',
  65             'monitor_time': 0.01,
  66             'sla': {'max_outage_time': 5}
  67         }
  68
  69     def test__basemonitor_start_wait_successful(self):
  70         ins = basemonitor.BaseMonitor(self.monitor_cfg, None)
  71         ins.start_monitor()
  72         ins.wait_monitor()
  73
  74     def test__basemonitor_all_successful(self):
  75         ins = self.MonitorSimple(self.monitor_cfg, None)
  76         ins.setup()
  77         ins.run()
  78         ins.verify_SLA()
  79
  80     @mock.patch(
  81         'yardstick.benchmark.scenarios.availability.monitor.basemonitor'
  82         '.multiprocessing')
  83     def test__basemonitor_func_false(self, mock_multiprocess):
  84         ins = self.MonitorSimple(self.monitor_cfg, None)
  85         ins.setup()
  86         mock_multiprocess.Event().is_set.return_value = False
  87         ins.run()
  88         ins.verify_SLA()
  89
  90     def test__basemonitor_getmonitorcls_successfule(self):
  91         cls = None
  92         try:
  93             cls = basemonitor.BaseMonitor.get_monitor_cls(self.monitor_cfg)
  94         except Exception:
  95             pass
  96         self.assertIsNone(cls)