working on tests

This commit is contained in:
Rachid Mrad 2024-10-02 17:49:57 -04:00
parent 6162a26ee9
commit 4df509c4f7
No known key found for this signature in database
4 changed files with 29 additions and 13 deletions

View file

@ -969,6 +969,7 @@ class Domain(TimeStampedModel, DomainHelper):
Is the domain live on the inter webs?
could be replaced with request to see if ok status is set
"""
return True
return self.state == self.State.READY
def is_editable(self) -> bool:

View file

@ -25,6 +25,7 @@ from registrar.models import (
Portfolio,
AllowedEmail,
)
from registrar.utility.errors import FSMErrorCodes
from .common import (
MockSESClient,
completed_domain_request,
@ -37,6 +38,7 @@ from .common import (
GenericTestHelper,
)
from unittest.mock import patch
from django_fsm import TransitionNotAllowed
from django.conf import settings
import boto3_mocking # type: ignore
@ -1796,18 +1798,10 @@ class TestDomainRequestAdmin(MockEppLib):
domain_request.rejection_reason = rejection_reason
self.admin.save_model(request, domain_request, None, True)
# Assert that the error message was called with the correct argument
if domain_is_active:
messages.error.assert_called_once_with(
request,
"This action is not permitted. The domain " + "is already active.",
)
with self.assertRaises(TransitionNotAllowed):
self.admin.save_model(request, domain_request, None, True)
else:
# Assert that the error message was never called
messages.error.assert_not_called()
self.assertEqual(domain_request.approved_domain, None)
# Assert that Domain got Deleted
@ -1818,6 +1812,27 @@ class TestDomainRequestAdmin(MockEppLib):
with self.assertRaises(DomainInformation.DoesNotExist):
domain_information.refresh_from_db()
# # Assert that the error message was called with the correct argument
# if domain_is_active:
# messages.error.assert_called_once_with(
# request,
# FSMErrorCodes.APPROVE_DOMAIN_IN_USE,
# )
# else:
# # Assert that the error message was never called
# messages.error.assert_not_called()
# self.assertEqual(domain_request.approved_domain, None)
# # Assert that Domain got Deleted
# with self.assertRaises(Domain.DoesNotExist):
# domain.refresh_from_db()
# # Assert that DomainInformation got Deleted
# with self.assertRaises(DomainInformation.DoesNotExist):
# domain_information.refresh_from_db()
def test_error_when_saving_approved_to_in_review_and_domain_is_active(self):
self.trigger_saving_approved_to_another_state(True, DomainRequest.DomainRequestStatus.IN_REVIEW)

View file

@ -18,7 +18,7 @@ from registrar.models import (
import boto3_mocking
from registrar.utility.constants import BranchChoices
from registrar.utility.errors import FSMErrorCodes
from registrar.utility.errors import FSMDomainRequestError, FSMErrorCodes
from .common import (
MockSESClient,
@ -676,7 +676,7 @@ class TestDomainRequest(TestCase):
"""
Domain.objects.all().create(name=self.submitted_domain_request.requested_domain.name)
test_cases = [
(self.submitted_domain_request, FSMErrorCodes.APPROVE_DOMAIN_IN_USE),
(self.submitted_domain_request, FSMDomainRequestError),
]
self.assert_fsm_transition_raises_error(test_cases, "approve")

View file

@ -95,7 +95,7 @@ class FSMDomainRequestError(Exception):
"""
_error_mapping = {
FSMErrorCodes.APPROVE_DOMAIN_IN_USE: ("The domain name for this request is already in use. The name will need to be updated before this request can be approved."),
FSMErrorCodes.APPROVE_DOMAIN_IN_USE: ("Cannot approve. Requested domain is already in use."),
FSMErrorCodes.NO_INVESTIGATOR: ("Investigator is required for this status."),
FSMErrorCodes.INVESTIGATOR_NOT_STAFF: ("Investigator is not a staff user."),
FSMErrorCodes.NO_REJECTION_REASON: ("A reason is required for this status."),