2

我有这个代码：

我的应用程序.py：

from flask import Flask
from flask.ext.sqlalchemy import SQLAlchemy

import os

app = Flask(__name__)
app.config["SQLALCHEMY_TRACK_MODIFICATIONS"] = False
app.config["BASE_DIR"] = os.path.abspath(os.path.dirname(__file__))
app.config["SQLALCHEMY_DATABASE_URI"] = "sqlite:///" + os.path.abspath(os.path.join(app.config["BASE_DIR"], "app.db"))

db = SQLAlchemy(app)

from user import User

# create all tables
db.create_all()

if not User.query.filter_by(username="test").first():
    dummy_user = User(username="test", password="", email="")
    db.session.add(dummy_user)
    db.session.commit()

用户.py：

from flask.ext.login import UserMixin

from my_app import db


class User(db.Model):
    __tablename__ = "users"

    id = db.Column(db.Integer, primary_key=True)
    username = db.Column(db.String(255), nullable=False, unique=True)
    email = db.Column(db.String(255), nullable=False, unique=True)
    password = db.Column(db.String(255), nullable=False)

测试.py：

from flask.ext.testing import TestCase
from my_app import app, db
from user import User

import os
import unittest


class MyTestCase(TestCase):
    def create_app(self):
        app.config["SQLALCHEMY_DATABASE_URI"] = "sqlite:///:memory:"
        app.config["TESTING"] = True
        return app

    def setUp(self):
        db.create_all()

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

    def test_dummy(self):
        assert User.query.count() == 0

if __name__ == '__main__':
    unittest.main()

我怎样才能使单元测试工作？

我希望我的单元测试使用与主应用程序分开的内存数据库，但显然我不能按照我尝试的方式去做（只需更改 SQLALCHEMY_DATABASE_URI 值）

提前致谢！

4

3 回答 3

2

听起来像是继承配置的完美案例！

您是否尝试过使用此处找到的模板？Config 基类包含所有不同环境通用的设置。但是您可以拥有一个使用内存数据库的开发环境。例如：

class Config:
  # pass

class DevConfig(Config):
  SQLALCHEMY_DATABASE_URI = 'sqlite://path_to_in_memory_db'

class ProductionConfig(Config):
  SQLALCHEMY_DATABASE_URI = 'postgresql://path_to_production_db'

还值得研究使用工厂模式来创建您的应用程序。

于 2016-02-19T04:51:28.273 回答
1

我使用环境变量来选择带有 pytest 的数据库。这在使用 docker-compose 为 dev 指定生产数据库而不是 sqllite 时效果很好。

# app.py
BASE_DIR = os.path.abspath(os.path.dirname(__file__))
DATABASE = os.environ.get("DB_URI", f"sqlite:///{os.path.join(BASE_DIR, 'app.db')}")

app = Flask(__name__)
app.config["SQLALCHEMY_DATABASE_URI"] = DATABASE

然后在 test.py 中注入 DB_URI 环境变量以在导入应用程序或数据库之前指定内存数据库：

import pytest
import os

os.environ['DB_URI'] = "sqlite:///:memory:"
from app import app, db
from models import Model1, Model2


@pytest.fixture
def client():
    client = app.test_client()

    with app.app_context():
        db.create_all()
        prepare_data()
    yield client

    db.drop_all()
于 2019-12-23T22:40:45.207 回答
0

好的，所以我找到了一个解决方案，或者说是一种解决方法：

在初始化 db 实例之前，我使用环境变量指定要加载的附加配置文件。

所以我最终这样做了：

app.config.from_object("app.config.production")
additional_config = os.environ.get("ADDITIONAL_CONFIG")
if additional_config:
    app.config.from_object(additional_config)

在我的 my_app.py 和我的 tests.py 中：

os.environ["ADDITIONAL_CONFIG"] = "app.config.testing"

from my_app import app, db

（在导入app对象之前定义环境变量当然很重要）

不管怎么说，还是要谢谢你。

于 2016-02-19T03:05:21.833 回答