generated from coulomb/repo-seed
86 lines
3.2 KiB
Python
86 lines
3.2 KiB
Python
import io
|
|
|
|
import factory
|
|
import pytest
|
|
from django.core.files.uploadedfile import SimpleUploadedFile
|
|
from django.urls import reverse
|
|
|
|
from vergabe_teilnahme.apps.ausschreibungen.tests import AusschreibungFactory
|
|
|
|
from .models import Dokument
|
|
|
|
|
|
class DokumentFactory(factory.django.DjangoModelFactory):
|
|
class Meta:
|
|
model = Dokument
|
|
|
|
ausschreibung = factory.SubFactory(AusschreibungFactory)
|
|
dateiname = factory.Sequence(lambda n: f'dokument_{n}.pdf')
|
|
kategorie = 'intern'
|
|
status = 'entwurf'
|
|
version = '1.0'
|
|
|
|
|
|
def _pdf_file(name='test.pdf'):
|
|
return SimpleUploadedFile(name, b'%PDF-1.4 fake content', content_type='application/pdf')
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_dokument_upload_valid(client, tmp_path, settings):
|
|
settings.MEDIA_ROOT = tmp_path
|
|
a = AusschreibungFactory()
|
|
url = reverse('ausschreibungen:dokumente:upload', kwargs={'ausschreibung_id': a.pk})
|
|
response = client.post(url, {'datei': _pdf_file(), 'kategorie': 'intern', 'version': '1.0'})
|
|
assert response.status_code == 302
|
|
assert Dokument.objects.filter(ausschreibung=a).exists()
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_dokument_upload_invalid_extension(client, tmp_path, settings):
|
|
settings.MEDIA_ROOT = tmp_path
|
|
a = AusschreibungFactory()
|
|
url = reverse('ausschreibungen:dokumente:upload', kwargs={'ausschreibung_id': a.pk})
|
|
bad_file = SimpleUploadedFile('malware.exe', b'MZ bad', content_type='application/octet-stream')
|
|
response = client.post(url, {'datei': bad_file, 'kategorie': 'intern', 'version': '1.0'})
|
|
assert response.status_code == 200
|
|
assert not Dokument.objects.filter(ausschreibung=a).exists()
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_dokument_upload_too_large(client, tmp_path, settings):
|
|
settings.MEDIA_ROOT = tmp_path
|
|
settings.MAX_UPLOAD_SIZE = 10
|
|
a = AusschreibungFactory()
|
|
url = reverse('ausschreibungen:dokumente:upload', kwargs={'ausschreibung_id': a.pk})
|
|
big_file = SimpleUploadedFile('big.pdf', b'%PDF' + b'x' * 100, content_type='application/pdf')
|
|
response = client.post(url, {'datei': big_file, 'kategorie': 'intern', 'version': '1.0'})
|
|
assert response.status_code == 200
|
|
assert not Dokument.objects.filter(ausschreibung=a).exists()
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_dokument_neue_version(client, tmp_path, settings):
|
|
settings.MEDIA_ROOT = tmp_path
|
|
altes_dok = DokumentFactory(version='1.0')
|
|
a = altes_dok.ausschreibung
|
|
url = reverse('ausschreibungen:dokumente:neue_version',
|
|
kwargs={'ausschreibung_id': a.pk, 'pk': altes_dok.pk})
|
|
response = client.post(url, {'datei': _pdf_file('v2.pdf'), 'version': '2.0'})
|
|
assert response.status_code == 302
|
|
altes_dok.refresh_from_db()
|
|
assert altes_dok.status == 'ersetzt'
|
|
assert Dokument.objects.filter(ausschreibung=a, version='2.0').exists()
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_dokument_finale_version(client):
|
|
dok = DokumentFactory(status='freigegeben')
|
|
a = dok.ausschreibung
|
|
url = reverse('ausschreibungen:dokumente:finale_version',
|
|
kwargs={'ausschreibung_id': a.pk, 'pk': dok.pk})
|
|
response = client.post(url)
|
|
assert response.status_code == 200
|
|
dok.refresh_from_db()
|
|
assert dok.finale_abgabeversion is True
|
|
assert dok.status == 'final_abgegeben'
|