mirror of
https://gitea.invidious.io/iv-org/shard-ameba.git
synced 2024-08-15 00:53:29 +00:00
119 lines
3.5 KiB
Crystal
119 lines
3.5 KiB
Crystal
require "../../../spec_helper"
|
|
|
|
module Ameba::AST
|
|
describe Assignment do
|
|
node = Crystal::NilLiteral.new
|
|
scope = Scope.new as_node "foo = 1"
|
|
variable = Variable.new(Crystal::Var.new("foo"), scope)
|
|
|
|
describe "#initialize" do
|
|
it "creates a new assignment with node and var" do
|
|
assignment = Assignment.new(node, variable, scope)
|
|
assignment.node.should_not be_nil
|
|
end
|
|
end
|
|
|
|
describe "#reference=" do
|
|
it "creates a new reference" do
|
|
assignment = Assignment.new(node, variable, scope)
|
|
assignment.referenced = true
|
|
assignment.referenced?.should be_true
|
|
end
|
|
end
|
|
|
|
describe "delegation" do
|
|
it "delegates locations" do
|
|
assignment = Assignment.new(node, variable, scope)
|
|
assignment.location.should eq node.location
|
|
assignment.end_location.should eq node.end_location
|
|
end
|
|
|
|
it "delegates to_s" do
|
|
assignment = Assignment.new(node, variable, scope)
|
|
assignment.to_s.should eq node.to_s
|
|
end
|
|
|
|
it "delegates scope" do
|
|
assignment = Assignment.new(node, variable, scope)
|
|
assignment.scope.should eq variable.scope
|
|
end
|
|
end
|
|
|
|
describe "#branch" do
|
|
it "returns the branch of the assignment" do
|
|
nodes = as_nodes %(
|
|
def method(a)
|
|
if a
|
|
a = 3 # --> Crystal::Expressions
|
|
puts a
|
|
end
|
|
end
|
|
)
|
|
|
|
scope = Scope.new nodes.def_nodes.first
|
|
variable = Variable.new(nodes.var_nodes.first, scope)
|
|
assignment = Assignment.new(nodes.assign_nodes.first, variable, scope)
|
|
assignment.branch.should_not be_nil
|
|
assignment.branch.not_nil!.node.class.should eq Crystal::Expressions
|
|
end
|
|
|
|
it "returns inner branch" do
|
|
nodes = as_nodes %(
|
|
def method(a, b)
|
|
if a
|
|
if b
|
|
a = 3 # --> Crystal::Assign
|
|
end
|
|
end
|
|
end
|
|
)
|
|
scope = Scope.new nodes.def_nodes.first
|
|
variable = Variable.new(nodes.var_nodes.first, scope)
|
|
assignment = Assignment.new(nodes.assign_nodes.first, variable, scope)
|
|
assignment.branch.should_not be_nil
|
|
assignment.branch.not_nil!.node.class.should eq Crystal::Assign
|
|
end
|
|
|
|
it "returns nil if assignment does not have a branch" do
|
|
nodes = as_nodes %(
|
|
def method(a)
|
|
a = 2
|
|
end
|
|
)
|
|
|
|
scope = Scope.new nodes.def_nodes.first
|
|
variable = Variable.new(nodes.var_nodes.first, scope)
|
|
assignment = Assignment.new(nodes.assign_nodes.first, variable, scope)
|
|
assignment.branch.should be_nil
|
|
end
|
|
end
|
|
|
|
describe "#transformed?" do
|
|
it "returns false if the assignment is not transformed by the compiler" do
|
|
nodes = as_nodes %(
|
|
def method(a)
|
|
a = 2
|
|
end
|
|
)
|
|
|
|
scope = Scope.new nodes.def_nodes.first
|
|
variable = Variable.new(nodes.var_nodes.first, scope)
|
|
assignment = Assignment.new(nodes.assign_nodes.first, variable, scope)
|
|
assignment.transformed?.should be_false
|
|
end
|
|
|
|
it "returns true if the assignment is transformed by the compiler" do
|
|
nodes = as_nodes %(
|
|
array.each do |(a, b)|
|
|
end
|
|
)
|
|
|
|
scope = Scope.new nodes.block_nodes.first
|
|
variable = Variable.new(nodes.var_nodes.first, scope)
|
|
assignment = Assignment.new(nodes.assign_nodes.first, variable, scope)
|
|
assignment.transformed?.should be_true
|
|
end
|
|
end
|
|
end
|
|
end
|