设置数据库以在Flask中进行测试

时间:2014-04-30 23:43:36

标签: python testing flask sqlalchemy

我正在开发我的第一个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

问题是:如何实现setUptearDown以便在我的测试中我可以使用我的模型和连接来测试数据库?如果我只导入db,它将适用于开发数据库。

如果它有帮助,我不需要从头开始创建测试数据库,我使用Flask-Migrate并且测试可以假设测试数据库已初始化并且为空。

欢迎任何评论,如果我的设计存在缺陷,我不介意重构。

1 个答案:

答案 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