From 8fa5944c9e4fd7c3d696cc0b29ce4ff62b9ac550 Mon Sep 17 00:00:00 2001 From: Erik Reid <erik.reid@geant.org> Date: Wed, 3 Jun 2020 12:56:40 +0200 Subject: [PATCH] unittest wip --- test/test_monitoring.py | 36 ++++++++++++++++++++++++++++++++++++ 1 file changed, 36 insertions(+) create mode 100644 test/test_monitoring.py diff --git a/test/test_monitoring.py b/test/test_monitoring.py new file mode 100644 index 00000000..c8bb739f --- /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() -- GitLab