mirror of
https://github.com/cisagov/manage.get.gov.git
synced 2025-08-12 20:49:41 +02:00
Update test_models.py
This commit is contained in:
parent
7c69205deb
commit
3ba4293334
1 changed files with 0 additions and 37 deletions
|
@ -31,46 +31,9 @@ from .common import (
|
||||||
create_test_user,
|
create_test_user,
|
||||||
)
|
)
|
||||||
from waffle.testutils import override_flag
|
from waffle.testutils import override_flag
|
||||||
|
|
||||||
import logging
|
|
||||||
|
|
||||||
from api.tests.common import less_console_noise_decorator
|
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):
|
class TestDomainInformation(TestCase):
|
||||||
"""Test the DomainInformation model, when approved or otherwise"""
|
"""Test the DomainInformation model, when approved or otherwise"""
|
||||||
|
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue