fiona-zhong

程序开发过程中,代码是为了完成需求,当代码编译通过后,能不能保证功能的正常实现,需要我们编写测试代码,模拟程序运行过程,检验功能是否符合预期。

单元测试主要面向一些功能单一的模块进行。

单元测试,实际上就是一些“断言”的代码,用于判断函数或对象的一个方法所产生的结果是否符合预期

python中,断言是申明结果为真的判定,如果表达式为假,程序会发生异常。

 

常用的断言有:

  assertTrue()  条件为True,则pass

  assertFalse()  条件为False,则pass

  assertEqual()  两个值相等,则pass

  assertNotEqual()  两个值不相等,则pass

  assertIsNone()  不存在,则pass

  assertIsNotNone  存在,则pass

 

Flask单元测试中用到unittest

基本的结构是:

  import unittest

  class Test(unittest.TestCase):

    def setUp(self):

      pass

    def tearDown(self):

      pass

    def test_something1(self):

      pass

    def test_something2(self):

      pass

  if __name__=="__main__":

    unittest.main()

 

上面代码执行顺序是:setUp()---->test_something1()---->tearDown()

          setUp()---->test_something2()----->tearDown()

setUp():

  用于初始化app,db等

tearDown():

  用于释放资源等

test_something():

  测试主题,函数名以test_开头

 

完整代码如下:

import unittest

from apps.model import db

class Test(unittest.TestCase)

  from apps import instance_app

  def setUp(self):

    # 实例化核心对象

    self.app = instance_app()

    # 更新server_name

    self.app.config.update(SERVER_NAME='your server name')

    # 获取应用上下文

    self.app_ctx = self.app.app_context()

    self.app_ctx.push()

    # 配置测试用数据库

    self.app.config["SQLALCHEMY_DATABASE_URI"] = "mysql+cymysql://用户名:密码@IP:PORT/DATABASE_NAME"

    # 开启测试标记,可以帮忙定位测试信息

    self.app.config("TESTING") = True

    # 创建表

    self.client = self.app.test_client()

    db.create_all()

  def tearDown(self):

    移除数据库会话对应,如果没有移除数据库会话对象,会导致程序一直运行,无法结束。

    db.session.remove()

    # 删除测试数据库的所有表

    db.drop_all()

  def test_case(self):

    res = self.client.get(url_for("endpoint"))

    res1 = self.client.post(url_for(endpoint), data={})

    # 断言:判断返回结果中包含test字符串  as_text=True是将结果转化为字符串

    self.assertTrue("test" in res.get_data(as_text=True))

    self.assertTrue(res1.status_code==200)

 

if __name__=="__main__":

  unittest.main()

    

 

分类:

flask

技术点:

相关文章: