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: if is_requesting_new_suborganization:
# Validate custom suborganization fields # Validate custom suborganization fields
if not cleaned_data.get("requested_suborganization"): 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"): 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"): 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: elif not suborganization:
self.add_error("sub_organization", "Select a suborganization.") self.add_error("sub_organization", "Suborganization is required.")
return cleaned_data return cleaned_data

View file

@ -1788,15 +1788,15 @@ class TestRequestingEntity(WebTest):
response = form.submit() response = form.submit()
self.app.set_cookie(settings.SESSION_COOKIE_NAME, session_id) 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 # Test missing custom suborganization details
form["portfolio_requesting_entity-is_requesting_new_suborganization"] = True form["portfolio_requesting_entity-is_requesting_new_suborganization"] = True
response = form.submit() response = form.submit()
self.app.set_cookie(settings.SESSION_COOKIE_NAME, session_id) 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, "Requested suborganization is required.", status_code=200)
self.assertContains(response, "Enter details for your city.", status_code=200) self.assertContains(response, "City is required.", status_code=200)
self.assertContains(response, "Enter details for your state or territory.", 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_feature", active=True)
@override_flag("organization_requests", active=True) @override_flag("organization_requests", active=True)