Skip to content
Snippets Groups Projects
Commit 37915f63 authored by Robert Latta's avatar Robert Latta
Browse files

Added test for get_circuits. Removed unused code

parent 0d552af7
No related branches found
No related tags found
No related merge requests found
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)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment