import pytest

from email_campaigns.reporting.exceptions import EmailReportingReportNotFoundError
from email_campaigns.reporting.handlers import (
    DeleteEmailReportingReportHandler,
    DeleteEmailReportingReportRequest,
)
from email_campaigns.reporting.models import EmailReportingReport

from tests.unit.types import CreateModel


class TestDeleteEmailReportingReportHandler:
    @pytest.mark.db
    def test_delete_report(
        self,
        handler: DeleteEmailReportingReportHandler,
        create_model: CreateModel,
        identity_id: str,
    ) -> None:
        report = create_model(
            EmailReportingReport,
            identity_id=identity_id,
            name="Test Report",
        )

        handler.handle(
            DeleteEmailReportingReportRequest(
                identity_id=identity_id,
                report_id=report.id,
            )
        )

        assert True

    @pytest.mark.db
    def test_delete_report_not_found(
        self,
        handler: DeleteEmailReportingReportHandler,
        identity_id: str,
    ) -> None:
        with pytest.raises(EmailReportingReportNotFoundError):
            handler.handle(
                DeleteEmailReportingReportRequest(
                    identity_id=identity_id,
                    report_id="non-existent-id",
                )
            )