我可以在python unittests中伪造/模拟我的模拟对象的类型吗

Question

在我的python代码中，我检查了参数之一的类型，以确保它是我期望的类型。 例如：

def myfunction(dbConnection):
    if (type(dbConnection)<>bpgsql.Connection):
        r['error'] += ' invalid database connection'

我想通过模拟连接进行测试。 有没有办法使模拟对象假装为正确的类型？

Answer 1

在所有应有的尊重下，你们似乎不太正确！

我可以使用鸭子输入，但是首先有一种方法可以完成我打算做的事情：

来自http://docs.python.org/dev/library/unittest.mock.html

使用类或实例作为spec或spec_set模拟对象能够通过isintance测试：

>>>
>>> mock = Mock(spec=SomeClass)
>>> isinstance(mock, SomeClass)
True
>>> mock = Mock(spec_set=SomeClass())
>>> isinstance(mock, SomeClass)
True

所以我的示例代码如下：

m = mock.MagicMock(spec=bpgsql.Connection)
isinstance(m, bpgsql.Connection) 

这返回True

话虽如此，我不是在要求在python中进行严格的类型检查，我说如果您需要检查它就可以做到，它也可以与测试和模拟一起使用。

Answer 2

这正是您不应该执行严格类型检查的确切原因！ 您应该将该行从代码中完全删除。

如果你不想这样做，写出来与大家希望有属性的抽象基类（ .connect() .cursor() ，...？），并检查isinstance这一点。

_{<>也已经过时了。 使用!= 。}