Add routes to test_project (#212)

* Add routes to test_project

* Add tests

* Format

Author: florimondmanca

test_project/project/urls.py

@@ -1,4 +1,11 @@
 from django.contrib import admin
 from django.urls import path
 
-urlpatterns = [path("admin/", admin.site.urls)]
+from . import views
+
+urlpatterns = [
+    path("admin/", admin.site.urls),
+    path("api/public/", views.PublicAPIView.as_view()),
+    path("api/protected/", views.ProtectedAPIView.as_view()),
+    path("api/protected/object/", views.ProtectedObjectAPIView.as_view()),
+]

test_project/project/views.py

@@ -0,0 +1,24 @@
+from rest_framework.request import Request
+from rest_framework.response import Response
+from rest_framework.views import APIView
+from test_project.heroes.permissions import HasHeroAPIKey
+
+
+class PublicAPIView(APIView):
+    def get(self, request: Request) -> Response:
+        return Response({"message": "Hello, world!"})
+
+
+class ProtectedAPIView(APIView):
+    permission_classes = [HasHeroAPIKey]
+
+    def get(self, request: Request) -> Response:
+        return Response({"message": "Hello, world!"})
+
+
+class ProtectedObjectAPIView(APIView):
+    permission_classes = [HasHeroAPIKey]
+
+    def get(self, request: Request) -> Response:
+        self.check_object_permissions(request, object())
+        return Response({"message": "Hello, world!"})

tests/conftest.py

@@ -39,8 +39,8 @@ def pytest_configure() -> None:
             ],
             "MIDDLEWARE": [
                 # Admin
-                "django.contrib.messages.middleware.MessageMiddleware",
                 "django.contrib.sessions.middleware.SessionMiddleware",
+                "django.contrib.messages.middleware.MessageMiddleware",
                 "django.contrib.auth.middleware.AuthenticationMiddleware",
             ],
             "ROOT_URLCONF": "test_project.project.urls",

tests/test_test_project.py

@@ -0,0 +1,22 @@
+import pytest
+from rest_framework.test import APIClient
+from test_project.heroes.models import Hero
+from test_project.heroes.permissions import HeroAPIKey
+
+
+@pytest.mark.django_db
+def test_test_project_routes() -> None:
+    batman = Hero.objects.create(name="Batman")
+    _, key = HeroAPIKey.objects.create_key(name="test", hero=batman)
+    headers = {"HTTP_AUTHORIZATION": f"Api-Key {key}"}
+
+    client = APIClient()
+
+    response = client.get("/api/public/", format="json")
+    assert response.status_code == 200
+
+    response = client.get("/api/protected/", format="json", **headers)
+    assert response.status_code == 200
+
+    response = client.get("/api/protected/object/", format="json", **headers)
+    assert response.status_code == 200