Merge pull request #261 from markkuleinio/fix-version-session

Fixes #260: Change Request.get_version() to use self.http_session

Author: Zach Moody

pynetbox/core/query.py

@@ -197,7 +197,7 @@ def get_version(self):
         headers = {
             "Content-Type": "application/json;",
         }
-        req = requests.get(self.normalize_url(self.base), headers=headers,)
+        req = self.http_session.get(self.normalize_url(self.base), headers=headers,)
         if req.ok:
             return req.headers.get("API-Version", "")
         else:

tests/test_api.py

@@ -51,7 +51,10 @@ class ResponseHeadersWithVersion:
         headers = {"API-Version": "1.999"}
         ok = True
 
-    @patch("requests.get", return_value=ResponseHeadersWithVersion())
+    @patch(
+        "pynetbox.core.query.requests.sessions.Session.get",
+        return_value=ResponseHeadersWithVersion(),
+    )
     def test_api_version(self, *_):
         api = pynetbox.api(host,)
         self.assertEqual(api.version, "1.999")
@@ -60,7 +63,10 @@ class ResponseHeadersWithoutVersion:
         headers = {}
         ok = True
 
-    @patch("requests.get", return_value=ResponseHeadersWithoutVersion())
+    @patch(
+        "pynetbox.core.query.requests.sessions.Session.get",
+        return_value=ResponseHeadersWithoutVersion(),
+    )
     def test_api_version_not_found(self, *_):
         api = pynetbox.api(host,)
         self.assertEqual(api.version, "")