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 yardstick.benchmark.scenarios.availability.monitor.monitor_command
  13
  14 import mock
  15 import unittest
  16
  17 from yardstick.benchmark.scenarios.availability.monitor import basemonitor
  18
  19
  20 @mock.patch('yardstick.benchmark.scenarios.availability.monitor.basemonitor.BaseMonitor')
  21 class MonitorMgrTestCase(unittest.TestCase):
  22
  23     def setUp(self):
  24         config = {
  25             'monitor_type': 'openstack-api',
  26             'key' : 'service_status'
  27         }
  28
  29         self.monitor_configs = []
  30         self.monitor_configs.append(config)
  31
  32     def test__MonitorMgr_setup_successful(self, mock_monitor):
  33         instance = basemonitor.MonitorMgr()
  34         instance.init_monitors(self.monitor_configs, None)
  35         instance.start_monitors()
  36         instance.wait_monitors()
  37
  38         ret = instance.verify_SLA()
  39
  40     def test_MonitorMgr_getitem(self, mock_monitor):
  41         monitorMgr = basemonitor.MonitorMgr()
  42         monitorMgr.init_monitors(self.monitor_configs, None)
  43         monitorIns = monitorMgr['service_status']
  44
  45 class BaseMonitorTestCase(unittest.TestCase):
  46
  47     class MonitorSimple(basemonitor.BaseMonitor):
  48         __monitor_type__ = "MonitorForTest"
  49         def setup(self):
  50             self.monitor_result = False
  51
  52         def monitor_func(self):
  53             return self.monitor_result
  54
  55     def setUp(self):
  56         self.monitor_cfg = {
  57             'monitor_type': 'MonitorForTest',
  58             'command_name': 'nova image-list',
  59             'monitor_time': 0.01,
  60             'sla': {'max_outage_time': 5}
  61         }
  62
  63     def test__basemonitor_start_wait_successful(self):
  64         ins = basemonitor.BaseMonitor(self.monitor_cfg, None)
  65         ins.start_monitor()
  66         ins.wait_monitor()
  67
  68
  69     def test__basemonitor_all_successful(self):
  70         ins = self.MonitorSimple(self.monitor_cfg, None)
  71         ins.setup()
  72         ins.run()
  73         ins.verify_SLA()
  74
  75     @mock.patch('yardstick.benchmark.scenarios.availability.monitor.basemonitor.multiprocessing')
  76     def test__basemonitor_func_false(self, mock_multiprocess):
  77         ins = self.MonitorSimple(self.monitor_cfg, None)
  78         ins.setup()
  79         mock_multiprocess.Event().is_set.return_value = False
  80         ins.run()
  81         ins.verify_SLA()
  82
  83     def test__basemonitor_getmonitorcls_successfule(self):
  84         cls = None
  85         try:
  86             cls = basemonitor.BaseMonitor.get_monitor_cls(self.monitor_cfg)
  87         except Exception:
  88             pass
  89         self.assertIsNone(cls)
  90