|
1 | 1 | from corsheaders.middleware import CorsMiddleware
|
2 | 2 | from django.conf import settings
|
3 |
| -from django.contrib.auth.models import AnonymousUser |
| 3 | +from django.contrib.auth.middleware import AuthenticationMiddleware |
| 4 | +from django.contrib.sessions.middleware import SessionMiddleware |
4 | 5 | from django.http import Http404
|
5 | 6 | from django.test import TestCase
|
6 | 7 | from django.test.client import RequestFactory
|
@@ -30,10 +31,14 @@ def setUp(self):
|
30 | 31 | privacy_level='public'
|
31 | 32 | )
|
32 | 33 |
|
| 34 | + def process_request_auth_middleware(self, request): |
| 35 | + SessionMiddleware().process_request(request) |
| 36 | + AuthenticationMiddleware().process_request(request) |
| 37 | + |
33 | 38 | def test_failey_cname(self):
|
34 | 39 | self.assertFalse(Domain.objects.filter(domain='my.host.com').exists())
|
35 | 40 | request = self.factory.get(self.url, HTTP_HOST='my.host.com')
|
36 |
| - request.user = AnonymousUser() |
| 41 | + self.process_request_auth_middleware(request) |
37 | 42 | r = self.middleware.process_request(request)
|
38 | 43 | self.assertEqual(r.status_code, 404)
|
39 | 44 | self.assertEqual(request.cname, True)
|
@@ -98,7 +103,7 @@ def test_domain_object(self):
|
98 | 103 | def test_domain_object_missing(self):
|
99 | 104 | self.domain = get(Domain, domain='docs.foobar2.com', project=self.pip)
|
100 | 105 | request = self.factory.get(self.url, HTTP_HOST='docs.foobar.com')
|
101 |
| - request.user = AnonymousUser() |
| 106 | + self.process_request_auth_middleware(request) |
102 | 107 | r = self.middleware.process_request(request)
|
103 | 108 | self.assertEqual(r.status_code, 404)
|
104 | 109 |
|
@@ -144,15 +149,15 @@ def test_use_subdomain(self):
|
144 | 149 | def test_long_bad_subdomain(self):
|
145 | 150 | domain = 'www.pip.readthedocs.org'
|
146 | 151 | request = self.factory.get(self.url, HTTP_HOST=domain)
|
147 |
| - request.user = AnonymousUser() |
| 152 | + self.process_request_auth_middleware(request) |
148 | 153 | res = self.middleware.process_request(request)
|
149 | 154 | self.assertEqual(res.status_code, 400)
|
150 | 155 |
|
151 | 156 | @override_settings(PRODUCTION_DOMAIN='readthedocs.org')
|
152 | 157 | def test_long_subdomain(self):
|
153 | 158 | domain = 'some.long.readthedocs.org'
|
154 | 159 | request = self.factory.get(self.url, HTTP_HOST=domain)
|
155 |
| - request.user = AnonymousUser() |
| 160 | + self.process_request_auth_middleware(request) |
156 | 161 | res = self.middleware.process_request(request)
|
157 | 162 | self.assertIsNone(res)
|
158 | 163 |
|
|
0 commit comments