时间戳

首页 > 解决方案 > 修补本地类引用

python - 修补本地类引用

问题描述

像许多人一样，我在模拟修补和正确路径方面遇到问题。具体来说，我的代码引用了同一文件中的另一个类，我无法修补该引用。

我有以下 python 文件，package/engine/dataflows/flow.py：

class Flow:

  def run(self, type):
    if type == 'A':
      method1()
    elif type == 'B':
      method2()
    else:
      backfill = Backfill()
      backfill.run()


class Backfill(Flow):

  def run(self):
    ...

还有一个测试文件 package/tests/engine/dataflows/test_Flow.py

import unittest
from unittest.mock import Mock, patch
from engine.dataflows.flow import Flow

class MockFlow(Flow):
  ...

class TestFlowRun(unittest.TestCase):

  def setUp(self):
    self.flow = MockFlow()

  def test_run_type_c(self):
    with patch('engine.dataflows.flow.Backfill') as mock_backfill:
      self.flow.run(type='C')
      assert mock_backfill.run.call_count == 1

该补丁的工作原理是在使用 pytest 运行时不会引发错误，但断言失败。我认为这是因为在初始化 MockFlow 时，基本上已经导入了对 Backfill 类的本地引用，但我一直无法想出一个修补路径来处理这个问题。

flow.py 的内容包括 Flow 基类和几个实现不同数据流模式的子类。它们位于同一个文件中，以便于理解和常见的依赖关系。

标签： pythonunit-testingmockingpytest

解决方案

问题是您正在检查一个类的 run() 函数，而不是该类的实例化。模拟的 Backfill 类将通过其构造函数/init 返回该类的实例。那就是您要检查的模拟/对象。

with patch('engine.dataflows.flow.Backfill') as mock_backfill:
    mocked_backfill_instance = mock_backfill.return_value
    self.flow.run(type='C')
    assert mocked_backfill_instance.run.call_count == 1

推荐阅读