mirror of
https://github.com/cisagov/manage.get.gov.git
synced 2025-05-19 02:49:21 +02:00
tentative changes to tests"
"
This commit is contained in:
parent
d4475f074e
commit
ffeed489c4
6 changed files with 39 additions and 6 deletions
|
@ -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
|
||||
|
|
|
@ -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)
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue