tests: better mysql initialization

tests/__init__.py

@@ -0,0 +1,16 @@
+from testcontainers.mysql import MySqlContainer
+import sqlalchemy
+from config import connections
+from scripts.database import seed_utils
+from utils import execute_non_query
+
+mysql = MySqlContainer("mysql:5.7.17", port=3306)
+mysql.start()
+connections["MYSQL"] = mysql.get_connection_url()
+
+engine = sqlalchemy.create_engine(mysql.get_connection_url())
+seed_commands = seed_utils.get_seed_commands()
+
+
+for statement in seed_commands.split(";"):
+    execute_non_query(engine, statement)

tests/test_integration.py

@@ -1,38 +1,11 @@
 from starlette.testclient import TestClient
-import sqlalchemy
-from scripts.database import seed_utils
 import ipdb
-from utils import execute_non_query, execute_raw_query
-from config import connections
 
-from testcontainers.mysql import MySqlContainer
-
-
-mysql = MySqlContainer("mysql:5.7.17", port=3306)
-mysql.start()
-engine = sqlalchemy.create_engine(mysql.get_connection_url())
-connections["MYSQL"] = mysql.get_connection_url()
 
 from main import app
 
 client = TestClient(app)
 
-# ipdb.set_trace()
-seed_commands = seed_utils.get_seed_commands()
-
-
-for statement in seed_commands.split(";"):
-    execute_non_query(engine, statement)
-
-
-def get_all_tabs():
-    sql = "select * from Tab"
-    return execute_raw_query(engine, sql)
-
-
-def test_simple_query():
-    assert len(get_all_tabs()) == 1
-
 
 def test_hello_world():
     response = client.get("/")