added test for db.get_session()

This commit is contained in:
Andrei Stoica 2022-10-15 12:51:08 -04:00
parent 3a45cfb02a
commit 6189de8039
2 changed files with 13 additions and 0 deletions

View File

@ -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

View File

@ -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