170 lines
3.6 KiB
Ruby
170 lines
3.6 KiB
Ruby
require 'pry'
|
|
|
|
# in case the tests call reset_defaults, ensure we reset them to
|
|
# amended (test friendly) values
|
|
class << Pry
|
|
alias_method :orig_reset_defaults, :reset_defaults
|
|
def reset_defaults
|
|
orig_reset_defaults
|
|
|
|
Pry.color = false
|
|
Pry.pager = false
|
|
Pry.config.should_load_rc = false
|
|
Pry.config.should_load_local_rc= false
|
|
Pry.config.should_load_plugins = false
|
|
Pry.config.history.should_load = false
|
|
Pry.config.history.should_save = false
|
|
Pry.config.correct_indent = false
|
|
Pry.config.hooks = Pry::Hooks.new
|
|
Pry.config.collision_warning = false
|
|
end
|
|
end
|
|
Pry.reset_defaults
|
|
|
|
# A global space for storing temporary state during tests.
|
|
|
|
module PryTestHelpers
|
|
|
|
module_function
|
|
|
|
# inject a variable into a binding
|
|
def inject_var(name, value, b)
|
|
Pry.current[:pry_local] = value
|
|
b.eval("#{name} = ::Pry.current[:pry_local]")
|
|
ensure
|
|
Pry.current[:pry_local] = nil
|
|
end
|
|
|
|
def constant_scope(*names)
|
|
names.each do |name|
|
|
Object.remove_const name if Object.const_defined?(name)
|
|
end
|
|
|
|
yield
|
|
ensure
|
|
names.each do |name|
|
|
Object.remove_const name if Object.const_defined?(name)
|
|
end
|
|
end
|
|
|
|
# Open a temp file and yield it to the block, closing it after
|
|
# @return [String] The path of the temp file
|
|
def temp_file(ext='.rb')
|
|
file = Tempfile.new(['pry', ext])
|
|
yield file
|
|
ensure
|
|
file.close(true) if file
|
|
File.unlink("#{file.path}c") if File.exists?("#{file.path}c") # rbx
|
|
end
|
|
|
|
def unindent(*args)
|
|
Pry::Helpers::CommandHelpers.unindent(*args)
|
|
end
|
|
|
|
def mock_command(cmd, args=[], opts={})
|
|
output = StringIO.new
|
|
ret = cmd.new(opts.merge(:output => output)).call_safely(*args)
|
|
Struct.new(:output, :return).new(output.string, ret)
|
|
end
|
|
|
|
def mock_exception(*mock_backtrace)
|
|
StandardError.new.tap do |e|
|
|
e.define_singleton_method(:backtrace) { mock_backtrace }
|
|
end
|
|
end
|
|
end
|
|
|
|
def pry_tester(*args, &block)
|
|
if args.length == 0 || args[0].is_a?(Hash)
|
|
args.unshift(Pry.toplevel_binding)
|
|
end
|
|
|
|
PryTester.new(*args).tap do |t|
|
|
(class << t; self; end).class_eval(&block) if block
|
|
end
|
|
end
|
|
|
|
def pry_eval(*eval_strs)
|
|
if eval_strs.first.is_a? String
|
|
binding = Pry.toplevel_binding
|
|
else
|
|
binding = Pry.binding_for(eval_strs.shift)
|
|
end
|
|
|
|
pry_tester(binding).eval(*eval_strs)
|
|
end
|
|
|
|
class PryTester
|
|
extend Forwardable
|
|
|
|
attr_reader :pry, :out
|
|
|
|
def_delegators :@pry, :eval_string, :eval_string=
|
|
|
|
def initialize(target = TOPLEVEL_BINDING, options = {})
|
|
@pry = Pry.new(options.merge(:target => target))
|
|
@history = options[:history]
|
|
|
|
@pry.inject_sticky_locals!
|
|
reset_output
|
|
end
|
|
|
|
def eval(*strs)
|
|
reset_output
|
|
result = nil
|
|
|
|
strs.flatten.each do |str|
|
|
str = "#{str.strip}\n"
|
|
@history.push str if @history
|
|
|
|
if @pry.process_command(str)
|
|
result = last_command_result_or_output
|
|
else
|
|
result = @pry.evaluate_ruby(str)
|
|
end
|
|
end
|
|
|
|
result
|
|
end
|
|
|
|
def push(*lines)
|
|
Array(lines).flatten.each do |line|
|
|
@pry.eval(line)
|
|
end
|
|
end
|
|
|
|
def push_binding(context)
|
|
@pry.push_binding context
|
|
end
|
|
|
|
def last_output
|
|
@out.string if @out
|
|
end
|
|
|
|
def process_command(command_str)
|
|
@pry.process_command(command_str) or raise "Not a valid command"
|
|
last_command_result_or_output
|
|
end
|
|
|
|
def last_command_result
|
|
result = Pry.current[:pry_cmd_result]
|
|
result.retval if result
|
|
end
|
|
|
|
protected
|
|
|
|
def last_command_result_or_output
|
|
result = last_command_result
|
|
if result != Pry::Command::VOID_VALUE
|
|
result
|
|
else
|
|
last_output
|
|
end
|
|
end
|
|
|
|
def reset_output
|
|
@out = StringIO.new
|
|
@pry.output = @out
|
|
end
|
|
end
|