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

add latest pymongo 3.9 as part of the CI

Browse Source
This commit is contained in:
Bastien Gérard
2019-10-04 21:30:32 +02:00
parent 02a920feea
commit e3cd553f82
4 changed files with 15 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
tests/queryset/test_queryset.py
View File

@@ -4641,43 +4641,35 @@ class TestQueryset(unittest.TestCase):
bars = Bar.objects.read_preference(ReadPreference.SECONDARY_PREFERRED)
self.assertEqual(bars._read_preference, ReadPreference.SECONDARY_PREFERRED)
self.assertEqual(
bars._cursor._Cursor__read_preference, ReadPreference.SECONDARY_PREFERRED
bars._cursor.collection.read_preference, ReadPreference.SECONDARY_PREFERRED
)
# Make sure that `.read_preference(...)` does accept string values.
self.assertRaises(TypeError, Bar.objects.read_preference, "Primary")
def assert_read_pref(qs, expected_read_pref):
self.assertEqual(qs._read_preference, expected_read_pref)
self.assertEqual(qs._cursor.collection.read_preference, expected_read_pref)
# Make sure read preference is respected after a `.skip(...)`.
bars = Bar.objects.skip(1).read_preference(ReadPreference.SECONDARY_PREFERRED)
self.assertEqual(bars._read_preference, ReadPreference.SECONDARY_PREFERRED)
self.assertEqual(
bars._cursor._Cursor__read_preference, ReadPreference.SECONDARY_PREFERRED
)
assert_read_pref(bars, ReadPreference.SECONDARY_PREFERRED)
# Make sure read preference is respected after a `.limit(...)`.
bars = Bar.objects.limit(1).read_preference(ReadPreference.SECONDARY_PREFERRED)
self.assertEqual(bars._read_preference, ReadPreference.SECONDARY_PREFERRED)
self.assertEqual(
bars._cursor._Cursor__read_preference, ReadPreference.SECONDARY_PREFERRED
)
assert_read_pref(bars, ReadPreference.SECONDARY_PREFERRED)
# Make sure read preference is respected after an `.order_by(...)`.
bars = Bar.objects.order_by("txt").read_preference(
ReadPreference.SECONDARY_PREFERRED
)
self.assertEqual(bars._read_preference, ReadPreference.SECONDARY_PREFERRED)
self.assertEqual(
bars._cursor._Cursor__read_preference, ReadPreference.SECONDARY_PREFERRED
)
assert_read_pref(bars, ReadPreference.SECONDARY_PREFERRED)
# Make sure read preference is respected after a `.hint(...)`.
bars = Bar.objects.hint([("txt", 1)]).read_preference(
ReadPreference.SECONDARY_PREFERRED
)
self.assertEqual(bars._read_preference, ReadPreference.SECONDARY_PREFERRED)
self.assertEqual(
bars._cursor._Cursor__read_preference, ReadPreference.SECONDARY_PREFERRED
)
assert_read_pref(bars, ReadPreference.SECONDARY_PREFERRED)
def test_read_preference_aggregation_framework(self):
class Bar(Document):