我最近在运行我的Flask应用程序的单元测试时遇到了一个问题,大约有100个单元测试。所有单元测试都将通过,但是当一次性运行它们时,它们会失败并显示以下错误:
OperationalError: (OperationalError) FATAL: remaining connection slots are reserved for non-replication superuser connections
所有内容都在本地机器的virtualbox/vagrant/ubuntu12.04实例中运行。我的postgres max_connections设置为100,因此我认为连接没有关闭,在运行了100次测试后,我使用了所有可用的连接。
这个人Flask unit tests with SQLAlchemy and PostgreSQL exhausts db connections看起来遇到了完全相同的问题。Mike/Zzzeek (sqlalchemy开发者)甚至回复说可能在create_app()中发生了某些事情,因此我也将其包含在下面。
这是否意味着我没有在某个地方关闭连接?所有这些错误都是由我的unittest的setUp()方法中的db.create_all()
触发的。
# test.py
class TestCase(DataMixin, Base):
"""Base test class"""
def create_app(self):
return create_app(TestConfig())
def setUp(self):
db.create_all()
def tearDown(self):
db.session.remove()
db.drop_all()
# app.py
def create_app(config=None):
app = Flask(__name__)
# Config
app.config.from_object(BaseConfig())
if config is not None:
app.config.from_object(config)
# Extensions
db.init_app(app)
mail.init_app(app)
bcrypt.init_app(app)
# Blueprints
app.register_blueprint(core_blueprint, url_prefix='/')
app.register_blueprint(accounts_blueprint, url_prefix='/account')
app.register_blueprint(admin_blueprint, url_prefix='/admin')
app.register_blueprint(cart_blueprint, url_prefix='/cart')
# Login Manager
login_manager.setup_app(app, add_context_processor=True)
login_manager.login_view = "accounts.login"
login_manager.user_callback = load_user
# Templates
app.jinja_env.globals['is_admin'] = is_admin
app.jinja_env.globals['is_staff'] = is_staff
@app.context_processor
def inject_cart():
cart = count = None
if current_user.is_authenticated():
cart = current_user.get_cart()
return dict(cart=cart)
# Error Handling
@app.errorhandler(404)
def page_not_found(error):
return render_template('404.html'), 404
return app