pub/mongoengine
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Use Queryset._query instead of Cursor.__spec for count()

Browse Source
This commit is contained in:
Bastien Gerard
2020-12-13 10:43:26 +01:00
parent aa3ff39ef8
commit c28bb517cb
4 changed files with 27 additions and 45 deletions
Download Patch File Download Diff File Expand all files Collapse all files

50
tests/document/test_instance.py
View File

@@ -2900,50 +2900,32 @@ class TestDocumentInstance(MongoDBTestCase):
# Checks
assert ",".join([str(b) for b in Book.objects.all()]) == "1,2,3,4,5,6,7,8,9"
# bob related books
assert (
",".join(
[
str(b)
for b in Book.objects.filter(
Q(extra__a=bob) | Q(author=bob) | Q(extra__b=bob)
)
]
)
== "1,2,3,4"
bob_books_qs = Book.objects.filter(
Q(extra__a=bob) | Q(author=bob) | Q(extra__b=bob)
)
assert [str(b) for b in bob_books_qs] == ["1", "2", "3", "4"]
assert bob_books_qs.count() == 4
# Susan & Karl related books
assert (
",".join(
[
str(b)
for b in Book.objects.filter(
Q(extra__a__all=[karl, susan])
| Q(author__all=[karl, susan])
| Q(extra__b__all=[karl.to_dbref(), susan.to_dbref()])
)
]
)
== "1"
susan_karl_books_qs = Book.objects.filter(
Q(extra__a__all=[karl, susan])
| Q(author__all=[karl, susan])
| Q(extra__b__all=[karl.to_dbref(), susan.to_dbref()])
)
assert [str(b) for b in susan_karl_books_qs] == ["1"]
assert susan_karl_books_qs.count() == 1
# $Where
assert (
",".join(
[
str(b)
for b in Book.objects.filter(
__raw__={
"$where": """
custom_qs = Book.objects.filter(
__raw__={
"$where": """
function(){
return this.name == '1' ||
this.name == '2';}"""
}
)
]
)
== "1,2"
}
)
assert [str(b) for b in custom_qs] == ["1", "2"]
assert custom_qs.count() == 2
def test_switch_db_instance(self):
register_connection("testdb-1", "mongoenginetest2")