# (C) Copyright 2017 Hewlett Packard Enterprise Development LP

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")