Writting py test for sqlalchemy app

Question

I am trying convert unit test into py test. I am using the unit test example

class TestCase(unittest.TestCase):
    def setUp(self):
        app.config[\'TESTIN         


        

Answer 1

I searched high and low for a well explained solution to use SqlAlchemy without Flask-SQLAlchemy and run tests with Pytest, so here's how i have achieved this:

1. Set up your engine & Session objects as per the docs. (I have opted for sessionmaker as i want to check in my app if the session is still available in the Flask's request thread pool, see: https://dev.to/nestedsoftware/flask-and-sqlalchemy-without-the-flask-sqlalchemy-extension-3cf8

2. Import your Base object from wherever you've created it in your app. This will create all the tables in your database defined by the engine.

3. Now we want to Yield a Session back to your unit tests. The idea is to setup before calling Yield & teardown after. Now, in your test you can create a table and populate it with some rows of data etc.

4. Now we must close the Session, this is important!

5. Now by calling Base.metadata.drop_all(bind=engine) we drop all the tables in the database ( we can define a table(s) to drop if required, default is: tables=None)

    engine = create_engine(create_db_connection_str(config), echo=True)
    Session = scoped_session(sessionmaker(bind=engine))
   
    @pytest.fixture(scope="function") # or "module" (to teardown at a module level)
    def db_session():
        Base.metadata.create_all(engine)
        session = Session()
        yield session
        session.close()
        Base.metadata.drop_all(bind=engine)
   

6. Now we can pass the function scoped fixture to each unit test:

    class TestNotebookManager:
        """
            Using book1.mon for this test suite
        """
        book_name = "book1"
   
        def test_load(self, client: FlaskClient, db_session) -> None:
            notebook = Notebook(name=self.book_name)
            db_session.add(book)
            db_session.commit()
            rv = client.get(f"/api/v1/manager/load?name={self.name}")
            assert "200" in rv.status