mirror of
https://github.com/cisagov/manage.get.gov.git
synced 2025-07-22 10:46:06 +02:00
PR suggestions - unit test
This commit is contained in:
parent
413bb53664
commit
8081e9252f
1 changed files with 54 additions and 0 deletions
|
@ -2198,3 +2198,57 @@ class TestDomainRequestIncomplete(TestCase):
|
|||
self.domain_request.generic_org_type = None
|
||||
self.domain_request.save()
|
||||
self.assertFalse(self.domain_request._form_complete(request))
|
||||
|
||||
|
||||
class TestPortfolio(TestCase):
|
||||
def setUp(self):
|
||||
self.user, _ = User.objects.get_or_create(
|
||||
username="intern@igorville.com", email="intern@igorville.com", first_name="Lava", last_name="World"
|
||||
)
|
||||
super().setUp()
|
||||
|
||||
def tearDown(self):
|
||||
super().tearDown()
|
||||
Portfolio.objects.all().delete()
|
||||
User.objects.all().delete()
|
||||
|
||||
def test_urbanization_field_resets_when_not_puetro_rico(self):
|
||||
"""The urbanization field should only be populated when the state is puetro rico.
|
||||
Otherwise, this field should be empty."""
|
||||
# Start out as PR, then change the field
|
||||
portfolio = Portfolio.objects.create(
|
||||
creator=self.user,
|
||||
organization_name="Test Portfolio",
|
||||
state_territory=DomainRequest.StateTerritoryChoices.PUERTO_RICO,
|
||||
urbanization="test",
|
||||
)
|
||||
|
||||
self.assertEqual(portfolio.urbanization, "test")
|
||||
self.assertEqual(portfolio.state_territory, DomainRequest.StateTerritoryChoices.PUERTO_RICO)
|
||||
|
||||
portfolio.state_territory = DomainRequest.StateTerritoryChoices.ALABAMA
|
||||
portfolio.save()
|
||||
|
||||
self.assertEqual(portfolio.urbanization, None)
|
||||
self.assertEqual(portfolio.state_territory, DomainRequest.StateTerritoryChoices.ALABAMA)
|
||||
|
||||
def test_can_add_urbanization_field(self):
|
||||
"""Ensures that you can populate the urbanization field when conditions are right"""
|
||||
# Create a portfolio that cannot have this field
|
||||
portfolio = Portfolio.objects.create(
|
||||
creator=self.user,
|
||||
organization_name="Test Portfolio",
|
||||
state_territory=DomainRequest.StateTerritoryChoices.ALABAMA,
|
||||
urbanization="test",
|
||||
)
|
||||
|
||||
# Implicitly check if this gets cleared on create. It should.
|
||||
self.assertEqual(portfolio.urbanization, None)
|
||||
self.assertEqual(portfolio.state_territory, DomainRequest.StateTerritoryChoices.ALABAMA)
|
||||
|
||||
portfolio.state_territory = DomainRequest.StateTerritoryChoices.PUERTO_RICO
|
||||
portfolio.urbanization = "test123"
|
||||
portfolio.save()
|
||||
|
||||
self.assertEqual(portfolio.urbanization, "test123")
|
||||
self.assertEqual(portfolio.state_territory, DomainRequest.StateTerritoryChoices.PUERTO_RICO)
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue