зеркало из https://github.com/github/ruby.git
115 строки
3.0 KiB
Ruby
115 строки
3.0 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require_relative "../test_helper"
|
|
|
|
module Prism
|
|
class CommandLineTest < TestCase
|
|
def test_command_line_p
|
|
program = Prism.parse("1", command_line: "p").value
|
|
statements = program.statements.body
|
|
|
|
assert_equal 2, statements.length
|
|
assert_kind_of CallNode, statements.last
|
|
assert_equal :print, statements.last.name
|
|
end
|
|
|
|
def test_command_line_n
|
|
program = Prism.parse("1", command_line: "n").value
|
|
statements = program.statements.body
|
|
|
|
assert_equal 1, statements.length
|
|
assert_kind_of WhileNode, statements.first
|
|
|
|
predicate = statements.first.predicate
|
|
assert_kind_of CallNode, predicate
|
|
assert_equal :gets, predicate.name
|
|
|
|
arguments = predicate.arguments.arguments
|
|
assert_equal 1, arguments.length
|
|
assert_equal :$/, arguments.first.name
|
|
end
|
|
|
|
def test_command_line_a
|
|
program = Prism.parse("1", command_line: "na").value
|
|
statements = program.statements.body
|
|
|
|
assert_equal 1, statements.length
|
|
assert_kind_of WhileNode, statements.first
|
|
|
|
statement = statements.first.statements.body.first
|
|
assert_kind_of GlobalVariableWriteNode, statement
|
|
assert_equal :$F, statement.name
|
|
end
|
|
|
|
def test_command_line_l
|
|
program = Prism.parse("1", command_line: "nl").value
|
|
statements = program.statements.body
|
|
|
|
assert_equal 1, statements.length
|
|
assert_kind_of WhileNode, statements.first
|
|
|
|
predicate = statements.first.predicate
|
|
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
|
|
assert_equal "chomp", arguments.last.elements.first.key.unescaped
|
|
end
|
|
|
|
def test_command_line_e
|
|
result = Prism.parse("1 if 2..3")
|
|
assert_equal 2, result.warnings.length
|
|
|
|
result = Prism.parse("1 if 2..3", command_line: "e")
|
|
assert_equal 0, result.warnings.length
|
|
end
|
|
|
|
def test_command_line_x_implicit
|
|
result = Prism.parse_statement(<<~RUBY, main_script: true)
|
|
#!/bin/bash
|
|
exit 1
|
|
|
|
#!/usr/bin/env ruby
|
|
1
|
|
RUBY
|
|
|
|
assert_kind_of IntegerNode, result
|
|
end
|
|
|
|
def test_command_line_x_explicit
|
|
result = Prism.parse_statement(<<~RUBY, command_line: "x")
|
|
exit 1
|
|
|
|
#!/usr/bin/env ruby
|
|
1
|
|
RUBY
|
|
|
|
assert_kind_of IntegerNode, result
|
|
end
|
|
|
|
def test_command_line_x_implicit_fail
|
|
result = Prism.parse(<<~RUBY, main_script: true)
|
|
#!/bin/bash
|
|
exit 1
|
|
RUBY
|
|
|
|
assert_equal 1, result.errors.length
|
|
assert_equal :load, result.errors.first.level
|
|
end
|
|
|
|
def test_command_line_x_explicit_fail
|
|
result = Prism.parse(<<~RUBY, command_line: "x")
|
|
exit 1
|
|
RUBY
|
|
|
|
assert_equal 1, result.errors.length
|
|
assert_equal :load, result.errors.first.level
|
|
end
|
|
end
|
|
end
|