我正在尝试将单元测试转换为 py 测试。我正在使用单元测试示例
class TestCase(unittest.TestCase):
def setUp(self):
app.config['TESTING'] = True
app.config['CSRF_ENABLED'] = False
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///' + os.path.join(basedir,
'test.db')
db.create_all()
def tearDown(self):
db.session.remove()
db.drop_all()
我不确定,它的 py 测试版本应该是什么。
最佳答案
我上下搜索了一个很好解释的解决方案,以使用 SqlAlchemy 没有 Flask-SQLAlchemy 并使用 Pytest 运行测试,所以我是如何实现这一目标的:
根据文档设置您的
engine
和Session
对象。 (我选择了 sessionmaker,因为如果 session 在 Flask 的请求线程池中仍然可用,我想检查我的应用程序,请参阅:https://dev.to/nestedsoftware/flask-and-sqlalchemy-without-the-flask-sqlalchemy-extension-3cf8从您在应用程序中创建它的任何位置导入您的
Base
对象。这将在您的数据库中创建由engine
定义的所有表。现在我们要将
Session
返回给您的单元测试。这个想法是在调用Yield
之前进行设置,之后进行拆卸。现在,在您的测试中,您可以创建一个表并用一些数据行等填充它。现在我们必须关闭 session ,这很重要!
现在通过调用
Base.metadata.drop_all(bind=engine)
我们删除数据库中的所有表(如果需要,我们可以定义要删除的表,默认为: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)
现在我们可以将函数范围的 fixture 传递给每个单元测试:
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
关于python - 为 sqlalchemy 应用程序编写 pytest,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/23325669/