diff --git a/alerts/tests.py b/alerts/tests.py index ce37343..4cf6f4a 100644 --- a/alerts/tests.py +++ b/alerts/tests.py @@ -1,7 +1,7 @@ from django.test import TestCase as DjangoTestCase from django.core import mail -from django.contrib.auth.models import User +from django.contrib.auth import get_user_model from django.template.loader import get_template from issues.factories import get_or_create_issue @@ -11,6 +11,8 @@ from events.factories import create_event from .tasks import send_new_issue_alert, send_regression_alert, send_unmute_alert from .views import DEBUG_CONTEXTS +User = get_user_model() + class TestAlertSending(DjangoTestCase): @@ -67,8 +69,8 @@ class TestAlertSending(DjangoTestCase): def test_txt_and_html_have_relevant_variables_defined(self): example_context = DEBUG_CONTEXTS["issue_alert"] - html_template = get_template("alerts/issue_alert.html") - text_template = get_template("alerts/issue_alert.txt") + html_template = get_template("mails/issue_alert.html") + text_template = get_template("mails/issue_alert.txt") unused_in_text = [ "base_url", # link to the site is not included at the top of the text template diff --git a/events/tests.py b/events/tests.py index 12d29a4..6cac928 100644 --- a/events/tests.py +++ b/events/tests.py @@ -1,7 +1,7 @@ import datetime from django.test import TestCase as DjangoTestCase, TransactionTestCase -from django.contrib.auth.models import User +from django.contrib.auth import get_user_model from django.conf import settings from django.utils import timezone @@ -11,6 +11,8 @@ from issues.factories import denormalized_issue_fields from .factories import create_event +User = get_user_model() + class ViewTests(TransactionTestCase): # we start with minimal "does this show something and not fully crash" tests and will expand from there. diff --git a/issues/tests.py b/issues/tests.py index e51b29b..db33566 100644 --- a/issues/tests.py +++ b/issues/tests.py @@ -8,7 +8,7 @@ from unittest.mock import patch from datetime import datetime, timezone from django.test import TestCase as DjangoTestCase, TransactionTestCase -from django.contrib.auth.models import User +from django.contrib.auth import get_user_model from django.test import tag from projects.models import Project, ProjectMembership @@ -25,6 +25,8 @@ from .regressions import is_regression, is_regression_2, issue_is_regression from .factories import denormalized_issue_fields from .utils import get_issue_grouper_for_data +User = get_user_model() + def fresh(obj): return type(obj).objects.get(pk=obj.pk)