1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
|
import pytest
from beanie.odm.operators.find.logical import And, Nor, Not, Or
from tests.odm.models import Sample
async def test_and():
q = And(Sample.integer == 1)
assert q == {"integer": 1}
q = And(Sample.integer == 1, Sample.nested.integer > 3)
assert q == {"$and": [{"integer": 1}, {"nested.integer": {"$gt": 3}}]}
async def test_not(preset_documents):
q = Not(Sample.integer == 1)
assert q == {"integer": {"$not": {"$eq": 1}}}
docs = await Sample.find(q).to_list()
assert len(docs) == 7
with pytest.raises(AttributeError):
q = Not(And(Sample.integer == 1, Sample.nested.integer > 3))
await Sample.find(q).to_list()
async def test_nor():
q = Nor(Sample.integer == 1)
assert q == {"$nor": [{"integer": 1}]}
q = Nor(Sample.integer == 1, Sample.nested.integer > 3)
assert q == {"$nor": [{"integer": 1}, {"nested.integer": {"$gt": 3}}]}
async def test_or():
q = Or(Sample.integer == 1)
assert q == {"integer": 1}
q = Or(Sample.integer == 1, Sample.nested.integer > 3)
assert q == {"$or": [{"integer": 1}, {"nested.integer": {"$gt": 3}}]}
|
