Skip to content

Commit 6f383ee

Browse files
committed
move range test to querying
1 parent 5d7fb5a commit 6f383ee

File tree

1 file changed

+10
-5
lines changed

1 file changed

+10
-5
lines changed

tests/model_fields_/test_embedded_model.py

Lines changed: 10 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -136,6 +136,15 @@ def test_order_and_group_by_embedded_field(self):
136136
],
137137
)
138138

139+
def test_range_query(self):
140+
sorted_obj = sorted(self.objs, key=lambda x: x.data.auto_now)
141+
queryset = Holder.objects.filter(
142+
data__auto_now__range=(sorted_obj[2].data.auto_now, sorted_obj[-1].data.auto_now)
143+
).order_by("data__integer")
144+
self.assertEqual(queryset.count(), 4)
145+
expected_objs = sorted(e.data.integer for e in sorted_obj[2:])
146+
self.assertQuerySetEqual(queryset, expected_objs, lambda h: h.data.integer)
147+
139148
def test_order_and_group_by_embedded_field_annotation(self):
140149
# Create repeated `data__integer` values.
141150
[Holder.objects.create(data=Data(integer=x)) for x in range(6)]
@@ -188,6 +197,7 @@ class MyModel(models.Model):
188197
"Embedded models must be a subclass of django_mongodb_backend.models.EmbeddedModel.",
189198
)
190199

200+
191201
class SubqueryExistsTests(TestCase):
192202
def setUpTestData(self):
193203
# Create test data
@@ -224,11 +234,6 @@ def test_in_subquery(self):
224234
self.assertEqual(queryset.count(), 3)
225235
self.assertQuerySetEqual(queryset, ["Book 2", "Book 3", "Book 4"], lambda book: book.name)
226236

227-
def test_range_query(self):
228-
queryset = Book.objects.filter(author__age__range=(25, 30)).order_by("author__name")
229-
self.assertEqual(queryset.count(), 2)
230-
self.assertQuerySetEqual(queryset, ["Alice", "Alice"], lambda book: book.author.name)
231-
232237
def test_exists_with_foreign_object(self):
233238
subquery = Library.objects.filter(best_seller=OuterRef("name"))
234239
queryset = Book.objects.filter(Exists(subquery))

0 commit comments

Comments
 (0)