import pytest
from django.forms import DateInput, Select, SelectMultiple

from engagements.forms import (
    EngagementCreateForm,
    EngagementEffortCreateForm,
    EngagementStrategyCreateForm,
    AssessmentCreateForm,
)
from engagements.models import EngagementType, RegulatoryCycle

pytestmark = pytest.mark.django_db


def test_basic_validation(engagement, user):
    form = EngagementEffortCreateForm(
        data={
            "is_planned": True,
            "proposed_start_date": "2022-10-10 10:00",
            "proposed_end_date": "2022-10-10 12:00",
            "engagement": engagement,
            "effort_type": "PLANNING",
            "officers": [user],
        }
    )
    assert not form.errors


def test_basic_validation_on_bad_entry(engagement, user):
    form = EngagementEffortCreateForm(
        data={
            "is_planned": True,
            "proposed_start_date": "20240-10-10 10:00",
            "proposed_end_date": "2022-10-10 12:00",
            "engagement": engagement,
            "effort_type": "bobbins",
            "officers": [user],
            "sub_instruments": [""],
        }
    )
    assert "effort_type" in form.errors
    assert "proposed_start_date" in form.errors


def test_form_fields():
    form = EngagementCreateForm()
    expected_fields = [
        "proposed_start_date",
        "proposed_end_date",
        "engagement_type",
        "external_party",
        "officers",
    ]
    assert list(form.fields.keys()) == expected_fields


def test_form_labels():
    form = EngagementCreateForm()
    assert form.fields["officers"].label == "Inspectors"


def test_form_help_texts():
    form = EngagementCreateForm()
    assert form.fields["proposed_start_date"].help_text == "YYYY-MM-DD"
    assert form.fields["proposed_end_date"].help_text == "YYYY-MM-DD"


def test_form_widgets():
    form = EngagementCreateForm()
    assert isinstance(form.fields["proposed_start_date"].widget, DateInput)
    assert isinstance(form.fields["proposed_end_date"].widget, DateInput)
    assert isinstance(form.fields["engagement_type"].widget, Select)
    assert isinstance(form.fields["officers"].widget, SelectMultiple)


def test_form_valid_data(engagement_type, external_party, user1, user2):
    form_data = {
        "proposed_start_date": "2023-01-01",
        "proposed_end_date": "2023-01-31",
        "engagement_type": engagement_type.id,
        "external_party": external_party.id,
        "officers": [user1.id, user2.id],
    }
    form = EngagementCreateForm(data=form_data)
    assert form.is_valid()


def test_form_invalid_dates(engagement_type, external_party, user1):
    form_data = {
        "proposed_start_date": "2023-01-31",
        "proposed_end_date": "2023-01-01",
        "engagement_type": engagement_type.id,
        "external_party": external_party.id,
        "officers": [user1.id],
    }
    form = EngagementCreateForm(data=form_data)
    assert not form.is_valid()


def test_form_missing_required_fields():
    form_data = {
        "proposed_start_date": "2023-01-01",
    }
    form = EngagementCreateForm(data=form_data)
    assert not form.is_valid()
    assert "engagement_type" in form.errors
    assert "external_party" in form.errors


def test_form_engagement_type_queryset(engagement_type):
    initial_data = {"engagement_type": EngagementType.objects.all()}
    form = EngagementCreateForm(initial=initial_data)
    assert form.fields["engagement_type"].queryset.first().name == initial_data["engagement_type"].first().name


def test_form_engagement_type_queryset_without_initial():
    form = EngagementCreateForm()
    assert list(form.fields["engagement_type"].queryset) == list(EngagementType.objects.all())


def test_create_engagement_strategy_form(org, user, regulatory_cycles):
    sy = RegulatoryCycle.objects.get(start_date="2022-01-01")
    ey = RegulatoryCycle.objects.get(start_date="2024-01-01")
    form_data = {
        "organisation": org,
        "start_year": sy,
        "end_year": ey,
        "description": "Example description",
        "inspector_sign_off": "2022-01-10",
        "owned_by": user,
        "reviewed_by": user,
        "management_sign_off": "2022-02-10",
        "status": "DRAFT",
    }
    form = EngagementStrategyCreateForm(data=form_data)
    assert form.is_valid()


def test_create_assessment_engagement_type_form():
    e_type = EngagementType(name="ASSESSMENT")
    form = AssessmentCreateForm({"initial": e_type})