c3lf-system-3/core/files/tests/v1/test_files.py

from django.test import TestCase, Client
from django.core.files.base import ContentFile

from files.models import File
from inventory.models import Event, Container, Item

client = Client()


class FileTestCase(TestCase):

    def setUp(self):
        super().setUp()
        self.event = Event.objects.create(slug='EVENT', name='Event')
        self.box = Container.objects.create(name='BOX')

    def test_create_file_raw(self):
        from hashlib import sha256
        content = b"foo"
        chash = sha256(content).hexdigest()
        item = Item.objects.create(container=self.box, event=self.event, description='1')
        file = File.objects.create(file=ContentFile(b"foo"), mime_type='text/plain', hash=chash, item=item)
        file.save()
        self.assertEqual(1, len(File.objects.all()))
        self.assertEqual(content, File.objects.all()[0].file.read())
        self.assertEqual(chash, File.objects.all()[0].hash)

    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/1/files')
        self.assertEqual(response.status_code, 200)
        self.assertEqual(response.json()[0]['hash'], item.hash)
        self.assertEqual(len(response.json()[0]['hash']), 64)
        self.assertEqual(len(File.objects.all()), 1)
        self.assertEqual(File.objects.all()[0].file.read(), b"foo")

    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/1/file/{item.hash}')
        self.assertEqual(response.status_code, 200)
        self.assertEqual(response.json()['hash'], item.hash)
        self.assertEqual(len(response.json()['hash']), 64)
        self.assertEqual(len(File.objects.all()), 1)
        self.assertEqual(File.objects.all()[0].file.read(), b"foo")

    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 = client.post('/api/1/file',
                               {'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)
        self.assertEqual(len(File.objects.all()), 1)
        self.assertEqual(File.objects.all()[0].file.read(), b"foo")

    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 = client.delete(f'/api/1/file/{file.hash}')
        self.assertEqual(response.status_code, 204)