Update test_models.py

This commit is contained in:
zandercymatics 2024-10-09 08:37:01 -06:00
parent 7c69205deb
commit 3ba4293334
No known key found for this signature in database
GPG key ID: FF4636ABEC9682B7

View file

@ -31,46 +31,9 @@ from .common import (
create_test_user,
)
from waffle.testutils import override_flag
import logging
from api.tests.common import less_console_noise_decorator
logger = logging.getLogger(__name__)
class TestPermissions(TestCase):
"""Test the User-Domain-Role connection."""
def setUp(self):
super().setUp()
self.mock_client = MockSESClient()
def tearDown(self):
super().tearDown()
self.mock_client.EMAILS_SENT.clear()
@boto3_mocking.patching
@less_console_noise_decorator
def test_approval_creates_role(self):
draft_domain, _ = DraftDomain.objects.get_or_create(name="igorville.gov")
user, _ = User.objects.get_or_create()
investigator, _ = User.objects.get_or_create(username="frenchtoast", is_staff=True)
domain_request = DomainRequest.objects.create(
creator=user, requested_domain=draft_domain, investigator=investigator
)
with boto3_mocking.clients.handler_for("sesv2", self.mock_client):
# skip using the submit method
domain_request.status = DomainRequest.DomainRequestStatus.SUBMITTED
domain_request.approve()
# should be a role for this user
domain = Domain.objects.get(name="igorville.gov")
self.assertTrue(UserDomainRole.objects.get(user=user, domain=domain))
class TestDomainInformation(TestCase):
"""Test the DomainInformation model, when approved or otherwise"""