fix testing configuration for newer Django versions

mixcloud · May 11, 2022 · 01b68d6 · 01b68d6
1 parent 27fde95
commit 01b68d6
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 4 deletions.
diff --git a/experiments/tests/test_webuser.py b/experiments/tests/test_webuser.py
@@ -227,7 +227,7 @@ def test_transfer_enrollments(self):
         # if the participant cache hasn't been wiped appropriately then the
         # session experiment user will be impacted instead of the authenticated
         # experiment user
-        ExperimentsRetentionMiddleware().process_response(request, HttpResponse())
+        ExperimentsRetentionMiddleware(request).process_response(request, HttpResponse())
         self.assertIsNotNone(Enrollment.objects.all()[0].last_seen)
 
 

diff --git a/experiments/tests/test_webuser_incorporate.py b/experiments/tests/test_webuser_incorporate.py
@@ -103,7 +103,7 @@ def _login(self):
     def test_visit_incorporate(self):
         alternative = participant(self.request).enroll(self.experiment.name, ['alternative'])
 
-        ExperimentsRetentionMiddleware().process_response(self.request, HttpResponse())
+        ExperimentsRetentionMiddleware(self.request).process_response(self.request, HttpResponse())
 
         self.assertEqual(
             dict(self.experiment_counter.participant_goal_frequencies(self.experiment,

diff --git a/testrunner.py b/testrunner.py
@@ -20,14 +20,16 @@ def runtests():
                         'django.contrib.contenttypes',
                         'django.contrib.sessions',
                         'django.contrib.admin',
+                        'django.contrib.messages',
                         'experiments',),
         ROOT_URLCONF='experiments.tests.urls',
-        MIDDLEWARE_CLASSES = (
+        MIDDLEWARE = (
             'django.contrib.sessions.middleware.SessionMiddleware',
             'django.middleware.csrf.CsrfViewMiddleware',
             'django.contrib.auth.middleware.AuthenticationMiddleware',
-            'django.contrib.auth.middleware.SessionAuthenticationMiddleware',
+            'django.contrib.messages.middleware.MessageMiddleware'
         ),
+        SECRET_KEY="foobarbaz",
         TEMPLATES = [
             {
                 'BACKEND': 'django.template.backends.django.DjangoTemplates',
@@ -36,6 +38,7 @@ def runtests():
                 'OPTIONS': {
                     'context_processors': [
                         'django.contrib.auth.context_processors.auth',
+                        'django.contrib.messages.context_processors.messages',
                     ],
                 },
             },