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

一、组织单元测试用例

看看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)

　　

运行结果如下：

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

但是这样还是没有解决添加用例的问题，此时使用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，大大简化了测试用例的查找与执行。