zydronium/manage.get.gov
1
0
Fork 0
mirror of https://github.com/cisagov/manage.get.gov.git synced 2025-08-14 05:29:43 +02:00
Code Issues Projects Releases Packages Wiki Activity

Update test_views_domain.py

Browse source
This commit is contained in:
zandercymatics 2024-03-07 14:44:13 -07:00
parent 8438067d79
commit 4e67733861
No known key found for this signature in database
GPG key ID: FF4636ABEC9682B7
1 changed files with 37 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

51
src/registrar/tests/test_views_domain.py
View file

@ -1245,7 +1245,6 @@ class TestDomainOrganization(TestDomainOverview):
self.assertEqual(self.domain_information.organization_type, tribal_org_type) self.assertEqual(self.domain_information.organization_type, tribal_org_type)
org_name_page = self.app.get(reverse("domain-org-name-address", kwargs={"pk": self.domain.id})) org_name_page = self.app.get(reverse("domain-org-name-address", kwargs={"pk": self.domain.id}))
print(f"what is the org name page? {org_name_page}")
form = org_name_page.forms[0] form = org_name_page.forms[0]
# Check the value of the input field # Check the value of the input field
@ -1284,38 +1283,62 @@ class TestDomainOrganization(TestDomainOverview):
# Check for the value we want to update # Check for the value we want to update
self.assertContains(success_result_page, "Faketown") self.assertContains(success_result_page, "Faketown")
def test_domain_org_name_address_form_federal_disabled(self): def test_domain_org_name_address_form_federal(self):
""" """
Tests if the federal_agency field is readonly Submitting a change to federal_agency is blocked for federal domains
""" """
# Set the current domain to a tribal organization with a preset value. # Set the current domain to a tribal organization with a preset value.
# Save first, so we can test if saving is unaffected (it should be). # Save first, so we can test if saving is unaffected (it should be).
federal_org_type = DomainInformation.OrganizationChoices.FEDERAL fed_org_type = DomainInformation.OrganizationChoices.FEDERAL
self.domain_information.organization_type = federal_org_type self.domain_information.organization_type = fed_org_type
self.domain_information.save() self.domain_information.save()
try: try:
# Add a federal agency. Defined as a tuple since this list may change order. self.domain_information.federal_agency = "AMTRAK"
self.domain_information.federal_agency = ("AMTRAK", "AMTRAK")
self.domain_information.save() self.domain_information.save()
except ValueError as err: except ValueError as err:
self.fail(f"A ValueError was caught during the test: {err}") self.fail(f"A ValueError was caught during the test: {err}")
self.assertEqual(self.domain_information.organization_type, federal_org_type) self.assertEqual(self.domain_information.organization_type, tribal_org_type)
org_name_page = self.app.get(reverse("domain-org-name-address", kwargs={"pk": self.domain.id})) org_name_page = self.app.get(reverse("domain-org-name-address", kwargs={"pk": self.domain.id}))
print(f"what is the org name page? {org_name_page}")
form = org_name_page.forms[0] form = org_name_page.forms[0]
# Check the value of the input field # Check the value of the input field
organization_name_input = form.fields["federal_agency"][0] agency_input = form.fields["federal_agency"][0]
self.assertEqual(agency_input.value, "AMTRAK")
# Check if the input field is disabled. # Check if the input field is disabled
# Webtest has some issues dealing with Selects, so we can't self.assertTrue("disabled" in agency_input.attrs)
# directly test the value but we can test its attributes. self.assertEqual(agency_input.attrs.get("disabled"), "")
# This is done in another test.
session_id = self.app.cookies[settings.SESSION_COOKIE_NAME]
org_name_page.form["federal_agency"] = "Department of State"
org_name_page.form["city"] = "Faketown"
self.app.set_cookie(settings.SESSION_COOKIE_NAME, session_id)
# Make the change. The agency should be unchanged, but city should be modifiable.
success_result_page = org_name_page.form.submit()
self.assertEqual(success_result_page.status_code, 200)
# Check for the old and new value
self.assertContains(success_result_page, "AMTRAK")
self.assertNotContains(success_result_page, "Department of State")
# Do another check on the form itself
form = success_result_page.forms[0]
# Check the value of the input field
organization_name_input = form.fields["federal_agency"][0]
self.assertEqual(organization_name_input.value, "AMTRAK")
# Check if the input field is disabled
self.assertTrue("disabled" in organization_name_input.attrs) self.assertTrue("disabled" in organization_name_input.attrs)
self.assertEqual(organization_name_input.attrs.get("disabled"), "") self.assertEqual(organization_name_input.attrs.get("disabled"), "")
# Check for the value we want to update
self.assertContains(success_result_page, "Faketown")
def test_federal_agency_submit_blocked(self): def test_federal_agency_submit_blocked(self):
""" """
Submitting a change to federal_agency is blocked for federal domains Submitting a change to federal_agency is blocked for federal domains