如何在测试用例中模拟uuid代？

Question

我有一个测试案例，在setUp中，我创建了一个对象，我想在模块uuid中模拟函数uuid4。现在

TEST_UUIDS = ['uuid_{}'.format(i) for i in range(10000)] 
UUID_POOL = iter(TEST_UUIDS) 

def generate_uuid() -> str: 
    return next(UUID_POOL) 

class MyTestCase(TestCase): 

    def setUp(self, **kwargs): 
     self._create_my_object 

    @patch.object(uuid, 'uuid4', generate_uuid) 
    def _create_my_object(self): 
     # Create an object using a uuid 

我的问题是，当我运行写两个测试案例，创建对象第二次，它得到其他的UUID是第一次。因此，结果取决于测试类中测试用例的数量和它们运行的​​顺序，这是我不想要的。

• 这是嘲笑uuid发电机的最好方法吗？
• 如何在每次setUp（或tearDown）调用时重置或重新创建迭代器？

Answer 1

答案比我想的要简单：只是不要使用迭代器！相反，请将uuids列表设置为模拟的side_effect。在`side_effect`

TEST_UUIDS = ['uuid_{}'.format(i) for i in range(10000)] 

class MyTestCase(TestCase): 

    def setUp(self, **kwargs): 
     self._create_my_object 

    @patch.object(uuid, 'uuid4', side_effect=TEST_UUIDS) 
    def _create_my_object(self): 
     # Create an object using a uuid