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)