部分模拟父子类
创始人
2024-12-24 05:30:27
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类中的同名方法。

相关内容

热门资讯

iwatch怎么连接安卓系统,... 你有没有想过,那款时尚又实用的iWatch,竟然只能和iPhone好上好?别急,今天就来给你揭秘,怎...
安卓系统怎么连不上carlif... 安卓系统无法连接CarLife的原因及解决方法随着智能手机的普及,CarLife这一车载互联功能为驾...
oppo手机安卓系统换成苹果系... OPPO手机安卓系统换成苹果系统:现实吗?如何操作?随着智能手机市场的不断发展,用户对于手机系统的需...
iphone系统与安卓系统更新... 最近是不是你也遇到了这样的烦恼?手机更新系统总是失败,急得你团团转。别急,今天就来给你揭秘为什么iP...
安卓平板改windows 系统... 你有没有想过,你的安卓平板电脑是不是也能变身成Windows系统的超级英雄呢?想象在同一个设备上,你...
安卓系统上滑按键,便捷生活与高... 你有没有发现,现在手机屏幕越来越大,操作起来却越来越方便了呢?这都得归功于安卓系统上的那些神奇的上滑...
安卓系统连接耳机模式,蓝牙、有... 亲爱的手机控们,你们有没有遇到过这种情况:手机突然变成了“耳机模式”,明明耳机没插,声音却只从耳机孔...
希沃系统怎么装安卓系统,解锁更... 亲爱的读者们,你是否也像我一样,对希沃一体机上的安卓系统充满了好奇呢?想象在教室里,你的希沃一体机不...
安装了Anaconda之后找不... 在安装Anaconda后,如果找不到Jupyter Notebook,可以尝试以下解决方法:检查环境...
安卓换鸿蒙系统会卡吗,体验流畅... 最近手机圈可是热闹非凡呢!不少安卓用户都在议论纷纷,说鸿蒙系统要来啦!那么,安卓手机换上鸿蒙系统后,...