From 37915f631776756ce695a0c0171975f31a0a2edc Mon Sep 17 00:00:00 2001
From: Robert Latta <robert.latta@geant.org>
Date: Fri, 4 Jan 2019 10:59:51 +0000
Subject: [PATCH] Added test for get_circuits. Removed unused code
---
test/test_opsdb.py | 40 ++++++++++++++--------------------------
1 file changed, 14 insertions(+), 26 deletions(-)
diff --git a/test/test_opsdb.py b/test/test_opsdb.py
index 5685b2a8..793543da 100644
--- a/test/test_opsdb.py
+++ b/test/test_opsdb.py
@@ -1,28 +1,6 @@
import inventory_provider.opsdb
-class MockedCrs:
- def __enter__(self):
- return "hello"
-
- def __exit__(self):
- return "Goodbye"
-
- def close(self):
- pass
-
- def execute(self):
- pass
-
-#
-# def test_get_circuits_to_monitor(mocker):
-# mocked_get_cursor = mocker.patch('inventory_provider.opsdb.cursor')
-# mocked_get_cursor.return_value.__enter__.return_value = None
-# mocked_execute = mocked_get_cursor. \
-# return_value.__enter__.return_value.execute
-# pass
-
-
def test_update_fields(mocker):
mocker.patch("inventory_provider.opsdb._juniper_field_update")
t = {"manufacturer": "juniper"}
@@ -121,9 +99,19 @@ def test_juniper_field_update():
def test_get_circuits(mocker):
mocked_get_cursor = mocker.patch("inventory_provider.opsdb.cursor")
- mocked_execute = mocked_get_cursor. \
- return_value.__enter__.return_value.execute
+ mocked_get_cursor.return_value.__enter__.return_value.execute
mocked_convert_to_dict = mocker.patch(
"inventory_provider.opsdb._convert_to_dict")
- mocked_convert_to_dict.return_value = ""
- pass
+ i = {"manufacturer": "infinera"}
+ j = {"manufacturer": "juniper"}
+ mocked_convert_to_dict.return_value = [i, j]
+
+ mocked_infinera_update = mocker.patch(
+ "inventory_provider.opsdb._infinera_field_update")
+ mocked_juniper_update = mocker.patch(
+ "inventory_provider.opsdb._juniper_field_update")
+
+ inventory_provider.opsdb.get_circuits(None)
+
+ mocked_infinera_update.assert_called_once_with(i)
+ mocked_juniper_update.assert_called_once_with(j)
--
GitLab