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'