added test for db.get_session()
This commit is contained in:
parent
3a45cfb02a
commit
6189de8039
|
|
@ -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
|
||||||
|
|
|
||||||
|
|
@ -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
|
||||||
Loading…
Reference in New Issue