From ce3ed3c0015d60a076333d5f4fed8fdea684fa02 Mon Sep 17 00:00:00 2001
From: imcompiling <develop@imcompiling.com>
Date: Mon, 3 Mar 2025 21:53:19 -0500
Subject: [PATCH] Just more testing.

---
 database/database_test.py | 29 ++++++++++++++++++++---------
 1 file changed, 20 insertions(+), 9 deletions(-)

diff --git a/database/database_test.py b/database/database_test.py
index b2ba0f0..1212c51 100644
--- a/database/database_test.py
+++ b/database/database_test.py
@@ -1,18 +1,29 @@
-from sqlmodel import Session, SQLModel, create_engine
+from sqlmodel import Session, SQLModel, create_engine, select
 
-from database.models import CategoryGroup, Guild
+from models import Guild
 
 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)
+    # 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()
 
-    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)
+    # anotherone = Guild(id=2, name="test guild 2", submission_channel=1, verification_channel=2)
+    # session.add(anotherone)
+    # group1 = CategoryGroup(name="group1", channel_id=1, guild_id=2, message_id=1)
+    # group2 = CategoryGroup(name="group2", channel_id=2, guild_id=2, message_id=2)
+    # session.add(group1)
+    # session.add(group2)
+    # session.commit()
 
-    session.commit()
+    stmt = select(Guild).where(Guild.id == 1)
+    guild = session.scalar(stmt)
+
+    print(guild.category_groups)