python写UT基础篇

最新推荐文章于 2024-04-20 00:02:09 发布
最新推荐文章于 2024-04-20 00:02:09 发布
阅读量4.3k 收藏 8
点赞数 2
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/wangliyao518/article/details/82387675
版权

定义:

单元测试，是指对软件中的最小可测试单元进行检查和验证。对于单元测试中单元的含义，一般来说，要根据实际情况去判定其具体含义，如C语言中单元指一个函数，Java里单元指一个类，图形化的软件中可以指一个窗口或一个菜单等。总的来说，单元就是人为规定的最小的被测功能模块, python中单元我认为也可以是一个函数

 

一般认为，在结构化程序时代，单元测试所说的单元是指函数，在当今的面向对象时代，单元测试所说的单元是指类。以我的实践来看，以类作为测试单位，复杂度高，可操作性较差，因此仍然主张以函数作为单元测试的测试单位，但可以用一个测试类来组织某个类的所有测试函数。单元测试不应过分强调面向对象，因为局部代码依然是结构化的。单元测试的工作量较大，简单实用高效才是硬道理。

简单例子:

既然单元测试是针对函数,针对源码,那首先需要有源码, 那先让我们那些段ssh登录 host做操作的代码

import re


class FsmAccess(object):

    def __init__(self, connection):
        self.con = connection

    def get_lmp_ip_address(self):
        result = ""
        stderr, stdout = self.con.exec_command("ifconfig eth3")
        pattern = r"(\d{1,4}.\d{1,4}.\d{1,4}.\d{1,4})"
        for line in stdout.split('\n'):
            ret = re.search(pattern, line)
            if ret:
                result = ret.groups()[0]
        return result

针对上面get_lmp_ip_address函数写单元测试, 我们看到它依赖一个外部组件 connection, 而这个connection在单元测试时就不应该真实的执行, 所以我们应该把它替换掉, 我们只是依赖connection这个对象中的exec_command函数输出的结果,然后有我们自己的运算输出一个想要的结果, 我们的单元测试这个get_lmp_ip_address的目的也只是为了验证我们对这个’ifconfig’ 命令输出值的正确解析

有了上面的理解,我们知道了需要替换掉exec_command, 不依赖第三方模块情况下,我们可以自己造一个假的这个对象,代码如下

import unittest
from fsm import FsmAccess

class MockConnection(object):

    def exec_command(self, command):
        return  0, 'eth3      Link encap:Ethernet  HWaddr 00:0F:BB:BA:99:CD  \ninet addr:10.0.2.2  Bcast:10.0.2.255  Mask:255.255.255.0\nUP BROADCAST RUNNING MULTICAST  MTU:1500  Metric:1\nRX packets:0 errors:0 dropped:0 overruns:0 frame:0\nTX packets:16 errors:0 dropped:0 overruns:0 carrier:0\ncollisions:0 txqueuelen:1000\nRX bytes:0 (0.0 B)  TX bytes:1152 (1.1 KiB)'

class TestKissConfigInterface(unittest.TestCase):

    def test_get_lmp_ip_add(self):
        mock_instant = MockConnection()
        fsm_instance = FsmAccess(connection=mock_instant)
        ret = fsm_instance.get_lmp_ip_address()
        self.assertEqual(ret, '10.0.2.2')

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

mock:

上面的UT 完全可以帮我们测试我们的函数功能, 但有一个问题是如果我们要测试很多的函数而且很多函数也都和get_lmp_ip_address一样有外部依赖我们就需要写很多这样假的类和方法, 这样效率明显很低下, 于是就有一些工具或者模块的出现, 这就是mock和patch 

import unittest
from fsm import FsmAccess
from mock import Mock


class TestKissConfigInterface(unittest.TestCase):

    def test_get_lmp_ip_add(self):
        con_mock = Mock()
        con_mock.exec_command.return_value = 0, 'eth3      Link encap:Ethernet  HWaddr 00:0F:BB:BA:99:CD  \ninet addr:10.0.2.2  Bcast:10.0.2.255  Mask:255.255.255.0\nUP BROADCAST RUNNING MULTICAST  MTU:1500  Metric:1\nRX packets:0 errors:0 dropped:0 overruns:0 frame:0\nTX packets:16 errors:0 dropped:0 overruns:0 carrier:0\ncollisions:0 txqueuelen:1000\nRX bytes:0 (0.0 B)  TX bytes:1152 (1.1 KiB)'
        fsm_instance = FsmAccess(connection=con_mock)
        ret = fsm_instance.get_lmp_ip_address()
        self.assertEqual(ret, '10.0.2.2')

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

上面新的测试代码比起上面不需要自己定义一个新类, 只需要用一个mock实例化一个对象,然后这个mock对象中就可直接写上我们的期望的函数名字和返回值

mock的方法还可以这样用:

import unittest
from fsm import FsmAccess
from mock import Mock


class TestKissConfigInterface(unittest.TestCase):

    def test_get_lmp_ip_add(self):
        fake_return_value = 0, 'eth3      Link encap:Ethernet  HWaddr 00:0F:BB:BA:99:CD  \ninet addr:10.0.2.2  Bcast:10.0.2.255  Mask:255.255.255.0\nUP BROADCAST RUNNING MULTICAST  MTU:1500  Metric:1\nRX packets:0 errors:0 dropped:0 overruns:0 frame:0\nTX packets:16 errors:0 dropped:0 overruns:0 carrier:0\ncollisions:0 txqueuelen:1000\nRX bytes:0 (0.0 B)  TX bytes:1152 (1.1 KiB)'
        fsm_instance = FsmAccess(connection=Mock())
        fsm_instance.con.exec_command = Mock(return_value=fake_return_value)
        ret = fsm_instance.get_lmp_ip_address()
        self.assertEqual(ret, '10.0.2.2')

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

patch:

 我们也可以用patch 方式来写UT代码, 如下:

import re


class FsmAccess(object):

    def __init__(self, connection):
        self._con = connection

    def con(self):
	    return self._con

    def get_lmp_ip_address(self):
        result = ""
        stderr, stdout = self.con.exec_command("ifconfig eth3")
        pattern = r"(\d{1,4}.\d{1,4}.\d{1,4}.\d{1,4})"
        for line in stdout.split('\n'):
            ret = re.search(pattern, line)
            if ret:
                result = ret.groups()[0]
        return result


import unittest
from fsm import FsmAccess
from mock import Mock, patch


class TestKissConfigInterface(unittest.TestCase):


    @patch('fsm.FsmAccess.con')
    def test_get_lmp_ip_add_patch(self, con_mock):
        con_mock.exec_command.return_value = (0, 'eth3      Link encap:Ethernet  HWaddr 00:0F:BB:BA:99:CD  \ninet addr:10.0.2.2  Bcast:10.0.2.255  Mask:255.255.255.0\nUP BROADCAST RUNNING MULTICAST  MTU:1500  Metric:1\nRX packets:0 errors:0 dropped:0 overruns:0 frame:0\nTX packets:16 errors:0 dropped:0 overruns:0 carrier:0\ncollisions:0 txqueuelen:1000\nRX bytes:0 (0.0 B)  TX bytes:1152 (1.1 KiB)')
        fsm_instance = FsmAccess(con_mock)
        ret = fsm_instance.get_lmp_ip_address()
        self.assertEqual(ret, '10.0.2.2')

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

如果我们对上述代码中的_con属性直接用patch 会是什么样的结果? 

import re


class FsmAccess(object):

    def __init__(self, connection):
        self._con = connection

    # def con(self):
	   #  return self._con

    def get_lmp_ip_address(self):
        result = ""
        stderr, stdout = self._con.exec_command("ifconfig eth3")
        pattern = r"(\d{1,4}.\d{1,4}.\d{1,4}.\d{1,4})"
        for line in stdout.split('\n'):
            ret = re.search(pattern, line)
            if ret:
                result = ret.groups()[0]
        return result


import unittest
from mock import Mock, patch


class TestKissConfigInterface(unittest.TestCase):


    @patch('__main__.FsmAccess._con')
    def test_get_lmp_ip_add_patch(self, con_mock):
        con_mock.exec_command.return_value = (0, 'eth3      Link encap:Ethernet  HWaddr 00:0F:BB:BA:99:CD  \ninet addr:10.0.2.2  Bcast:10.0.2.255  Mask:255.255.255.0\nUP BROADCAST RUNNING MULTICAST  MTU:1500  Metric:1\nRX packets:0 errors:0 dropped:0 overruns:0 frame:0\nTX packets:16 errors:0 dropped:0 overruns:0 carrier:0\ncollisions:0 txqueuelen:1000\nRX bytes:0 (0.0 B)  TX bytes:1152 (1.1 KiB)')
        fsm_instance = FsmAccess(con_mock)
        ret = fsm_instance.get_lmp_ip_address()
        self.assertEqual(ret, '10.0.2.2')

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

上面代码会报下面的错

  File "C:\Python27\lib\site-packages\mock\mock.py", line 1369, in __enter__
    original, local = self.get_original()
  File "C:\Python27\lib\site-packages\mock\mock.py", line 1343, in get_original
    "%s does not have the attribute %r" % (target, name)
AttributeError: <class '__main__.FsmAccess'> does not have the attribute '_con'

----------------------------------------------------------------------
Ran 1 test in 0.001s

其实只需要在测试函数中的patch里面加一个参数就可以pass

import unittest
from mock import Mock, patch


class TestKissConfigInterface(unittest.TestCase):


    @patch('__main__.FsmAccess._con', create=True)
    def test_get_lmp_ip_add_patch(self, con_mock):
        con_mock.exec_command.return_value = (0, 'eth3      Link encap:Ethernet  HWaddr 00:0F:BB:BA:99:CD  \ninet addr:10.0.2.2  Bcast:10.0.2.255  Mask:255.255.255.0\nUP BROADCAST RUNNING MULTICAST  MTU:1500  Metric:1\nRX packets:0 errors:0 dropped:0 overruns:0 frame:0\nTX packets:16 errors:0 dropped:0 overruns:0 carrier:0\ncollisions:0 txqueuelen:1000\nRX bytes:0 (0.0 B)  TX bytes:1152 (1.1 KiB)')
        fsm_instance = FsmAccess(con_mock)
        ret = fsm_instance.get_lmp_ip_address()
        self.assertEqual(ret, '10.0.2.2')

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

主要的使用场景是, 有的人喜欢把代码写成如下的样子:

import paramiko
class FsmAccess(object):

    def __init__(self):
        pass

    def connect(self, host, port, username, password):
        self.con = paramiko.SSHClient()
        self.con.set_missing_host_key_policy(paramiko.AutoAddPolicy())
        self.con.connect(host, port, username, password)

    def get_lmp_ip_address(self):
        result = ""
        stderr, stdout = self.con.exec_command("ifconfig eth3")
        pattern = r"(\d{1,4}.\d{1,4}.\d{1,4}.\d{1,4})"
        for line in stdout.split('\n'):
            ret = re.search(pattern, line)
            if ret:
                result = ret.groups()[0]
        return result

    def disconnect(self):
        self.con.close()

我不想来评断上面代码是否写的够好, 我只是来谈谈这么使用patch, 以及如何来测试这个connect函数(get_lmp_ip_address已经讲过, 下面就比再写了)

 

import unittest
from mock import Mock, patch


class TestFsm(unittest.TestCase):


    @patch('paramiko.SSHClient.set_missing_host_key_policy')
    @patch('paramiko.SSHClient.connect')
    def test_connect(self, con_mock, set_missing_host_key_policy_mock):
        fsm_instance = FsmAccess()
        fsm_instance.connect('127.0.0.1', 22, 'ute', 'ute')
        fsm_instance.con.set_missing_host_key_policy.assert_called()
        fsm_instance.con.connect.assert_called_with('127.0.0.1', 22, 'ute', 'ute')


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

上面的connect函数由于没有任何的逻辑封装,所以UT代码其实意义不是很大, 我们确实一方面是为了覆盖率, 其实另外一方面也是为了保证这个函数关键的调用部分不被别人轻易移调, patch函数也很简单, patch里面的参数就是声明我们需要替代掉的函数’路径’, 它会被对应def中测试函数中参数代表的对象所替换, 我们这个例子中参数没做任何的赋值或者声明, 所以它就是none

但上面的例子, 有一个问题, 如果我们手误把assert_called_with写错, 你也会发现这个用例依旧是pass, 为了解决这个问题, 我们需要把上面的代码写成下面的样子

import unittest
from mock import Mock, patch


class TestFsm(unittest.TestCase):


    @patch('paramiko.SSHClient.set_missing_host_key_policy')
    @patch('paramiko.SSHClient', autospec=True)
    def test_connect(self, con_mock, set_missing_host_key_policy_mock):
        fsm_instance = FsmAccess()
        fsm_instance.connect('127.0.0.1', 22, 'ute', 'ute')
        fsm_instance.con.set_missing_host_key_policy.assert_called()
        con_mock.return_value.connect.assert_called_with('127.0.0.1', 22, 'ute', 'ute')

 

 

wangliyao518
关注
Python的标准UT组合
seven的专栏
09-24 3846
最近研究如何给Python代码做UT，从开发角度保证代码质量。 Python提供了极为方便的库用来做UT，以下三个是比较经典的组合： Unittest Python-mock Coverage Unittest是UT的framework，提供了标准类及方法供testcase调用 Python-mock用于mock控制，可用在UT中排除外部API的干扰 Coverage
pythoncqt_python基础篇
weixin_39757040的博客
12-11 1万+
python脚本开头#!/usr/bin/env python# -*- coding: utf-8 -*print "你好，世界"不要问为什么，记住就好了变量定于的规则变量名只能是字母、数字或下划线的任意组合变量名的第一个字符不能是数字以下关键字不能声明为变量名['and', 'as', 'assert', 'break', 'class', 'continue', 'def', 'del', ...
Python中使用unittest做UT
杰瑞的专栏
02-15 2883
待测试的类(Widget.py) # Widget.py # Python 2.7.6 class Widget: def __init__(self, size = (40,40)): self.size = size def getSize(self): return self.size def reSize(self,wi
pyhton的UT测试介绍
04-13
Python UT测试，Python单元测试，Python测试的几个实例
Practice In Python UT & Coverage
ace_fei的专栏
01-13 1211
Practice In Python UT & CoveragePython unittestBelow is my python unittest template:import unittest import os import string from random import sample from mock.mock import patch#mock toggle mock_flag
Python中unittest模块做UT（单元测试）使用实例
12-23
待测试的类(Widget.py) # Widget.py # Python 2.7.6 class Widget: def __init__(self, size = (40,40)): self.size = size def getSize(self): return self.size def reSize(self,width,height): if width <0 or height < 0: raise ValueError, 'illegal size' else: self.size = (width,h
Python基础篇：print、import、input与赋值语句详解
本文档是针对初学者的一篇Python基础教程，由专注于网络安全领域的博主编写，旨在帮助读者掌握Python中的基本语法和常用功能，特别是print、import、input和赋值语句。以下是文章的主要知识点： 1. print语句： - ...
100天精通Python（基础篇）——第32天：日志标准库logging【文末送书两本】(1)
最新发布
2401_83704159的博客
04-20 706
一、日志基础知识==========================================================================1、日志是什么2、日志的作用通过记录和分析日志可以了解一个系统 或软件程序运行情况是否正常，也可以在应用程序出现故障时快速定位问题。例如：开发者可通过在控制台上输出的各种日志进行程序调试；运维人员在接收到报警或各种问题反馈后，进行问题排查时通常都会先看各种日志，大部分问题都可在日志中找到答案。日志的作用可总结为3点：1、程序调试2、了解软件程序运
python取模1e9+7_Python的基础语法（一）
weixin_39684454的博客
12-15 3065
0. 前言最近正在重新整理python的基础知识，以便更好地学习新知识。这一部分主要是讲述python基础语法的演示。下面的语法都是基于python3的语法。1. 注释注释：python的注释方法有两种：单行注释： # 你添加的注释多行注释： 使用三引号‘’‘你添加的注释你添加的注释你添加的注释’‘’中文注释：为了让自己写的代码注释中的中文字符不变成乱码，一般我们会在第一行加上#coding=ut...
python接口测试
热门推荐
JOJOY的博客
11-21 2万+
python+request+unittest+HTMLTestRunner 首先介绍一下python的requests模块： requests的使用介绍：requests快速入门 环境说明： 1.WIN 7,   64位操作系统 2.Python3.4.3 （pip-8.1.2） 3.Requests  --->python setup.py install 4.Unittes
python UT相关
weixin_33874713的博客
08-28 165
2019独角兽企业重金招聘Python工程师标准>>> ...
python UT验证/调试工具 icecream
qq_44963876的博客
08-02 351
python UT验证/调试工具 icecream1.icecream 安装2.使用方法 1.icecream 安装 pip install icecream 2.使用方法 from icecream import ic ic.disable() ic.enable() # 输入使能 # ic.configureOutput(includeContext=True) #详细信息 def sumSquare(a, b): return a * a + b * b+a/b if __name
合格的后端Coder都应该写好UT和Mock测试
Newerwu的专栏
02-03 985
关于测试 测试都包括哪些 广义的测试包括UT、IT、压力测试、硬件测试等等，这里重点讨论Unit Test即单元测试。 啥是UT 单元测试（又称为模块测试, Unit Testing）是针对程序模块（软件设计的最小单位）来进行正确性检验的测试工作。程序单元是应用的最小可测试部件。在过程化编程中，一个单元就是单个程序、函数、过程等；对于面向对象编程，最小单元就是方法，包括基类（超类）、抽象类、或者派生类（子类）中的方法。 简而言之就是覆盖你的代码的一些测试用例，不依赖于任何第三方的服务依赖，如H
Python mock UT
walkman19900204的专栏
05-31 534
*mock 的作用非常强大，需要重点研究这个类* **为什么用 mock.patch.object****with as 的用法，先执行with后面的expression ，然后传递值给as 后面的expression ，再执行冒号后的expression**
python大数加法、需要ut,python处理大数字的方法
weixin_42102358的博客
03-26 145
这篇文章主要介绍了python处理大数字的方法,涉及Python递归操作的相关技巧,需要的朋友可以参考下本文实例讲述了python处理大数字的方法。分享给大家供大家参考。具体实现方法如下：def getFactorial(n):"""returns the factorial of n"""if n == 0:return 1else:k = n * getFactorial(n-1)return...
UT源码 149
azhe0815的博客
03-10 211
（2）NextDate函数问题 NextDate函数说明一种复杂的关系，即输入变量之间逻辑关系的复杂性 NextDate函数包含三个变量month、day和year，函数的输出为输入日期后一天的日期。 要求输入变量month、day和year均为整数值，并且满足下列条件： 条件1 1≤ month ≤12 否则输出，月份超出范围 条件2 1≤ d...
UT编写记录
danfu_wu的博客
05-19 3299
UT编写规则https://zhuanlan.zhihu.com/p/424117483 1、遵循AIR原则 A（automation，自动化）：单元测试应该是全自动执行的，并且非交互。 I （Independence，独立性）：单元测试用例之间决不能互相调用，也不能依赖执行的先后次序。 --------反例：method2需要依赖method1的执行，将执行结果作为method2的输入。 R（Repeatable，可重复）：单元测试是可以重复执行的，不能受到外界环境的影响。 2、结构 a、.
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值