From 6d384217fc49624c8cea60efa6794cc02d3b5927 Mon Sep 17 00:00:00 2001
From: Erik Reid <erik.reid@geant.org>
Date: Thu, 3 Jan 2019 12:45:28 +0100
Subject: [PATCH] fix handling of snmp result list
---
test/test_celery_worker.py | 9 ++++++++-
1 file changed, 8 insertions(+), 1 deletion(-)
diff --git a/test/test_celery_worker.py b/test/test_celery_worker.py
index 86113eb5..7e882d73 100644
--- a/test/test_celery_worker.py
+++ b/test/test_celery_worker.py
@@ -2,6 +2,7 @@
just checks that the worker methods call the right functions
and some data ends up in the right place ... otherwise not very detailed
"""
+import json
import logging
from lxml import etree
@@ -17,6 +18,12 @@ class MockedRedis(object):
pass
def hset(self, name, key, value):
+ assert isinstance(value, str)
+ if key == 'netconf':
+ etree.fromstring(value)
+ elif key == 'snmp-interfaces':
+ obj = json.loads(value)
+ assert isinstance(obj, dict)
MockedRedis.db.setdefault(name, {})[key] = value
@@ -53,7 +60,7 @@ def test_snmp_refresh_interfaces(
mocked_worker_module, mocker, cached_test_data):
def _mocked_snmp_interfaces(hostname, community, _):
- return cached_test_data[hostname]["snmp-interfaces"]
+ return json.loads(cached_test_data[hostname]["snmp-interfaces"])
mocker.patch(
'inventory_provider.tasks.worker.snmp.get_router_interfaces',
--
GitLab