简单聊一聊关于UnitTest的二三事(三)

上一篇 / 下一篇  2021-06-02 10:52:09 / 个人分类:单元测试

六、数据驱动DDT(Data Driven Tests)
在介绍数据驱动之前,我们先来看一下下面这个示例,代码的内容特别特别的简单,用来模拟自动化测试时输入用户名和密码(哈哈,我知道这样模拟有点勉强!!!):
1、打开百度
2、第一次在输入框中输入java,然后清除,再输入123456,关闭浏览器
3、第二次在输入框中输入selenium,然后清除,输入abcdef,关闭浏览器。
import unittest
from selenium import webdriver
import time


class UnitForTestddt(unittest.TestCase):

    def setUp(self) -> None:
        self.driver = webdriver.Chrome()
        self.driver.get('http://www.baidu.com')

    def test_ddt1(self):
        self.driver.find_element_by_id('kw').send_keys('java')
        time.sleep(1)
        self.driver.find_element_by_id('kw').clear()
        time.sleep(1)
        self.driver.find_element_by_id('kw').send_keys('123456')
        time.sleep(1)

    def test_ddt2(self):
        self.driver.find_element_by_id('kw').send_keys('selenium')
        time.sleep(1)
        self.driver.find_element_by_id('kw').clear()
        time.sleep(1)
        self.driver.find_element_by_id('kw').send_keys('abcdef')
        time.sleep(1)

    def tearDown(self) -> None:
        self.driver.quit()


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

由代码可以看出,中间的测试用例部分很多代码是重复的,造成代码冗余,为了解决代码冗余,我们采用数据驱动的方式;同时,为了方便后期代码的维护,进行数据与代码的分离。UnitTest没有自带数据的驱动功能,如果在使用UnitTest的同时又想使用数据驱动,那么就可以使用DDT来完成。

使用方法如下:
(1)ddt.data:装饰测试方法,参数是一系列的值,比如元组等;
(2)ddt.file_data:装饰测试方法,参数是文件名,测试数据保存在参数文件中。文件类型可以是JSON或者YAML;
(3)ddt.unpack:当ddt传递复杂的数据结构时使用,通常称为解包。
下面分别将这几个方法进行示例演示,在使用ddt之前我们需要导包:import ddt

1. ddt.data方法
import unittest
from selenium import webdriver
import time
import ddt


@ddt.ddt
class UnitForTestddt(unittest.TestCase):

    def setUp(self) -> None:
        self.driver = webdriver.Chrome()
        self.driver.get('http://www.baidu.com')

    @ddt.data('java', 'selenium', 'python')
    def test_ddt(self, a):
        self.driver.find_element_by_id('kw').send_keys(a)
        time.sleep(1)

    def tearDown(self) -> None:
        self.driver.quit()


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

以上示例中,传递数据的格式为元组,元组中包含了三个元素,浏览器共打开了三次,而三次中分别输入了java、selenium、python,很显然可以看出,代码中只编写了一次测试用例,而实际结果中却执行了三次,减少了代码的冗余,实现了不同的输入条件执行相同的测试用例。
ddt.data在使用过程中,需要注意一下几点:
(1)导包:import ddt;
(2)在测试类之前加装饰方法:@ddt.ddt;
(3)在测试用例之前加 @ddt.data(),传入数据。

2. ddt.unpack方法
我们再看一个示例,如下:
import unittest
from selenium import webdriver
import time
import ddt

@ddt.ddt
class UnitForTestddt(unittest.TestCase):

    def setUp(self) -> None:
        self.driver = webdriver.Chrome()
        self.driver.get('http://www.baidu.com')

    @ddt.data(['java', '123456'], ['python', '666666'])
    @ddt.unpack
    def test_ddt1(self, username, pwd):
        self.driver.find_element_by_id('kw').send_keys(username)
        time.sleep(1)
        self.driver.find_element_by_id('kw').clear()
        time.sleep(1)
        self.driver.find_element_by_id('kw').send_keys(pwd)
        time.sleep(1)

    def tearDown(self) -> None:
        self.driver.quit()

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

当我们传递的内容为list列表时,需要添加@ddt.unpack进行解包,否则会运行错误,示例中传递了两个list,用例执行了两次,分别将java、python传递给了username,将123456,666666传递给了pwd。

3. 从文件中读取数据
在实际项目中,一般测试数据都比较多,测试数据都会写在文件中,下面我们以.txt文件为例,介绍一下如何从文件中读取数据。
在当前包下新建一个testdata.txt文件,文件的内容如下:
java,123456
python,666666
selenium,777777

现在在测试用例中读取testdata.txt中的数据,代码如下:
import unittest
from selenium import webdriver
import time
import ddt

@ddt.ddt
class UnitForTestddt(unittest.TestCase):
    # 读取文件内容
    def read_file():
        file = open('testdata.txt', 'r', encoding='utf-8')
        li = []
        for line in file.readlines():
            li.append(line.strip('\n').split(','))
            file.close()
        return li

    def setUp(self) -> None:
        self.driver = webdriver.Chrome()
        self.driver.get('http://www.baidu.com')

    # 一个*表示以元祖的形式去解读,两个*表示以字典的形式去解读
    @ddt.data(*read_file())
    @ddt.unpack
    def test_ddt1(self, username, pwd):
        self.driver.find_element_by_id('kw').send_keys(username)
        time.sleep(1)
        self.driver.find_element_by_id('kw').clear()
        time.sleep(1)
        self.driver.find_element_by_id('kw').send_keys(pwd)
        time.sleep(1)

    def tearDown(self) -> None:
        self.driver.quit()

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

以上读取txt文件的方式,是基于Python中open方法、readlines方法等原本的处理方式去读取,并未实现完全的兼容,而ddt中直接兼容了yaml文件的读取。

4. 从YAML文件中读取数据(file_data)
首先yaml的安装:pip install pyyaml
在当前包下新建一个testdata.yaml文件,文件的内容(编写时注意格式)如下:
-
  username: java
  pwd: 123456
-
  username: python
  pwd: 666666
-
  username: selenium
  pwd: 777777

现在在测试用例中读取testdata.yaml中的数据,代码如下:
import unittest
from selenium import webdriver
import time
import ddt

@ddt.ddt
class UnitForTestddt(unittest.TestCase):

    def setUp(self) -> None:
        self.driver = webdriver.Chrome()
        self.driver.get('http://www.baidu.com')

    # 读取文件中的数据
    @ddt.file_data('testdata.yaml')
    def test_ddt(self,**user):
        self.driver.find_element_by_id('kw').send_keys(user.get('username'))
        time.sleep(1)
        self.driver.find_element_by_id('kw').clear()
        time.sleep(1)
        self.driver.find_element_by_id('kw').send_keys(user.get('pwd'))
        time.sleep(1)

    def tearDown(self) -> None:
        self.driver.quit()

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

七、UnitTest生成测试报告(HTMLTestRunner)
批量执行完用例后,生成的测试报告是文本形式的,不够直观,为了更好的展示测试报告,最好是生成HTML格式的。unittest里面是不能生成html格式报告的,需要导入一个第三方的模块:HTMLTestRunner

环境搭建
(1)下载HTMLTestRunner.py,导入到Python中的lib文件夹下;
(2)修改部分源码
由于HTMLTestRunner.py是基于python2开发,为了使其支持python3的环境,需要对其内容进行部分修改;
#第94行
import StringIO 修改为:import io

#第539行
self.outputBuffer = StringIO.StringIO() 修改为:self.outputBuffer = io.StringIO()

#第631行
print >>sys.stderr, '\nTime Elapsed: %s' % (self.stopTime-self.startTime)修改为:print(sys.stderr, '\nTime Elapsed: %s' % (self.stopTime-self.startTime))

#第642行
if not rmap.has_key(cls):修改为:if not cls in rmap:

#第766行
uo = o.decode('latin-1')修改为:uo = o

#第772行
ue = e.decode('latin-1')修改为:ue = e

(3)导包:from HTMLTestRunner import HTMLTestRunner。
示例代码如下:
# 导入unittest的包
import unittest
# 导入测试类
from unittest_demo.unit_for_testA import UnitForTestA
# 导入生成报告所需要的包
from HTMLTestRunner import HTMLTestRunner
import os
# 创建一个测试套件
suite = unittest.TestSuite()

# 添加测试用例的第二种方法
cases = [UnitForTestA('test_1'), UnitForTestA('test_2'), UnitForTestA('test_3')]

# 集成测试报告
report_name = '测试报告名称'
report_title = '测试报告标题'
report_desc = '测试报告描述'
report_path = './report/'
report_file = report_path + 'report.html'
if not os.path.exists(report_path):
    os.mkdir(report_path)
else:
    pass
with open(report_file, 'wb') as report:
    suite.addTests(cases)
    runner = HTMLTestRunner(stream=report, title=report_title, description=report_desc)
    runner.run(suite)

执行完成后,在当前的目录下会生成一个report/report.html文件,通过浏览器打开,界面如下:

八、后记
以上是作为一个测试小白关于python中UnitTest的理解,烦请各位大佬们不吝赐教,在软件测试方面我还有很多很多知识要学习,希望未来一起加油!!!

TAG: unittest Unittest

 

评分:0

我来说两句

Open Toolbar