diff --git a/test/test_opsdb.py b/test/test_opsdb.py index 5685b2a82604e04c0e8d2a1f8b597a301ca7c4ec..793543da7da1a49ee3cb97f348fe8767663548ee 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)