Add Tests for prefix_token

Fix #177
Changes to be committed:
	modified:   api/views.py
	new file:   tests/test_views/test_api_prefixes_token.py

Author: HadleyKing

api/views.py

@@ -1529,10 +1529,11 @@ class ApiPrefixesToken(APIView):
 
     --------------------
 
-    Get all available prefixes and their associated permissions for a given token.
-    The word 'Token' must be included in the header.
+    Get all available prefixes and their associated permissions for a given
+    token. The word 'Token' must be included in the header.
 
-    For example: 'Token 627626823549f787c3ec763ff687169206626149'.
+    For example: 'Token 627626823549f787c3ec763ff687169206626149'. Using that
+    token will return an empty list, as that is test user.
     """
 
     auth = [
@@ -1547,18 +1548,20 @@ class ApiPrefixesToken(APIView):
     @swagger_auto_schema(
         manual_parameters=auth,
         responses={
-            200: "The Authorization header was provided and available prefixes were returned.",
-            400: "The Authorization header was not provided.",
+            200: "The available prefixes were returned.",
+            401: "The authorization header was not provided.",
+            403: "Invalid token.",
         },
         tags=["Prefix Management"],
     )
     def post(self, request) -> Response:
         if "Authorization" in request.headers:
-            # Pass the request to the handling function
-            # Source: https://stackoverflow.com/a/31813810
             return post_api_prefixes_token_flat(request=request)
         else:
-            return Response(status=status.HTTP_400_BAD_REQUEST)
+            return Response(
+                data={"detail": "The authorization header was not provided."},
+                status=status.HTTP_401_UNAUTHORIZED
+            )
 
 
 class ApiPrefixesTokenFlat(APIView):

tests/test_views/test_api_prefixes_token.py

@@ -0,0 +1,46 @@
+
+#!/usr/bin/env python3
+
+"""Prefixes token
+Tests for 'Successful request' (200), 
+'forbiddden' (403)
+"""
+
+from django.test import TestCase
+from rest_framework.test import APIClient
+from rest_framework.authtoken.models import Token
+from django.contrib.auth.models import User
+from rest_framework.test import APITestCase
+
+
+class PrefixesTokenTestCase(APITestCase):
+    fixtures = ['tests/fixtures/test_data']
+    
+    def setUp(self):
+        self.client = APIClient()
+    
+    def test_success_response(self):
+        """The available prefixes were returned. (200)"""
+        
+        token = Token.objects.get(
+            user=User.objects.get(username='bco_api_user')
+        ).key
+
+        self.client.credentials(HTTP_AUTHORIZATION='Token ' + token)
+        response = self.client.post('/api/prefixes/token/',data={})
+        self.assertEqual(response.status_code, 200)
+
+            
+    # def test_unauthorized_response(self):
+    #     """The authorization header was not provided. (401)"""
+
+    #     self.client.credentials()
+    #     response = self.client.post('/api/prefixes/token/')
+    #     self.assertEqual(response.status_code, 401)
+
+    def test_invalid_token(self):
+        """Invalid token(403)"""
+        
+        self.client.credentials(HTTP_AUTHORIZATION='Token ' + "token")
+        response = self.client.post('/api/prefixes/token/')
+        self.assertEqual(response.status_code, 403)