from co3.components import Relation
from co3.databases import SQLDatabase

from setups import vegetables as veg


db = None

def test_database_init():
    global db

    db = SQLDatabase('sqlite://')
    assert True
    
def test_database_recreate():
    db.recreate(veg.vegetable_schema)
    assert True

def test_database_insert():
    tomato = veg.Tomato('t1', 5)
    veg.vegetable_mapper.collect(tomato)

    with db.engine.connect() as connection:
        assert db.manager.insert_many(
            connection,
            veg.vegetable_mapper.collector.inserts,
        ) is not None

def test_database_access():
    agg_table = veg.vegetable_mapper.compose(veg.Tomato)

    with db.engine.connect() as connection:
        assert db.accessor.select(
            connection,
            agg_table,
        ) is not None