diff --git a/src/api/tests/common.py b/src/api/tests/common.py index 1a8c32526..d975a8012 100644 --- a/src/api/tests/common.py +++ b/src/api/tests/common.py @@ -2,6 +2,7 @@ import os import logging from contextlib import contextmanager +from django.contrib.auth import get_user_model def get_handlers(): @@ -63,3 +64,16 @@ def less_console_noise_decorator(func): return func(*args, **kwargs) return wrapper + + +def create_user(): + username = "restricted_user" + first_name = "First" + last_name = "Last" + email = "restricted@example.com" + phone = "8003111234" + title = "title" + user = get_user_model().objects.create( + username=username, first_name=first_name, last_name=last_name, email=email, phone=phone, title=title + ) + return user diff --git a/src/api/tests/test_available.py b/src/api/tests/test_available.py index b85ea6335..a23b18820 100644 --- a/src/api/tests/test_available.py +++ b/src/api/tests/test_available.py @@ -6,7 +6,7 @@ from django.contrib.auth import get_user_model from django.test import RequestFactory from ..views import available, check_domain_available -from .common import less_console_noise +from .common import less_console_noise, create_user from registrar.tests.common import MockEppLib from registrar.utility.errors import GenericError, GenericErrorCodes from unittest.mock import call @@ -126,7 +126,7 @@ class AvailableAPITest(MockEppLib): def setUp(self): super().setUp() - self.user = get_user_model().objects.create(username="username") + self.user = create_user() def test_available_get(self): self.client.force_login(self.user) diff --git a/src/registrar/tests/common.py b/src/registrar/tests/common.py index c75fa1940..c1058b031 100644 --- a/src/registrar/tests/common.py +++ b/src/registrar/tests/common.py @@ -845,6 +845,7 @@ def create_superuser(): last_name="last", is_staff=True, password=p, + phone="8003111234", ) # Retrieve the group or create it if it doesn't exist group, _ = UserGroup.objects.get_or_create(name="full_access_group") @@ -862,7 +863,9 @@ def create_user(): first_name="first", last_name="last", is_staff=True, + title="title", password=p, + phone="8003111234", ) # Retrieve the group or create it if it doesn't exist group, _ = UserGroup.objects.get_or_create(name="cisa_analysts_group") @@ -878,8 +881,15 @@ def create_test_user(): email = "info@example.com" phone = "8003111234" title = "test title" + password = "password" user = get_user_model().objects.create( - username=username, first_name=first_name, last_name=last_name, email=email, phone=phone, title=title + username=username, + first_name=first_name, + last_name=last_name, + email=email, + phone=phone, + title=title, + password=password, ) return user diff --git a/src/registrar/tests/test_reports.py b/src/registrar/tests/test_reports.py index ed2b75791..24dfe7dbd 100644 --- a/src/registrar/tests/test_reports.py +++ b/src/registrar/tests/test_reports.py @@ -350,6 +350,7 @@ class ExportDataTest(MockDbForIndividualTests, MockEppLib): # Get the csv content csv_content = self._run_domain_data_type_user_export(request) + print(csv_content) # We expect only domains associated with the user's portfolio self.assertIn(self.domain_1.name, csv_content) @@ -389,6 +390,8 @@ class ExportDataTest(MockDbForIndividualTests, MockEppLib): DomainDataTypeUser.export_data_to_csv(csv_file, request=request) # Reset the CSV file's position to the beginning csv_file.seek(0) + print(csv_file) + print(csv_file.seek(0)) # Read the content into a variable csv_content = csv_file.read() diff --git a/src/registrar/tests/test_views.py b/src/registrar/tests/test_views.py index a615d4207..14b894a9c 100644 --- a/src/registrar/tests/test_views.py +++ b/src/registrar/tests/test_views.py @@ -494,7 +494,13 @@ class HomeTests(TestWithUser): phone = "8003111234" status = User.RESTRICTED restricted_user = get_user_model().objects.create( - username=username, first_name=first_name, last_name=last_name, email=email, phone=phone, status=status + username=username, + first_name=first_name, + last_name=last_name, + email=email, + phone=phone, + status=status, + title="title", ) self.client.force_login(restricted_user) response = self.client.get("/request/", follow=True) diff --git a/src/registrar/tests/test_views_domain.py b/src/registrar/tests/test_views_domain.py index c57cbe483..87ecd9b04 100644 --- a/src/registrar/tests/test_views_domain.py +++ b/src/registrar/tests/test_views_domain.py @@ -723,7 +723,7 @@ class TestDomainManagers(TestDomainOverview): email_address = "mayor@igorville.gov" invitation, _ = DomainInvitation.objects.get_or_create(domain=self.domain, email=email_address) - other_user = User() + other_user = create_user() other_user.save() self.client.force_login(other_user) mock_client = MagicMock() @@ -759,7 +759,7 @@ class TestDomainManagers(TestDomainOverview): new_user.on_each_login() # Now load the home page and make sure our domain appears there - home_page = self.app.get(reverse("home")) + home_page = self.app.get(reverse("user-profile")) self.assertContains(home_page, self.domain.name)