python - 如何在测试用例中模拟 uuid 生成？

Question

我有一个测试用例，在其中的设置中，我创建了一个对象，我想为其模拟模块 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

答案比我想象的要简单:只是不要使用迭代器!相反，将 uuid 列表设置为模拟的 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

编辑

我发现了一种更好的方式来将其编写为上下文管理器，它允许根据上下文为 uuid 添加前缀。

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

def uuid_prefix(prefix: str):
    return patch.object(uuid, 'uuid4', side_effect=['{}_{}'.format(prefix, x) for x in TEST_UUIDS])

class MyTestCase(TestCase):

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

    def _create_my_object(self):
        with uuid_prefix('obj_a'):
            # Create an object A using a uuid
        with uuid_prefix('obj_b'):
            # Create an object B using a uuid

说明:我正在使用 patch.object(uuid, 'uuid4') 模拟函数 uuid.uuid4。在其中，我将副作用定义为列表。如果你的副作用是一个列表，它可以被看作是该函数在后续调用中的返回值列表，所以第一次调用函数 uuid4() 时，它返回的第一个元素该列表，第二次是第二个元素，等等。如果在 with-context 中我生成 10 个对象 A，UUID 将是 'obj_a_uuid_0' 到 'obj_a_uuid_9'.