【妙用协程】 - 单元测试的setUp和tearDown

MartinDai 发布于2019-07-24 17:43 / 2498人阅读

摘要：一般的做法是把这些动作写在和的两个方法里，单元测试框架会负责在开始和结束的时候调用这两个方法。从视觉上无法直观的指导原来和是一对的。然后再把这个小的上下文附着到主测试逻辑上这里利用了单元测试的的特性，把转化为回调在的时候就设置好。

很多测试都需要在启动的时候做一些事情，然后在结束的时候再把做的事情给清理了。一般的做法是把这些动作写在setUp和tearDown的两个方法里，单元测试框架会负责在开始和结束的时候调用这两个方法。

class SomeTest(unittest.case.TestCase):
    def setUp(self):
        super(SomeTest, self).setUp()
        setup_db()

    def tearDown(self):
        clean_db()
        super(SomeTest, self).tearDown()

这种写法有好几个烦人的地方。首先是Logic Locality不好的问题：setup_db()和clean_db()是分在两处的，中间可能隔着很长一段代码。从视觉上无法直观的指导setup_db()原来和clean_db()是一对的。
其次是很难重用的问题（上纲上线的话就是复杂度不好管理的问题），为了避免重复写公共的setUp和tearDown一般会抽取出一个UsingDbTest这样的基类。这样所有的子类必须记得super(xxx, self).setUp()，否则就会覆盖掉基类的setUp。其次在需要有多个维度的东西需要复用的时候，比如有一个UsingDbTest的基类，有一个UsingNetworkTest的基类，难道让子类继承两个基类么（mixin是不是有点过于复杂了？）。
使用generator可以很好的解决这个问题。首先我们写一个方法来做setUp和tearDown：

@contextlib.contextmanager
def using_db():
    setup_db()
    yield
    clean_db()

这样可以非常清晰地知道setup_db和clean_db是一对的。然后再把这个小的上下文附着到主测试逻辑上：

def apply_context(test, contextmanager):
    contextmanager.__enter__()
    test.addCleanup(lambda: contextmanager.__exit__(None, None, None))

class SomeTest(unittest.case.TestCase):
    def setUp(self):
        apply_context(self, using_db())

这里利用了单元测试的addCleanup的特性，把tearDown转化为回调在setUpd的时候就设置好。利用这种方式，我们可以用组合的方式而不是继承的方式来复用公共的setUp和tearDown的逻辑了。

专线服务私有云单元测试单元测试规范单元测试图片测试java php单元测试’

文章版权归作者所有，未经允许请勿转载,若此文章存在违规行为，您可以联系管理员删除。

转载请注明本文地址：https://www.ucloud.cn/yun/37401.html

谈谈项目的重构与测试

这篇文章摘自我的博客, 欢迎大家没事去逛逛~ 背景这几个月我开发了公司里的一个restful webservice，起初技术选型的时候是采用了flask框架。虽然flask是一个同步的框架，但是可以配合gevent或者其它方式运行在异步的容器中(测试链接)，效果看上去也还可以，因此就采用了这种方式。后面阅读了tornado的源码，也去了解了各种协程框架以及运行的原理。总感觉flask的这种同步...

Lavender 2019-07-24 18:17 评论0 收藏0
谈谈项目的重构与测试

这篇文章摘自我的博客, 欢迎大家没事去逛逛~ 背景这几个月我开发了公司里的一个restful webservice，起初技术选型的时候是采用了flask框架。虽然flask是一个同步的框架，但是可以配合gevent或者其它方式运行在异步的容器中(测试链接)，效果看上去也还可以，因此就采用了这种方式。后面阅读了tornado的源码，也去了解了各种协程框架以及运行的原理。总感觉flask的这种同步...

wuaiqiu 2019-06-26 16:58 评论0 收藏0
python学习笔记- 单元测试，UnitTest

摘要：所谓的单元测试，就是对一个模块，一个函数，或则是一个类进行正确性检测的一类测试工作。当然，单元测试也会让代码量大大增加。编写单元测试代码需要引入的包。再所有单元测试开始前运行函数在所有单元测试运行后运行。所谓的单元测试，就是对一个模块，一个函数，或则是一个类进行正确性检测的一类测试工作。以测试驱动的开发方式叫做测试驱动开发(Test Drived Development). 这种开...

k00baa 2019-07-25 11:18 评论0 收藏0