部分模拟父子类
创始人
2024-12-24 05:30:11
0次

部分模拟是指在单元测试过程中,部分模拟一个对象,而不改变其实例中未被模拟的部分,这在只需要测试某个方法的情况下非常有用。但在涉及到父子类的情况下,可能会遇到一些问题。

假设有如下的父类和子类:

class Parent:
    def foo(self):
        print("parent foo")

class Child(Parent):
    def bar(self):
        self.foo()
        print("child bar")

现在我们想要只模拟Child类的bar方法,我们可以使用unittest中的mock模块来部分模拟:

from unittest.mock import patch

class TestChild(unittest.TestCase):
    def test_bar(self):
        with patch.object(Child, "foo") as mock_foo:
            c = Child()
            c.bar()
            mock_foo.assert_called_once()

然而,mock_foo.assert_called_once()将会失败并报错,因为调用的方法实际上是Parent类中的foo方法,而不是Child类。这是因为一旦覆盖子类中的方法,就无法正确地调用其父类中的同名方法。

解决方法是将被模拟的方法重命名,以避免与父类中同名方法的冲突。例如,我们可以将Child类中的foo方法重命名为_child_foo方法:

class Child(Parent):
    def _child_foo(self):
        print("child foo")
        
    def bar(self):
        self._child_foo()
        print("child bar")

现在,我们将它们打包在一起:

class TestChild(unittest.TestCase):
    def test_bar(self):
        with patch.object(Child, "_child_foo") as mock_foo:
            c = Child()
            c.bar()
            mock_foo.assert_called_once()

现在,我们可以成功地部分模拟Child类的bar方法,而不会影响Parent类中的同名方法。

相关内容

热门资讯

实操分享辅助挂!we-poke... 本地部署 AI 黑科技工具迎来新版本,不用上传文件到云端就能完成we-poker辅助解析、 AI 处...
热门推荐透视!we poker... 一款 AI 游戏攻略黑科技工具上线,查攻略效率直接开挂。它整合版本更新、装备搭配、BOSS 打法、牌...
玩家交流透视挂!wpk透视是真... AI 黑科技赛道又上新,这款智能处理工具堪称效率开挂,专门面向短视频创作者打造。它打通脚本撰写、AI...
AI教程辅助!wepoker有... AI教程辅助!wepoker有透视底牌吗,微扑克开挂技巧插件(往昔是真的有辅助)主打本地运行的 AI...
最新技巧!辅助挂!WPK购买,... 最新技巧!辅助挂!WPK购买,WPK可以开挂吗安装(确实真的是有辅助挂)AI 是一款可以让一直输的玩...
今日焦点透视挂!wpk德州局怎... 一款 AI 游戏攻略黑科技工具上线,查攻略效率直接开挂。它整合版本更新、装备搭配、BOSS 打法、牌...
2分钟细说透视挂!wepoke... 您好,2分钟细说透视挂!wepoker底牌透视,德扑之星怎么开挂工具(素来真的是有透视挂)这款游戏可...
玩家必备教程透视挂!WPK工具... 玩家必备教程透视挂!WPK工具,WPK可以装挂吗插件(从来有透视挂)主打本地运行的 AI 黑科技工具...
热点推荐辅助!wejoker辅... wejoker辅助软件AI 游戏攻略黑科技上线本地缓存功能,离线查阅能力开挂。提前下载对应游戏攻略数...
盘点一款辅助器!微扑克游戏辅助... 微扑克游戏辅助器AI 游戏攻略黑科技上线本地缓存功能,离线查阅能力开挂。提前下载对应游戏攻略数据包,...