mirror of
https://github.com/ruby/ruby.git
synced 2025-09-15 16:44:01 +02:00
74 lines
2 KiB
Ruby
74 lines
2 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require_relative "test_helper"
|
|
require "stringio"
|
|
|
|
module Prism
|
|
class ParseStreamTest < TestCase
|
|
def test_single_line
|
|
io = StringIO.new("1 + 2")
|
|
result = Prism.parse_stream(io)
|
|
|
|
assert result.success?
|
|
assert_kind_of Prism::CallNode, result.value.statements.body.first
|
|
end
|
|
|
|
def test_multi_line
|
|
io = StringIO.new("1 + 2\n3 + 4")
|
|
result = Prism.parse_stream(io)
|
|
|
|
assert result.success?
|
|
assert_kind_of Prism::CallNode, result.value.statements.body.first
|
|
assert_kind_of Prism::CallNode, result.value.statements.body.last
|
|
end
|
|
|
|
def test_multi_read
|
|
io = StringIO.new("a" * 4096 * 4)
|
|
result = Prism.parse_stream(io)
|
|
|
|
assert result.success?
|
|
assert_kind_of Prism::CallNode, result.value.statements.body.first
|
|
end
|
|
|
|
def test___END__
|
|
io = StringIO.new("1 + 2\n3 + 4\n__END__\n5 + 6")
|
|
result = Prism.parse_stream(io)
|
|
|
|
assert result.success?
|
|
assert_equal 2, result.value.statements.body.length
|
|
assert_equal "5 + 6", io.read
|
|
end
|
|
|
|
def test_false___END___in_string
|
|
io = StringIO.new("1 + 2\n3 + 4\n\"\n__END__\n\"\n5 + 6")
|
|
result = Prism.parse_stream(io)
|
|
|
|
assert result.success?
|
|
assert_equal 4, result.value.statements.body.length
|
|
end
|
|
|
|
def test_false___END___in_regexp
|
|
io = StringIO.new("1 + 2\n3 + 4\n/\n__END__\n/\n5 + 6")
|
|
result = Prism.parse_stream(io)
|
|
|
|
assert result.success?
|
|
assert_equal 4, result.value.statements.body.length
|
|
end
|
|
|
|
def test_false___END___in_list
|
|
io = StringIO.new("1 + 2\n3 + 4\n%w[\n__END__\n]\n5 + 6")
|
|
result = Prism.parse_stream(io)
|
|
|
|
assert result.success?
|
|
assert_equal 4, result.value.statements.body.length
|
|
end
|
|
|
|
def test_false___END___in_heredoc
|
|
io = StringIO.new("1 + 2\n3 + 4\n<<-EOF\n__END__\nEOF\n5 + 6")
|
|
result = Prism.parse_stream(io)
|
|
|
|
assert result.success?
|
|
assert_equal 4, result.value.statements.body.length
|
|
end
|
|
end
|
|
end
|