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
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
|
import pytest
from beanie import After, Before
from tests.odm.models import (
DocumentWithActions,
DocumentWithActions2,
InheritedDocumentWithActions,
)
class TestActions:
@pytest.mark.parametrize(
"doc_class",
[
DocumentWithActions,
DocumentWithActions2,
InheritedDocumentWithActions,
],
)
async def test_actions_insert(self, doc_class):
test_name = f"test_actions_insert_{doc_class}"
sample = doc_class(name=test_name)
await sample.insert()
assert sample.name != test_name
assert sample.name == test_name.capitalize()
assert sample.num_1 == 1
assert sample.num_2 == 9
@pytest.mark.parametrize(
"doc_class",
[
DocumentWithActions,
DocumentWithActions2,
InheritedDocumentWithActions,
],
)
async def test_actions_replace(self, doc_class):
test_name = f"test_actions_replace_{doc_class}"
sample = doc_class(name=test_name)
await sample.insert()
await sample.replace()
assert sample.num_1 == 2
assert sample.num_3 == 99
@pytest.mark.parametrize(
"doc_class",
[
DocumentWithActions,
DocumentWithActions2,
InheritedDocumentWithActions,
],
)
async def test_skip_actions_insert(self, doc_class):
test_name = f"test_skip_actions_insert_{doc_class}"
sample = doc_class(name=test_name)
await sample.insert(skip_actions=[After, "capitalize_name"])
# capitalize_name has been skipped
assert sample.name == test_name
# add_one has not been skipped
assert sample.num_1 == 1
# num_2_change has been skipped
assert sample.num_2 == 10
@pytest.mark.parametrize(
"doc_class",
[
DocumentWithActions,
DocumentWithActions2,
InheritedDocumentWithActions,
],
)
async def test_skip_actions_replace(self, doc_class):
test_name = f"test_skip_actions_replace{doc_class}"
sample = doc_class(name=test_name)
await sample.insert()
await sample.replace(skip_actions=[Before, "num_3_change"])
# add_one has been skipped
assert sample.num_1 == 1
# num_3_change has been skipped
assert sample.num_3 == 100
@pytest.mark.parametrize(
"doc_class",
[
DocumentWithActions,
DocumentWithActions2,
InheritedDocumentWithActions,
],
)
async def test_actions_delete(self, doc_class):
test_name = f"test_actions_delete_{doc_class}"
sample = doc_class(name=test_name)
await sample.delete()
assert sample.Inner.inner_num_1 == 1
assert sample.Inner.inner_num_2 == 2
@pytest.mark.parametrize(
"doc_class",
[
DocumentWithActions,
DocumentWithActions2,
InheritedDocumentWithActions,
],
)
async def test_actions_update(self, doc_class):
test_name = f"test_actions_update_{doc_class}"
sample = doc_class(name=test_name)
await sample.insert()
await sample.update({"$set": {"name": "new_name"}})
assert sample.name == "new_name"
assert sample.num_1 == 1
assert sample.num_2 == 9
assert sample._private_num == 101
await sample.set({"name": "awesome_name"})
assert sample._private_num == 102
assert sample.num_2 == 9
assert sample.name == "awesome_name"
@pytest.mark.parametrize(
"doc_class",
[
DocumentWithActions,
DocumentWithActions2,
InheritedDocumentWithActions,
],
)
async def test_actions_save(self, doc_class):
test_name = f"test_actions_save_{doc_class}"
sample = doc_class(name=test_name)
await sample.save()
assert sample.num_1 == 1
|
