使用mocha,有没有办法存根许多参数?

时间:2012-03-09 17:48:58

标签: ruby-on-rails ruby testing mocha

我们假设我有这个课程

class Foo
  def bar(param1=nil, param2=nil, param3=nil)
     :bar1 if param1
     :bar2 if param2
     :bar3 if param3
  end
end

我可以使用以下方法存根整条方法:

Foo.any_instance.expects(:bar).at_least_once.returns(false)

但是如果我只想在bar方法的param1为真时存根,我就找不到办法:

我也看过with和has_entry,它似乎只适用于单个参数。

我期待这样的功能。

Foo.any_instance.expects(:bar).with('true',:any,:any).returns(:baz1)
Foo.any_instance.expects(:bar).with(any,'some',:any).returns(:baz2)

由于

............................................... ....编辑以下........................................... ..

谢谢,纳什

不熟悉rspec,所以我尝试使用any_instance进行单元测试,看起来似乎有效

require 'test/unit'
require 'mocha'

class FooTest < Test::Unit::TestCase 

  def test_bar_stub 
    foo = Foo.new
    p foo.bar(1)

    Foo.any_instance.stubs(:bar).with { |*args| args[0]=='hee' }.returns('hee')
    Foo.any_instance.stubs(:bar).with { |*args| args[1]=='haa' }.returns('haa')
    Foo.any_instance.stubs(:bar).with { |*args| args[2]!=nil   }.returns('heehaa')

    foo = Foo.new
    p foo.bar('hee')
    p foo.bar('sth', 'haa')
    p foo.bar('sth', 'haa', 'sth')
  end

end

1 个答案:

答案 0 :(得分:9)

如果我找对你,可以是:

class Foo
  def bar(param1=nil, param2=nil, param3=nil)
     :bar1 if param1
     :bar2 if param2
     :bar3 if param3
  end
end

describe Foo do
  it "returns 0 for all gutter game" do
    foo = Foo.new
    foo.stub(:bar).with { |*args| args[0] }.and_return(:bar1)
    foo.stub(:bar).with { |*args| args[1] }.and_return(:bar2)
    foo.stub(:bar).with { |*args| args[2] }.and_return(:bar3)
    foo.bar(true).should == :bar1
    foo.bar('blah', true).should == :bar2
    foo.bar('blah', 'blah', true).should == :bar3
  end
end