php-parser/node/stmt/t_if_test.go

147 lines
3.6 KiB
Go
Raw Normal View History

2018-02-08 21:31:38 +00:00
package stmt_test
import (
"bytes"
"testing"
"github.com/z7zmey/php-parser/node/expr"
"github.com/z7zmey/php-parser/node"
"github.com/z7zmey/php-parser/node/stmt"
"github.com/z7zmey/php-parser/php5"
"github.com/z7zmey/php-parser/php7"
)
func TestIf(t *testing.T) {
src := `<? if ($a) {}`
expected := &stmt.StmtList{
Stmts: []node.Node{
&stmt.If{
2018-03-18 14:50:19 +00:00
Cond: &expr.Variable{VarName: &node.Identifier{Value: "a"}},
2018-02-08 21:31:38 +00:00
Stmt: &stmt.StmtList{Stmts: []node.Node{}},
},
},
}
actual, _, _ := php7.Parse(bytes.NewBufferString(src), "test.php")
assertEqual(t, expected, actual)
actual, _, _ = php5.Parse(bytes.NewBufferString(src), "test.php")
assertEqual(t, expected, actual)
}
func TestElseIf(t *testing.T) {
src := `<? if ($a) {} elseif ($b) {}
`
expected := &stmt.StmtList{
Stmts: []node.Node{
&stmt.If{
2018-03-18 14:50:19 +00:00
Cond: &expr.Variable{VarName: &node.Identifier{Value: "a"}},
2018-02-08 21:31:38 +00:00
Stmt: &stmt.StmtList{Stmts: []node.Node{}},
ElseIf: []node.Node{
&stmt.ElseIf{
2018-03-18 14:50:19 +00:00
Cond: &expr.Variable{VarName: &node.Identifier{Value: "b"}},
2018-02-08 21:31:38 +00:00
Stmt: &stmt.StmtList{Stmts: []node.Node{}},
},
},
},
},
}
actual, _, _ := php7.Parse(bytes.NewBufferString(src), "test.php")
assertEqual(t, expected, actual)
actual, _, _ = php5.Parse(bytes.NewBufferString(src), "test.php")
assertEqual(t, expected, actual)
}
func TestElse(t *testing.T) {
src := `<? if ($a) {} else {}`
expected := &stmt.StmtList{
Stmts: []node.Node{
&stmt.If{
2018-03-18 14:50:19 +00:00
Cond: &expr.Variable{VarName: &node.Identifier{Value: "a"}},
2018-02-08 21:31:38 +00:00
Stmt: &stmt.StmtList{Stmts: []node.Node{}},
Else: &stmt.Else{
Stmt: &stmt.StmtList{Stmts: []node.Node{}},
},
},
},
}
actual, _, _ := php7.Parse(bytes.NewBufferString(src), "test.php")
assertEqual(t, expected, actual)
actual, _, _ = php5.Parse(bytes.NewBufferString(src), "test.php")
assertEqual(t, expected, actual)
}
func TestElseElseIf(t *testing.T) {
src := `<? if ($a) {} elseif ($b) {} elseif ($c) {} else {}`
expected := &stmt.StmtList{
Stmts: []node.Node{
&stmt.If{
2018-03-18 14:50:19 +00:00
Cond: &expr.Variable{VarName: &node.Identifier{Value: "a"}},
2018-02-08 21:31:38 +00:00
Stmt: &stmt.StmtList{Stmts: []node.Node{}},
ElseIf: []node.Node{
&stmt.ElseIf{
2018-03-18 14:50:19 +00:00
Cond: &expr.Variable{VarName: &node.Identifier{Value: "b"}},
2018-02-08 21:31:38 +00:00
Stmt: &stmt.StmtList{Stmts: []node.Node{}},
},
&stmt.ElseIf{
2018-03-18 14:50:19 +00:00
Cond: &expr.Variable{VarName: &node.Identifier{Value: "c"}},
2018-02-08 21:31:38 +00:00
Stmt: &stmt.StmtList{Stmts: []node.Node{}},
},
},
Else: &stmt.Else{
Stmt: &stmt.StmtList{Stmts: []node.Node{}},
},
},
},
}
actual, _, _ := php7.Parse(bytes.NewBufferString(src), "test.php")
assertEqual(t, expected, actual)
actual, _, _ = php5.Parse(bytes.NewBufferString(src), "test.php")
assertEqual(t, expected, actual)
}
func TestElseIfElseIfElse(t *testing.T) {
src := `<? if ($a) {} elseif ($b) {} else if ($c) {} else {}`
expected := &stmt.StmtList{
Stmts: []node.Node{
&stmt.If{
2018-03-18 14:50:19 +00:00
Cond: &expr.Variable{VarName: &node.Identifier{Value: "a"}},
2018-02-08 21:31:38 +00:00
Stmt: &stmt.StmtList{Stmts: []node.Node{}},
ElseIf: []node.Node{
&stmt.ElseIf{
2018-03-18 14:50:19 +00:00
Cond: &expr.Variable{VarName: &node.Identifier{Value: "b"}},
2018-02-08 21:31:38 +00:00
Stmt: &stmt.StmtList{Stmts: []node.Node{}},
},
},
Else: &stmt.Else{
Stmt: &stmt.If{
2018-03-18 14:50:19 +00:00
Cond: &expr.Variable{VarName: &node.Identifier{Value: "c"}},
2018-02-08 21:31:38 +00:00
Stmt: &stmt.StmtList{Stmts: []node.Node{}},
Else: &stmt.Else{
Stmt: &stmt.StmtList{Stmts: []node.Node{}},
},
},
},
},
},
}
actual, _, _ := php7.Parse(bytes.NewBufferString(src), "test.php")
assertEqual(t, expected, actual)
actual, _, _ = php5.Parse(bytes.NewBufferString(src), "test.php")
assertEqual(t, expected, actual)
}