stash

core/tickets/tests/v2/test_tickets.py

@@ -2,16 +2,24 @@ from datetime import datetime, timedelta
 
 from django.test import TestCase, Client
 
+from authentication.models import ExtendedUser
 from mail.models import Email
 from tickets.models import IssueThread, StateChange, Comment
-
-client = Client()
+from django.contrib.auth.models import User
+from knox.models import AuthToken
 
 
 class IssueApiTest(TestCase):
 
+    def setUp(self):
+        super().setUp()
+        self.user = ExtendedUser.objects.create_user('testuser', 'test', 'test')
+        self.user.save()
+        self.token = AuthToken.objects.create(user=self.user)
+        self.client = Client(headers={'Authorization': 'Token ' + self.token[1]})
+
     def test_issues_empty(self):
-        response = client.get('/api/2/tickets/')
+        response = self.client.get('/api/2/tickets/')
         self.assertEqual(response.status_code, 200)
         self.assertEqual(response.json(), [])
 
@@ -48,7 +56,7 @@ class IssueApiTest(TestCase):
             timestamp=now + timedelta(seconds=3),
         )
 
-        response = client.get('/api/2/tickets/')
+        response = self.client.get('/api/2/tickets/')
         self.assertEqual(response.status_code, 200)
         self.assertEqual(len(response.json()), 1)
         self.assertEqual(response.json()[0]['id'], issue.id)