Fixing tests after rebasing

Author: johannwagner

cosmo/clients/netbox.py

@@ -1,8 +1,4 @@
-import json
 import time
-from multiprocessing.pool import Pool
-from string import Template
-from urllib.parse import urlencode
 
 import requests
 
@@ -15,12 +11,12 @@ def __init__(self, url, token):
         self.url = url
         self.token = token
 
-        v = self.query_version()
+        self.version = self.query_version()
 
-        if v.startswith("wc_3."):
+        if self.version.startswith("wc_3."):
             print("[INFO] Using version 3.x strategy...")
             self.child_client = NetboxV3Strategy(url, token)
-        elif v.startswith("4."):
+        elif self.version.startswith("4."):
             print("[INFO] Using version 4.x strategy...")
             self.child_client = NetboxV4Strategy(url, token)
         else:

cosmo/netboxclient.py

@@ -1,7 +1,7 @@
 import pytest
 
 import cosmo.tests.utils as utils
-from cosmo.netboxclient import NetboxClient
+from cosmo.clients.netbox import NetboxClient
 
 TEST_URL = 'https://netbox.example.com'
 TEST_TOKEN = 'token123'
@@ -21,6 +21,7 @@ def test_case_get_data(mocker):
         "device_list": [],
         "l2vpn_list": [],
         "vrf_list": [],
+        "loopbacks": {},
     }
     utils.RequestResponseMock.patchNetboxClient(mocker)
 

cosmo/tests/test_netboxclient.py

@@ -52,7 +52,8 @@ def patchPostFunc(url, json, **kwargs):
             request_lists = [
                 "device_list",
                 "vrf_list",
-                "l2vpn_list"
+                "l2vpn_list",
+                "interface_list"
             ]
             retVal = dict()