tentative changes to tests"

"
This commit is contained in:
asaki222 2024-09-16 16:43:15 -04:00
parent d4475f074e
commit ffeed489c4
No known key found for this signature in database
GPG key ID: 2C4F802060E06EA4
6 changed files with 39 additions and 6 deletions

View file

@ -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

View file

@ -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)

View file

@ -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

View file

@ -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()

View file

@ -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)

View file

@ -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)