我正在开发我的第一个Flask应用程序。这是我的侧面项目,所以我专注于良好的实践和设计,并花时间。我有点坚持测试 - 我在文档中找到了一些示例,这里有SO,但它们要么不适用于我的应用程序,要么看起来不像Pythonic /设计得好。
相关的代码片段是:
# application module __init__.py
def create_app(config):
app = Flask(__name__)
app.config.from_object('config.%s' % config.title())
return app
config = os.getenv('CONFIG', 'development')
app = create_app(config)
db = SQLAlchemy(app)
# config.py
class Testing(Base):
TESTING = True
SQLALCHEMY_DATABASE_URI = \
'sqlite:///' + os.path.join(_basedir, 'testing.sqlite')
# models.py
class User(db.Model):
__tablename__ = 'user'
id = db.Column(db.Integer, primary_key=True)
username = db.Column(db.String(60), unique=True, nullable=False)
password_hash = db.Column(db.String(60), nullable=False)
# testing.py
class TestCase(unittest.TestCase):
def setUp(self):
self.app = create_app('testing')
# TODO: create connection to testing db
def tearDown(self):
# TODO: abort transaction
pass
问题是:如何实现setUp
和tearDown
以便在我的测试中我可以使用我的模型和连接来测试数据库?如果我只导入db
,它将适用于开发数据库。
如果它有帮助,我不需要从头开始创建测试数据库,我使用Flask-Migrate并且测试可以假设测试数据库已初始化并且为空。
欢迎任何评论,如果我的设计存在缺陷,我不介意重构。
答案 0 :(得分:5)
看起来就像你应该能够运行CONFIG=Testing python -m unittest discover
并让一切正常运转。您可能想要更改的唯一想法是,而不是在测试中调用create_app
,只需从__init __导入它.py:
# testing.py
from . import config, db
class TestCase(unittest.TestCase):
def setUp(self):
self.app = create_app(config)
# db is properly set up to use the testing config
# but any *changes* you make to the db object
# (e. g. monkey-patching) will persist between tests
db.create_all()
def tearDown(self):
db.session.remove()
db.drop_all()
有关示例,请参阅here。