olog/co3
1
0
Fork 0
Code Issues 3 Pull Requests Packages Projects Releases Wiki Activity

loosen type Mapper args, add more Database tests

Browse Source
This commit is contained in:
Sam G.
2024-04-18 12:31:16 -07:00
parent b369428eb3
commit 1f89f0dee8
6 changed files with 61 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tests/test_co3.py
View File

@@ -7,14 +7,14 @@ def test_mapper_getters():
veg_comp = veg.vegetable_schema.get_component('vegetable')
tom_comp = veg.vegetable_schema.get_component('tomato')
assert veg.vegetable_mapper.get_attribute_comp(veg.Vegetable) is veg_comp
assert veg.vegetable_mapper.get_attribute_comp(veg.Tomato) is tom_comp
assert veg.vegetable_mapper.get_attr_comp(veg.Vegetable) is veg_comp
assert veg.vegetable_mapper.get_attr_comp(veg.Tomato) is tom_comp
tom_aging = veg.vegetable_schema.get_component('tomato_aging_states')
tom_cooking = veg.vegetable_schema.get_component('tomato_cooking_states')
assert veg.vegetable_mapper.get_collation_comp(veg.Tomato, 'aging') is tom_aging
assert veg.vegetable_mapper.get_collation_comp(veg.Tomato, 'cooking') is tom_cooking
assert veg.vegetable_mapper.get_coll_comp(veg.Tomato, 'aging') is tom_aging
assert veg.vegetable_mapper.get_coll_comp(veg.Tomato, 'cooking') is tom_cooking
def test_mapper_attach():
assert veg.vegetable_mapper.attach(

18
tests/test_database.py
View File

@@ -20,6 +20,24 @@ def test_database_insert():
tomato = veg.Tomato('t1', 5)
veg.vegetable_mapper.collect(tomato)
# test instance as arg
tom_comp = veg.vegetable_mapper.get_attr_comp(tomato)
inserts = veg.vegetable_mapper.collector.collect_inserts()
tom_inserts = inserts.get(tom_comp)
assert tom_inserts is not None
with db.engine.connect() as connection:
assert db.manager.insert(
connection,
tom_comp,
tom_inserts,
) is not None
def test_database_insert_many():
tomato = veg.Tomato('t2', 5)
veg.vegetable_mapper.collect(tomato)
with db.engine.connect() as connection:
assert db.manager.insert_many(
connection,

8
tests/test_mapper.py
View File

@@ -7,14 +7,14 @@ def test_mapper_getters():
veg_comp = veg.vegetable_schema.get_component('vegetable')
tom_comp = veg.vegetable_schema.get_component('tomato')
assert veg.vegetable_mapper.get_attribute_comp(veg.Vegetable) is veg_comp
assert veg.vegetable_mapper.get_attribute_comp(veg.Tomato) is tom_comp
assert veg.vegetable_mapper.get_attr_comp(veg.Vegetable) is veg_comp
assert veg.vegetable_mapper.get_attr_comp(veg.Tomato) is tom_comp
tom_aging = veg.vegetable_schema.get_component('tomato_aging_states')
tom_cooking = veg.vegetable_schema.get_component('tomato_cooking_states')
assert veg.vegetable_mapper.get_collation_comp(veg.Tomato, 'aging') is tom_aging
assert veg.vegetable_mapper.get_collation_comp(veg.Tomato, 'cooking') is tom_cooking
assert veg.vegetable_mapper.get_coll_comp(veg.Tomato, 'aging') is tom_aging
assert veg.vegetable_mapper.get_coll_comp(veg.Tomato, 'cooking') is tom_cooking
def test_mapper_attach():
assert veg.vegetable_mapper.attach(