cleaning up some code

This commit is contained in:
asaki222 2024-09-17 11:27:09 -04:00
parent 2679f0ac99
commit 171894d692
No known key found for this signature in database
GPG key ID: 2C4F802060E06EA4
3 changed files with 7 additions and 17 deletions

View file

@ -2,7 +2,6 @@ import os
import logging
from contextlib import contextmanager
from django.contrib.auth import get_user_model
def get_handlers():
@ -64,16 +63,3 @@ 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, create_user
from .common import less_console_noise
from registrar.tests.common import MockEppLib
from registrar.utility.errors import GenericError, GenericErrorCodes
from unittest.mock import call
@ -126,7 +126,9 @@ class AvailableAPITest(MockEppLib):
def setUp(self):
super().setUp()
self.user = create_user()
username = "username"
title = "title"
self.user = get_user_model().objects.create(username=username, title=title)
def test_available_get(self):
self.client.force_login(self.user)