add user agent and version check to server (#6)

tests/test_api_compat.py

@@ -0,0 +1,88 @@
+import json
+import pytest
+
+from util import ClientCase
+from test_api import setup_module, teardown_module
+from peeringdb_server.models import REFTAG_MAP, User
+from rest_framework.test import APIClient, force_authenticate
+
+
+class TestAPIClientCompat(ClientCase):
+    @classmethod
+    def setUpTestData(cls):
+        super(TestAPIClientCompat, cls).setUpTestData()
+        cls.superuser = User.objects.create_user("su", "su@localhost", "su",
+                                                 is_superuser=True)
+        cls.org = REFTAG_MAP["org"].objects.create(name="Test Org",
+                                                   status="ok")
+
+    @property
+    def expected_compat_err_str(self):
+        return "Your client version is incompatible with server version of the api, please install peeringdb>={},<={} {}>={},<={}".format(
+            "0.6", "0.6.5", "django_peeringdb", "0.6", "0.6.5")
+
+    def _compat(self, ua_c, ua_be, error):
+        if ua_c and ua_be:
+            useragent = "PeeringDB/{} django_peeringdb/{}".format(ua_c, ua_be)
+            self.client = APIClient(HTTP_USER_AGENT=useragent)
+        else:
+            self.client = APIClient()
+        self.client.force_authenticate(self.superuser)
+
+        r = self.client.get("/api/org/1", format="json")
+        content = json.loads(r.content)
+        if error:
+            assert r.status_code == 400
+            assert content["meta"]["error"] == self.expected_compat_err_str
+        else:
+            assert r.status_code == 200
+
+        r = self.client.post("/api/net", {
+            "org_id": 1,
+            "name": "Test net",
+            "asn": 9000000
+        }, format="json")
+        content = json.loads(r.content)
+        if error:
+            assert r.status_code == 400
+            assert content["meta"]["error"] == self.expected_compat_err_str
+            net = {"id": 1}
+        else:
+            assert r.status_code == 201
+            net = content["data"][0]
+            del net["org"]
+
+        r = self.client.put("/api/net/{}".format(net["id"]), net,
+                            format="json")
+        content = json.loads(r.content)
+        if error:
+            assert r.status_code == 400
+            assert content["meta"]["error"] == self.expected_compat_err_str
+        else:
+            assert r.status_code == 200
+
+        r = self.client.delete("/api/net/{}".format(net["id"]), {},
+                               format="json")
+        if error:
+            content = json.loads(r.content)
+            assert r.status_code == 400
+            assert content["meta"]["error"] == self.expected_compat_err_str
+        else:
+            assert r.status_code == 204
+
+        REFTAG_MAP["net"].objects.all().delete()
+
+    def test_incompatible(self):
+        self._compat("0.5.0", "0.4.0", True)
+        self._compat("0.6.0", "0.5.0", True)
+        self._compat("0.5.0", "0.6.0", True)
+        self._compat("0.7.0", "0.6.0", True)
+        self._compat("0.6.0", "0.7.0", True)
+
+    def test_compatible(self):
+        self._compat("0.6.0", "0.6.0", False)
+        self._compat("0.6", "0.6", False)
+        self._compat("0.6.1", "0.6.1", False)
+        self._compat("0.6", "0.6.1", False)
+        self._compat("0.6.1", "0.6", False)
+        self._compat(None, None, False)