From cca39135abc6ab1045b03fe8123808c2a5ebcd78 Mon Sep 17 00:00:00 2001 From: Erik Reid <erik.reid@geant.org> Date: Wed, 3 Jun 2020 13:30:20 +0200 Subject: [PATCH] pep8 --- inventory_provider/tasks/common.py | 1 + inventory_provider/tasks/worker.py | 3 ++- test/test_monitoring.py | 7 +++++-- 3 files changed, 8 insertions(+), 3 deletions(-) diff --git a/inventory_provider/tasks/common.py b/inventory_provider/tasks/common.py index 2d7f699f..ac85afb7 100644 --- a/inventory_provider/tasks/common.py +++ b/inventory_provider/tasks/common.py @@ -66,6 +66,7 @@ TASK_LOG_SCHEMA = { 'additionalProperties': False } + def get_latch(r): latch = r.get('db:latch') if latch is None: diff --git a/inventory_provider/tasks/worker.py b/inventory_provider/tasks/worker.py index cf861ded..48058c7c 100644 --- a/inventory_provider/tasks/worker.py +++ b/inventory_provider/tasks/worker.py @@ -111,7 +111,8 @@ def snmp_refresh_interfaces(self, hostname, community): def netconf_refresh_config(self, hostname): try: - netconf_doc = juniper.load_config(hostname, InventoryTask.config["ssh"]) + netconf_doc = juniper.load_config( + hostname, InventoryTask.config["ssh"]) netconf_str = etree.tostring(netconf_doc, encoding='unicode') except ConnectionError: msg = f'error loading netconf data from {hostname}' diff --git a/test/test_monitoring.py b/test/test_monitoring.py index f25d3793..66fa8285 100644 --- a/test/test_monitoring.py +++ b/test/test_monitoring.py @@ -21,6 +21,7 @@ TEST_LOG_EVENTS = [ {'type': 'task-error', 'uuid': 'AAAA', 'clock': 777} ] + @contextlib.contextmanager def mocked_connection(): yield CONNECTION_FINGERPRINT @@ -56,8 +57,10 @@ def backend_db(): @mock.patch('inventory_provider.tasks.monitor.app.events.State', MockedState) -@mock.patch('inventory_provider.tasks.monitor.app.connection', mocked_connection) -@mock.patch('inventory_provider.tasks.monitor.app.events.Receiver', MockedReceiver) +@mock.patch( + 'inventory_provider.tasks.monitor.app.connection', mocked_connection) +@mock.patch( + 'inventory_provider.tasks.monitor.app.events.Receiver', MockedReceiver) def test_latchdb(data_config_filename, mocked_redis): os.environ['INVENTORY_PROVIDER_CONFIG_FILENAME'] = data_config_filename -- GitLab