模拟出任何python类实例上的方法

Question

我想在生产代码中模拟某个类的任何实例的方法，以便于测试。Python中是否有任何库可以简化此过程？

基本上，我想执行以下操作，但是在Python中（以下代码是Ruby，使用Mocha库）：

  def test_stubbing_an_instance_method_on_all_instances_of_a_class
    Product.any_instance.stubs(:name).returns('stubbed_name')
    assert_equal 'stubbed_name', SomeClassThatUsesProduct.get_new_product_name
  end

从上面需要注意的重要一点是，我需要在类级别上对其进行模拟，因为实际上我需要对我正在测试的事物创建的实例上的方法进行模拟。

用例：

我有一个类QueryMaker，它在的实例上调用方法RemoteAPI。我想模拟出RemoteAPI.get_data_from_remote_server返回常量的方法。我如何在测试中执行此操作，而不必在RemoteAPI代码中放入特殊情况以检查其运行在什么环境中。

我想要行动的示例：

# a.py
class A(object):
    def foo(self):
        return "A's foo"

# b.py
from a import A

class B(object):
    def bar(self):
        x = A()
        return x.foo()

# test.py
from a import A
from b import B

def new_foo(self):
    return "New foo"

A.foo = new_foo

y = B()
if y.bar() == "New foo":
    print "Success!"

Answer 1

最简单的方法可能是使用类方法。您确实应该使用实例方法，但是创建它们很麻烦，而内置函数可以创建类方法。使用类方法，您的存根将获得对类（而不是实例）的引用作为第一个参数，但是由于它是存根，因此这可能无关紧要。所以：

Product.name = classmethod(lambda cls: "stubbed_name")

请注意，lambda的签名必须与您要替换的方法的签名匹配。另外，当然，由于Python（例如Ruby）是一种动态语言，因此无法保证有人在您获得实例之前不会将您的存根方法切换为其他方法，尽管我希望您很快就会知道如果发生这种情况。

编辑：在进一步调查中，您可以省略classmethod()：

Product.name = lambda self: "stubbed_name"

我试图尽可能地保留原始方法的行为，但实际上似乎并不是必需的（无论如何也不会保留我所希望的行为）。