2

我正在尝试对我的 Flask Web 应用程序进行单元测试。我正在使用我在 Flask 的 Udemy 课程中看到的模式和类似于在线 Flask Mega-Tutorial 的模式（http://blog.miguelgrinberg.com/post/the-flask-mega-tutorial-part-vii-单元测试）。我遇到的问题是测试实际上并没有创建它自己的数据库——而是它使用了生产数据库并将其搞砸了。

下面是 tests.py 脚本的样子：

import os,sys
sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
basedir = os.path.abspath(os.path.dirname(__file__))

import unittest
from myapp import app, db
from user.models import User

class UserTest(unittest.TestCase):

    def setUp(self):
        self.db_uri = 'sqlite:///' + os.path.join(basedir, 'test.db')
        app.config['TESTING'] = True
        app.config['WTF_CSRF_ENABLED'] = False
        app.config['SQL_ALCHEMY_DATABASE_URI'] = self.db_uri
        self.app = app.test_client()
        db.create_all()

    def tearDown(self):
        db.session.remove()
        db.drop_all()

    def test_models(self):
        #Create a customer user
        user = User("John Doe", "jdoe@jdoe.com", "jdoe", "password", is_consultant=False)
        db.session.add(user)
        db.session.commit()

        #Create 2 consultant users
        user1 = User("Jane Doe", "janedoe@gg.com", "janedoe", "password", is_consultant=True)
        db.session.add(user1)
        user2 = User("Nikola Tesla", "nikola@tesla.com", "nikola", "password", is_consultant=True)
        db.session.add(user2)
        db.session.commit()

        #Check that all users exist
        assert len(User.query.all()) is 3

我的应用程序 init 在同一个文件夹中，如下所示：

from flask import Flask
from flask.ext.sqlalchemy import SQLAlchemy
from flask.ext.migrate import Migrate
from flask.ext.login import LoginManager

app = Flask(__name__)
app.config.from_object('config')
db = SQLAlchemy(app)

# flask-login extension setup
login_manager = LoginManager()
login_manager.init_app(app)

# migration setup
migrate = Migrate(app, db)

from user import views

我不明白发生了什么事。它从不创建 test.db SQLite 数据库。它总是创建 app.db 生产数据库。如果它在那里，它完全搞砸了数据库。之后，如果我执行 python manage.py runserver - 它不再起作用。它说找不到表。因为拆解删除了所有表格。到底是怎么回事？我该如何解决？

4

1 回答 1

5

天哪，我想通了。我为数据库 URI 设置了错误的键。它应该是：app.config['SQLALCHEMY_DATABASE_URI'] = self.db_uri.

所以一切都很好。做就是了：

class UserTest(unittest.TestCase):

    def setUp(self):
        self.db_uri = 'sqlite:///' + os.path.join(basedir, 'test.db')
        app.config['TESTING'] = True
        app.config['WTF_CSRF_ENABLED'] = False
        app.config['SQLALCHEMY_DATABASE_URI'] = self.db_uri
        self.app = app.test_client()
        db.create_all()

一切都按预期工作。

我通过在测试中设置断点并查看 app.config 是什么来检查发生了什么——我看到既有一个SQL_ALCHEMY_DATABASE_URI键（它什么也没做，我正在设置）和一个SQLALCHEMY_DATABASE_URI键（这个是最重要的）。

于 2015-07-30T21:50:36.337 回答