1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
|
require 'rspec/support/source/node'
class RSpec::Support::Source
RSpec.describe Node, :if => RSpec::Support::RubyFeatures.ripper_supported? do
let(:root_node) do
Node.new(sexp)
end
let(:sexp) do
require 'ripper'
Ripper.sexp(source)
end
let(:source) { <<-END }
variable = do_something(1, 2)
variable.do_anything do |arg|
puts arg
end
END
# [:program,
# [[:assign,
# [:var_field, [:@ident, "variable", [1, 6]]],
# [:method_add_arg,
# [:fcall, [:@ident, "do_something", [1, 17]]],
# [:arg_paren,
# [:args_add_block,
# [[:@int, "1", [1, 30]], [:@int, "2", [1, 33]]],
# false]]]],
# [:method_add_block,
# [:call,
# [:var_ref, [:@ident, "variable", [2, 6]]],
# :".",
# [:@ident, "do_anything", [2, 15]]],
# [:do_block,
# [:block_var,
# [:params, [[:@ident, "arg", [2, 31]]], nil, nil, nil, nil, nil, nil],
# false],
# [[:command,
# [:@ident, "puts", [3, 8]],
# [:args_add_block, [[:var_ref, [:@ident, "arg", [3, 13]]]], false]]]]]]]
describe '#args' do
context 'when the sexp args consist of direct child sexps' do
let(:target_node) do
root_node.find { |node| node.type == :method_add_arg }
end
it 'returns the child nodes' do
expect(target_node.args).to match([
an_object_having_attributes(:type => :fcall),
an_object_having_attributes(:type => :arg_paren)
])
end
end
context 'when the sexp args include an array of sexps' do
let(:target_node) do
root_node.find { |node| node.type == :args_add_block }
end
it 'returns pseudo expression sequence node for the array' do
expect(target_node.args).to match([
an_object_having_attributes(:type => :_expression_sequence),
false
])
end
end
end
describe '#each_ancestor' do
let(:target_node) do
root_node.find { |node| node.type == :arg_paren }
end
it 'yields ancestor nodes from parent to root' do
expect { |b| target_node.each_ancestor(&b) }.to yield_successive_args(
an_object_having_attributes(:type => :method_add_arg),
an_object_having_attributes(:type => :assign),
an_object_having_attributes(:type => :_expression_sequence),
an_object_having_attributes(:type => :program)
)
end
end
describe '#location' do
context 'with identifier type node' do
let(:target_node) do
root_node.find { |node| node.type == :@ident }
end
it 'returns a Location object with line and column numbers' do
expect(target_node.location).to have_attributes(:line => 1, :column => 6)
end
end
context 'with non-identifier type node' do
let(:target_node) do
root_node.find { |node| node.type == :assign }
end
it 'returns nil' do
expect(target_node.location).to be_nil
end
end
end
describe '#inspect' do
it 'returns a string including class name and node type' do
expect(root_node.inspect).to eq('#<RSpec::Support::Source::Node program>')
end
end
end
end
|
