from database_setup.models import CategoryGroup, Guild
from sqlmodel import Session, SQLModel, create_engine

url = "sqlite:///database.db"
engine = create_engine(url)
SQLModel.metadata.create_all(engine)

with Session(engine) as session:
    guild = Guild(id=1, name="test guild", submission_channel=1, verification_channel=2)
    session.add(guild)

    group1 = CategoryGroup(name="group1", channel_id=1, guild_id=1, message_id=1)
    group2 = CategoryGroup(name="group2", channel_id=2, guild_id=1, message_id=2)
    session.add(group1)
    session.add(group2)

    session.commit()