使用RSpec期望方法调用和原始方法的代理

时间:2011-03-16 02:11:28

标签: ruby-on-rails rspec bdd controllers

我想发现缺少BDD:包括ActiveRecord :: Base.find方法的参数。所以我的想法是在规范中有这样的东西:

ActiveRecord::Base.should_receive(:find).once.and_proxy_to_original_method
parent = SomeClass.find 34
parent.child.should be_loaded
parent.other_children.should be_loaded

如果#child或#other_children关联未加载,则期望失败,例如: “预期ActiveRecord :: Base.find被调用一次但是它被调用了2次以下args:1。......; 2. ...”

有没有人知道是否有一些像这样的匹配器或者如何制作它。

由于

1 个答案:

答案 0 :(得分:5)

认为我遇到了同样的问题here。在你的特殊情况下,我会这样做,我觉得很干净。

original_method = ActiveRecord::Base.method(:find)
ActiveRecord::Base.should_receive(:find).once do (*args)
  original_method.call(*args)
end

我相信您可以扩展Rspec Mocks::MessageExpectation类以包含and_proxy_to_original_method方法,不应该太难,但我没看过。