selenium

本文按照循序渐进的方法介绍测试用例的组织方式，实际工作中，可以借鉴方法三

待测文件：

calculate.py文件
 1 # 计算器类
class Count:
    def __init__(self, a, b):
        self.a = a
        self.b = b

    # 计算加法
    def add(self):
        return self.a + self.b

    # 计算减法
    def sub(self):
        return self.a - self.b

测试文件：

方法一：直接创建测试类，进行对应功能的测试

from calculate import Count
import unittest

class TestAdd(unittest.TestCase):   # 创建TestAdd类，测试add()功能
    def setUp(self):
        print('test add start------------')

    def tearDown(self):
        print('test add end--------------')

    def test_add_1(self):
        print('add 1')
        self.assertEqual(Count(2, 4).add(), 6)

    def test_add_2(self):
        print('add 2')
        self.assertEqual(Count(200, 400).add(), 600)

class TestSub(unittest.TestCase):    # 创建TestSub类，测试sub()功能
    def setUp(self):
        print('test sub start------------')

    def tearDown(self):
        print('test sub end--------------')

    def test_sub_1(self):
        print('sub 1')
        self.assertEqual(Count(2, 4).sub(), -2)

    def test_sub_2(self):
        print('sub 1')
        self.assertEqual(Count(200, 400).sub(), -200)


if __name__ == '__main__':
    # 构造测试集
    suite = unittest.TestSuite()
    suite.addTest(TestAdd('test_add_1'))
    suite.addTest(TestAdd('test_add_2'))
    suite.addTest(TestSub('test_sub_1'))
    suite.addTest(TestSub('test_sub_2'))

    # 运行测试集合
    runner = unittest.TextTestRunner()
    runner.run(suite)

setUp()和tearDown()方法作用于每个测试用例的开始与结束，执行结果如下：

test add start------------
add 1
test add end--------------
test add start------------
add 2
test add end--------------
test sub start------------
sub 1
test sub end--------------
test sub start------------
sub 2
test sub end--------------


Ran 4 tests in 0.008s

OK

方法二：将共同的setUp()和tearDown()提取出来

如果每个类中的setUp()和tearDown()所做的事情都是一样的，就可以封装一个自己的测试类，如下：

将注意力放在具体测试用例的编写上，无须关注setUp()和tearDown()所做的事情

from calculate import Count
import unittest

class MyTest(unittest.TestCase):  # 封装自己的测试类，继承TestCase类
    def setUp(self):
        print('test start------------')

    def tearDown(self):
        print('test end--------------')

class TestAdd(MyTest):   # 继承MyTest类
    def test_add_1(self):
        print('add 1')
        self.assertEqual(Count(2, 4).add(), 6)

    def test_add_2(self):
        print('add 2')
        self.assertEqual(Count(200, 400).add(), 600)

class TestSub(MyTest):   # 继承MyTest类
    def test_sub_1(self):
        print('sub 1')
        self.assertEqual(Count(2, 4).sub(), -2)

    def test_sub_2(self):
        print('sub 2')
        self.assertEqual(Count(200, 400).sub(), -200)


if __name__ == '__main__':
    # 构造测试集
    suite = unittest.TestSuite()
    suite.addTest(TestAdd('test_add_1'))
    suite.addTest(TestAdd('test_add_2'))
    suite.addTest(TestSub('test_sub_1'))
    suite.addTest(TestSub('test_sub_2'))

    # 运行测试集合
    runner = unittest.TextTestRunner()
    runner.run(suite)

方法三：将不同功能的测试用例，分散到不同的测试文件中

随着软件功能的不断增加，对应的测试用例，也会呈指数级增长。如果测试用例达到上百个，把所有测试用例都写在一个测试文件中，那么这个文件会越来越臃肿，后期维护也比较麻烦。需要将这些用例，按照所测试的功能进行拆分，分散到不同的测试文件中。

不仅可以分散到不同的文件，甚至可以分散到不同的测试目录中，测试文件中还可以将不同的小功能划分为不同的测试类，在类下编写测试用例，整体结构更加清晰。（建议：一个测试类，对应一个被测功能）

上例中，拆分后的目录结构如下：

 实现代码如下：

 testadd.py
 1 from testpro.calculate import Count
import unittest

class TestAdd(unittest.TestCase):
    def setUp(self):
        print('add start')

    def test_add_1(self):
        print('add 1')
        self.assertEqual(Count(2, 4).add(), 6)

    def test_add_2(self):
        print('add 2')
        self.assertEqual(Count(1, 2).add(), 3)

    def tearDown(self):
        print('add end')


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

 testsub.py
 1 from testpro.calculate import Count
import unittest

class TestSub(unittest.TestCase):
    def setUp(self):
        print('sub start')

    def test_sub_1(self):
        print('sub 1')
        self.assertEqual(Count(2, 4).sub(), -2)

    def test_sub_2(self):
        print('sub 2')
        self.assertEqual(Count(1, 2).sub(), -1)

    def tearDown(self):
        print('sub end')


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

 runtest.py
 1 from testpro.testadd import TestAdd
from testpro.testsub import TestSub
import unittest


if __name__ == '__main__':
    suite = unittest.TestSuite()
    suite.addTest(TestAdd("test_add_1"))
    suite.addTest(TestAdd("test_add_2"))
    suite.addTest(TestSub("test_sub_1"))
    suite.addTest(TestSub("test_sub_2"))

    runner = unittest.TextTestRunner()
    runner.run(suite)

执行结果如下：

add start
add 1
add end
add start
add 2
add end
sub start
sub 1
sub end
sub start
sub 2
sub end
....
----------------------------------------------------------------------
Ran 4 tests in 0.001s

OK