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
|
from tests.odm.models import DocumentTestModelWithSoftDelete
async def test_get_item(document_soft_delete_not_inserted):
# insert a document with soft delete
result = await document_soft_delete_not_inserted.insert()
# get from db by id
document = await DocumentTestModelWithSoftDelete.get(document_id=result.id)
assert document.is_deleted() is False
assert document.deleted_at is None
assert document.test_int == result.test_int
assert document.test_str == result.test_str
# # delete the document
await document.delete()
assert document.is_deleted() is True
# check if document exist with `.get()`
document = await DocumentTestModelWithSoftDelete.get(document_id=result.id)
assert document is None
# check document exist in trashed
results = (
await DocumentTestModelWithSoftDelete.find_many_in_all().to_list()
)
assert len(results) == 1
async def test_find_one(document_soft_delete_not_inserted):
result = await document_soft_delete_not_inserted.insert()
# # delete the document
await result.delete()
# check if document exist with `.find_one()`
document = await DocumentTestModelWithSoftDelete.find_one(
DocumentTestModelWithSoftDelete.id == result.id
)
assert document is None
async def test_find(documents_soft_delete_not_inserted):
# insert 3 documents
inserted_docs = []
for doc in documents_soft_delete_not_inserted:
result = await doc.insert()
inserted_docs.append(result)
# use `.find_many()` to get them all
results = await DocumentTestModelWithSoftDelete.find().to_list()
assert len(results) == 3
# delete one of them
await inserted_docs[0].delete()
# check items in with `.find_many()`
results = await DocumentTestModelWithSoftDelete.find_many().to_list()
assert len(results) == 2
founded_documents_id = [doc.id for doc in results]
assert inserted_docs[0].id not in founded_documents_id
# check in trashed items
results = (
await DocumentTestModelWithSoftDelete.find_many_in_all().to_list()
)
assert len(results) == 3
async def test_find_many(documents_soft_delete_not_inserted):
# insert 2 documents
item_1 = await documents_soft_delete_not_inserted[0].insert()
item_2 = await documents_soft_delete_not_inserted[1].insert()
# use `.find_many()` to get them all
results = await DocumentTestModelWithSoftDelete.find_many().to_list()
assert len(results) == 2
# delete one of them
await item_1.delete()
# check items in with `.find_many()`
results = await DocumentTestModelWithSoftDelete.find_many().to_list()
assert len(results) == 1
assert results[0].id == item_2.id
# check in trashed items
results = (
await DocumentTestModelWithSoftDelete.find_many_in_all().to_list()
)
assert len(results) == 2
async def test_hard_delete(document_soft_delete_not_inserted):
result = await document_soft_delete_not_inserted.insert()
await result.hard_delete()
# check items in with `.find_many()`
results = await DocumentTestModelWithSoftDelete.find_many().to_list()
assert len(results) == 0
# check in trashed
results = (
await DocumentTestModelWithSoftDelete.find_many_in_all().to_list()
)
assert len(results) == 0
|
