summaryrefslogtreecommitdiff
path: root/test/prism/api
diff options
context:
space:
mode:
Diffstat (limited to 'test/prism/api')
-rw-r--r--test/prism/api/command_line_test.rb7
-rw-r--r--test/prism/api/freeze_test.rb65
-rw-r--r--test/prism/api/parse_stream_test.rb51
-rw-r--r--test/prism/api/parse_test.rb97
4 files changed, 211 insertions, 9 deletions
diff --git a/test/prism/api/command_line_test.rb b/test/prism/api/command_line_test.rb
index a313845ead..e53d18703a 100644
--- a/test/prism/api/command_line_test.rb
+++ b/test/prism/api/command_line_test.rb
@@ -52,6 +52,9 @@ module Prism
assert_kind_of CallNode, predicate
assert_equal :gets, predicate.name
+ arguments = predicate.arguments
+ assert arguments.contains_keywords?
+
arguments = predicate.arguments.arguments
assert_equal 2, arguments.length
assert_equal :$/, arguments.first.name
@@ -67,7 +70,7 @@ module Prism
end
def test_command_line_x_implicit
- result = Prism.parse_statement(<<~RUBY)
+ result = Prism.parse_statement(<<~RUBY, main_script: true)
#!/bin/bash
exit 1
@@ -90,7 +93,7 @@ module Prism
end
def test_command_line_x_implicit_fail
- result = Prism.parse(<<~RUBY)
+ result = Prism.parse(<<~RUBY, main_script: true)
#!/bin/bash
exit 1
RUBY
diff --git a/test/prism/api/freeze_test.rb b/test/prism/api/freeze_test.rb
new file mode 100644
index 0000000000..bf91792e69
--- /dev/null
+++ b/test/prism/api/freeze_test.rb
@@ -0,0 +1,65 @@
+# frozen_string_literal: true
+
+require_relative "../test_helper"
+
+module Prism
+ class FreezeTest < TestCase
+ def test_parse
+ assert_frozen(Prism.parse("1 + 2; %i{foo} + %i{bar}", freeze: true))
+ end
+
+ def test_offsets_usable
+ node = Prism.parse_statement("1 + 2", freeze: true)
+ assert_equal(1, node.start_line)
+ end
+
+ def test_lex
+ assert_frozen(Prism.lex("1 + 2; %i{foo} + %i{bar}", freeze: true))
+ end
+
+ def test_parse_lex
+ assert_frozen(Prism.parse_lex("1 + 2; %i{foo} + %i{bar}", freeze: true))
+ assert_frozen(Prism.parse_lex("# encoding: euc-jp\n%i{foo}", freeze: true))
+ end
+
+ def test_parse_comments
+ assert_frozen(Prism.parse_comments("# comment", freeze: true))
+ end
+
+ def test_parse_stream
+ assert_frozen(Prism.parse_stream(StringIO.new("1 + 2; %i{foo} + %i{bar}"), freeze: true))
+ end
+
+ if !ENV["PRISM_BUILD_MINIMAL"]
+ def test_dump
+ assert_frozen(Prism.dump("1 + 2; %i{foo} + %i{bar}", freeze: true))
+ end
+ end
+
+ private
+
+ def assert_frozen_each(value)
+ assert_predicate value, :frozen?
+
+ value.instance_variables.each do |name|
+ case (child = value.instance_variable_get(name))
+ when Array
+ child.each { |item| assert_frozen_each(item) }
+ when Hash
+ child.each { |key, item| assert_frozen_each(key); assert_frozen_each(item) }
+ else
+ assert_frozen_each(child)
+ end
+ end
+ end
+
+ if defined?(Ractor.shareable?)
+ def assert_frozen(value)
+ assert_frozen_each(value)
+ assert Ractor.shareable?(value), -> { binding.irb }
+ end
+ else
+ alias assert_frozen assert_frozen_each
+ end
+ end
+end
diff --git a/test/prism/api/parse_stream_test.rb b/test/prism/api/parse_stream_test.rb
index 1c068c617c..3bc86fbd61 100644
--- a/test/prism/api/parse_stream_test.rb
+++ b/test/prism/api/parse_stream_test.rb
@@ -30,16 +30,28 @@ module Prism
end
def test___END__
- io = StringIO.new("1 + 2\n3 + 4\n__END__\n5 + 6")
+ io = StringIO.new(<<~RUBY)
+ 1 + 2
+ 3 + 4
+ __END__
+ 5 + 6
+ RUBY
result = Prism.parse_stream(io)
assert result.success?
assert_equal 2, result.value.statements.body.length
- assert_equal "5 + 6", io.read
+ assert_equal "5 + 6\n", io.read
end
def test_false___END___in_string
- io = StringIO.new("1 + 2\n3 + 4\n\"\n__END__\n\"\n5 + 6")
+ io = StringIO.new(<<~RUBY)
+ 1 + 2
+ 3 + 4
+ "
+ __END__
+ "
+ 5 + 6
+ RUBY
result = Prism.parse_stream(io)
assert result.success?
@@ -47,7 +59,14 @@ module Prism
end
def test_false___END___in_regexp
- io = StringIO.new("1 + 2\n3 + 4\n/\n__END__\n/\n5 + 6")
+ io = StringIO.new(<<~RUBY)
+ 1 + 2
+ 3 + 4
+ /
+ __END__
+ /
+ 5 + 6
+ RUBY
result = Prism.parse_stream(io)
assert result.success?
@@ -55,7 +74,14 @@ module Prism
end
def test_false___END___in_list
- io = StringIO.new("1 + 2\n3 + 4\n%w[\n__END__\n]\n5 + 6")
+ io = StringIO.new(<<~RUBY)
+ 1 + 2
+ 3 + 4
+ %w[
+ __END__
+ ]
+ 5 + 6
+ RUBY
result = Prism.parse_stream(io)
assert result.success?
@@ -63,7 +89,14 @@ module Prism
end
def test_false___END___in_heredoc
- io = StringIO.new("1 + 2\n3 + 4\n<<-EOF\n__END__\nEOF\n5 + 6")
+ io = StringIO.new(<<~RUBY)
+ 1 + 2
+ 3 + 4
+ <<-EOF
+ __END__
+ EOF
+ 5 + 6
+ RUBY
result = Prism.parse_stream(io)
assert result.success?
@@ -71,7 +104,11 @@ module Prism
end
def test_nul_bytes
- io = StringIO.new("1 # \0\0\0 \n2 # \0\0\0\n3")
+ io = StringIO.new(<<~RUBY)
+ 1 # \0\0\0\t
+ 2 # \0\0\0
+ 3
+ RUBY
result = Prism.parse_stream(io)
assert result.success?
diff --git a/test/prism/api/parse_test.rb b/test/prism/api/parse_test.rb
index 6ad3829de0..c9a47c1a61 100644
--- a/test/prism/api/parse_test.rb
+++ b/test/prism/api/parse_test.rb
@@ -79,6 +79,103 @@ module Prism
end
end
+ def test_parse_directory
+ error = nil
+
+ begin
+ Prism.parse_file(__dir__)
+ rescue SystemCallError => error
+ end
+
+ assert_kind_of Errno::EISDIR, error
+ end
+
+ def test_partial_script
+ assert Prism.parse_failure?("break")
+ assert Prism.parse_success?("break", partial_script: true)
+
+ assert Prism.parse_failure?("next")
+ assert Prism.parse_success?("next", partial_script: true)
+
+ assert Prism.parse_failure?("redo")
+ assert Prism.parse_success?("redo", partial_script: true)
+
+ assert Prism.parse_failure?("yield")
+ assert Prism.parse_success?("yield", partial_script: true)
+ end
+
+ def test_version
+ assert Prism.parse_success?("1 + 1", version: "3.3")
+ assert Prism.parse_success?("1 + 1", version: "3.3.0")
+ assert Prism.parse_success?("1 + 1", version: "3.3.1")
+ assert Prism.parse_success?("1 + 1", version: "3.3.9")
+ assert Prism.parse_success?("1 + 1", version: "3.3.10")
+
+ assert Prism.parse_success?("1 + 1", version: "3.4")
+ assert Prism.parse_success?("1 + 1", version: "3.4.0")
+ assert Prism.parse_success?("1 + 1", version: "3.4.9")
+ assert Prism.parse_success?("1 + 1", version: "3.4.10")
+
+ assert Prism.parse_success?("1 + 1", version: "3.5")
+ assert Prism.parse_success?("1 + 1", version: "3.5.0")
+
+ assert Prism.parse_success?("1 + 1", version: "4.0")
+ assert Prism.parse_success?("1 + 1", version: "4.0.0")
+
+ assert Prism.parse_success?("1 + 1", version: "4.1")
+ assert Prism.parse_success?("1 + 1", version: "4.1.0")
+
+ assert Prism.parse_success?("1 + 1", version: "latest")
+
+ # Test edge case
+ error = assert_raise(ArgumentError) { Prism.parse("1 + 1", version: "latest2") }
+ assert_equal "invalid version: latest2", error.message
+
+ assert_raise ArgumentError do
+ Prism.parse("1 + 1", version: "3.3.a")
+ end
+
+ # Not supported version (too old)
+ assert_raise ArgumentError do
+ Prism.parse("1 + 1", version: "3.2.0")
+ end
+
+ # Not supported version (too new)
+ assert_raise ArgumentError do
+ Prism.parse("1 + 1", version: "3.6.0")
+ end
+ end
+
+ def test_version_current
+ if RUBY_VERSION >= "3.3"
+ assert Prism.parse_success?("1 + 1", version: "current")
+ else
+ assert_raise(CurrentVersionError) { Prism.parse_success?("1 + 1", version: "current") }
+ end
+ end
+
+ def test_nearest
+ assert Prism.parse_success?("1 + 1", version: "nearest")
+ end
+
+ def test_scopes
+ assert_kind_of Prism::CallNode, Prism.parse_statement("foo")
+ assert_kind_of Prism::LocalVariableReadNode, Prism.parse_statement("foo", scopes: [[:foo]])
+ assert_kind_of Prism::LocalVariableReadNode, Prism.parse_statement("foo", scopes: [Prism.scope(locals: [:foo])])
+
+ assert Prism.parse_failure?("foo(*)")
+ assert Prism.parse_success?("foo(*)", scopes: [Prism.scope(forwarding: [:*])])
+
+ assert Prism.parse_failure?("foo(**)")
+ assert Prism.parse_success?("foo(**)", scopes: [Prism.scope(forwarding: [:**])])
+
+ assert Prism.parse_failure?("foo(&)")
+ assert Prism.parse_success?("foo(&)", scopes: [Prism.scope(forwarding: [:&])])
+
+ assert Prism.parse_failure?("foo(...)")
+ assert Prism.parse_success?("foo(...)", scopes: [Prism.scope(forwarding: [:"..."])])
+ end
+
private
def find_source_file_node(program)
generated by cgit v1.2.3 (git 2.25.1) at 2026-06-06 20:23:35 +0000