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

reformat w black

Browse Source
This commit is contained in:
Bastien Gérard
2020-03-03 23:51:11 +01:00
parent ba34589065
commit ff4d57032a
3 changed files with 31 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
tests/queryset/test_queryset.py
View File

@@ -4671,41 +4671,33 @@ class TestQueryset(unittest.TestCase):
bars = list(Bar.objects.read_concern(None))
assert bars == [bar]
bars = Bar.objects.read_concern(ReadConcern(level='local'))
assert bars._read_concern == ReadConcern(level='local')
assert (
bars._cursor.collection.read_concern
== ReadConcern(level='local')
)
bars = Bar.objects.read_concern(ReadConcern(level="local"))
assert bars._read_concern == ReadConcern(level="local")
assert bars._cursor.collection.read_concern == ReadConcern(level="local")
# Make sure that `.read_concern(...)` does accept string values.
with pytest.raises(TypeError):
Bar.objects.read_concern('local')
Bar.objects.read_concern("local")
def assert_read_concern(qs, expected_read_concern):
assert qs._read_concern == expected_read_concern
assert qs._cursor.collection.read_concern == expected_read_concern
# Make sure read concern is respected after a `.skip(...)`.
bars = Bar.objects.skip(1).read_concern(ReadConcern('majority'))
assert_read_concern(bars, ReadConcern('majority'))
bars = Bar.objects.skip(1).read_concern(ReadConcern("majority"))
assert_read_concern(bars, ReadConcern("majority"))
# Make sure read concern is respected after a `.limit(...)`.
bars = Bar.objects.limit(1).read_concern(ReadConcern('majority'))
assert_read_concern(bars, ReadConcern('majority'))
bars = Bar.objects.limit(1).read_concern(ReadConcern("majority"))
assert_read_concern(bars, ReadConcern("majority"))
# Make sure read concern is respected after an `.order_by(...)`.
bars = Bar.objects.order_by("txt").read_concern(
ReadConcern('majority')
)
assert_read_concern(bars, ReadConcern('majority'))
bars = Bar.objects.order_by("txt").read_concern(ReadConcern("majority"))
assert_read_concern(bars, ReadConcern("majority"))
# Make sure read concern is respected after a `.hint(...)`.
bars = Bar.objects.hint([("txt", 1)]).read_concern(
ReadConcern('majority')
)
assert_read_concern(bars, ReadConcern('majority'))
bars = Bar.objects.hint([("txt", 1)]).read_concern(ReadConcern("majority"))
assert_read_concern(bars, ReadConcern("majority"))
def test_json_simple(self):
class Embedded(EmbeddedDocument):