Selenium 2自动化测试实战29（组织单元测试用例和discover更多测试用例）

时间：2021-05-11 11:50:20

一、组织单元测试用例

看看unittest单元测试框架是如何扩展和组织新增的测试用例
以之前的calculator.py文件为例，为其扩展sub（）方法，用来计算两个数相减的结果。

#coding:utf-8

#计算器类

class Count():

    def __init__(self,a,b):

        self.a=int(a)

        self.b=int(b)

    #计算加法

    def sum(self):

        return self.a+self.b

    #计算减法

    def sub(self):

        return self.a-self.b

因为对计算器类又新增了减法功能（sub），所以需要针对新功能编写测试用例，扩展后的test.py如下

#coding:utf-8

from Demo1 import Count

import unittest

class TestSum(unittest.TestCase):

    def setUp(self):

        a = input("please input the date:")

        self.a = int(a)

        b = input("please input the other date:")

        self.b = int(b)

    def testSum(self):

        self.assertEqual((Count(self.a,self.b).sum()),5,msg="the date is not Equal")

    def testSum2(self):

        self.assertEqual((Count(self.a,self.b).sum()),15,msg="the date is not Equal")

    def tearDown(self):

        print"test sum end"

class TestSub(unittest.TestCase):

    def setUp(self):

        a = input("please input the date:")

        self.a=int(a)

        b = input("please input the other date:")

        self.b = int(b)

    def testSub(self):

        self.assertEqual((Count(self.a,self.b).sub()), 5, msg="the date is not Equal")

    def testSub2(self):

        self.assertEqual((Count(self.a,self.b).sub()), 15, msg="the date is not Equal")

    def tearDown(self):

        print"test sub end"

if __name__=="__main__":

    #构造测试集

    suite=unittest.TestSuite()

    suite.addTest(TestSum("testSum"))

    suite.addTest(TestSum("testSum2"))

    suite.addTest(TestSub("testSub"))

    suite.addTest(TestSub("testSub2"))

    #运行测试集

    runner=unittest.TextTestRunner()

    runner.run(suite)

上例中创建了TestAdd（）和TestSub（）两个测试类，分别测试calculator.py文件中的add（）和sub（）两个功能。通过TestSuite类的addTest（）方法把不同测试类中的测试方法组装到测试套件中。

运行结果如下所示：

D:\Python27\python.exe F:/study/Demo2/testDemo1/test.py

please input the date:2

please input the other date:3

test sum end

.please input the date:4

please input the other date:5

Ftest sum end

please input the date:6

please input the other date:1

.test sub end

please input the date:20

please input the other date:10

test sub end

F

======================================================================

FAIL: testSum2 (__main__.TestSum)

----------------------------------------------------------------------

Traceback (most recent call last):

File "F:/study/Demo2/testDemo1/test.py", line 18, in testSum2

self.assertEqual((Count(self.a,self.b).sum()),15,msg="the date is not Equal")

AssertionError: the date is not Equal

======================================================================

FAIL: testSub2 (__main__.TestSub)

----------------------------------------------------------------------

Traceback (most recent call last):

File "F:/study/Demo2/testDemo1/test.py", line 35, in testSub2

self.assertEqual((Count(self.a,self.b).sub()), 15, msg="the date is not Equal")

AssertionError: the date is not Equal

----------------------------------------------------------------------

Ran 4 tests in 21.417s

FAILED (failures=2)

Process finished with exit code 0

通过测试结果看到，setUp（）和tearDown（）方法分别作用于每个测试用例的开始与结束。尝试着封装一个自己的测试类。

#test.py

#coding:utf-8

import unittest

from Demo1 import Count

class Mytest(unittest.TestCase):

    def setUp(self):

        a = input("please input the date:")

        self.a = int(a)

        b = input("please input the other date:")

        self.b = int(b)

    def tearDown(self):

        print"test sum end"

class TestSum(Mytest):

    def testSum(self):

        self.assertEqual((Count(self.a,self.b).sum()),5,msg="the date is not Equal")

    def testSum2(self):

        self.assertEqual((Count(self.a,self.b).sum()),15,msg="the date is not Equal")

class TestSub(Mytest):

    def testSub(self):

        self.assertEqual((Count(self.a,self.b).sub()), 5, msg="the date is not Equal")

    def testSub2(self):

        self.assertEqual((Count(self.a,self.b).sub()), 15, msg="the date is not Equal")

if __name__=="__main__":

    # 构造测试集

    suite = unittest.TestSuite()

    suite.addTest(TestSum("testSum"))

    suite.addTest(TestSum("testSum2"))

    suite.addTest(TestSub("testSub"))

    suite.addTest(TestSub("testSub2"))

    # 运行测试集

    runner = unittest.TextTestRunner()

    runner.run(suite)

创建MyTest（）类的好处显而易见，对于测试类和测试方法来说，应将注意力放在具体用例的编写上，不需要关心setUp（）和tearDown（）所做的事情，不过，前提条件是setUp（）和tearDown（）所做的事情是每个用例都需要的。

 二、discover更多测试用例

对上例中test.py文件的测试用例进行拆分，拆分后的目录结果如下：

Testpro/
--count.py
--testsum.py
--testsub.py
--runtest.py
文件拆分后的代码如下：

#count.py



#coding:utf-8

#计算器类

class Count():

    def __init__(self,a,b):

        self.a=int(a)

        self.b=int(b)

    #计算加法

    def sum(self):

        return self.a+self.b

    #计算减法

    def sub(self):

        return self.a-self.b

#testsum.py



#coding:utf-8

from Demo1 import Count

import unittest

class TestSum(unittest.TestCase):

    def setUp(self):

        a = input("please input the date:")

        self.a = int(a)

        b = input("please input the other date:")

        self.b = int(b)

    def tearDown(self):

        print"test sum end"

    def testSum(self):

        self.assertEqual((Count(self.a,self.b).sum()),5,msg="the date is not Equal")

    def testSum2(self):

        self.assertEqual((Count(self.a, self.b).sum()),15, msg="the date is not Equal")

if __name__=="__main__":

    unittest.main()

　

#testsub.py



#coding:utf-8

from Demo1 import Count

import unittest

class TestSub(unittest.TestCase):

    def setUp(self):

        a = input("please input the date:")

        self.a = int(a)

        b = input("please input the other date:")

        self.b = int(b)

    def tearDown(self):

        print"test sum end"

    def testSub(self):

        self.assertEqual((Count(self.a,self.b).sub()),5,msg="the date is not Equal")

    def testSub2(self):

        self.assertEqual((Count(self.a, self.b).sub()),15, msg="the date is not Equal")

if __name__=="__main__":

    unittest.main()

　　

#runtest.py



#coding:utf-8

import unittest

#加载测试文件

import testsum

import testsub

#构造测试集

suite=unittest.TestSuite()

suite.addTest(testsum.TestSum("testSum"))

suite.addTest(testsum.TestSum("testSum2"))

suite.addTest(testsub.TestSub("testSub"))

suite.addTest(testsub.TestSub("testSub2"))

#执行测试

if __name__=="__main__":

    runner=unittest.TextTestRunner()

    runner.run(suite)

　　

运行结果如下：

Selenium 2自动化测试实战29（组织单元测试用例和discover更多测试用例）

这样的拆分带来了好处，可以根据不同的功能创建不同的测试用例，甚至是不同的测试目录，测试文件中还可以将不同的小功能划分为不同的测试类，在类下编写测试用例，整体结构更清晰。

但是这样还是没有解决添加用例的问题，此时使用TestLoader类中提供的discover()方法可以解决这个问题。

TestLoader

该类负责根据各种标准加载测试用例，并将它们返回给测试套件，正常情况下，不需要创建这个类的实例。unittest提供了可以共享的defaultTestLoader类，可以使用其子类和方法创建实例，discover()方法就是其中之一。

discover（start_dir，pattern=‘test*.py’,top_level_dir=None）

找到指定目录下所有测试模块，并可递归查到子目录下的测试模块，只有匹配到文件名才能加载。如果启动的不是顶层目录，那么顶层目录必须单独指定。

start_dir：要测试的模块名或测试用例目录。

pattern=‘test*.py’：表示用例文件名的匹配原则。此处匹配文件名以“test”开头的“.py"类型的文件，星号“*”表示任意多个字符。

top_level_dir=None：测试模块的顶层目录，如果没有顶层目录，默认为None。

现在通过discover()方法重新实现runtest.py文件的功能。　　
#runtest.py
#coding:utf-8

import unittest

#定义测试用例的目录为当前目录

test_dir='./'

discover=unittest.defaultTestLoader.discover(test_dir,pattern='test*.py')

if __name__=="__main__":

    runner=unittest.TextTestRunner()

    runner.run(discover)

Discover()方法会自动根据测试目录（test_dir）匹配查找测试用例文件（test*.py），并将查找到的测试用例组装到测试套件中，因此，可以直接通过run（）方法执行discover，大大简化了测试用例的查找与执行。

相关文章