1
0
Fork 0
mirror of https://github.com/pry/pry.git synced 2022-11-09 12:35:05 -05:00
pry--pry/lib/pry/test/helper.rb

170 lines
3.6 KiB
Ruby
Raw Normal View History

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)
2013-01-15 17:29:23 -05:00
Pry.current[:pry_local] = value
b.eval("#{name} = ::Pry.current[:pry_local]")
ensure
2013-01-15 17:29:23 -05:00
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
2011-12-31 10:21:58 -05:00
def mock_exception(*mock_backtrace)
StandardError.new.tap do |e|
e.define_singleton_method(:backtrace) { mock_backtrace }
end
2011-01-19 03:50:45 -05:00
end
end
def pry_tester(*args, &block)
2012-09-09 02:12:25 -04:00
if args.length == 0 || args[0].is_a?(Hash)
args.unshift(Pry.toplevel_binding)
end
PryTester.new(*args).tap do |t|
2012-07-24 08:49:30 -04:00
(class << t; self; end).class_eval(&block) if block
2012-07-24 04:36:53 -04:00
end
end
def pry_eval(*eval_strs)
if eval_strs.first.is_a? String
2012-09-09 01:29:00 -04:00
binding = Pry.toplevel_binding
else
binding = Pry.binding_for(eval_strs.shift)
end
pry_tester(binding).eval(*eval_strs)
end
2012-07-24 04:36:53 -04:00
class PryTester
extend Forwardable
2012-07-24 08:49:30 -04:00
attr_reader :pry, :out
2012-07-24 04:36:53 -04:00
2012-12-28 01:06:50 -05:00
def_delegators :@pry, :eval_string, :eval_string=
2012-07-24 04:36:53 -04:00
2012-12-18 02:54:07 -05:00
def initialize(target = TOPLEVEL_BINDING, options = {})
@pry = Pry.new(options.merge(:target => target))
@history = options[:history]
2012-12-18 02:54:07 -05:00
@pry.inject_sticky_locals!
reset_output
2012-07-24 04:36:53 -04:00
end
def eval(*strs)
reset_output
2012-07-24 08:49:30 -04:00
result = nil
strs.flatten.each do |str|
str = "#{str.strip}\n"
@history.push str if @history
2012-07-24 04:36:53 -04:00
if @pry.process_command(str)
result = last_command_result_or_output
2012-07-24 04:36:53 -04:00
else
2012-07-24 08:49:30 -04:00
result = @pry.evaluate_ruby(str)
2012-07-24 04:36:53 -04:00
end
2012-07-24 08:49:30 -04:00
end
result
end
2012-12-28 01:06:50 -05:00
def push(*lines)
Array(lines).flatten.each do |line|
@pry.eval(line)
end
end
2012-12-18 02:58:02 -05:00
def push_binding(context)
@pry.push_binding context
end
2012-07-24 08:49:30 -04:00
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
2012-07-24 08:49:30 -04:00
def last_command_result
2013-01-15 17:29:23 -05:00
result = Pry.current[:pry_cmd_result]
2012-07-24 08:49:30 -04:00
result.retval if result
2012-07-24 04:36:53 -04:00
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
2012-07-24 04:36:53 -04:00
end