diff --git a/test/test_monitoring.py b/test/test_monitoring.py new file mode 100644 index 0000000000000000000000000000000000000000..c8bb739fe6b2788a3a8fa0bcd52bda7285f62065 --- /dev/null +++ b/test/test_monitoring.py @@ -0,0 +1,36 @@ +import contextlib +import os +from unittest import mock +from inventory_provider.tasks import monitor + +CONNECTION_FINGERPRINT = "SDF@#$@#" + + +@contextlib.contextmanager +def mocked_connection(): + yield CONNECTION_FINGERPRINT + + +class MockedState(): + def __init__(self): + pass + + def event(self, e): + pass + + +class MockedReceiver(): + def __init__(self, connection, handlers): + assert connection == CONNECTION_FINGERPRINT + self.handlers = handlers + + def capture(self, **kwargs): + pass + + +@mock.patch('inventory_provider.tasks.monitor.app.events.State', MockedState) +@mock.patch('inventory_provider.tasks.monitor.app.connection', mocked_connection) +def test_latchdb(data_config_filename, mocked_redis): + + os.environ['INVENTORY_PROVIDER_CONFIG_FILENAME'] = data_config_filename + monitor.run()