summaryrefslogtreecommitdiff
path: root/test/prism/api
diff options
context:
space:
mode:
Diffstat (limited to 'test/prism/api')
-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.rb41
3 files changed, 149 insertions, 8 deletions
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 ee8061c98c..c9a47c1a61 100644
--- a/test/prism/api/parse_test.rb
+++ b/test/prism/api/parse_test.rb
@@ -116,6 +116,15 @@ module Prism
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
@@ -133,10 +142,40 @@ module Prism
# Not supported version (too new)
assert_raise ArgumentError do
- Prism.parse("1 + 1", version: "3.5.0")
+ 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-05 04:33:56 +0000