相关单元测试

Question

我有一个程序，该程序根据输入的参数对象和存储在数据库中的信息给出输出。

我正在尝试使用2个测试用例测试该程序，其中第一个不提供任何参数。 然后，我的程序将创建一个默认参数对象，并将其作为输出的一部分返回，然后第二次测试将对参数对象进行特定更改，并将其作为输入返回给程序。

目前，我通过将两个测试用例合并为一个来做到这一点，以便能够将参数对象保存在局部变量中并对其进行更改并将其用于第二个：

def test_function:
    testData = {'name':"Test"}        
    response = self.app.post('/test', data=testData)
    # make assert statements for first response
    parameters = json.loads(response.data)['parameters']
    parameters['foo'] = "bar"
    testData['parameters'] = parameters
    response = self.app.post('/test', data=testData)
    # make assert statements for second response

有什么办法可以在两个单独的测试用例中做到这一点？ 或更重要的是，我应该在两个单独的测试用例中执行此操作，因为如果第一个不工作，第二个将无法运行？

我不能为第二个测试用例硬编码参数对象，因为它依赖于数据库中的数据，并且进行测试的部分原因是看程序是否针对不同/没有数据工作。

Answer 1

如果要使用已知的开始状态开发多个测试，请使用setUp函数。

这是python网站上的一个简单示例：

import random
import unittest

class TestSequenceFunctions(unittest.TestCase):

    def setUp(self):
        self.seq = range(10)

    def test_shuffle(self):
        # make sure the shuffled sequence does not lose any elements
        random.shuffle(self.seq)
        self.seq.sort()
        self.assertEqual(self.seq, range(10))

        # should raise an exception for an immutable sequence
        self.assertRaises(TypeError, random.shuffle, (1,2,3))

    def test_choice(self):
        element = random.choice(self.seq)
        self.assertTrue(element in self.seq)

    def test_sample(self):
        with self.assertRaises(ValueError):
            random.sample(self.seq, 20)
        for element in random.sample(self.seq, 5):
            self.assertTrue(element in self.seq)

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