From 6189de8039ef50816bfad3213d44a59e50bbd5cd Mon Sep 17 00:00:00 2001
From: Andrei Stoica <andrei.stoica.365@gmail.com>
Date: Sat, 15 Oct 2022 12:51:08 -0400
Subject: [PATCH] added test for db.get_session()

---
 test/test_db.py           | 6 ++++++
 test/test_insert_sites.py | 7 +++++++
 2 files changed, 13 insertions(+)
 create mode 100644 test/test_insert_sites.py

diff --git a/test/test_db.py b/test/test_db.py
index 9bde118..f40c5ce 100644
--- a/test/test_db.py
+++ b/test/test_db.py
@@ -2,6 +2,7 @@ import inspect
 from recipe_graph import db
 from sqlalchemy import select
 from sqlalchemy.exc import SQLAlchemyError
+from sqlalchemy.orm import sessionmaker
 import sqlalchemy
 
 import pytest
@@ -49,6 +50,11 @@ def test_db_connection(engine):
     finally:
         assert connected == True
 
+def test_get_session():
+    session = db.get_session()
+    eng = db.get_engine()
+    session == sessionmaker(eng)
+
 
 def test_db_classes(tables):
     assert len(tables) > 0
diff --git a/test/test_insert_sites.py b/test/test_insert_sites.py
new file mode 100644
index 0000000..a792f5f
--- /dev/null
+++ b/test/test_insert_sites.py
@@ -0,0 +1,7 @@
+import inspect
+from recipe_graph import insert_sites
+from sqlalchemy import select
+from sqlalchemy.exc import SQLAlchemyError
+import sqlalchemy
+
+import pytest