Skip to content

Commit bc77322

Browse files
committed
Update unit tests
1 parent 8913a74 commit bc77322

File tree

1 file changed

+14
-14
lines changed

1 file changed

+14
-14
lines changed

tests/queryset/test_queryset.py

Lines changed: 14 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -4739,33 +4739,33 @@ class Bar(Document):
47394739
bars = list(Bar.objects.read_concern(None))
47404740
assert bars == [bar]
47414741

4742-
bars = Bar.objects.read_concern(ReadConcern(level="local"))
4743-
assert bars._read_concern == ReadConcern(level="local")
4744-
assert bars._cursor.collection.read_concern == ReadConcern(level="local")
4742+
bars = Bar.objects.read_concern({"level": "local"})
4743+
assert bars._read_concern.document == {"level": "local"}
4744+
assert bars._cursor.collection.read_concern.document == {"level": "local"}
47454745

4746-
# Make sure that `.read_concern(...)` does accept string values.
4746+
# Make sure that `.read_concern(...)` does not accept string values.
47474747
with pytest.raises(TypeError):
47484748
Bar.objects.read_concern("local")
47494749

47504750
def assert_read_concern(qs, expected_read_concern):
4751-
assert qs._read_concern == expected_read_concern
4752-
assert qs._cursor.collection.read_concern == expected_read_concern
4751+
assert qs._read_concern.document == expected_read_concern
4752+
assert qs._cursor.collection.read_concern.document == expected_read_concern
47534753

47544754
# Make sure read concern is respected after a `.skip(...)`.
4755-
bars = Bar.objects.skip(1).read_concern(ReadConcern("majority"))
4756-
assert_read_concern(bars, ReadConcern("majority"))
4755+
bars = Bar.objects.skip(1).read_concern({"level": "local"})
4756+
assert_read_concern(bars, {"level": "local"})
47574757

47584758
# Make sure read concern is respected after a `.limit(...)`.
4759-
bars = Bar.objects.limit(1).read_concern(ReadConcern("majority"))
4760-
assert_read_concern(bars, ReadConcern("majority"))
4759+
bars = Bar.objects.limit(1).read_concern({"level": "local"})
4760+
assert_read_concern(bars, {"level": "local"})
47614761

47624762
# Make sure read concern is respected after an `.order_by(...)`.
4763-
bars = Bar.objects.order_by("txt").read_concern(ReadConcern("majority"))
4764-
assert_read_concern(bars, ReadConcern("majority"))
4763+
bars = Bar.objects.order_by("txt").read_concern({"level": "local"})
4764+
assert_read_concern(bars, {"level": "local"})
47654765

47664766
# Make sure read concern is respected after a `.hint(...)`.
4767-
bars = Bar.objects.hint([("txt", 1)]).read_concern(ReadConcern("majority"))
4768-
assert_read_concern(bars, ReadConcern("majority"))
4767+
bars = Bar.objects.hint([("txt", 1)]).read_concern({"level": "majority"})
4768+
assert_read_concern(bars, {"level": "majority"})
47694769

47704770
def test_json_simple(self):
47714771
class Embedded(EmbeddedDocument):

0 commit comments

Comments
 (0)