stash

core/files/api_v1.py

@@ -16,6 +16,8 @@ class FileViewSet(viewsets.ModelViewSet):
     serializer_class = FileSerializer
     queryset = File.objects.all()
     lookup_field = 'hash'
+    permission_classes = []
+    authentication_classes = []
 
 
 router = routers.SimpleRouter(trailing_slash=False)

core/files/migrations/0001_initial.py

@@ -1,4 +1,4 @@
-# Generated by Django 4.2.7 on 2023-11-18 11:28
+# Generated by Django 4.2.7 on 2023-12-09 02:13
 
 from django.db import migrations, models
 import django.db.models.deletion

core/files/tests/v2/test_files.py

@@ -1,22 +1,26 @@
 from django.test import TestCase, Client
 
+from authentication.models import ExtendedUser
 from files.models import File
 from inventory.models import Event, Container, Item
-
-client = Client()
+from knox.models import AuthToken
 
 
 class FileTestCase(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]})
         self.event = Event.objects.create(slug='EVENT', name='Event')
         self.box = Container.objects.create(name='BOX')
 
     def test_list_files(self):
         import base64
         item = File.objects.create(data="data:text/plain;base64," + base64.b64encode(b"foo").decode('utf-8'))
-        response = client.get('/api/2/files/')
+        response = self.client.get('/api/2/files/')
         self.assertEqual(response.status_code, 200)
         self.assertEqual(response.json()[0]['hash'], item.hash)
         self.assertEqual(len(response.json()[0]['hash']), 64)
@@ -24,7 +28,7 @@ class FileTestCase(TestCase):
     def test_one_file(self):
         import base64
         item = File.objects.create(data="data:text/plain;base64," + base64.b64encode(b"foo").decode('utf-8'))
-        response = client.get(f'/api/2/files/{item.hash}/')
+        response = self.client.get(f'/api/2/files/{item.hash}/')
         self.assertEqual(response.status_code, 200)
         self.assertEqual(response.json()['hash'], item.hash)
         self.assertEqual(len(response.json()['hash']), 64)
@@ -33,7 +37,7 @@ class FileTestCase(TestCase):
         import base64
         Item.objects.create(container=self.box, event=self.event, description='1')
         item = Item.objects.create(container=self.box, event=self.event, description='2')
-        response = client.post('/api/2/files/',
+        response = self.client.post('/api/2/files/',
                                {'data': "data:text/plain;base64," + base64.b64encode(b"foo").decode('utf-8')},
                                content_type='application/json')
         self.assertEqual(response.status_code, 201)
@@ -45,5 +49,5 @@ class FileTestCase(TestCase):
         File.objects.create(item=item, data="data:text/plain;base64," + base64.b64encode(b"foo").decode('utf-8'))
         file = File.objects.create(item=item, data="data:text/plain;base64," + base64.b64encode(b"bar").decode('utf-8'))
         self.assertEqual(len(File.objects.all()), 2)
-        response = client.delete(f'/api/2/files/{file.hash}/')
+        response = self.client.delete(f'/api/2/files/{file.hash}/')
         self.assertEqual(response.status_code, 204)