使用 py.test 对 python 代码进行测试

其实之前我记得有人对我说过，写代码甚至能够先写完测试以后部署好了再开始写逻辑代码。我以为有点吃惊和奇怪，这不是扯淡吗？

可是此次在完成了积分支付第一阶段开发以后我意识到，这可能并非开玩笑，特别是项目可能还涉及到之前的代码进行迁移，如何保证代码迁移以后还能继续正常运行？

越是大的项目，在这个上面的维护成本就越高，若是是一人开发，既要负责测试又要负责开发，没有单元测试对这种任务来讲简直就是灾难。

以前在编写openapi模块的时候也为那边的代码写过测试，可是对py.test也没有使用什么其它的特性，此次用到感受比较通用的几个函数和用法因此纪录一下。

 

上个例子：

import pytest
@pytest.mark.parametrize("package_price, package_ident, begin_time, end_time", [
    (-800, 1, datetime.datetime.now(), datetime.datetime.now()+datetime.timedelta(days=30)),
    (800, 1, datetime.datetime.now(), datetime.datetime.now()+datetime.timedelta(days=30)),
], ids=["-钱购买爱尝鲜", "+钱购买爱尝鲜"])

来分析一下这个pytest.mark.parametrize("", [])

第一个字符串参数里面能够传你想要初始化的字段name以逗号分隔，注意它们都在一个字符串里。

后面的数组以tuple为一个元素，每一个tuple里面能够初始化在前面string里面写的各参数，一样以逗号分隔。

ids是能够指定单组参数也就是一个tuple为一个单位的这组测试用例的id，这个会在测试的时候显示出来就像这样。

这样咱们就能经过本身手动构造参数愉快的开始测试了。

可是其实除了这种简单的参数，真正在测试的时候是比较难构造的是指定的obejct对象。

 

这里能够引入一个叫mock的库

下面看个例子

from mock import MagicMock
order = MagicMock(id=order_id, shop_id=shop_id, user_id=user1.id, finish_time=order_finish_time)
gio = MagicMock(id=gio_id, order_id=order_id, goods_id=goods_id, goods_type=goods_type,
                    user_id=user1.id, shop_id=shop_id,
                    is_allowed_to_review=lambda: True, is_reviewed=lambda: False, order=order,
                    goods_paid_cash_amount=goods_paid_cash_amount)
gio.__class__ = GoodsInOrder

你用一个叫mock的库来轻易构造那些看似复杂的须要传递大量参数的对象。上面那些参事能够在parametrize里面构造好而后传入MagicMock里要初始化的对象里面去，就能按照你的想法初始化出这个对象。给测试带来极大便利。

mock库在python3.4中已加入标准库。不过在python2.7里面还须要经过pip install mock安装。

 

下面贴上一个完整的例子.

@pytest.mark.parametrize("package_price, package_ident, begin_time, end_time", [
    (-800, 1, datetime.datetime.now(), datetime.datetime.now()+datetime.timedelta(days=30)),
    (800, 1, datetime.datetime.now(), datetime.datetime.now()+datetime.timedelta(days=30)),
], ids=["-钱买不起爱尝鲜", "+钱也买不起爱尝鲜"])
def test_can_not_buy_fresh_member(package_price, package_ident, begin_time, end_time, user4):
    with pytest.raises(NoEnoughAvailablePoints):
        decrease_points_by_fresh_member(user4.id, package_price, package_ident, begin_time, end_time)

 

 Reference:

https://wangshunping.gitbooks.io/testing_python/content/chapter3/Mock_and_Patch.html  Mock和Patch小技巧