Merge pull request #319 from ignatenkobrain/master

Fix tests
netbox-community · Dec 30, 2020 · 7d93576 · 7d93576
2 parents 636ddeb + 8a3ff35
commit 7d93576
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/tests/test_api.py b/tests/test_api.py
@@ -81,7 +81,7 @@ def json(self):
             }
 
     @patch(
-        "pynetbox.core.query.requests.sessions.Session.get",
+        "requests.sessions.Session.get",
         return_value=ResponseWithStatus(),
     )
     def test_api_status(self, *_):

diff --git a/tests/test_dcim.py b/tests/test_dcim.py
@@ -326,7 +326,7 @@ def test_get_all(self, mock):
         )
 
     @patch(
-        "pynetbox.core.query.requests.sessions.Session.get",
+        "requests.sessions.Session.get",
         side_effect=[
             Response(fixture="dcim/interface.json"),
             Response(fixture="dcim/interface_trace.json"),

diff --git a/tests/test_users.py b/tests/test_users.py
@@ -25,7 +25,7 @@ class Tests(unittest.TestCase):
 
         def test_get_all(self):
             with patch(
-                "pynetbox.core.query.requests.sessions.Session.get",
+                "requests.sessions.Session.get",
                 return_value=Response(fixture="{}/{}.json".format(self.app, self.name)),
             ) as mock:
                 ret = getattr(nb, self.name).all()
@@ -43,7 +43,7 @@ def test_get_all(self):
 
         def test_filter(self):
             with patch(
-                "pynetbox.core.query.requests.sessions.Session.get",
+                "requests.sessions.Session.get",
                 return_value=Response(fixture="{}/{}.json".format(self.app, self.name)),
             ) as mock:
                 ret = getattr(nb, self.name).filter(name="test")
@@ -61,7 +61,7 @@ def test_filter(self):
 
         def test_get(self):
             with patch(
-                "pynetbox.core.query.requests.sessions.Session.get",
+                "requests.sessions.Session.get",
                 return_value=Response(
                     fixture="{}/{}.json".format(self.app, self.name[:-1])
                 ),
@@ -83,7 +83,7 @@ class UsersTestCase(Generic.Tests):
     name = "users"
 
     @patch(
-        "pynetbox.core.query.requests.sessions.Session.get",
+        "requests.sessions.Session.get",
         return_value=Response(fixture="users/user.json"),
     )
     def test_repr(self, _):