Skip to content

Commit e8b1396

Browse files
committed
Edits.
1 parent cb67abe commit e8b1396

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

tests/basic_/tests.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,5 @@
1+
from operator import attrgetter
2+
13
from django.db.models import Value
24
from django.test import TestCase
35

@@ -80,12 +82,12 @@ def test_update_tuple_value(self):
8082

8183
class QueryDollarPrefixTests(MongoTestCaseMixin, TestCase):
8284
def test_query_injection(self):
83-
"""$-prefixed Value() expressions are correctly handled on update."""
85+
"""$-prefixed Value() expressions are correctly handled on query."""
8486
Author.objects.create(name="Gustavo")
8587
Author.objects.create(name="Walter")
8688
with self.assertNumQueries(1) as ctx:
8789
qs = list(Author.objects.annotate(a_value=Value("$name")))
88-
self.assertTrue(all(v.a_value == "$name" for v in qs))
90+
self.assertQuerySetEqual(qs, ["$name"] * 2, attrgetter("a_value"))
8991
self.assertAggregateQuery(
9092
ctx.captured_queries[0]["sql"],
9193
"basic__author",

0 commit comments

Comments
 (0)