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

common.py update

Browse source
This commit is contained in:
zandercymatics 2024-05-02 15:06:51 -06:00
parent 093f596041
commit 44dec2b73a
No known key found for this signature in database
GPG key ID: FF4636ABEC9682B7
1 changed files with 7 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

26
src/registrar/tests/common.py
View file

@ -753,14 +753,12 @@ def mock_user():
def create_superuser(): def create_superuser():
"""Creates a user with admin-level permissions"""
User = get_user_model() User = get_user_model()
p = "adminpass" p = "adminpass"
user = User.objects.create_user( user = User.objects.create_user(
username="superuser", username="superuser",
email="admin@example.com", email="admin@example.com",
is_staff=True, is_staff=True,
is_superuser=True,
password=p, password=p,
) )
# Retrieve the group or create it if it doesn't exist # Retrieve the group or create it if it doesn't exist
@ -770,8 +768,7 @@ def create_superuser():
return user return user
def create_staffuser(): def create_user():
"""Creates a user with staff level permissions"""
User = get_user_model() User = get_user_model()
p = "userpass" p = "userpass"
user = User.objects.create_user( user = User.objects.create_user(
@ -787,25 +784,12 @@ def create_staffuser():
return user return user
def create_user():
"""Creates a user with no special permissions"""
User = get_user_model()
p = "userpass"
user = User.objects.create_user(
username="regularuser",
email="regularuser@example.com",
is_staff=False,
is_superuser=False,
password=p,
)
return user
def create_ready_domain(): def create_ready_domain():
domain, _ = Domain.objects.get_or_create(name="city.gov", state=Domain.State.READY) domain, _ = Domain.objects.get_or_create(name="city.gov", state=Domain.State.READY)
return domain return domain
# TODO in 1793: Remove the federal agency/updated federal agency fields
def completed_domain_request( def completed_domain_request(
has_other_contacts=True, has_other_contacts=True,
has_current_website=True, has_current_website=True,
@ -820,6 +804,8 @@ def completed_domain_request(
generic_org_type="federal", generic_org_type="federal",
is_election_board=False, is_election_board=False,
organization_type=None, organization_type=None,
federal_agency=None,
updated_federal_agency=None,
): ):
"""A completed domain request.""" """A completed domain request."""
if not user: if not user:
@ -856,6 +842,7 @@ def completed_domain_request(
last_name="Bob", last_name="Bob",
is_staff=True, is_staff=True,
) )
domain_request_kwargs = dict( domain_request_kwargs = dict(
generic_org_type=generic_org_type, generic_org_type=generic_org_type,
is_election_board=is_election_board, is_election_board=is_election_board,
@ -873,6 +860,8 @@ def completed_domain_request(
creator=user, creator=user,
status=status, status=status,
investigator=investigator, investigator=investigator,
federal_agency=federal_agency,
updated_federal_agency=updated_federal_agency,
) )
if has_about_your_organization: if has_about_your_organization:
domain_request_kwargs["about_your_organization"] = "e-Government" domain_request_kwargs["about_your_organization"] = "e-Government"
@ -881,7 +870,6 @@ def completed_domain_request(
if organization_type: if organization_type:
domain_request_kwargs["organization_type"] = organization_type domain_request_kwargs["organization_type"] = organization_type
domain_request, _ = DomainRequest.objects.get_or_create(**domain_request_kwargs) domain_request, _ = DomainRequest.objects.get_or_create(**domain_request_kwargs)
if has_other_contacts: if has_other_contacts: