55 lines
2.6 KiB
Python
55 lines
2.6 KiB
Python
from django.test import TestCase, Client
|
|
from django.contrib.auth.models import Permission
|
|
|
|
from authentication.models import ExtendedUser
|
|
from files.models import File
|
|
from inventory.models import Event, Container, Item
|
|
from knox.models import AuthToken
|
|
|
|
|
|
class FileTestCase(TestCase):
|
|
|
|
def setUp(self):
|
|
super().setUp()
|
|
self.user = ExtendedUser.objects.create_user('testuser', 'test', 'test')
|
|
self.user.user_permissions.add(*Permission.objects.all())
|
|
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 = 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)
|
|
|
|
def test_one_file(self):
|
|
import base64
|
|
item = File.objects.create(data="data:text/plain;base64," + base64.b64encode(b"foo").decode('utf-8'))
|
|
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)
|
|
|
|
def test_create_file(self):
|
|
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 = 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)
|
|
self.assertEqual(len(response.json()['hash']), 64)
|
|
|
|
def test_delete_file(self):
|
|
import base64
|
|
item = Item.objects.create(container=self.box, event=self.event, description='1')
|
|
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 = self.client.delete(f'/api/2/files/{file.hash}/')
|
|
self.assertEqual(response.status_code, 204)
|