peeringdb-peeringdb/tests/test_api_compat.py

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().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 = f"PeeringDB/{ua_c} django_peeringdb/{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)