diff --git a/{{cookiecutter.project_slug}}/{{cookiecutter.project_slug}}/users/tests/test_models.py b/{{cookiecutter.project_slug}}/{{cookiecutter.project_slug}}/users/tests/test_models.py index e7180db16..3194be1fd 100644 --- a/{{cookiecutter.project_slug}}/{{cookiecutter.project_slug}}/users/tests/test_models.py +++ b/{{cookiecutter.project_slug}}/{{cookiecutter.project_slug}}/users/tests/test_models.py @@ -1,8 +1,8 @@ import pytest -from django.contrib.auth import get_user_model + +from {{ cookiecutter.project_slug }}.users.models import User pytestmark = pytest.mark.django_db -User = get_user_model() def test_user_get_absolute_url(user: User): diff --git a/{{cookiecutter.project_slug}}/{{cookiecutter.project_slug}}/users/tests/test_urls.py b/{{cookiecutter.project_slug}}/{{cookiecutter.project_slug}}/users/tests/test_urls.py index f5cc874a4..aab6d0a87 100644 --- a/{{cookiecutter.project_slug}}/{{cookiecutter.project_slug}}/users/tests/test_urls.py +++ b/{{cookiecutter.project_slug}}/{{cookiecutter.project_slug}}/users/tests/test_urls.py @@ -1,9 +1,9 @@ import pytest -from django.contrib.auth import get_user_model from django.urls import resolve, reverse +from {{ cookiecutter.project_slug }}.users.models import User + pytestmark = pytest.mark.django_db -User = get_user_model() def test_detail(user: User): diff --git a/{{cookiecutter.project_slug}}/{{cookiecutter.project_slug}}/users/tests/test_views.py b/{{cookiecutter.project_slug}}/{{cookiecutter.project_slug}}/users/tests/test_views.py index 075cbf8df..59b83b83c 100644 --- a/{{cookiecutter.project_slug}}/{{cookiecutter.project_slug}}/users/tests/test_views.py +++ b/{{cookiecutter.project_slug}}/{{cookiecutter.project_slug}}/users/tests/test_views.py @@ -8,6 +8,7 @@ from django.http.response import Http404 from django.test import RequestFactory from {{ cookiecutter.project_slug }}.users.forms import UserChangeForm +from {{ cookiecutter.project_slug }}.users.models import User from {{ cookiecutter.project_slug }}.users.tests.factories import UserFactory from {{ cookiecutter.project_slug }}.users.views import ( UserRedirectView, @@ -16,7 +17,6 @@ from {{ cookiecutter.project_slug }}.users.views import ( ) pytestmark = pytest.mark.django_db -User = get_user_model() class TestUserUpdateView: