import pytest
import uuid
from django.core.files.uploadedfile import SimpleUploadedFile
from django.urls import reverse
from django.core.files.storage import default_storage

from ..models import ImageMemo, User


@pytest.mark.django_db
class TestDeleteDocument:

    @pytest.fixture
    def setup_users(self):
        self.user1 = User.objects.create_user(
            email="user1@test.com", password="password1"
        )
        self.user2 = User.objects.create_user(
            email="user2@test.com", password="password2"
        )

    @pytest.fixture
    def setup_document(self, setup_users):
        img_content = b"fake image content"
        test_image = SimpleUploadedFile(
            name="test_image.jpg", content=img_content, content_type="image/jpeg"
        )
        self.image_memo = ImageMemo.objects.create(
            author=self.user1,
            image=test_image,
        )

    def test_delete_document_success(self, client, setup_document):
        client.force_login(self.user1)
        url = reverse("delete_document", kwargs={"pk": str(self.image_memo.id)})
        response = client.post(url)

        assert response.status_code == 302
        assert response.url == reverse("dashboard")
        assert not ImageMemo.objects.filter(id=self.image_memo.id).exists()
        assert not default_storage.exists(self.image_memo.image.name)

    def test_delete_nonexistent_document(self, client, setup_users):
        client.force_login(self.user1)
        url = reverse(
            "delete_document", kwargs={"pk": "facade00-0000-4000-a000-000000000000"}
        )
        response = client.post(url)

        assert response.status_code == 404
        assert response.content == b"Document not found"

    def test_delete_other_users_document(self, client, setup_document):
        client.force_login(self.user2)
        url = reverse("delete_document", kwargs={"pk": str(self.image_memo.id)})
        response = client.post(url)

        assert response.status_code == 404
        assert response.content == b"Document not found"
        assert ImageMemo.objects.filter(id=self.image_memo.id).exists()

    def test_delete_document_missing_file(self, client, setup_document):
        # Remove the file manually
        default_storage.delete(self.image_memo.image.name)

        client.force_login(self.user1)
        url = reverse("delete_document", kwargs={"pk": str(self.image_memo.id)})
        response = client.post(url)

        assert response.status_code == 302
        assert response.url == reverse("dashboard")
        assert not ImageMemo.objects.filter(id=self.image_memo.id).exists()

    # ...