Skip to content

Commit 5e292f4

Browse files
author
Ivan Katanovic
committed
Fix tests
1 parent 21849a7 commit 5e292f4

File tree

3 files changed

+18
-15
lines changed

3 files changed

+18
-15
lines changed

daguerre/tests/unit/test_views.py

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
from django.contrib.auth.models import AnonymousUser
44
from django.http import Http404
55
from django.test import RequestFactory
6-
from django.utils.encoding import force_text
6+
from django.utils.encoding import force_str
77

88
from daguerre.helpers import AdjustmentHelper
99
from daguerre.models import Area
@@ -100,7 +100,7 @@ def test_get__pk(self):
100100

101101
self.assertEqual(response.status_code, 200)
102102
self.assertEqual(response['Content-Type'], "application/json")
103-
data = json.loads(force_text(response.content))
103+
data = json.loads(force_str(response.content))
104104
self.assertEqual(data, area.serialize())
105105

106106
def test_get__pk__wrong(self):
@@ -128,7 +128,7 @@ def test_get__no_pk(self):
128128

129129
self.assertEqual(response.status_code, 200)
130130
self.assertEqual(response['Content-Type'], "application/json")
131-
data = json.loads(force_text(response.content))
131+
data = json.loads(force_str(response.content))
132132
self.assertEqual(data, [area1.serialize(), area2.serialize()])
133133

134134
def test_post__no_change_perms(self):
@@ -141,7 +141,7 @@ def test_post__no_change_perms(self):
141141
response = view.post(request)
142142

143143
self.assertEqual(response.status_code, 403)
144-
self.assertEqual(force_text(response.content), '')
144+
self.assertEqual(force_str(response.content), '')
145145

146146
def test_post__invalid_params(self):
147147
area = self.create_area(x2=50, y2=50)
@@ -200,7 +200,7 @@ def test_post__update(self):
200200
self.assertEqual(response.status_code, 200)
201201
self.assertEqual(response['Content-Type'], "application/json")
202202
self.assertEqual(Area.objects.count(), 1)
203-
data = json.loads(force_text(response.content))
203+
data = json.loads(force_str(response.content))
204204
new_area = Area.objects.get(pk=area.pk, storage_path=area.storage_path)
205205
self.assertEqual(data, new_area.serialize())
206206
self.assertNotEqual(data, old_serialize)
@@ -234,7 +234,7 @@ def test_post__update__invalid(self):
234234

235235
self.assertEqual(response.status_code, 400)
236236
self.assertEqual(Area.objects.count(), 1)
237-
data = json.loads(force_text(response.content))
237+
data = json.loads(force_str(response.content))
238238
self.assertEqual(list(data.keys()), ['error'])
239239

240240
def test_post__add(self):
@@ -266,7 +266,7 @@ def test_post__add(self):
266266
self.assertEqual(response.status_code, 200)
267267
self.assertEqual(response['Content-Type'], "application/json")
268268
self.assertEqual(Area.objects.count(), 2)
269-
data = json.loads(force_text(response.content))
269+
data = json.loads(force_str(response.content))
270270
new_area = Area.objects.exclude(pk=area.pk).get()
271271
self.assertEqual(data, new_area.serialize())
272272
del data['storage_path']
@@ -299,7 +299,7 @@ def test_post__add__no_perms(self):
299299

300300
self.assertEqual(response.status_code, 403)
301301
self.assertEqual(Area.objects.count(), 1)
302-
self.assertEqual(force_text(response.content), '')
302+
self.assertEqual(force_str(response.content), '')
303303

304304
def test_delete__no_perms(self):
305305
area = self.create_area(x2=50, y2=50)
@@ -319,7 +319,7 @@ def test_delete__no_perms(self):
319319

320320
self.assertEqual(response.status_code, 403)
321321
self.assertEqual(Area.objects.count(), 1)
322-
self.assertEqual(force_text(response.content), '')
322+
self.assertEqual(force_str(response.content), '')
323323

324324
def test_delete__no_pk(self):
325325
area = self.create_area(x2=50, y2=50)
@@ -356,5 +356,5 @@ def test_delete(self):
356356
response = view.delete(request)
357357

358358
self.assertEqual(response.status_code, 200)
359-
self.assertEqual(force_text(response.content), '')
359+
self.assertEqual(force_str(response.content), '')
360360
self.assertEqual(Area.objects.count(), 0)

daguerre/views.py

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,10 @@
1010
from daguerre.models import Area
1111

1212

13+
def is_ajax(request):
14+
return request.headers.get('x-requested-with') == 'XMLHttpRequest'
15+
16+
1317
class AdjustedImageRedirectView(View):
1418
"""
1519
Returns a redirect to an :attr:`~AdjustedImage.adjusted` file,
@@ -46,7 +50,7 @@ class AjaxAdjustmentInfoView(AdjustedImageRedirectView):
4650
secure = False
4751

4852
def get(self, request, *args, **kwargs):
49-
if not request.is_ajax():
53+
if not is_ajax(request):
5054
raise Http404("Request is not AJAX.")
5155

5256
helper = self.get_helper(generate=False)
@@ -77,7 +81,7 @@ def has_delete_permission(self, request):
7781
return self.has_permission(request.user, 'delete', Area)
7882

7983
def get(self, request, *args, **kwargs):
80-
if not request.is_ajax():
84+
if not is_ajax(request):
8185
raise Http404("Request is not AJAX.")
8286

8387
storage_path = self.kwargs['storage_path']
@@ -96,7 +100,7 @@ def get(self, request, *args, **kwargs):
96100
return HttpResponse(json.dumps(data), content_type="application/json")
97101

98102
def post(self, request, *args, **kwargs):
99-
if not request.is_ajax():
103+
if not is_ajax(request):
100104
raise Http404("Request is not AJAX.")
101105

102106
if not self.has_change_permission(request):
@@ -138,7 +142,7 @@ def post(self, request, *args, **kwargs):
138142
status=status)
139143

140144
def delete(self, request, *args, **kwargs):
141-
if not request.is_ajax():
145+
if not is_ajax(request):
142146
raise Http404("Request is not AJAX.")
143147

144148
if self.kwargs['pk'] is None:

test_project/requirements-4.2.txt

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,2 @@
11
Django==4.2.27
2-
pytz==2019.3
32
-r requirements.txt

0 commit comments

Comments
 (0)