php-parser/node/scalar/t_visitor_test.go

105 lines
2.4 KiB
Go
Raw Normal View History

2018-01-12 13:13:05 +00:00
package scalar_test
import (
"testing"
2019-02-25 16:11:28 +00:00
"gotest.tools/assert"
2018-01-12 13:13:05 +00:00
"github.com/z7zmey/php-parser/node"
"github.com/z7zmey/php-parser/node/scalar"
2018-01-17 16:58:45 +00:00
"github.com/z7zmey/php-parser/walker"
2018-01-12 13:13:05 +00:00
)
var nameNodesTests = []struct {
node node.Node // node
expectedVisitedKeys []string // visited keys
expectedAttributes map[string]interface{}
}{
{
&scalar.String{Value: "foo"},
[]string{},
map[string]interface{}{"Value": "foo"},
},
{
&scalar.Lnumber{Value: "0"},
[]string{},
map[string]interface{}{"Value": "0"},
},
{
&scalar.Dnumber{Value: "0.1"},
[]string{},
map[string]interface{}{"Value": "0.1"},
},
{
&scalar.MagicConstant{Value: "__DIR__"},
[]string{},
map[string]interface{}{"Value": "__DIR__"},
},
{
&scalar.EncapsedStringPart{Value: "foo"},
[]string{},
map[string]interface{}{"Value": "foo"},
},
{
&scalar.Encapsed{Parts: []node.Node{&scalar.EncapsedStringPart{Value: "foo"}}},
[]string{"Parts"},
nil,
},
2018-04-05 10:47:36 +00:00
{
&scalar.Heredoc{Label: "LBL", Parts: []node.Node{&scalar.EncapsedStringPart{Value: "foo"}}},
[]string{"Parts"},
map[string]interface{}{"Label": "LBL"},
},
2018-01-12 13:13:05 +00:00
}
type visitorMock struct {
visitChildren bool
visitedKeys []string
}
2018-02-20 17:52:07 +00:00
func (v *visitorMock) EnterNode(n walker.Walkable) bool { return v.visitChildren }
2018-06-18 20:29:52 +00:00
func (v *visitorMock) LeaveNode(n walker.Walkable) {}
func (v *visitorMock) EnterChildNode(key string, w walker.Walkable) {
2018-01-12 13:13:05 +00:00
v.visitedKeys = append(v.visitedKeys, key)
}
2018-06-18 20:29:52 +00:00
func (v *visitorMock) LeaveChildNode(key string, w walker.Walkable) {}
func (v *visitorMock) EnterChildList(key string, w walker.Walkable) {
v.visitedKeys = append(v.visitedKeys, key)
}
func (v *visitorMock) LeaveChildList(key string, w walker.Walkable) {}
2018-01-12 13:13:05 +00:00
func TestNameVisitorDisableChildren(t *testing.T) {
for _, tt := range nameNodesTests {
2019-02-25 16:11:28 +00:00
v := &visitorMock{false, []string{}}
2018-01-12 13:13:05 +00:00
tt.node.Walk(v)
expected := []string{}
actual := v.visitedKeys
2019-02-25 16:11:28 +00:00
assert.DeepEqual(t, expected, actual)
2018-01-12 13:13:05 +00:00
}
}
func TestNameVisitor(t *testing.T) {
for _, tt := range nameNodesTests {
2019-02-25 16:11:28 +00:00
v := &visitorMock{true, []string{}}
2018-01-12 13:13:05 +00:00
tt.node.Walk(v)
expected := tt.expectedVisitedKeys
actual := v.visitedKeys
2019-02-25 16:11:28 +00:00
assert.DeepEqual(t, expected, actual)
2018-01-12 13:13:05 +00:00
}
}
// test Attributes()
func TestNameAttributes(t *testing.T) {
for _, tt := range nameNodesTests {
expected := tt.expectedAttributes
actual := tt.node.Attributes()
2019-02-25 16:11:28 +00:00
assert.DeepEqual(t, expected, actual)
2018-01-12 13:13:05 +00:00
}
}