2012-03-09 44 views
6

让我们假设我有这个类使用摩卡,有没有一种方法来存留很多参数?

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) 

但是如果我只想存根时,杆法的参数1是真的,我不能找到一个办法:

我也看着,和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 
+0

预期,而不是存根对我来说也能正常工作。 Foo.any_instance.expects(:bar).with {| * args | args [0] =='hee'} .returns('hee') – allenhwkim 2012-03-09 21:08:34

回答

7

如果我得到你的权利也可以是这样的:

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 
相关问题