mirror of
https://github.com/ruby/ruby.git
synced 2025-08-24 13:34:17 +02:00
2796e4ece2
This PR implements the once node on interpolated regexes. There is a bug in Prism where the interpolated regex with the once flag only works when there is not a local variable so the test uses a "1". We'll need to fix that.
1140 lines
36 KiB
Ruby
1140 lines
36 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
# This file is organized to match itemization in https://github.com/ruby/prism/issues/1335
|
|
module Prism
|
|
class TestCompilePrism < Test::Unit::TestCase
|
|
# Subclass is used for tests which need it
|
|
class Subclass; end
|
|
############################################################################
|
|
# Literals #
|
|
############################################################################
|
|
|
|
def test_FalseNode
|
|
assert_prism_eval("false")
|
|
end
|
|
|
|
def test_FloatNode
|
|
assert_prism_eval("1.2")
|
|
assert_prism_eval("1.2e3")
|
|
assert_prism_eval("+1.2e+3")
|
|
assert_prism_eval("-1.2e-3")
|
|
end
|
|
|
|
def test_ImaginaryNode
|
|
assert_prism_eval("1i")
|
|
assert_prism_eval("+1.0i")
|
|
assert_prism_eval("1ri")
|
|
end
|
|
|
|
def test_IntegerNode
|
|
assert_prism_eval("1")
|
|
assert_prism_eval("+1")
|
|
assert_prism_eval("-1")
|
|
assert_prism_eval("0x10")
|
|
assert_prism_eval("0b10")
|
|
assert_prism_eval("0o10")
|
|
assert_prism_eval("010")
|
|
end
|
|
|
|
def test_NilNode
|
|
assert_prism_eval("nil")
|
|
end
|
|
|
|
def test_RationalNode
|
|
assert_prism_eval("1.2r")
|
|
assert_prism_eval("+1.2r")
|
|
end
|
|
|
|
def test_SelfNode
|
|
assert_prism_eval("self")
|
|
end
|
|
|
|
def test_SourceEncodingNode
|
|
assert_prism_eval("__ENCODING__")
|
|
end
|
|
|
|
def test_SourceFileNode
|
|
assert_prism_eval("__FILE__")
|
|
end
|
|
|
|
def test_SourceLineNode
|
|
ruby_eval = RubyVM::InstructionSequence.compile("__LINE__").eval
|
|
prism_eval = RubyVM::InstructionSequence.compile_prism("__LINE__").eval
|
|
|
|
assert_equal ruby_eval, prism_eval
|
|
end
|
|
|
|
def test_TrueNode
|
|
assert_prism_eval("true")
|
|
end
|
|
|
|
############################################################################
|
|
# Reads #
|
|
############################################################################
|
|
|
|
def test_BackReferenceReadNode
|
|
assert_prism_eval("$+")
|
|
end
|
|
|
|
def test_ClassVariableReadNode
|
|
assert_prism_eval("class Prism::TestCompilePrism; @@pit = 1; @@pit; end")
|
|
end
|
|
|
|
def test_ConstantPathNode
|
|
assert_prism_eval("Prism::TestCompilePrism")
|
|
end
|
|
|
|
def test_ConstantReadNode
|
|
assert_prism_eval("Prism")
|
|
end
|
|
|
|
def test_DefinedNode
|
|
assert_prism_eval("defined? nil")
|
|
assert_prism_eval("defined? self")
|
|
assert_prism_eval("defined? true")
|
|
assert_prism_eval("defined? false")
|
|
assert_prism_eval("defined? [A, B, C]")
|
|
assert_prism_eval("defined? 'str'")
|
|
assert_prism_eval("defined? a && b")
|
|
assert_prism_eval("defined? a || b")
|
|
|
|
assert_prism_eval("defined? @a")
|
|
assert_prism_eval("defined? $a")
|
|
assert_prism_eval("defined? @@a")
|
|
assert_prism_eval("defined? A")
|
|
assert_prism_eval("defined? ::A")
|
|
assert_prism_eval("defined? A::B")
|
|
assert_prism_eval("defined? A::B::C")
|
|
assert_prism_eval("defined? yield")
|
|
assert_prism_eval("defined? super")
|
|
|
|
assert_prism_eval("defined? X = 1")
|
|
assert_prism_eval("defined? X *= 1")
|
|
assert_prism_eval("defined? X /= 1")
|
|
assert_prism_eval("defined? X &= 1")
|
|
assert_prism_eval("defined? X ||= 1")
|
|
|
|
assert_prism_eval("defined? $X = 1")
|
|
assert_prism_eval("defined? $X *= 1")
|
|
assert_prism_eval("defined? $X /= 1")
|
|
assert_prism_eval("defined? $X &= 1")
|
|
assert_prism_eval("defined? $X ||= 1")
|
|
|
|
assert_prism_eval("defined? @@X = 1")
|
|
assert_prism_eval("defined? @@X *= 1")
|
|
assert_prism_eval("defined? @@X /= 1")
|
|
assert_prism_eval("defined? @@X &= 1")
|
|
assert_prism_eval("defined? @@X ||= 1")
|
|
|
|
assert_prism_eval("defined? @X = 1")
|
|
assert_prism_eval("defined? @X *= 1")
|
|
assert_prism_eval("defined? @X /= 1")
|
|
assert_prism_eval("defined? @X &= 1")
|
|
assert_prism_eval("defined? @X ||= 1")
|
|
|
|
assert_prism_eval("x = 1; defined? x = 1")
|
|
assert_prism_eval("x = 1; defined? x *= 1")
|
|
assert_prism_eval("x = 1; defined? x /= 1")
|
|
assert_prism_eval("x = 1; defined? x &= 1")
|
|
assert_prism_eval("x = 1; defined? x ||= 1")
|
|
|
|
assert_prism_eval("if defined? A; end")
|
|
end
|
|
|
|
def test_GlobalVariableReadNode
|
|
assert_prism_eval("$pit = 1; $pit")
|
|
end
|
|
|
|
def test_InstanceVariableReadNode
|
|
assert_prism_eval("class Prism::TestCompilePrism; @pit = 1; @pit; end")
|
|
end
|
|
|
|
def test_LocalVariableReadNode
|
|
assert_prism_eval("pit = 1; pit")
|
|
end
|
|
|
|
def test_NumberedReferenceReadNode
|
|
assert_prism_eval("$1")
|
|
assert_prism_eval("$99999")
|
|
end
|
|
|
|
############################################################################
|
|
# Writes #
|
|
############################################################################
|
|
|
|
def test_ClassVariableAndWriteNode
|
|
assert_prism_eval("class Prism::TestCompilePrism; @@pit = 0; @@pit &&= 1; end")
|
|
end
|
|
|
|
def test_ClassVariableOperatorWriteNode
|
|
assert_prism_eval("class Prism::TestCompilePrism; @@pit = 0; @@pit += 1; end")
|
|
end
|
|
|
|
def test_ClassVariableOrWriteNode
|
|
assert_prism_eval("class Prism::TestCompilePrism; @@pit = 1; @@pit ||= 0; end")
|
|
assert_prism_eval("class Prism::TestCompilePrism; @@pit = nil; @@pit ||= 1; end")
|
|
end
|
|
|
|
def test_ClassVariableWriteNode
|
|
assert_prism_eval("class Prism::TestCompilePrism; @@pit = 1; end")
|
|
end
|
|
|
|
def test_ConstantAndWriteNode
|
|
assert_prism_eval("Constant = 1; Constant &&= 1")
|
|
end
|
|
|
|
def test_ConstantOperatorWriteNode
|
|
assert_prism_eval("Constant = 1; Constant += 1")
|
|
end
|
|
|
|
def test_ConstantOrWriteNode
|
|
assert_prism_eval("Constant = 1; Constant ||= 1")
|
|
end
|
|
|
|
def test_ConstantWriteNode
|
|
# We don't call assert_prism_eval directly in this case becuase we
|
|
# don't want to assign the constant mutliple times if we run
|
|
# with `--repeat-count`
|
|
# Instead, we eval manually here, and remove the constant to
|
|
constant_name = "YCT"
|
|
source = "#{constant_name} = 1"
|
|
prism_eval = RubyVM::InstructionSequence.compile_prism(source).eval
|
|
assert_equal prism_eval, 1
|
|
Object.send(:remove_const, constant_name)
|
|
end
|
|
|
|
def test_ConstantPathWriteNode
|
|
assert_prism_eval("Prism::CPWN = 1")
|
|
assert_prism_eval("::CPWN = 1")
|
|
end
|
|
|
|
def test_ConstantPathAndWriteNode
|
|
assert_prism_eval("Prism::CPAWN = 1; Prism::CPAWN &&= 2")
|
|
assert_prism_eval("Prism::CPAWN &&= 1")
|
|
end
|
|
|
|
def test_ConstantPathOrWriteNode
|
|
assert_prism_eval("Prism::CPOrWN = nil; Prism::CPOrWN ||= 1")
|
|
assert_prism_eval("Prism::CPOrWN ||= 1")
|
|
end
|
|
|
|
def test_ConstantPathOperatorWriteNode
|
|
assert_prism_eval("Prism::CPOWN = 0; Prism::CPOWN += 1")
|
|
end
|
|
|
|
def test_GlobalVariableAndWriteNode
|
|
assert_prism_eval("$pit = 0; $pit &&= 1")
|
|
end
|
|
|
|
def test_GlobalVariableOperatorWriteNode
|
|
assert_prism_eval("$pit = 0; $pit += 1")
|
|
end
|
|
|
|
def test_GlobalVariableOrWriteNode
|
|
assert_prism_eval("$pit ||= 1")
|
|
end
|
|
|
|
def test_GlobalVariableWriteNode
|
|
assert_prism_eval("$pit = 1")
|
|
end
|
|
|
|
def test_InstanceVariableAndWriteNode
|
|
assert_prism_eval("@pit = 0; @pit &&= 1")
|
|
end
|
|
|
|
def test_InstanceVariableOperatorWriteNode
|
|
assert_prism_eval("@pit = 0; @pit += 1")
|
|
end
|
|
|
|
def test_InstanceVariableOrWriteNode
|
|
assert_prism_eval("@pit ||= 1")
|
|
end
|
|
|
|
def test_InstanceVariableWriteNode
|
|
assert_prism_eval("class Prism::TestCompilePrism; @pit = 1; end")
|
|
end
|
|
|
|
def test_LocalVariableAndWriteNode
|
|
assert_prism_eval("pit = 0; pit &&= 1")
|
|
end
|
|
|
|
def test_LocalVariableOperatorWriteNode
|
|
assert_prism_eval("pit = 0; pit += 1")
|
|
end
|
|
|
|
def test_LocalVariableOrWriteNode
|
|
assert_prism_eval("pit ||= 1")
|
|
end
|
|
|
|
def test_LocalVariableWriteNode
|
|
assert_prism_eval("pit = 1")
|
|
end
|
|
|
|
def test_MatchWriteNode
|
|
assert_prism_eval("/(?<foo>bar)(?<baz>bar>)/ =~ 'barbar'")
|
|
assert_prism_eval("/(?<foo>bar)/ =~ 'barbar'")
|
|
end
|
|
|
|
############################################################################
|
|
# Multi-writes #
|
|
############################################################################
|
|
|
|
def test_ClassVariableTargetNode
|
|
assert_prism_eval("class Prism::TestCompilePrism; @@pit, @@pit1 = 1; end")
|
|
end
|
|
|
|
def test_ConstantTargetNode
|
|
# We don't call assert_prism_eval directly in this case becuase we
|
|
# don't want to assign the constant mutliple times if we run
|
|
# with `--repeat-count`
|
|
# Instead, we eval manually here, and remove the constant to
|
|
constant_names = ["YCT", "YCT2"]
|
|
source = "#{constant_names.join(",")} = 1"
|
|
prism_eval = RubyVM::InstructionSequence.compile_prism(source).eval
|
|
assert_equal prism_eval, 1
|
|
constant_names.map { |name|
|
|
Object.send(:remove_const, name)
|
|
}
|
|
end
|
|
|
|
def test_ConstantPathTargetNode
|
|
verbose = $VERBOSE
|
|
# Create some temporary nested constants
|
|
Object.send(:const_set, "MyFoo", Object)
|
|
Object.const_get("MyFoo").send(:const_set, "Bar", Object)
|
|
|
|
constant_names = ["MyBar", "MyFoo::Bar", "MyFoo::Bar::Baz"]
|
|
source = "#{constant_names.join(",")} = Object"
|
|
iseq = RubyVM::InstructionSequence.compile_prism(source)
|
|
$VERBOSE = nil
|
|
prism_eval = iseq.eval
|
|
$VERBOSE = verbose
|
|
assert_equal prism_eval, Object
|
|
ensure
|
|
## Teardown temp constants
|
|
Object.const_get("MyFoo").send(:remove_const, "Bar")
|
|
Object.send(:remove_const, "MyFoo")
|
|
Object.send(:remove_const, "MyBar")
|
|
$VERBOSE = verbose
|
|
end
|
|
|
|
def test_GlobalVariableTargetNode
|
|
assert_prism_eval("$pit, $pit1 = 1")
|
|
end
|
|
|
|
def test_InstanceVariableTargetNode
|
|
assert_prism_eval("class Prism::TestCompilePrism; @pit, @pit1 = 1; end")
|
|
end
|
|
|
|
def test_LocalVariableTargetNode
|
|
assert_prism_eval("pit, pit1 = 1")
|
|
end
|
|
|
|
def test_MultiTargetNode
|
|
assert_prism_eval("a, (b, c) = [1, 2, 3]")
|
|
assert_prism_eval("a, (b, c) = [1, 2, 3]; a")
|
|
assert_prism_eval("a, (b, c) = [1, 2, 3]; b")
|
|
assert_prism_eval("a, (b, c) = [1, 2, 3]; c")
|
|
assert_prism_eval("a, (b, c) = [1, [2, 3]]; c")
|
|
assert_prism_eval("(a, (b, c, d, e), f, g), h = [1, [2, 3]], 4, 5, [6, 7]; c")
|
|
end
|
|
|
|
def test_MultiWriteNode
|
|
assert_prism_eval("foo, bar = [1, 2]")
|
|
assert_prism_eval("foo, *, bar = [1, 2]")
|
|
assert_prism_eval("foo, bar = 1, 2")
|
|
assert_prism_eval("foo, *, bar = 1, 2")
|
|
assert_prism_eval("foo, *, bar = 1, 2, 3, 4")
|
|
assert_prism_eval("a, b, *, d = 1, 2, 3, 4")
|
|
assert_prism_eval("a, b, *, d = 1, 2")
|
|
assert_prism_eval("(a, b), *, c = [1, 3], 4, 5")
|
|
assert_prism_eval("(a, b), *, c = [1, 3], 4, 5; a")
|
|
assert_prism_eval("(a, b), *, c = [1, 3], 4, 5; b")
|
|
assert_prism_eval("(a, b), *, c = [1, 3], 4, 5; c")
|
|
assert_prism_eval("a, *, (c, d) = [1, 3], 4, 5; a")
|
|
assert_prism_eval("a, *, (c, d) = [1, 3], 4, 5; c")
|
|
assert_prism_eval("(a, b, c), *, (d, e) = [1, 3], 4, 5, [6, 7]")
|
|
assert_prism_eval("(a, b, c), *, (d, e) = [1, 3], 4, 5, [6, 7]; b")
|
|
assert_prism_eval("(a, b, c), *, (d, e) = [1, 3], 4, 5, [6, 7]; d")
|
|
assert_prism_eval("((a, *, b), *, (c, *, (d, *, e, f, g))), *, ((h, i, *, j), *, (k, l, m, *, n, o, p), q, r) = 1; a")
|
|
end
|
|
|
|
############################################################################
|
|
# String-likes #
|
|
############################################################################
|
|
|
|
def test_EmbeddedStatementsNode
|
|
assert_prism_eval('"foo #{to_s} baz"')
|
|
end
|
|
|
|
def test_EmbeddedVariableNode
|
|
assert_prism_eval('class Prism::TestCompilePrism; @pit = 1; "#@pit"; end')
|
|
assert_prism_eval('class Prism::TestCompilePrism; @@pit = 1; "#@@pit"; end')
|
|
assert_prism_eval('$pit = 1; "#$pit"')
|
|
end
|
|
|
|
def test_InterpolatedMatchLastLineNode
|
|
assert_prism_eval("$pit = '.oo'; if /\#$pit/mix; end")
|
|
end
|
|
|
|
def test_InterpolatedRegularExpressionNode
|
|
assert_prism_eval('$pit = 1; /1 #$pit 1/')
|
|
assert_prism_eval('$pit = 1; /#$pit/i')
|
|
assert_prism_eval('/1 #{1 + 2} 1/')
|
|
assert_prism_eval('/1 #{"2"} #{1 + 2} 1/')
|
|
end
|
|
|
|
def test_InterpolatedStringNode
|
|
assert_prism_eval('$pit = 1; "1 #$pit 1"')
|
|
assert_prism_eval('"1 #{1 + 2} 1"')
|
|
end
|
|
|
|
def test_InterpolatedSymbolNode
|
|
assert_prism_eval('$pit = 1; :"1 #$pit 1"')
|
|
assert_prism_eval(':"1 #{1 + 2} 1"')
|
|
end
|
|
|
|
def test_InterpolatedXStringNode
|
|
assert_prism_eval('`echo #{1}`')
|
|
assert_prism_eval('`printf #{"100"}`')
|
|
end
|
|
|
|
def test_MatchLastLineNode
|
|
assert_prism_eval("if /foo/; end")
|
|
assert_prism_eval("if /foo/i; end")
|
|
assert_prism_eval("if /foo/x; end")
|
|
assert_prism_eval("if /foo/m; end")
|
|
assert_prism_eval("if /foo/im; end")
|
|
assert_prism_eval("if /foo/mx; end")
|
|
assert_prism_eval("if /foo/xi; end")
|
|
assert_prism_eval("if /foo/ixm; end")
|
|
end
|
|
|
|
def test_RegularExpressionNode
|
|
assert_prism_eval('/pit/')
|
|
assert_prism_eval('/pit/i')
|
|
assert_prism_eval('/pit/x')
|
|
assert_prism_eval('/pit/m')
|
|
assert_prism_eval('/pit/im')
|
|
assert_prism_eval('/pit/mx')
|
|
assert_prism_eval('/pit/xi')
|
|
assert_prism_eval('/pit/ixm')
|
|
|
|
assert_prism_eval('/pit/u')
|
|
assert_prism_eval('/pit/e')
|
|
assert_prism_eval('/pit/s')
|
|
assert_prism_eval('/pit/n')
|
|
|
|
assert_prism_eval('/pit/me')
|
|
assert_prism_eval('/pit/ne')
|
|
|
|
assert_prism_eval('2.times.map { /#{1}/o }')
|
|
end
|
|
|
|
def test_StringConcatNode
|
|
assert_prism_eval('"Prism" "::" "TestCompilePrism"')
|
|
end
|
|
|
|
def test_StringNode
|
|
assert_prism_eval('"pit"')
|
|
end
|
|
|
|
def test_SymbolNode
|
|
assert_prism_eval(":pit")
|
|
end
|
|
|
|
def test_XStringNode
|
|
assert_prism_eval(<<~RUBY)
|
|
class Prism::TestCompilePrism
|
|
def self.`(command) = command * 2
|
|
`pit`
|
|
end
|
|
RUBY
|
|
end
|
|
|
|
############################################################################
|
|
# Structures #
|
|
############################################################################
|
|
|
|
def test_ArrayNode
|
|
assert_prism_eval("[]")
|
|
assert_prism_eval("[1, 2, 3]")
|
|
assert_prism_eval("%i[foo bar baz]")
|
|
assert_prism_eval("%w[foo bar baz]")
|
|
end
|
|
|
|
def test_AssocNode
|
|
assert_prism_eval("{ foo: :bar }")
|
|
end
|
|
|
|
def test_AssocSplatNode
|
|
assert_prism_eval("foo = { a: 1 }; { **foo }")
|
|
assert_prism_eval("foo = { a: 1 }; bar = foo; { **foo, b: 2, **bar, c: 3 }")
|
|
assert_prism_eval("foo = { a: 1 }; { b: 2, **foo, c: 3}")
|
|
end
|
|
|
|
def test_HashNode
|
|
assert_prism_eval("{}")
|
|
assert_prism_eval("{ a: :a }")
|
|
assert_prism_eval("{ a: :a, b: :b }")
|
|
assert_prism_eval("a = 1; { a: a }")
|
|
assert_prism_eval("a = 1; { a: }")
|
|
assert_prism_eval("{ to_s: }")
|
|
assert_prism_eval("{ Prism: }")
|
|
assert_prism_eval("[ Prism: [:b, :c]]")
|
|
end
|
|
|
|
def test_ImplicitNode
|
|
assert_prism_eval("{ to_s: }")
|
|
end
|
|
|
|
def test_RangeNode
|
|
assert_prism_eval("1..2")
|
|
assert_prism_eval("1...2")
|
|
assert_prism_eval("..2")
|
|
assert_prism_eval("...2")
|
|
assert_prism_eval("1..")
|
|
assert_prism_eval("1...")
|
|
end
|
|
|
|
def test_SplatNode
|
|
assert_prism_eval("*b = []")
|
|
end
|
|
|
|
############################################################################
|
|
# Jumps #
|
|
############################################################################
|
|
|
|
def test_AndNode
|
|
assert_prism_eval("true && 1")
|
|
assert_prism_eval("false && 1")
|
|
end
|
|
|
|
def test_CaseNode
|
|
assert_prism_eval("case :a; when :a; 1; else; 2; end")
|
|
assert_prism_eval("case :a; when :b; 1; else; 2; end")
|
|
assert_prism_eval("case :a; when :a; 1; else; 2; end")
|
|
assert_prism_eval("case :a; when :a; end")
|
|
assert_prism_eval("case :a; when :b, :c; end")
|
|
assert_prism_eval("case; when :a; end")
|
|
assert_prism_eval("case; when :a, :b; 1; else; 2 end")
|
|
assert_prism_eval("case :a; when :b; else; end")
|
|
assert_prism_eval("b = 1; case :a; when b; else; end")
|
|
end
|
|
|
|
def test_ElseNode
|
|
assert_prism_eval("if false; 0; else; 1; end")
|
|
assert_prism_eval("if true; 0; else; 1; end")
|
|
assert_prism_eval("true ? 1 : 0")
|
|
assert_prism_eval("false ? 0 : 1")
|
|
end
|
|
|
|
def test_FlipFlopNode
|
|
assert_prism_eval("not (1 == 1) .. (2 == 2)")
|
|
assert_prism_eval("not (1 == 1) ... (2 == 2)")
|
|
end
|
|
|
|
def test_IfNode
|
|
assert_prism_eval("if true; 1; end")
|
|
assert_prism_eval("1 if true")
|
|
assert_prism_eval('a = b = 1; if a..b; end')
|
|
assert_prism_eval('if "a".."b"; end')
|
|
assert_prism_eval('if "a"..; end')
|
|
assert_prism_eval('if .."b"; end')
|
|
assert_prism_eval('if ..1; end')
|
|
assert_prism_eval('if 1..; end')
|
|
assert_prism_eval('if 1..2; end')
|
|
end
|
|
|
|
def test_OrNode
|
|
assert_prism_eval("true || 1")
|
|
assert_prism_eval("false || 1")
|
|
end
|
|
|
|
def test_UnlessNode
|
|
assert_prism_eval("1 unless true")
|
|
assert_prism_eval("1 unless false")
|
|
assert_prism_eval("unless true; 1; end")
|
|
assert_prism_eval("unless false; 1; end")
|
|
end
|
|
|
|
def test_UntilNode
|
|
assert_prism_eval("a = 0; until a == 1; a = a + 1; end")
|
|
end
|
|
|
|
def test_WhileNode
|
|
assert_prism_eval("a = 0; while a != 1; a = a + 1; end")
|
|
end
|
|
|
|
def test_ForNode
|
|
assert_prism_eval("for i in [1,2] do; i; end")
|
|
assert_prism_eval("for @i in [1,2] do; @i; end")
|
|
assert_prism_eval("for $i in [1,2] do; $i; end")
|
|
|
|
# TODO: multiple assignment in ForNode index
|
|
#assert_prism_eval("for i, j in {a: 'b'} do; i; j; end")
|
|
end
|
|
|
|
############################################################################
|
|
# Throws #
|
|
############################################################################
|
|
|
|
def test_BeginNode
|
|
assert_prism_eval("begin; 1; end")
|
|
assert_prism_eval("begin; end; 1")
|
|
end
|
|
|
|
def test_BreakNode
|
|
assert_prism_eval("while true; break; end")
|
|
assert_prism_eval("while true; break 1; end")
|
|
end
|
|
|
|
def test_EnsureNode
|
|
assert_prism_eval("begin; 1; ensure; 2; end")
|
|
assert_prism_eval("begin; 1; begin; 3; ensure; 4; end; ensure; 2; end")
|
|
end
|
|
|
|
def test_NextNode
|
|
# TODO:
|
|
# assert_prism_eval("2.times do |i|; next if i == 1; end")
|
|
end
|
|
|
|
def test_RedoNode
|
|
# TODO:
|
|
# assert_prism_eval(<<-CODE
|
|
# counter = 0
|
|
|
|
# 5.times do |i|
|
|
# counter += 1
|
|
# if i == 2 && counter < 3
|
|
# redo
|
|
# end
|
|
# end
|
|
# CODE
|
|
# )
|
|
end
|
|
|
|
def test_ReturnNode
|
|
assert_prism_eval("def return_node; return 1; end")
|
|
end
|
|
|
|
############################################################################
|
|
# Scopes/statements #
|
|
############################################################################
|
|
|
|
def test_BlockNode
|
|
assert_prism_eval("[1, 2, 3].each { |num| num }")
|
|
|
|
assert_prism_eval("[].tap { _1 }")
|
|
end
|
|
|
|
def test_ClassNode
|
|
assert_prism_eval("class PrismClassA; end")
|
|
assert_prism_eval("class PrismClassA; end; class PrismClassB < PrismClassA; end")
|
|
assert_prism_eval("class PrismClassA; end; class PrismClassA::PrismClassC; end")
|
|
assert_prism_eval(<<-HERE
|
|
class PrismClassA; end
|
|
class PrismClassA::PrismClassC; end
|
|
class PrismClassB; end
|
|
class PrismClassB::PrismClassD < PrismClassA::PrismClassC; end
|
|
HERE
|
|
)
|
|
end
|
|
|
|
# Many of these tests are versions of tests at bootstraptest/test_method.rb
|
|
def test_DefNode
|
|
assert_prism_eval("def prism_test_def_node; end")
|
|
assert_prism_eval("a = Object.new; def a.prism_singleton; :ok; end; a.prism_singleton")
|
|
assert_prism_eval("def self.prism_test_def_node() 1 end; prism_test_def_node()")
|
|
assert_prism_eval("def self.prism_test_def_node(a,b) [a, b] end; prism_test_def_node(1,2)")
|
|
assert_prism_eval("def self.prism_test_def_node(a,x=7,y=1) x end; prism_test_def_node(7,1)")
|
|
|
|
# rest argument
|
|
assert_prism_eval("def self.prism_test_def_node(*a) a end; prism_test_def_node().inspect")
|
|
assert_prism_eval("def self.prism_test_def_node(*a) a end; prism_test_def_node(1).inspect")
|
|
assert_prism_eval("def self.prism_test_def_node(x,y,*a) a end; prism_test_def_node(7,7,1,2).inspect")
|
|
assert_prism_eval("def self.prism_test_def_node(x,y=7,*a) a end; prism_test_def_node(7).inspect")
|
|
assert_prism_eval("def self.prism_test_def_node(x,y,z=7,*a) a end; prism_test_def_node(7,7).inspect")
|
|
assert_prism_eval("def self.prism_test_def_node(x,y,z=7,zz=7,*a) a end; prism_test_def_node(7,7,7).inspect")
|
|
|
|
# block argument
|
|
assert_prism_eval("def self.prism_test_def_node(&block) block end; prism_test_def_node{}.class")
|
|
assert_prism_eval("def self.prism_test_def_node(&block) block end; prism_test_def_node().inspect")
|
|
assert_prism_eval("def self.prism_test_def_node(a,b=7,*c,&block) b end; prism_test_def_node(7,1).inspect")
|
|
assert_prism_eval("def self.prism_test_def_node(a,b=7,*c,&block) c end; prism_test_def_node(7,7,1).inspect")
|
|
|
|
# splat
|
|
assert_prism_eval("def self.prism_test_def_node(a) a end; prism_test_def_node(*[1])")
|
|
assert_prism_eval("def self.prism_test_def_node(x,a) a end; prism_test_def_node(7,*[1])")
|
|
assert_prism_eval("def self.prism_test_def_node(x,y,a) a end; prism_test_def_node(7,7,*[1])")
|
|
assert_prism_eval("def self.prism_test_def_node(x,y,a,b,c) a end; prism_test_def_node(7,7,*[1,7,7])")
|
|
|
|
# recursive call
|
|
assert_prism_eval("def self.prism_test_def_node(n) n == 0 ? 1 : prism_test_def_node(n-1) end; prism_test_def_node(5)")
|
|
|
|
# instance method
|
|
assert_prism_eval("class PrismTestDefNode; def prism_test_def_node() 1 end end; PrismTestDefNode.new.prism_test_def_node")
|
|
assert_prism_eval("class PrismTestDefNode; def prism_test_def_node(*a) a end end; PrismTestDefNode.new.prism_test_def_node(1).inspect")
|
|
|
|
# block argument
|
|
assert_prism_eval(<<-CODE
|
|
def self.prism_test_def_node(&block) prism_test_def_node2(&block) end
|
|
def self.prism_test_def_node2() yield 1 end
|
|
prism_test_def_node2 {|a| a }
|
|
CODE
|
|
)
|
|
end
|
|
|
|
def test_LambdaNode
|
|
assert_prism_eval("-> { to_s }.call")
|
|
end
|
|
|
|
def test_ModuleNode
|
|
assert_prism_eval("module M; end")
|
|
assert_prism_eval("module M::N; end")
|
|
assert_prism_eval("module ::O; end")
|
|
end
|
|
|
|
def test_ParenthesesNode
|
|
assert_prism_eval("()")
|
|
assert_prism_eval("(1)")
|
|
end
|
|
|
|
def test_PreExecutionNode
|
|
# BEGIN {} must be defined at the top level, so we need to manually
|
|
# call the evals here instead of calling `assert_prism_eval`
|
|
ruby_eval = RubyVM::InstructionSequence.compile("BEGIN { a = 1 }; 2").eval
|
|
prism_eval = RubyVM::InstructionSequence.compile_prism("BEGIN { a = 1 }; 2").eval
|
|
assert_equal ruby_eval, prism_eval
|
|
|
|
ruby_eval = RubyVM::InstructionSequence.compile("b = 2; BEGIN { a = 1 }; a + b").eval
|
|
prism_eval = RubyVM::InstructionSequence.compile_prism("b = 2; BEGIN { a = 1 }; a + b").eval
|
|
assert_equal ruby_eval, prism_eval
|
|
end
|
|
|
|
def test_PostExecutionNode
|
|
assert_prism_eval("END { 1 }")
|
|
assert_prism_eval("END { @b }; @b = 1")
|
|
assert_prism_eval("END { @b; 0 }; @b = 1")
|
|
end
|
|
|
|
def test_ProgramNode
|
|
assert_prism_eval("")
|
|
assert_prism_eval("1")
|
|
end
|
|
|
|
def test_SingletonClassNode
|
|
assert_prism_eval("class << self; end")
|
|
end
|
|
|
|
def test_StatementsNode
|
|
assert_prism_eval("1")
|
|
end
|
|
|
|
def test_YieldNode
|
|
assert_prism_eval("def prism_test_yield_node; yield; end")
|
|
assert_prism_eval("def prism_test_yield_node; yield 1, 2; end")
|
|
|
|
# Test case where there's a call directly after the yield call
|
|
assert_prism_eval("def prism_test_yield_node; yield; 1; end")
|
|
assert_prism_eval("def prism_test_yield_node; yield 1, 2; 1; end")
|
|
end
|
|
|
|
############################################################################
|
|
# Calls / arugments #
|
|
############################################################################
|
|
|
|
def test_ArgumentsNode
|
|
# assert_prism_eval("[].push 1")
|
|
end
|
|
|
|
def test_BlockArgumentNode
|
|
assert_prism_eval("1.then(&:to_s)")
|
|
end
|
|
|
|
def test_BlockLocalVariableNode
|
|
assert_prism_eval(<<-CODE
|
|
pm_var = "outer scope variable"
|
|
|
|
1.times { |;pm_var| pm_var = "inner scope variable"; pm_var }
|
|
CODE
|
|
)
|
|
|
|
assert_prism_eval(<<-CODE
|
|
pm_var = "outer scope variable"
|
|
|
|
1.times { |;pm_var| pm_var = "inner scope variable"; pm_var }
|
|
pm_var
|
|
CODE
|
|
)
|
|
end
|
|
|
|
def test_CallNode
|
|
assert_prism_eval("to_s")
|
|
|
|
# with arguments
|
|
assert_prism_eval("eval '1'")
|
|
|
|
# with arguments and popped
|
|
assert_prism_eval("eval '1'; 1")
|
|
end
|
|
|
|
def test_CallAndWriteNode
|
|
assert_prism_eval(<<-CODE
|
|
class PrismTestSubclass; end
|
|
def PrismTestSubclass.test_call_and_write_node; end;
|
|
PrismTestSubclass.test_call_and_write_node &&= 1
|
|
CODE
|
|
)
|
|
|
|
assert_prism_eval(<<-CODE
|
|
def PrismTestSubclass.test_call_and_write_node
|
|
"str"
|
|
end
|
|
def PrismTestSubclass.test_call_and_write_node=(val)
|
|
val
|
|
end
|
|
PrismTestSubclass.test_call_and_write_node &&= 1
|
|
CODE
|
|
)
|
|
|
|
assert_prism_eval(<<-CODE
|
|
def self.test_call_and_write_node; end;
|
|
self.test_call_and_write_node &&= 1
|
|
CODE
|
|
)
|
|
|
|
assert_prism_eval(<<-CODE
|
|
def self.test_call_and_write_node
|
|
"str"
|
|
end
|
|
def self.test_call_and_write_node=(val)
|
|
val
|
|
end
|
|
self.test_call_and_write_node &&= 1
|
|
CODE
|
|
)
|
|
end
|
|
|
|
def test_CallOrWriteNode
|
|
assert_prism_eval(<<-CODE
|
|
class PrismTestSubclass; end
|
|
def PrismTestSubclass.test_call_or_write_node; end;
|
|
def PrismTestSubclass.test_call_or_write_node=(val)
|
|
val
|
|
end
|
|
PrismTestSubclass.test_call_or_write_node ||= 1
|
|
CODE
|
|
)
|
|
|
|
assert_prism_eval(<<-CODE
|
|
def PrismTestSubclass.test_call_or_write_node
|
|
"str"
|
|
end
|
|
PrismTestSubclass.test_call_or_write_node ||= 1
|
|
CODE
|
|
)
|
|
|
|
assert_prism_eval(<<-CODE
|
|
def self.test_call_or_write_node; end;
|
|
def self.test_call_or_write_node=(val)
|
|
val
|
|
end
|
|
self.test_call_or_write_node ||= 1
|
|
CODE
|
|
)
|
|
|
|
assert_prism_eval(<<-CODE
|
|
def self.test_call_or_write_node
|
|
"str"
|
|
end
|
|
self.test_call_or_write_node ||= 1
|
|
CODE
|
|
)
|
|
end
|
|
|
|
def test_CallOperatorWriteNode
|
|
assert_prism_eval(<<-CODE
|
|
class PrismTestSubclass; end
|
|
def PrismTestSubclass.test_call_operator_write_node
|
|
2
|
|
end
|
|
def PrismTestSubclass.test_call_operator_write_node=(val)
|
|
val
|
|
end
|
|
PrismTestSubclass.test_call_operator_write_node += 1
|
|
CODE
|
|
)
|
|
end
|
|
|
|
def test_ForwardingArgumentsNode
|
|
# http://ci.rvm.jp/results/trunk-iseq_binary@ruby-sp2-docker/4779277
|
|
#
|
|
# expected:
|
|
# == disasm: #<ISeq:prism_test_forwarding_arguments_node1@<compiled>:2 (2,8)-(4,11)>
|
|
# local table (size: 1, argc: 0 [opts: 0, rest: -1, post: 0, block: -1, kw: -1@-1, kwrest: 1])
|
|
# [ 1] "..."@0
|
|
# 0000 putself ( 3)
|
|
# 0001 getlocal_WC_0 ?@-2
|
|
# 0003 splatarray false
|
|
# 0005 getblockparamproxy ?@-1, 0
|
|
# 0008 send <calldata!mid:prism_test_forwarding_arguments_node, argc:1, ARGS_SPLAT|ARGS_BLOCKARG|FCALL>, nil
|
|
# 0011 leave ( 2)
|
|
# actual:
|
|
# == disasm: #<ISeq:prism_test_forwarding_arguments_node1@<compiled>:2 (2,8)-(4,11)>
|
|
# local table (size: 1, argc: 0 [opts: 0, rest: -1, post: 0, block: -1, kw: -1@-1, kwrest: 1])
|
|
# [ 1] "..."@0
|
|
# 0000 putself ( 3)
|
|
# 0001 getlocal_WC_0 ?@-2
|
|
# 0003 splatarray false
|
|
# 0005 getblockparamproxy "!"@-1, 0
|
|
# 0008 send <calldata!mid:prism_test_forwarding_arguments_node, argc:1, ARGS_SPLAT|ARGS_BLOCKARG|FCALL>, nil
|
|
# 0011 leave ( 2)
|
|
|
|
omit "fails on trunk-iseq_binary"
|
|
|
|
assert_prism_eval(<<-CODE)
|
|
def prism_test_forwarding_arguments_node(...); end;
|
|
def prism_test_forwarding_arguments_node1(...)
|
|
prism_test_forwarding_arguments_node(...)
|
|
end
|
|
CODE
|
|
|
|
assert_prism_eval(<<-CODE)
|
|
def prism_test_forwarding_arguments_node(...); end;
|
|
def prism_test_forwarding_arguments_node1(a, ...)
|
|
prism_test_forwarding_arguments_node(1,2, 3, ...)
|
|
end
|
|
CODE
|
|
end
|
|
|
|
def test_ForwardingSuperNode
|
|
assert_prism_eval("class Forwarding; def to_s; super; end; end")
|
|
assert_prism_eval("class Forwarding; def eval(code); super { code }; end; end")
|
|
end
|
|
|
|
def test_KeywordHashNode
|
|
assert_prism_eval("[a: [:b, :c]]")
|
|
end
|
|
|
|
def test_SuperNode
|
|
assert_prism_eval("def to_s; super 1; end")
|
|
assert_prism_eval("def to_s; super(); end")
|
|
assert_prism_eval("def to_s; super('a', :b, [1,2,3]); end")
|
|
assert_prism_eval("def to_s; super(1, 2, 3, &:foo); end")
|
|
end
|
|
|
|
############################################################################
|
|
# Methods / parameters #
|
|
############################################################################
|
|
|
|
def test_AliasGlobalVariableNode
|
|
assert_prism_eval("alias $prism_foo $prism_bar")
|
|
end
|
|
|
|
def test_AliasMethodNode
|
|
assert_prism_eval("alias :prism_a :to_s")
|
|
end
|
|
|
|
def test_BlockParameterNode
|
|
assert_prism_eval("def prism_test_block_parameter_node(&bar) end")
|
|
assert_prism_eval("->(b, c=1, *d, e, &f){}")
|
|
end
|
|
|
|
def test_BlockParametersNode
|
|
assert_prism_eval("Object.tap { || }")
|
|
assert_prism_eval("[1].map { |num| num }")
|
|
assert_prism_eval("[1].map { |a; b| b = 2; a + b}")
|
|
end
|
|
|
|
def test_FowardingParameterNode
|
|
assert_prism_eval("def prism_test_forwarding_parameter_node(...); end")
|
|
end
|
|
|
|
def test_KeywordRestParameterNode
|
|
assert_prism_eval("def prism_test_keyword_rest_parameter_node(a, **b); end")
|
|
end
|
|
|
|
def test_NoKeywordsParameterNode
|
|
assert_prism_eval("def prism_test_no_keywords(**nil); end")
|
|
assert_prism_eval("def prism_test_no_keywords(a, b = 2, **nil); end")
|
|
end
|
|
|
|
def test_OptionalParameterNode
|
|
assert_prism_eval("def prism_test_optional_param_node(bar = nil); end")
|
|
end
|
|
|
|
def test_OptionalKeywordParameterNode
|
|
assert_prism_eval("def prism_test_optional_keyword_param_node(bar: nil); end")
|
|
end
|
|
|
|
def test_ParametersNode
|
|
assert_prism_eval("def prism_test_parameters_node(bar, baz); end")
|
|
assert_prism_eval("def prism_test_parameters_node(a, b = 2); end")
|
|
end
|
|
|
|
def test_RequiredParameterNode
|
|
assert_prism_eval("def prism_test_required_param_node(bar); end")
|
|
assert_prism_eval("def prism_test_required_param_node(foo, bar); end")
|
|
end
|
|
|
|
def test_RequiredKeywordParameterNode
|
|
assert_prism_eval("def prism_test_required_param_node(bar:); end")
|
|
assert_prism_eval("def prism_test_required_param_node(foo:, bar:); end")
|
|
assert_prism_eval("-> a, b = 1, c:, d:, &e { a }")
|
|
end
|
|
|
|
def test_RestParameterNode
|
|
assert_prism_eval("def prism_test_rest_parameter_node(*a); end")
|
|
end
|
|
|
|
def test_UndefNode
|
|
assert_prism_eval("def prism_undef_node_1; end; undef prism_undef_node_1")
|
|
assert_prism_eval(<<-HERE
|
|
def prism_undef_node_2
|
|
end
|
|
def prism_undef_node_3
|
|
end
|
|
undef prism_undef_node_2, prism_undef_node_3
|
|
HERE
|
|
)
|
|
assert_prism_eval(<<-HERE
|
|
def prism_undef_node_4
|
|
end
|
|
undef :'prism_undef_node_#{4}'
|
|
HERE
|
|
)
|
|
end
|
|
|
|
############################################################################
|
|
# Pattern matching #
|
|
############################################################################
|
|
|
|
def test_AlternationPatternNode
|
|
assert_prism_eval("1 in 1 | 2")
|
|
assert_prism_eval("1 in 2 | 1")
|
|
assert_prism_eval("1 in 2 | 3 | 4 | 1")
|
|
assert_prism_eval("1 in 2 | 3")
|
|
end
|
|
|
|
def test_MatchPredicateNode
|
|
assert_prism_eval("1 in 1")
|
|
assert_prism_eval("1.0 in 1.0")
|
|
assert_prism_eval("1i in 1i")
|
|
assert_prism_eval("1r in 1r")
|
|
|
|
assert_prism_eval("\"foo\" in \"foo\"")
|
|
assert_prism_eval("\"foo \#{1}\" in \"foo \#{1}\"")
|
|
|
|
assert_prism_eval("false in false")
|
|
assert_prism_eval("nil in nil")
|
|
assert_prism_eval("self in self")
|
|
assert_prism_eval("true in true")
|
|
|
|
assert_prism_eval("5 in 0..10")
|
|
assert_prism_eval("5 in 0...10")
|
|
|
|
assert_prism_eval("[\"5\"] in %w[5]")
|
|
|
|
assert_prism_eval("Prism in Prism")
|
|
assert_prism_eval("Prism in ::Prism")
|
|
|
|
assert_prism_eval(":prism in :prism")
|
|
assert_prism_eval("%s[prism\#{1}] in %s[prism\#{1}]")
|
|
assert_prism_eval("\"foo\" in /.../")
|
|
assert_prism_eval("\"foo1\" in /...\#{1}/")
|
|
assert_prism_eval("4 in ->(v) { v.even? }")
|
|
|
|
assert_prism_eval("5 in foo")
|
|
|
|
assert_prism_eval("1 in 2")
|
|
end
|
|
|
|
def test_PinnedExpressionNode
|
|
assert_prism_eval("4 in ^(4)")
|
|
end
|
|
|
|
def test_PinnedVariableNode
|
|
assert_prism_eval("module Prism; @@prism = 1; 1 in ^@@prism; end")
|
|
assert_prism_eval("module Prism; @prism = 1; 1 in ^@prism; end")
|
|
assert_prism_eval("$prism = 1; 1 in ^$prism")
|
|
assert_prism_eval("prism = 1; 1 in ^prism")
|
|
end
|
|
|
|
############################################################################
|
|
# Miscellaneous #
|
|
############################################################################
|
|
|
|
def test_ScopeNode
|
|
assert_separately(%w[], "#{<<-'begin;'}\n#{<<-'end;'}")
|
|
begin;
|
|
def compare_eval(source)
|
|
ruby_eval = RubyVM::InstructionSequence.compile("module A; " + source + "; end").eval
|
|
prism_eval = RubyVM::InstructionSequence.compile_prism("module B; " + source + "; end").eval
|
|
|
|
assert_equal ruby_eval, prism_eval
|
|
end
|
|
|
|
def assert_prism_eval(source)
|
|
$VERBOSE, verbose_bak = nil, $VERBOSE
|
|
|
|
begin
|
|
compare_eval(source)
|
|
|
|
# Test "popped" functionality
|
|
compare_eval("#{source}; 1")
|
|
ensure
|
|
$VERBOSE = verbose_bak
|
|
end
|
|
end
|
|
assert_prism_eval("a = 1; 1.times do; { a: }; end")
|
|
assert_prism_eval("a = 1; def foo(a); a; end")
|
|
end;
|
|
end
|
|
|
|
############################################################################
|
|
# Errors #
|
|
############################################################################
|
|
|
|
def test_MissingNode
|
|
# TODO
|
|
end
|
|
|
|
############################################################################
|
|
# Encoding #
|
|
############################################################################
|
|
|
|
def test_encoding
|
|
assert_prism_eval('"però"')
|
|
assert_prism_eval(":però")
|
|
end
|
|
|
|
private
|
|
|
|
def compare_eval(source)
|
|
source = "class Prism::TestCompilePrism\n#{source}\nend"
|
|
|
|
ruby_eval = RubyVM::InstructionSequence.compile(source).eval
|
|
prism_eval = RubyVM::InstructionSequence.compile_prism(source).eval
|
|
|
|
if ruby_eval.is_a? Proc
|
|
assert_equal ruby_eval.class, prism_eval.class
|
|
else
|
|
assert_equal ruby_eval, prism_eval
|
|
end
|
|
end
|
|
|
|
def assert_prism_eval(source)
|
|
$VERBOSE, verbose_bak = nil, $VERBOSE
|
|
|
|
begin
|
|
compare_eval(source)
|
|
|
|
# Test "popped" functionality
|
|
compare_eval("#{source}; 1")
|
|
ensure
|
|
$VERBOSE = verbose_bak
|
|
end
|
|
end
|
|
end
|
|
end
|