php-parser/node/stmt/alt_if_test.go
2018-01-17 19:18:35 +02:00

135 lines
2.8 KiB
Go

package stmt_test
import (
"bytes"
"reflect"
"testing"
"github.com/z7zmey/php-parser/node/expr"
"github.com/kylelemons/godebug/pretty"
"github.com/z7zmey/php-parser/node"
"github.com/z7zmey/php-parser/node/stmt"
"github.com/z7zmey/php-parser/parser"
)
func assertEqual(t *testing.T, expected interface{}, actual interface{}) {
if !reflect.DeepEqual(expected, actual) {
diff := pretty.Compare(expected, actual)
if diff != "" {
t.Errorf("diff: (-expected +actual)\n%s", diff)
} else {
t.Errorf("expected and actual are not equal\n")
}
}
}
func TestAltIf(t *testing.T) {
src := `<?
if ($a) :
endif;
`
expected := &stmt.StmtList{
Stmts: []node.Node{
&stmt.AltIf{
Cond: &expr.Variable{VarName: &node.Identifier{Value: "$a"}},
Stmt: &stmt.StmtList{Stmts: []node.Node{}},
},
},
}
actual, _, _ := parser.ParsePhp7(bytes.NewBufferString(src), "test.php")
assertEqual(t, expected, actual)
}
func TestAltElseIf(t *testing.T) {
src := `<?
if ($a) :
elseif ($b):
endif;
`
expected := &stmt.StmtList{
Stmts: []node.Node{
&stmt.AltIf{
Cond: &expr.Variable{VarName: &node.Identifier{Value: "$a"}},
Stmt: &stmt.StmtList{Stmts: []node.Node{}},
ElseIf: []node.Node{
&stmt.AltElseIf{
Cond: &expr.Variable{VarName: &node.Identifier{Value: "$b"}},
Stmt: &stmt.StmtList{Stmts: []node.Node{}},
},
},
},
},
}
actual, _, _ := parser.ParsePhp7(bytes.NewBufferString(src), "test.php")
assertEqual(t, expected, actual)
}
func TestAltElse(t *testing.T) {
src := `<?
if ($a) :
else:
endif;
`
expected := &stmt.StmtList{
Stmts: []node.Node{
&stmt.AltIf{
Cond: &expr.Variable{VarName: &node.Identifier{Value: "$a"}},
Stmt: &stmt.StmtList{Stmts: []node.Node{}},
Else: &stmt.AltElse{
Stmt: &stmt.StmtList{Stmts: []node.Node{}},
},
},
},
}
actual, _, _ := parser.ParsePhp7(bytes.NewBufferString(src), "test.php")
assertEqual(t, expected, actual)
}
func TestAltElseElseIf(t *testing.T) {
src := `<?
if ($a) :
elseif ($b):
elseif ($c):
else:
endif;
`
expected := &stmt.StmtList{
Stmts: []node.Node{
&stmt.AltIf{
Cond: &expr.Variable{VarName: &node.Identifier{Value: "$a"}},
Stmt: &stmt.StmtList{Stmts: []node.Node{}},
ElseIf: []node.Node{
&stmt.AltElseIf{
Cond: &expr.Variable{VarName: &node.Identifier{Value: "$b"}},
Stmt: &stmt.StmtList{Stmts: []node.Node{}},
},
&stmt.AltElseIf{
Cond: &expr.Variable{VarName: &node.Identifier{Value: "$c"}},
Stmt: &stmt.StmtList{Stmts: []node.Node{}},
},
},
Else: &stmt.AltElse{
Stmt: &stmt.StmtList{Stmts: []node.Node{}},
},
},
},
}
actual, _, _ := parser.ParsePhp7(bytes.NewBufferString(src), "test.php")
assertEqual(t, expected, actual)
}