RuntimeError: No application found. Either work inside a view function or push an application contex

python + flask + flask-sqlalchemy 实现mysql数据库操作

python编写测试用例，用于实现MySQL的查询

测试用例如下:

class MySQLTest(unittest.TestCase):

  def test_queryAll(self):
      uas = db.session.query(UserAgent).all()
      self.assertIsNotNone(uas)

      for ua in uas:
        print("\n")
        print(ua)


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

---

一、问题描述

出现下面的报错信息

RuntimeError: No application found. Either work inside a view function or push an application context. See http://flask-sqlalchemy.pocoo.org/contexts/.

大致意思是：没找到应用，要么在视图函数中工作，要么推送应用程序上下文

第一反应，应该和java web应用中编写测试用例，找不到应用上下文有关

---

二、问题分析

测试用例是单独启动一个线程去执行数据库操作，相对于flask的应用时独立的，无法获取flask应用上下文的配置信息（包括数据库连接信息），所以需要推送应用程序上下文（注入上下文对象）

这也是为什么在视图函数中可以正常工作，因为视图函数是运行在flask 的web容器中的。

---

三、解决方案

python中用 with 语句来管理上下文

with 语句实质是上下文管理。
1、上下文管理协议。包含方法__enter__() 和 exit()，支持该协议对象要实现这两个方法。
2、上下文管理器，定义执行with语句时要建立的运行时上下文，负责执行with语句块上下文中的进入与退出操作。
3、进入上下文的时候执行__enter__方法，如果设置as var语句，var变量接受__enter__()方法返回值。
4、如果运行时发生了异常，就退出上下文管理器。调用管理器__exit__方法。

推送应用程序上下文，就是在线程开始执行代码的最上方，
加入 with app.app_context():

class MySQLTest(unittest.TestCase):

  def test_queryAll(self):
    # 导入上下文对象
    with app.app_context():
      uas = db.session.query(UserAgent).all()
      self.assertIsNotNone(uas)

      for ua in uas:
        print("\n")
        print(ua)


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

四、结果验证

加上 with app.app_context():
再写自己的程序逻辑就可以啦 ,测试用例运行成功