lib/rib/test/multiline.rb in rib-1.1.3 vs lib/rib/test/multiline.rb in rib-1.1.4

@@ -1,12 +1,12 @@
 
 shared :setup_multiline do
   def setup_shell
     @shell = Rib::Shell.new(
       :binding => Object.new.instance_eval{binding}).before_loop
-    stub(@shell).print
-    stub(@shell).puts
+    stub(@shell).print.with_any_args
+    stub(@shell).puts .with_any_args
   end
 
   def setup_input str
     if readline?
       mock(::Readline).readline(is_a(String), true){
@@ -17,10 +17,10 @@
     end
   end
 
   def input str
     setup_input(str)
-    mock.proxy(@shell).throw(:rib_multiline)
+    mock(@shell).throw(:rib_multiline).proxy
   end
 
   def input_done str, err=nil
     setup_input(str)
     if err