From d31967799cbf1705fbc69dff1add67c1599896f1 Mon Sep 17 00:00:00 2001
From: Conrad Irwin <conrad.irwin@gmail.com>
Date: Tue, 18 Dec 2012 00:10:52 -0800
Subject: [PATCH] Remove more rep() from spec/pry_spec.rb

---
 spec/helpers/mock_pry.rb |  3 ++-
 spec/pry_spec.rb         | 55 ++++++++--------------------------------
 2 files changed, 12 insertions(+), 46 deletions(-)

diff --git a/spec/helpers/mock_pry.rb b/spec/helpers/mock_pry.rb
index a96e7af3..e9c1c452 100644
--- a/spec/helpers/mock_pry.rb
+++ b/spec/helpers/mock_pry.rb
@@ -1,12 +1,13 @@
 def mock_pry(*args)
   args.flatten!
   binding = args.first.is_a?(Binding) ? args.shift : binding()
+  options = args.last.is_a?(Hash) ? args.pop : {}
 
   input = InputTester.new(*args)
   output = StringIO.new
 
   redirect_pry_io(input, output) do
-    binding.pry
+    binding.pry(options)
   end
 
   output.string
diff --git a/spec/pry_spec.rb b/spec/pry_spec.rb
index 0e6c8dda..0c008130 100644
--- a/spec/pry_spec.rb
+++ b/spec/pry_spec.rb
@@ -112,64 +112,35 @@ describe Pry do
       end
 
       it 'should work with multi-line input' do
-        o = Object.new
-
-        pry_tester = Pry.new(:input => InputTester.new("x = ", "1 + 4"), :output => @str_output)
-        pry_tester.rep(o)
-        @str_output.string.should =~ /5/
+        mock_pry("x = ", "1 + 4").should =~ /5/
       end
 
       it 'should define a nested class under Hello and not on top-level or Pry' do
-        pry_tester = Pry.new(:input => InputTester.new("class Nested", "end"), :output => StringIO.new)
-        pry_tester.rep(Hello)
+        mock_pry(Pry.binding_for(Hello), "class Nested", "end")
         Hello.const_defined?(:Nested).should == true
       end
 
       it 'should suppress output if input ends in a ";" and is an Exception object (single line)' do
-        o = Object.new
-
-        pry_tester = Pry.new(:input => InputTester.new("Exception.new;"), :output => @str_output)
-        pry_tester.rep(o)
-        @str_output.string.should == ""
+        mock_pry("Exception.new;").should == ""
       end
 
       it 'should suppress output if input ends in a ";" (single line)' do
-        o = Object.new
-
-        pry_tester = Pry.new(:input => InputTester.new("x = 5;"), :output => @str_output)
-        pry_tester.rep(o)
-        @str_output.string.should == ""
+        mock_pry("x = 5;").should == ""
       end
 
       it 'should suppress output if input ends in a ";" (multi-line)' do
-        o = Object.new
-
-        pry_tester = Pry.new(:input => InputTester.new("def self.blah", ":test", "end;"), :output => @str_output)
-        pry_tester.rep(o)
-        @str_output.string.should == ""
+        mock_pry("def self.blah", ":test", "end;").should == ""
       end
 
       it 'should be able to evaluate exceptions normally' do
-        o = Exception.new
-
         was_called = false
-        pry_tester = Pry.new(:input => InputTester.new("self"),
-                             :output => @str_output,
-                             :exception_handler => proc { was_called = true })
-
-        pry_tester.rep(o)
+        mock_pry("RuntimeError.new", :exception_handler => proc{ was_called = true })
         was_called.should == false
       end
 
       it 'should notice when exceptions are raised' do
-        o = Exception.new
-
         was_called = false
-        pry_tester = Pry.new(:input => InputTester.new("raise self"),
-                             :output => @str_output,
-                             :exception_handler => proc { was_called = true })
-
-        pry_tester.rep(o)
+        mock_pry("raise RuntimeError", :exception_handler => proc{ was_called = true })
         was_called.should == true
       end
 
@@ -252,16 +223,10 @@ describe Pry do
         end
 
         it 'store exceptions' do
-          res   = []
-          input = InputTester.new *["foo!","self << _in_[-1] << _out_[-1]"]
-          Pry.new(
-            :input => input,
-            :output => StringIO.new,
-            :memory_size => 1000
-          ).repl
+          mock_pry("foo!", "Pad.in = _in_[-1]; Pad.out = _out_[-1]")
 
-          res.first.should == "foo!\n"
-          res.last.should.be.kind_of NoMethodError
+          Pad.in.should == "foo!\n"
+          Pad.out.should.be.kind_of NoMethodError
         end
       end