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
|
from beanie import Link
from tests.odm.models import (
Bicycle,
Bike,
Bus,
Car,
Doc2NonRoot,
DocNonRoot,
Owner,
Vehicle,
)
class TestInheritance:
async def test_inheritance(self, db):
bicycle_1 = await Bicycle(color="white", frame=54, wheels=29).insert()
bicycle_2 = await Bicycle(color="red", frame=52, wheels=28).insert()
bike_1 = await Bike(color="black", fuel="gasoline").insert()
car_1 = await Car(color="grey", body="sedan", fuel="gasoline").insert()
car_2 = await Car(
color="white", body="crossover", fuel="diesel"
).insert()
bus_1 = await Bus(
color="white", seats=80, body="bus", fuel="diesel"
).insert()
bus_2 = await Bus(
color="yellow", seats=26, body="minibus", fuel="diesel"
).insert()
white_vehicles = await Vehicle.find(
Vehicle.color == "white", with_children=True
).to_list()
cars_only = await Car.find().to_list()
cars_and_buses = await Car.find(
Car.fuel == "diesel", with_children=True
).to_list()
big_bicycles = await Bicycle.find(Bicycle.wheels > 28).to_list()
await Bike.find().update({"$set": {Bike.color: "yellow"}})
sedan = await Car.find_one(Car.body == "sedan")
sedan.color = "yellow"
await sedan.save()
# get using Vehicle should return Bike instance
updated_bike = await Vehicle.get(bike_1.id, with_children=True)
assert isinstance(sedan, Car)
assert isinstance(updated_bike, Bike)
assert updated_bike.color == "yellow"
assert Bus._parent is Car
assert len(big_bicycles) == 1
assert big_bicycles[0].wheels > 28
assert len(white_vehicles) == 3
assert len(cars_only) == 2
assert {Car, Bus} == set(i.__class__ for i in cars_and_buses)
assert {Bicycle, Car, Bus} == set(i.__class__ for i in white_vehicles)
white_vehicles_2 = await Car.find(Vehicle.color == "white").to_list()
assert len(white_vehicles_2) == 1
for i in cars_and_buses:
assert i.fuel == "diesel"
for e in (bicycle_1, bicycle_2, bike_1, car_1, car_2, bus_1, bus_2):
assert isinstance(e, Vehicle)
await e.delete()
async def test_links(self, db):
car_1 = await Car(color="grey", body="sedan", fuel="gasoline").insert()
car_2 = await Car(
color="white", body="crossover", fuel="diesel"
).insert()
bus_1 = await Bus(
color="white", seats=80, body="bus", fuel="diesel"
).insert()
owner = await Owner(name="John").insert()
owner.vehicles = [car_1, car_2, bus_1]
await owner.save()
# re-fetch from DB w/o links
owner = await Owner.get(owner.id)
assert {Link} == set(i.__class__ for i in owner.vehicles)
await owner.fetch_all_links()
assert {Car, Bus} == set(i.__class__ for i in owner.vehicles)
# re-fetch from DB with resolved links
owner = await Owner.get(owner.id, fetch_links=True)
assert {Car, Bus} == set(i.__class__ for i in owner.vehicles)
for e in (owner, car_1, car_2, bus_1):
await e.delete()
def test_non_root_inheritance(self):
assert DocNonRoot._class_id is None
assert Doc2NonRoot._class_id is None
assert DocNonRoot.get_collection_name() == "DocNonRoot"
assert Doc2NonRoot.get_collection_name() == "Doc2NonRoot"
def test_class_ids(self):
assert Vehicle._class_id == "Vehicle"
assert Vehicle.get_collection_name() == "Vehicle"
assert Car._class_id == "Vehicle.Car"
assert Car.get_collection_name() == "Vehicle"
assert Bus._class_id == "Vehicle.Car.Bus"
assert Bus.get_collection_name() == "Vehicle"
assert Bike._class_id == "Vehicle.Bike"
assert Bike.get_collection_name() == "Vehicle"
assert Bicycle._class_id == "Vehicle.Bicycle"
assert Bicycle.get_collection_name() == "Vehicle"
assert Owner._class_id is None
|
