import datetime
from http import HTTPStatus
import pytest
from django.test import RequestFactory
from django.urls import reverse
from engagements import models, views
from engagements.models import EngagementStrategy, RegulatoryCycle
from engagements.utils import populate_database
pytestmark = pytest.mark.django_db
@pytest.fixture
def test_data():
return populate_database()
@pytest.fixture
def request_factory():
return RequestFactory()
def test_dscs_for_ep(client, test_data, request_factory):
org = test_data["orgs"][0]
et = models.EngagementType.objects.get(name="INSPECTION")
si = test_data["sub_instruments"][0]
si2 = test_data["sub_instruments"][2]
si3 = test_data["sub_instruments"][3]
engagement = models.Engagement.objects.create(
proposed_start_date=datetime.date(2022, 10, 10),
proposed_end_date=datetime.date(2022, 10, 10),
engagement_type=et,
external_party=org,
)
ef1 = models.EngagementEffort.objects.create(
is_planned=True,
effort_type="REGULATION",
proposed_start_date=datetime.date(2022, 10, 10),
proposed_end_date=datetime.date(2022, 10, 10),
engagement=engagement,
)
ef1.sub_instruments.add(si, si2, si3)
url = reverse("engagements:plan_for_org", kwargs={"orgslug": org.slug})
client.force_login(test_data["superuser"])
response = client.get(url)
assert response.status_code == HTTPStatus.OK
assert response.context["entity"]
assert response.context["entity"].name == org.name
assert si in response.context["dscs"]
assert si2 in response.context["dscs"]
assert si3 in response.context["dscs"]
def test_get_blank_form(client, test_data, request_factory):
url = reverse("engagements:effort_create", kwargs={"eid": 1, "etype": "PLANNING"})
client.force_login(test_data["superuser"])
request = request_factory.get(url)
request.user = test_data["superuser"]
response = views.engagement_effort_create(request, eid=1, etype="PLANNING")
assert response.status_code == HTTPStatus.OK
# def test_get_form_to_create_engagement_strategy(client, request_factory):
# url = reverse("engagements:es-create")
# client.force_login(test_data["superuser"])
# request = request_factory.get(url)
# request.user = test_data["superuser"]
# response = views.CreateEngagementStrategy()
# assert response.status_code == HTTPStatus.OK
def test_create_engagement_strategy(client, user, org, regulatory_cycles):
# Define the URL for the create view
url = reverse("engagements:es-create")
client.force_login(user)
sy = RegulatoryCycle.objects.get(start_date__year="2022")
ey = RegulatoryCycle.objects.get(start_date__year="2024")
# Define sample data for the 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",
}
# Send a POST request to the view
response = client.post(url, data)
# Check that the response redirects (status code 302) after successful creation
assert response.status_code == 200
# Check that a new EngagementStrategy object was created in the database
assert EngagementStrategy.objects.filter(organisation__name="MOD").exists()
# def test_create_view_uses_correct_form(self):
# # Define the URL for the create view
# url = reverse("engagement_strategy_create")
#
# # Send a GET request to the view
# response = self.client.get(url)
#
# # Check that the response uses the correct form class
# self.assertIsInstance(response.context["form"], EngagementStrategyCreateForm)
