@@ -9,8 +9,12 @@ module Spec __mock_proxy.add_negative_message_expectation(caller(1)[0], sym.to_sym, &block) end - def stub!(sym, opts={}) - __mock_proxy.add_stub(caller(1)[0], sym.to_sym, opts) + def stub!(sym_or_hash, opts={}) + if Hash === sym_or_hash + sym_or_hash.each {|method, value| stub!(method).and_return value } + else + __mock_proxy.add_stub(caller(1)[0], sym_or_hash.to_sym, opts) + end end def received_message?(sym, *args, &block) #:nodoc: lib/spec/mocks/methods.rb @@ -44,6 +44,12 @@ module Spec @instance.rspec_verify end.should_not raise_error end + + it "should handle multiple stubbed methods" do + @instance.stub!(:msg1 => 1, :msg2 => 2) + @instance.msg1.should == 1 + @instance.msg2.should == 2 + end it "should clear itself when verified" do @instance.stub!(:this_should_go).and_return(:blah) spec/spec/mocks/stub_spec.rb dc359c78147166bc9688ea49a5ebf237478e6f24 Pat Maddox pat.maddox@gmail.com http://github.com/dchelimsky/rspec/commit/5fd78361039d069449f91dc506ae2bf3d9d616de 5fd78361039d069449f91dc506ae2bf3d9d616de 2008-07-25T09:05:21-07:00 2008-07-25T09:05:21-07:00 added ability to stub multiple methods in one stub! call 498c5e97c890a7d6150ba6ab41e9f9dcc1291af5 Pat Maddox pat.maddox@gmail.com