diff --git a/test/test_db.py b/test/test_db.py new file mode 100644 index 0000000..6d6bfdd --- /dev/null +++ b/test/test_db.py @@ -0,0 +1,20 @@ +from sqlite3 import connect +from recipe_graph import db +from sqlalchemy.exc import SQLAlchemyError +import sqlalchemy + +import pytest + +@pytest.fixture +def engine() -> sqlalchemy.engine.Engine: + return db.get_engine() + +def test_db_connection(engine): + connected = False + try: + engine.connect() + connected = True + except (SQLAlchemyError): + pass + finally: + assert(connected == True)