21
21
psycopg = None
22
22
23
23
from ..base import BaseMultiDBTestCase , BaseTestCase
24
- from ..models import PostgresJSON
24
+ from ..models import Binary , PostgresJSON
25
25
26
26
27
27
def sql_call (use_iterator = False ):
@@ -149,7 +149,7 @@ def test_non_ascii_query(self):
149
149
self .assertEqual (len (self .panel ._queries ), 2 )
150
150
151
151
# non-ASCII bytes parameters
152
- list (User .objects .filter (username = "café" .encode ()))
152
+ list (Binary .objects .filter (field__in = [ "café" .encode ()] ))
153
153
self .assertEqual (len (self .panel ._queries ), 3 )
154
154
155
155
response = self .panel .process_request (self .request )
@@ -335,7 +335,7 @@ def test_insert_content(self):
335
335
Test that the panel only inserts content after generate_stats and
336
336
not the process_request.
337
337
"""
338
- list (User .objects .filter (username = "café" . encode () ))
338
+ list (User .objects .filter (username = "café" ))
339
339
response = self .panel .process_request (self .request )
340
340
# ensure the panel does not have content yet.
341
341
self .assertNotIn ("café" , self .panel .content )
@@ -351,7 +351,7 @@ def test_insert_locals(self):
351
351
Test that the panel inserts locals() content.
352
352
"""
353
353
local_var = "<script>alert('test');</script>" # noqa: F841
354
- list (User .objects .filter (username = "café" . encode () ))
354
+ list (User .objects .filter (username = "café" ))
355
355
response = self .panel .process_request (self .request )
356
356
self .panel .generate_stats (self .request , response )
357
357
self .assertIn ("local_var" , self .panel .content )
@@ -365,7 +365,7 @@ def test_not_insert_locals(self):
365
365
"""
366
366
Test that the panel does not insert locals() content.
367
367
"""
368
- list (User .objects .filter (username = "café" . encode () ))
368
+ list (User .objects .filter (username = "café" ))
369
369
response = self .panel .process_request (self .request )
370
370
self .panel .generate_stats (self .request , response )
371
371
self .assertNotIn ("djdt-locals" , self .panel .content )
0 commit comments