kotovalexarian-likes-github/ruby--ruby
1
0
Fork 0
mirror of https://github.com/ruby/ruby.git synced 2022-11-09 12:17:21 -05:00
Code Releases Activity

[ruby/irb] Add execute_lines to reduce command tests' boilerplate

Browse source 
code
(https://github.com/ruby/irb/pull/436)

https://github.com/ruby/irb/commit/1595337149
This commit is contained in:
Stan Lo 2022-11-08 17:19:55 +00:00 committed by git
parent 0df47fdaf9
commit b7b78f062f
1 changed files with 159 additions and 268 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

427
test/irb/test_cmd.rb
View file

@ -219,145 +219,139 @@ module TestIRB
end end
def test_measure def test_measure
IRB.init_config(nil) conf = {
IRB.conf[:PROMPT] = { PROMPT: {
DEFAULT: { DEFAULT: {
PROMPT_I: '> ', PROMPT_I: '> ',
PROMPT_S: '> ', PROMPT_S: '> ',
PROMPT_C: '> ', PROMPT_C: '> ',
PROMPT_N: '> ' PROMPT_N: '> '
} }
},
PROMPT_MODE: :DEFAULT,
MEASURE: false
} }
IRB.conf[:VERBOSE] = false
IRB.conf[:PROMPT_MODE] = :DEFAULT c = Class.new(Object)
IRB.conf[:MEASURE] = false out, err = execute_lines(
input = TestInputMethod.new([
"3\n", "3\n",
"measure\n", "measure\n",
"3\n", "3\n",
"measure :off\n", "measure :off\n",
"3\n", "3\n",
]) conf: conf,
c = Class.new(Object) main: c
irb = IRB::Irb.new(IRB::WorkSpace.new(c.new), input) )
irb.context.return_format = "=> %s\n"
out, err = capture_output do
irb.eval_input
end
assert_empty err assert_empty err
assert_match(/\A=> 3\nTIME is added\.\n=> nil\nprocessing time: .+\n=> 3\n=> nil\n=> 3\n/, out) assert_match(/\A=> 3\nTIME is added\.\n=> nil\nprocessing time: .+\n=> 3\n=> nil\n=> 3\n/, out)
assert_empty(c.class_variables) assert_empty(c.class_variables)
end end
def test_measure_enabled_by_rc def test_measure_enabled_by_rc
IRB.init_config(nil) conf = {
IRB.conf[:PROMPT] = { PROMPT: {
DEFAULT: { DEFAULT: {
PROMPT_I: '> ', PROMPT_I: '> ',
PROMPT_S: '> ', PROMPT_S: '> ',
PROMPT_C: '> ', PROMPT_C: '> ',
PROMPT_N: '> ' PROMPT_N: '> '
} }
},
PROMPT_MODE: :DEFAULT,
MEASURE: true
} }
IRB.conf[:VERBOSE] = false
IRB.conf[:PROMPT_MODE] = :DEFAULT out, err = execute_lines(
IRB.conf[:MEASURE] = true
input = TestInputMethod.new([
"3\n", "3\n",
"measure :off\n", "measure :off\n",
"3\n", "3\n",
]) conf: conf,
irb = IRB::Irb.new(IRB::WorkSpace.new(Object.new), input) )
irb.context.return_format = "=> %s\n"
out, err = capture_output do
irb.eval_input
end
assert_empty err assert_empty err
assert_match(/\Aprocessing time: .+\n=> 3\n=> nil\n=> 3\n/, out) assert_match(/\Aprocessing time: .+\n=> 3\n=> nil\n=> 3\n/, out)
end end
def test_measure_enabled_by_rc_with_custom def test_measure_enabled_by_rc_with_custom
IRB.init_config(nil) measuring_proc = proc { |line, line_no, &block|
IRB.conf[:PROMPT] = {
DEFAULT: {
PROMPT_I: '> ',
PROMPT_S: '> ',
PROMPT_C: '> ',
PROMPT_N: '> '
}
}
IRB.conf[:VERBOSE] = false
IRB.conf[:PROMPT_MODE] = :DEFAULT
IRB.conf[:MEASURE] = true
IRB.conf[:MEASURE_PROC][:CUSTOM] = proc { |line, line_no, &block|
time = Time.now time = Time.now
result = block.() result = block.()
puts 'custom processing time: %fs' % (Time.now - time) if IRB.conf[:MEASURE] puts 'custom processing time: %fs' % (Time.now - time) if IRB.conf[:MEASURE]
result result
} }
input = TestInputMethod.new([ conf = {
PROMPT: {
DEFAULT: {
PROMPT_I: '> ',
PROMPT_S: '> ',
PROMPT_C: '> ',
PROMPT_N: '> '
}
},
PROMPT_MODE: :DEFAULT,
MEASURE: true,
MEASURE_PROC: { CUSTOM: measuring_proc }
}
out, err = execute_lines(
"3\n", "3\n",
"measure :off\n", "measure :off\n",
"3\n", "3\n",
]) conf: conf,
irb = IRB::Irb.new(IRB::WorkSpace.new(Object.new), input) )
irb.context.return_format = "=> %s\n"
out, err = capture_output do
irb.eval_input
end
assert_empty err assert_empty err
assert_match(/\Acustom processing time: .+\n=> 3\n=> nil\n=> 3\n/, out) assert_match(/\Acustom processing time: .+\n=> 3\n=> nil\n=> 3\n/, out)
end end
def test_measure_with_custom def test_measure_with_custom
IRB.init_config(nil) measuring_proc = proc { |line, line_no, &block|
IRB.conf[:PROMPT] = {
DEFAULT: {
PROMPT_I: '> ',
PROMPT_S: '> ',
PROMPT_C: '> ',
PROMPT_N: '> '
}
}
IRB.conf[:VERBOSE] = false
IRB.conf[:PROMPT_MODE] = :DEFAULT
IRB.conf[:MEASURE] = false
IRB.conf[:MEASURE_PROC][:CUSTOM] = proc { |line, line_no, &block|
time = Time.now time = Time.now
result = block.() result = block.()
puts 'custom processing time: %fs' % (Time.now - time) if IRB.conf[:MEASURE] puts 'custom processing time: %fs' % (Time.now - time) if IRB.conf[:MEASURE]
result result
} }
input = TestInputMethod.new([ conf = {
PROMPT: {
DEFAULT: {
PROMPT_I: '> ',
PROMPT_S: '> ',
PROMPT_C: '> ',
PROMPT_N: '> '
}
},
PROMPT_MODE: :DEFAULT,
MEASURE: false,
MEASURE_PROC: { CUSTOM: measuring_proc }
}
out, err = execute_lines(
"3\n", "3\n",
"measure\n", "measure\n",
"3\n", "3\n",
"measure :off\n", "measure :off\n",
"3\n", "3\n",
]) conf: conf
irb = IRB::Irb.new(IRB::WorkSpace.new(Object.new), input) )
irb.context.return_format = "=> %s\n"
out, err = capture_output do
irb.eval_input
end
assert_empty err assert_empty err
assert_match(/\A=> 3\nCUSTOM is added\.\n=> nil\ncustom processing time: .+\n=> 3\n=> nil\n=> 3\n/, out) assert_match(/\A=> 3\nCUSTOM is added\.\n=> nil\ncustom processing time: .+\n=> 3\n=> nil\n=> 3\n/, out)
end end
def test_measure_with_proc def test_measure_with_proc
IRB.init_config(nil) conf = {
IRB.conf[:PROMPT] = { PROMPT: {
DEFAULT: { DEFAULT: {
PROMPT_I: '> ', PROMPT_I: '> ',
PROMPT_S: '> ', PROMPT_S: '> ',
PROMPT_C: '> ', PROMPT_C: '> ',
PROMPT_N: '> ' PROMPT_N: '> '
} }
},
PROMPT_MODE: :DEFAULT,
MEASURE: false,
} }
IRB.conf[:VERBOSE] = false c = Class.new(Object)
IRB.conf[:PROMPT_MODE] = :DEFAULT out, err = execute_lines(
IRB.conf[:MEASURE] = false
input = TestInputMethod.new([
"3\n", "3\n",
"measure { |context, code, line_no, &block|\n", "measure { |context, code, line_no, &block|\n",
" result = block.()\n", " result = block.()\n",
@ -373,56 +367,38 @@ module TestIRB
"3\n", "3\n",
"measure :off\n", "measure :off\n",
"3\n", "3\n",
]) conf: conf,
c = Class.new(Object) main: c
irb = IRB::Irb.new(IRB::WorkSpace.new(c.new), input) )
irb.context.return_format = "=> %s\n"
out, err = capture_output do
irb.eval_input
end
assert_empty err assert_empty err
assert_match(/\A=> 3\nBLOCK is added\.\n=> nil\naaa\n=> 3\nBLOCK is added.\naaa\n=> nil\nbbb\n=> 3\n=> nil\n=> 3\n/, out) assert_match(/\A=> 3\nBLOCK is added\.\n=> nil\naaa\n=> 3\nBLOCK is added.\naaa\n=> nil\nbbb\n=> 3\n=> nil\n=> 3\n/, out)
assert_empty(c.class_variables) assert_empty(c.class_variables)
end end
def test_irb_source def test_irb_source
IRB.init_config(nil)
File.write("#{@tmpdir}/a.rb", "a = 'hi'\n") File.write("#{@tmpdir}/a.rb", "a = 'hi'\n")
input = TestInputMethod.new([ out, err = execute_lines(
"a = 'bug17564'\n", "a = 'bug17564'\n",
"a\n", "a\n",
"irb_source '#{@tmpdir}/a.rb'\n", "irb_source '#{@tmpdir}/a.rb'\n",
"a\n", "a\n",
]) )
IRB.conf[:VERBOSE] = false
IRB.conf[:PROMPT_MODE] = :SIMPLE
irb = IRB::Irb.new(IRB::WorkSpace.new(self), input)
IRB.conf[:MAIN_CONTEXT] = irb.context
out, err = capture_output do
irb.eval_input
end
assert_empty err assert_empty err
assert_pattern_list([ assert_pattern_list([
/=> "bug17564"\n/, /=> "bug17564"\n/,
/=> "bug17564"\n/, /=> "bug17564"\n/,
/ => "hi"\n/, / => "hi"\n/,
/ => nil\n/, / => nil\n/,
/=> "hi"\n/, /=> "hi"\n/,
], out) ], out)
end end
def test_help def test_help
IRB.init_config(nil) out, _ = execute_lines(
input = TestInputMethod.new([ "help 'String#gsub'\n",
"help 'String#gsub'\n", "\n",
"\n", )
])
IRB.conf[:PROMPT_MODE] = :SIMPLE
IRB.conf[:VERBOSE] = false
irb = IRB::Irb.new(IRB::WorkSpace.new(self), input)
out, _ = capture_output do
irb.eval_input
end
# the former is what we'd get without document content installed, like on CI # the former is what we'd get without document content installed, like on CI
# the latter is what we may get locally # the latter is what we may get locally
@ -434,18 +410,11 @@ module TestIRB
end end
def test_help_without_rdoc def test_help_without_rdoc
IRB.init_config(nil) out, _ = without_rdoc do
input = TestInputMethod.new([ execute_lines(
"help 'String#gsub'\n", "help 'String#gsub'\n",
"\n", "\n",
]) )
IRB.conf[:PROMPT_MODE] = :SIMPLE
IRB.conf[:VERBOSE] = false
irb = IRB::Irb.new(IRB::WorkSpace.new(self), input)
out, _ = capture_output do
without_rdoc do
irb.eval_input
end
end end
# if it fails to require rdoc, it only returns the command object # if it fails to require rdoc, it only returns the command object
@ -456,21 +425,13 @@ module TestIRB
end end
def test_irb_load def test_irb_load
IRB.init_config(nil)
File.write("#{@tmpdir}/a.rb", "a = 'hi'\n") File.write("#{@tmpdir}/a.rb", "a = 'hi'\n")
input = TestInputMethod.new([ out, err = execute_lines(
"a = 'bug17564'\n", "a = 'bug17564'\n",
"a\n", "a\n",
"irb_load '#{@tmpdir}/a.rb'\n", "irb_load '#{@tmpdir}/a.rb'\n",
"a\n", "a\n",
]) )
IRB.conf[:VERBOSE] = false
IRB.conf[:PROMPT_MODE] = :SIMPLE
irb = IRB::Irb.new(IRB::WorkSpace.new(self), input)
IRB.conf[:MAIN_CONTEXT] = irb.context
out, err = capture_output do
irb.eval_input
end
assert_empty err assert_empty err
assert_pattern_list([ assert_pattern_list([
/=> "bug17564"\n/, /=> "bug17564"\n/,
@ -482,7 +443,7 @@ module TestIRB
end end
def test_ls def test_ls
input = TestInputMethod.new([ out, err = execute_lines(
"class P\n", "class P\n",
" def m() end\n", " def m() end\n",
" def m2() end\n", " def m2() end\n",
@ -508,16 +469,8 @@ module TestIRB
"obj.extend M2\n", "obj.extend M2\n",
"def obj.m5() end\n", "def obj.m5() end\n",
"ls obj\n", "ls obj\n",
]) )
IRB.init_config(nil)
workspace = IRB::WorkSpace.new(self)
IRB.conf[:VERBOSE] = false
irb = IRB::Irb.new(workspace, input)
IRB.conf[:MAIN_CONTEXT] = irb.context
irb.context.return_format = "=> %s\n"
out, err = capture_output do
irb.eval_input
end
assert_empty err assert_empty err
assert_match(/^instance variables:\s+@a\n/m, out) assert_match(/^instance variables:\s+@a\n/m, out)
assert_match(/P#methods:\s+m\n/m, out) assert_match(/P#methods:\s+m\n/m, out)
@ -528,18 +481,9 @@ module TestIRB
end end
def test_ls_with_no_singleton_class def test_ls_with_no_singleton_class
input = TestInputMethod.new([ out, err = execute_lines(
"ls 42", "ls 42",
]) )
IRB.init_config(nil)
workspace = IRB::WorkSpace.new(self)
IRB.conf[:VERBOSE] = false
irb = IRB::Irb.new(workspace, input)
IRB.conf[:MAIN_CONTEXT] = irb.context
irb.context.return_format = "=> %s\n"
out, err = capture_output do
irb.eval_input
end
assert_empty err assert_empty err
assert_match(/Comparable#methods:\s+/, out) assert_match(/Comparable#methods:\s+/, out)
assert_match(/Numeric#methods:\s+/, out) assert_match(/Numeric#methods:\s+/, out)
@ -547,70 +491,34 @@ module TestIRB
end end
def test_show_source def test_show_source
input = TestInputMethod.new([ out, err = execute_lines(
"show_source IRB.conf\n", "show_source IRB.conf\n",
]) )
IRB.init_config(nil)
workspace = IRB::WorkSpace.new(self)
IRB.conf[:VERBOSE] = false
irb = IRB::Irb.new(workspace, input)
IRB.conf[:MAIN_CONTEXT] = irb.context
irb.context.return_format = "=> %s\n"
out, err = capture_output do
irb.eval_input
end
assert_empty err assert_empty err
assert_match(%r[/irb\.rb], out) assert_match(%r[/irb\.rb], out)
end end
def test_show_source_method def test_show_source_method
input = TestInputMethod.new([ out, err = execute_lines(
"p show_source('IRB.conf')\n", "p show_source('IRB.conf')\n",
]) )
IRB.init_config(nil)
workspace = IRB::WorkSpace.new(self)
IRB.conf[:VERBOSE] = false
irb = IRB::Irb.new(workspace, input)
IRB.conf[:MAIN_CONTEXT] = irb.context
irb.context.return_format = "=> %s\n"
out, err = capture_output do
irb.eval_input
end
assert_empty err assert_empty err
assert_match(%r[/irb\.rb], out) assert_match(%r[/irb\.rb], out)
end end
def test_show_source_string def test_show_source_string
input = TestInputMethod.new([ out, err = execute_lines(
"show_source 'IRB.conf'\n", "show_source 'IRB.conf'\n",
]) )
IRB.init_config(nil)
workspace = IRB::WorkSpace.new(self)
IRB.conf[:VERBOSE] = false
irb = IRB::Irb.new(workspace, input)
IRB.conf[:MAIN_CONTEXT] = irb.context
irb.context.return_format = "=> %s\n"
out, err = capture_output do
irb.eval_input
end
assert_empty err assert_empty err
assert_match(%r[/irb\.rb], out) assert_match(%r[/irb\.rb], out)
end end
def test_show_source_alias def test_show_source_alias
input = TestInputMethod.new([ out, err = execute_lines(
"$ 'IRB.conf'\n", "$ 'IRB.conf'\n",
]) conf: { COMMAND_ALIASES: { :'$' => :show_source } }
IRB.init_config(nil) )
IRB.conf[:COMMAND_ALIASES] = { :'$' => :show_source }
workspace = IRB::WorkSpace.new(Object.new)
IRB.conf[:VERBOSE] = false
irb = IRB::Irb.new(workspace, input)
IRB.conf[:MAIN_CONTEXT] = irb.context
irb.context.return_format = "=> %s\n"
out, err = capture_output do
irb.eval_input
end
assert_empty err assert_empty err
assert_match(%r[/irb\.rb], out) assert_match(%r[/irb\.rb], out)
end end
@ -623,79 +531,62 @@ module TestIRB
end end
end end
EOS EOS
input = TestInputMethod.new([
out, err = execute_lines(
"show_source 'TestIRB::ExtendCommand#show_source_test_method'\n", "show_source 'TestIRB::ExtendCommand#show_source_test_method'\n",
]) )
IRB.init_config(nil)
workspace = IRB::WorkSpace.new(self)
IRB.conf[:VERBOSE] = false
irb = IRB::Irb.new(workspace, input)
IRB.conf[:MAIN_CONTEXT] = irb.context
irb.context.return_format = "=> %s\n"
out, err = capture_output do
irb.eval_input
end
assert_empty err assert_empty err
assert_include(out, code) assert_include(out, code)
end end
def test_whereami def test_whereami
input = TestInputMethod.new([ out, err = execute_lines(
"whereami\n", "whereami\n",
]) )
IRB.init_config(nil)
workspace = IRB::WorkSpace.new(self)
IRB.conf[:VERBOSE] = false
irb = IRB::Irb.new(workspace, input)
IRB.conf[:MAIN_CONTEXT] = irb.context
irb.context.return_format = "=> %s\n"
out, err = capture_output do
irb.eval_input
end
assert_empty err assert_empty err
assert_match(/^From: .+ @ line \d+ :\n/, out) assert_match(/^From: .+ @ line \d+ :\n/, out)
end end
def test_whereami_alias def test_whereami_alias
input = TestInputMethod.new([ out, err = execute_lines(
"@\n", "@\n",
]) conf: { COMMAND_ALIASES: { :'@' => :whereami } }
IRB.init_config(nil) )
IRB.conf[:COMMAND_ALIASES] = { :'@' => :whereami }
workspace = IRB::WorkSpace.new(Object.new)
IRB.conf[:VERBOSE] = false
irb = IRB::Irb.new(workspace, input)
IRB.conf[:MAIN_CONTEXT] = irb.context
out, err = capture_output do
irb.eval_input
end
assert_empty err assert_empty err
assert_match(/^From: .+ @ line \d+ :\n/, out) assert_match(/^From: .+ @ line \d+ :\n/, out)
end end
def test_vars_with_aliases def test_vars_with_aliases
input = TestInputMethod.new([ @foo = "foo"
$bar = "bar"
out, err = execute_lines(
"@foo\n", "@foo\n",
"$bar\n", "$bar\n",
]) conf: { COMMAND_ALIASES: { :'$' => :show_source, :'@' => :whereami } }
IRB.init_config(nil) )
IRB.conf[:COMMAND_ALIASES] = {
:'@' => :whereami,
:'$' => :show_source,
}
main = Object.new
main.instance_variable_set(:@foo, "foo")
$bar = "bar"
workspace = IRB::WorkSpace.new(main)
IRB.conf[:VERBOSE] = false
irb = IRB::Irb.new(workspace, input)
IRB.conf[:MAIN_CONTEXT] = irb.context
out, err = capture_output do
irb.eval_input
end
assert_empty err assert_empty err
assert_match(/"foo"/, out) assert_match(/"foo"/, out)
assert_match(/"bar"/, out) assert_match(/"bar"/, out)
ensure
remove_instance_variable(:@foo)
$bar = nil
end
private
def execute_lines(*lines, conf: {}, main: self)
IRB.init_config(nil)
IRB.conf[:VERBOSE] = false
IRB.conf[:PROMPT_MODE] = :SIMPLE
IRB.conf.merge!(conf)
input = TestInputMethod.new(lines)
irb = IRB::Irb.new(IRB::WorkSpace.new(main), input)
irb.context.return_format = "=> %s\n"
IRB.conf[:MAIN_CONTEXT] = irb.context
capture_output do
irb.eval_input
end
end end
end end
end end