Fix unit test

This commit is contained in:
zandercymatics 2024-11-01 11:40:19 -06:00
parent cf0fec178a
commit e4c15ee28b
No known key found for this signature in database
GPG key ID: FF4636ABEC9682B7
2 changed files with 8 additions and 8 deletions

View file

@ -115,13 +115,13 @@ class RequestingEntityForm(RegistrarForm):
if is_requesting_new_suborganization:
# Validate custom suborganization fields
if not cleaned_data.get("requested_suborganization"):
self.add_error("requested_suborganization", "Requested suborganization is required")
self.add_error("requested_suborganization", "Requested suborganization is required.")
if not cleaned_data.get("suborganization_city"):
self.add_error("suborganization_city", "City is required")
self.add_error("suborganization_city", "City is required.")
if not cleaned_data.get("suborganization_state_territory"):
self.add_error("suborganization_state_territory", "State, territory, or military post is required")
self.add_error("suborganization_state_territory", "State, territory, or military post is required.")
elif not suborganization:
self.add_error("sub_organization", "Select a suborganization.")
self.add_error("sub_organization", "Suborganization is required.")
return cleaned_data

View file

@ -1788,15 +1788,15 @@ class TestRequestingEntity(WebTest):
response = form.submit()
self.app.set_cookie(settings.SESSION_COOKIE_NAME, session_id)
self.assertContains(response, "Select a suborganization.", status_code=200)
self.assertContains(response, "Suborganization is required.", status_code=200)
# Test missing custom suborganization details
form["portfolio_requesting_entity-is_requesting_new_suborganization"] = True
response = form.submit()
self.app.set_cookie(settings.SESSION_COOKIE_NAME, session_id)
self.assertContains(response, "Enter details for your organization name.", status_code=200)
self.assertContains(response, "Enter details for your city.", status_code=200)
self.assertContains(response, "Enter details for your state or territory.", status_code=200)
self.assertContains(response, "Requested suborganization is required.", status_code=200)
self.assertContains(response, "City is required.", status_code=200)
self.assertContains(response, "State, territory, or military post is required.", status_code=200)
@override_flag("organization_feature", active=True)
@override_flag("organization_requests", active=True)