From e9c2b96a2528bcd921d4b37dae1d19bbe6b2d87a Mon Sep 17 00:00:00 2001 From: Erik Reid <erik.reid@geant.org> Date: Fri, 4 Sep 2020 17:42:19 +0200 Subject: [PATCH] pep8 --- inventory_provider/routes/common.py | 5 ++++- inventory_provider/routes/poller.py | 7 ++++--- test/per_router/test_poller_routes.py | 4 ++-- test/test_general_poller_routes.py | 5 +++-- 4 files changed, 13 insertions(+), 8 deletions(-) diff --git a/inventory_provider/routes/common.py b/inventory_provider/routes/common.py index ecabd233..275b9765 100644 --- a/inventory_provider/routes/common.py +++ b/inventory_provider/routes/common.py @@ -160,7 +160,10 @@ def _redis_client_proc(key_queue, value_queue, config_params, doc_type): def _load_redis_docs( - config_params, key_pattern, num_threads=10, doc_type=_DECODE_TYPE_JSON): + config_params, + key_pattern, + num_threads=10, + doc_type=_DECODE_TYPE_JSON): """ load all docs from redis and decode as `doc_type` diff --git a/inventory_provider/routes/poller.py b/inventory_provider/routes/poller.py index 583f2a04..b6f1f66f 100644 --- a/inventory_provider/routes/poller.py +++ b/inventory_provider/routes/poller.py @@ -2,7 +2,7 @@ import json import logging import re -from flask import Blueprint, Response, jsonify, current_app +from flask import Blueprint, Response, current_app from inventory_provider import juniper from inventory_provider.routes import common @@ -73,7 +73,7 @@ def _load_services(hostname=None): m = re.match(r'^opsdb:interface_services:([^:]+):(.+)', doc['key']) if not m: - logger.warning(f'can''t parse redis service key {doc["key"]}') + logger.warning(f'can\'t parse redis service key {doc["key"]}') # there are some weird records (dtn*, dp1*) continue @@ -84,6 +84,7 @@ def _load_services(hostname=None): return result + def _load_interfaces(hostname): key_pattern = f'netconf:{hostname}*' if hostname else 'netconf:*' for doc in common.load_xml_docs( @@ -111,7 +112,7 @@ def _load_poller_interfaces(hostname=None): snmp_indexes = _load_snmp_indexes(hostname) bundles = _load_interface_bundles(hostname) - services= _load_services(hostname) + services = _load_services(hostname) for ifc in _load_interfaces(hostname): diff --git a/test/per_router/test_poller_routes.py b/test/per_router/test_poller_routes.py index 8d380553..6879805e 100644 --- a/test/per_router/test_poller_routes.py +++ b/test/per_router/test_poller_routes.py @@ -32,7 +32,7 @@ INTERFACE_LIST_SCHEMA = { 'minimum': 1 }, 'bundle': { - 'type' : 'array', + 'type': 'array', 'items': {'type': 'string'} }, 'bundle-parents': { @@ -59,7 +59,7 @@ INTERFACE_LIST_SCHEMA = { def test_router_interfaces(router, client): rv = client.post( - "/poller/interfaces/" + router, + f'/poller/interfaces/{router}', headers=DEFAULT_REQUEST_HEADERS) assert rv.status_code == 200 diff --git a/test/test_general_poller_routes.py b/test/test_general_poller_routes.py index e14491ab..ada06035 100644 --- a/test/test_general_poller_routes.py +++ b/test/test_general_poller_routes.py @@ -59,7 +59,7 @@ INTERFACE_LIST_SCHEMA = { 'minimum': 1 }, 'bundle': { - 'type' : 'array', + 'type': 'array', 'items': {'type': 'string'} }, 'bundle-parents': { @@ -83,6 +83,7 @@ INTERFACE_LIST_SCHEMA = { 'items': {'$ref': '#/definitions/interface'} } + @pytest.mark.parametrize('category', ['mdvpn', 'lhcone', 'MDVpn', 'LHCONE']) def test_service_category(client, mocked_worker_module, category): worker._build_service_category_interface_list() @@ -107,7 +108,7 @@ def test_service_category_not_found(client, mocked_worker_module, category): def test_get_all_interfaces(client): rv = client.get( - f'/poller/interfaces', + '/poller/interfaces', headers=DEFAULT_REQUEST_HEADERS) assert rv.status_code == 200 assert rv.is_json -- GitLab