54 lines
2.0 KiB
Python
54 lines
2.0 KiB
Python
# (C) Copyright 2017 Hewlett Packard Enterprise Development LP
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
from unittest import TestCase
|
|
from eventlet.green import time
|
|
|
|
from monasca_agent.collector.checks.services_checks import ServicesCheck
|
|
|
|
from monasca_agent.collector.checks.services_checks import Status
|
|
|
|
class DummyServiceCheck(ServicesCheck):
|
|
def __init__(self, name, init_config, agent_config, instances=None):
|
|
super(DummyServiceCheck, self).__init__(name, init_config, agent_config, instances)
|
|
|
|
def _check(self, instance):
|
|
w = instance.get('service_wait', 5)
|
|
time.sleep(w)
|
|
return Status.UP, "UP"
|
|
|
|
def set_timeout(self, timeout):
|
|
self.timeout = timeout
|
|
|
|
class TestServicesCheck(TestCase):
|
|
def setUp(self):
|
|
TestCase.setUp(self)
|
|
pass
|
|
|
|
def test_service_check_timeout(self):
|
|
|
|
init_config = {}
|
|
agent_config = {'timeout': 4}
|
|
instances = []
|
|
for wait in [1,2, 6, 8]:
|
|
instances.append({'service_wait': wait, 'name' : 'dummy %d' % wait})
|
|
self.dummy_service = DummyServiceCheck("dummy service", init_config, agent_config, instances=instances)
|
|
self.dummy_service.run()
|
|
time.sleep(10)
|
|
self.assertEqual(self.dummy_service.statuses['dummy 1'][0], Status.UP)
|
|
self.assertEqual(self.dummy_service.statuses['dummy 2'][0], Status.UP)
|
|
|
|
self.assertEqual(self.dummy_service.statuses['dummy 6'][0], "FAILURE")
|
|
self.assertEqual(self.dummy_service.statuses['dummy 8'][0], "FAILURE")
|
|
|