有没有办法用Rspec存根包含模块的方法?

时间:2014-06-25 12:30:03

标签: ruby rspec mixins stubbing

我有一个包含在另一个模块中的模块,它们都实现了相同的方法。 我想保留包含模块的方法,如下所示:

module M
  def foo
    :M
  end
end

module A
  class << self
    include M

    def foo
      super
    end
  end
end

describe "trying to stub the included method" do
  before { allow(M).to receive(:foo).and_return(:bar) }

  it "should be stubbed when calling M" do
    expect(M.foo).to eq :bar
  end

  it "should be stubbed when calling A" do
    expect(A.foo).to eq :bar
  end
end

第一个测试是通过,但第二个测试输出:

Failure/Error: expect(A.foo).to eq :bar

   expected: :bar
        got: :M

为什么存根不能在这种情况下工作? 有没有不同的方法来实现这一目标?

谢谢!

------------------------------------- UPDATE --------- -------------------------

谢谢!使用allow_any_instance_of(M)解决了这个问题。 我的下一个问题是 - 如果我使用prepend而不包括会发生什么?请参阅以下代码:

module M
  def foo
    super
  end
end

module A
  class << self
    prepend M

    def foo
      :A
    end
  end
end

describe "trying to stub the included method" do
  before { allow_any_instance_of(M).to receive(:foo).and_return(:bar) }

  it "should be stubbed when calling A" do
    expect(A.foo).to eq :bar
  end
end 

这次,使用allow_any_instance_of(M)会导致无限循环。这是为什么?

1 个答案:

答案 0 :(得分:32)

请注意,您无法直接拨打M.foo!您的代码似乎才有效,因为您嘲笑M.foo返回:bar

当您打开A元类(class << self)以包含M时,您必须模拟M的任何实例,即添加到before块:

allow_any_instance_of(M).to receive(:foo).and_return(:bar)

module M
  def foo
    :M
  end
end

module A
  class << self
    include M

    def foo
      super
    end
  end
end

describe "trying to stub the included method" do
  before do
    allow(M).to receive(:foo).and_return(:bar)
    allow_any_instance_of(M).to receive(:foo).and_return(:bar)
  end


  it "should be stubbed when calling M" do
    expect(M.foo).to eq :bar
  end

  it "should be stubbed when calling A" do
    expect(A.foo).to eq :bar
  end
end