Add test for Api.status()

netbox-community · Dec 27, 2020 · 7e5f473 · 7e5f473
1 parent 6c1d1e2
commit 7e5f473
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/tests/test_api.py b/tests/test_api.py
@@ -70,3 +70,20 @@ class ResponseHeadersWithoutVersion:
     def test_api_version_not_found(self, *_):
         api = pynetbox.api(host,)
         self.assertEqual(api.version, "")
+
+
+class ApiStatusTestCase(unittest.TestCase):
+    class ResponseWithStatus:
+        ok = True
+        def json(self):
+            return {
+                "netbox-version": "0.9.9",
+            }
+
+    @patch(
+        "pynetbox.core.query.requests.sessions.Session.get",
+        return_value=ResponseWithStatus(),
+    )
+    def test_api_status(self, *_):
+        api = pynetbox.api(host,)
+        self.assertEqual(api.status()["netbox-version"], "0.9.9")