45 lines
789 B
Ruby
45 lines
789 B
Ruby
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(options)
|
|
end
|
|
|
|
output.string
|
|
end
|
|
|
|
# Set I/O streams. Out defaults to an anonymous StringIO.
|
|
def redirect_pry_io(new_in, new_out = StringIO.new)
|
|
old_in = Pry.input
|
|
old_out = Pry.output
|
|
|
|
Pry.input = new_in
|
|
Pry.output = new_out
|
|
begin
|
|
yield
|
|
ensure
|
|
Pry.input = old_in
|
|
Pry.output = old_out
|
|
end
|
|
end
|
|
|
|
class InputTester
|
|
def initialize(*actions)
|
|
@orig_actions = actions.dup
|
|
@actions = actions
|
|
end
|
|
|
|
def readline(*)
|
|
@actions.shift
|
|
end
|
|
|
|
def rewind
|
|
@actions = @orig_actions.dup
|
|
end
|
|
end
|