mirror of
https://github.com/cisagov/manage.get.gov.git
synced 2025-07-07 11:43:30 +02:00
cleaning up some code
This commit is contained in:
parent
2679f0ac99
commit
171894d692
3 changed files with 7 additions and 17 deletions
|
@ -2,7 +2,6 @@ import os
|
||||||
import logging
|
import logging
|
||||||
|
|
||||||
from contextlib import contextmanager
|
from contextlib import contextmanager
|
||||||
from django.contrib.auth import get_user_model
|
|
||||||
|
|
||||||
|
|
||||||
def get_handlers():
|
def get_handlers():
|
||||||
|
@ -64,16 +63,3 @@ def less_console_noise_decorator(func):
|
||||||
return func(*args, **kwargs)
|
return func(*args, **kwargs)
|
||||||
|
|
||||||
return wrapper
|
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
|
|
||||||
|
|
|
@ -6,7 +6,7 @@ from django.contrib.auth import get_user_model
|
||||||
from django.test import RequestFactory
|
from django.test import RequestFactory
|
||||||
|
|
||||||
from ..views import available, check_domain_available
|
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.tests.common import MockEppLib
|
||||||
from registrar.utility.errors import GenericError, GenericErrorCodes
|
from registrar.utility.errors import GenericError, GenericErrorCodes
|
||||||
from unittest.mock import call
|
from unittest.mock import call
|
||||||
|
@ -126,7 +126,9 @@ class AvailableAPITest(MockEppLib):
|
||||||
|
|
||||||
def setUp(self):
|
def setUp(self):
|
||||||
super().setUp()
|
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):
|
def test_available_get(self):
|
||||||
self.client.force_login(self.user)
|
self.client.force_login(self.user)
|
||||||
|
|
|
@ -535,8 +535,10 @@ class MockDb(TestCase):
|
||||||
first_name = "First"
|
first_name = "First"
|
||||||
last_name = "Last"
|
last_name = "Last"
|
||||||
email = "info@example.com"
|
email = "info@example.com"
|
||||||
|
title = "title"
|
||||||
|
phone = "8080102431"
|
||||||
cls.user = get_user_model().objects.create(
|
cls.user = get_user_model().objects.create(
|
||||||
username=username, first_name=first_name, last_name=last_name, email=email, title="title", phone="8080102431"
|
username=username, first_name=first_name, last_name=last_name, email=email, title=title, phone=phone
|
||||||
)
|
)
|
||||||
|
|
||||||
current_date = get_time_aware_date(datetime(2024, 4, 2))
|
current_date = get_time_aware_date(datetime(2024, 4, 2))
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue