如何使用SQLite在Flask中设置测试脚本？

Question

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

这是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", "[email protected]", "jdoe", "password", is_consultant=False)
        db.session.add(user)
        db.session.commit()

        #Create 2 consultant users
        user1 = User("Jane Doe", "[email protected]", "janedoe", "password", is_consultant=True)
        db.session.add(user1)
        user2 = User("Nikola Tesla", "[email protected]", "nikola", "password", is_consultant=True)
        db.session.add(user2)
        db.session.commit()

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

我的app 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 - 它不再起作用了。它说没找到桌子。因为拆解掉了所有的桌子。到底是怎么回事？我该如何解决？

Answer 1

我想出来的Omigod。我为数据库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键（这是重要的）。