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 recipe_graph import db
from sqlalchemy import select from sqlalchemy import select
from sqlalchemy.exc import SQLAlchemyError from sqlalchemy.exc import SQLAlchemyError
from sqlalchemy.orm import sessionmaker
import sqlalchemy import sqlalchemy
import pytest import pytest
@ -49,6 +50,11 @@ def test_db_connection(engine):
finally: finally:
assert connected == True assert connected == True
def test_get_session():
session = db.get_session()
eng = db.get_engine()
session == sessionmaker(eng)
def test_db_classes(tables): def test_db_classes(tables):
assert len(tables) > 0 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