php-parser/node/stmt/t_function_test.go

383 lines
8.0 KiB
Go
Raw Normal View History

2018-02-09 09:56:49 +00:00
package stmt_test
import (
"bytes"
"testing"
2018-02-12 21:10:53 +00:00
"github.com/z7zmey/php-parser/node/name"
"github.com/z7zmey/php-parser/node/scalar"
2018-06-24 07:19:44 +00:00
"github.com/z7zmey/php-parser/position"
2018-02-12 21:10:53 +00:00
2018-02-09 09:56:49 +00:00
"github.com/z7zmey/php-parser/node"
2018-02-12 13:08:08 +00:00
"github.com/z7zmey/php-parser/node/expr"
2018-02-09 09:56:49 +00:00
"github.com/z7zmey/php-parser/node/stmt"
"github.com/z7zmey/php-parser/php5"
"github.com/z7zmey/php-parser/php7"
)
func TestSimpleFunction(t *testing.T) {
src := `<? function foo() {}`
2018-05-02 09:14:24 +00:00
expected := &node.Root{
2018-06-24 07:19:44 +00:00
Position: &position.Position{
StartLine: 1,
EndLine: 1,
StartPos: 4,
EndPos: 20,
},
2018-02-09 09:56:49 +00:00
Stmts: []node.Node{
&stmt.Function{
2018-06-24 07:19:44 +00:00
Position: &position.Position{
StartLine: 1,
EndLine: 1,
StartPos: 4,
EndPos: 20,
},
2018-02-12 21:10:53 +00:00
ReturnsRef: false,
2018-02-09 09:56:49 +00:00
PhpDocComment: "",
2018-06-24 07:19:44 +00:00
FunctionName: &node.Identifier{
Position: &position.Position{
StartLine: 1,
EndLine: 1,
StartPos: 13,
EndPos: 15,
},
Value: "foo",
},
Stmts: []node.Node{},
2018-02-09 09:56:49 +00:00
},
},
}
2018-04-10 12:23:13 +00:00
php7parser := php7.NewParser(bytes.NewBufferString(src), "test.php")
php7parser.Parse()
actual := php7parser.GetRootNode()
2018-02-09 09:56:49 +00:00
assertEqual(t, expected, actual)
2018-04-10 12:23:13 +00:00
php5parser := php5.NewParser(bytes.NewBufferString(src), "test.php")
php5parser.Parse()
actual = php5parser.GetRootNode()
2018-02-09 09:56:49 +00:00
assertEqual(t, expected, actual)
}
2018-02-09 11:59:19 +00:00
func TestFunctionReturn(t *testing.T) {
src := `<? function foo() {return;}`
2018-05-02 09:14:24 +00:00
expected := &node.Root{
2018-06-24 07:19:44 +00:00
Position: &position.Position{
StartLine: 1,
EndLine: 1,
StartPos: 4,
EndPos: 27,
},
2018-02-09 11:59:19 +00:00
Stmts: []node.Node{
&stmt.Function{
2018-06-24 07:19:44 +00:00
Position: &position.Position{
StartLine: 1,
EndLine: 1,
StartPos: 4,
EndPos: 27,
},
2018-02-12 21:10:53 +00:00
ReturnsRef: false,
2018-02-09 11:59:19 +00:00
PhpDocComment: "",
2018-06-24 07:19:44 +00:00
FunctionName: &node.Identifier{
Position: &position.Position{
StartLine: 1,
EndLine: 1,
StartPos: 13,
EndPos: 15,
},
Value: "foo",
},
2018-02-09 11:59:19 +00:00
Stmts: []node.Node{
2018-06-24 07:19:44 +00:00
&stmt.Return{
Position: &position.Position{
StartLine: 1,
EndLine: 1,
StartPos: 20,
EndPos: 26,
},
},
2018-02-09 11:59:19 +00:00
},
},
},
}
2018-04-10 12:23:13 +00:00
php7parser := php7.NewParser(bytes.NewBufferString(src), "test.php")
php7parser.Parse()
actual := php7parser.GetRootNode()
2018-02-09 11:59:19 +00:00
assertEqual(t, expected, actual)
2018-04-10 12:23:13 +00:00
php5parser := php5.NewParser(bytes.NewBufferString(src), "test.php")
php5parser.Parse()
actual = php5parser.GetRootNode()
2018-02-09 11:59:19 +00:00
assertEqual(t, expected, actual)
}
2018-02-12 13:08:08 +00:00
func TestFunctionReturnVar(t *testing.T) {
2018-02-12 21:10:53 +00:00
src := `<? function foo(array $a, callable $b) {return $a;}`
2018-02-12 13:08:08 +00:00
2018-05-02 09:14:24 +00:00
expected := &node.Root{
2018-06-24 07:19:44 +00:00
Position: &position.Position{
StartLine: 1,
EndLine: 1,
StartPos: 4,
EndPos: 51,
},
2018-02-12 13:08:08 +00:00
Stmts: []node.Node{
&stmt.Function{
2018-06-24 07:19:44 +00:00
Position: &position.Position{
StartLine: 1,
EndLine: 1,
StartPos: 4,
EndPos: 51,
},
2018-02-12 21:10:53 +00:00
ReturnsRef: false,
2018-02-12 13:08:08 +00:00
PhpDocComment: "",
2018-06-24 07:19:44 +00:00
FunctionName: &node.Identifier{
Position: &position.Position{
StartLine: 1,
EndLine: 1,
StartPos: 13,
EndPos: 15,
},
Value: "foo",
},
2018-02-12 21:10:53 +00:00
Params: []node.Node{
&node.Parameter{
2018-06-24 07:19:44 +00:00
Position: &position.Position{
StartLine: 1,
EndLine: 1,
StartPos: 17,
EndPos: 24,
},
ByRef: false,
Variadic: false,
VariableType: &node.Identifier{
Position: &position.Position{
StartLine: 1,
EndLine: 1,
StartPos: 17,
EndPos: 21,
},
Value: "array",
},
Variable: &expr.Variable{
Position: &position.Position{
StartLine: 1,
EndLine: 1,
StartPos: 23,
EndPos: 24,
},
VarName: &node.Identifier{
Position: &position.Position{
StartLine: 1,
EndLine: 1,
StartPos: 23,
EndPos: 24,
},
Value: "a",
},
},
2018-02-12 21:10:53 +00:00
},
&node.Parameter{
2018-06-24 07:19:44 +00:00
Position: &position.Position{
StartLine: 1,
EndLine: 1,
StartPos: 27,
EndPos: 37,
},
ByRef: false,
Variadic: false,
VariableType: &node.Identifier{
Position: &position.Position{
StartLine: 1,
EndLine: 1,
StartPos: 27,
EndPos: 34,
},
Value: "callable",
},
Variable: &expr.Variable{
Position: &position.Position{
StartLine: 1,
EndLine: 1,
StartPos: 36,
EndPos: 37,
},
VarName: &node.Identifier{
Position: &position.Position{
StartLine: 1,
EndLine: 1,
StartPos: 36,
EndPos: 37,
},
Value: "b",
},
},
2018-02-12 21:10:53 +00:00
},
},
2018-02-12 13:08:08 +00:00
Stmts: []node.Node{
&stmt.Return{
2018-06-24 07:19:44 +00:00
Position: &position.Position{
StartLine: 1,
EndLine: 1,
StartPos: 41,
EndPos: 50,
},
Expr: &expr.Variable{
Position: &position.Position{
StartLine: 1,
EndLine: 1,
StartPos: 48,
EndPos: 49,
},
VarName: &node.Identifier{
Position: &position.Position{
StartLine: 1,
EndLine: 1,
StartPos: 48,
EndPos: 49,
},
Value: "a",
},
},
2018-02-12 13:08:08 +00:00
},
},
},
},
}
2018-04-10 12:23:13 +00:00
php7parser := php7.NewParser(bytes.NewBufferString(src), "test.php")
php7parser.Parse()
actual := php7parser.GetRootNode()
2018-02-12 13:08:08 +00:00
assertEqual(t, expected, actual)
2018-04-10 12:23:13 +00:00
php5parser := php5.NewParser(bytes.NewBufferString(src), "test.php")
php5parser.Parse()
actual = php5parser.GetRootNode()
2018-02-12 13:08:08 +00:00
assertEqual(t, expected, actual)
}
2018-02-09 09:56:49 +00:00
func TestRefFunction(t *testing.T) {
2018-02-09 11:59:19 +00:00
src := `<? function &foo() {return 1;}`
2018-02-09 09:56:49 +00:00
2018-05-02 09:14:24 +00:00
expected := &node.Root{
2018-06-24 07:19:44 +00:00
Position: &position.Position{
StartLine: 1,
EndLine: 1,
StartPos: 4,
EndPos: 30,
},
2018-02-09 09:56:49 +00:00
Stmts: []node.Node{
&stmt.Function{
2018-06-24 07:19:44 +00:00
Position: &position.Position{
StartLine: 1,
EndLine: 1,
StartPos: 4,
EndPos: 30,
},
2018-02-12 21:10:53 +00:00
ReturnsRef: true,
2018-02-09 09:56:49 +00:00
PhpDocComment: "",
2018-06-24 07:19:44 +00:00
FunctionName: &node.Identifier{
Position: &position.Position{
StartLine: 1,
EndLine: 1,
StartPos: 14,
EndPos: 16,
},
Value: "foo",
},
2018-02-09 11:59:19 +00:00
Stmts: []node.Node{
&stmt.Return{
2018-06-24 07:19:44 +00:00
Position: &position.Position{
StartLine: 1,
EndLine: 1,
StartPos: 21,
EndPos: 29,
},
Expr: &scalar.Lnumber{
Position: &position.Position{
StartLine: 1,
EndLine: 1,
StartPos: 28,
EndPos: 28,
},
Value: "1",
},
2018-02-09 11:59:19 +00:00
},
},
2018-02-09 09:56:49 +00:00
},
},
}
2018-04-10 12:23:13 +00:00
php7parser := php7.NewParser(bytes.NewBufferString(src), "test.php")
php7parser.Parse()
actual := php7parser.GetRootNode()
2018-02-09 09:56:49 +00:00
assertEqual(t, expected, actual)
2018-04-10 12:23:13 +00:00
php5parser := php5.NewParser(bytes.NewBufferString(src), "test.php")
php5parser.Parse()
actual = php5parser.GetRootNode()
2018-02-09 09:56:49 +00:00
assertEqual(t, expected, actual)
}
func TestReturnTypeFunction(t *testing.T) {
src := `<? function &foo(): void {}`
2018-05-02 09:14:24 +00:00
expected := &node.Root{
2018-06-24 07:19:44 +00:00
Position: &position.Position{
StartLine: 1,
EndLine: 1,
StartPos: 4,
EndPos: 27,
},
2018-02-09 09:56:49 +00:00
Stmts: []node.Node{
&stmt.Function{
2018-06-24 07:19:44 +00:00
Position: &position.Position{
StartLine: 1,
EndLine: 1,
StartPos: 4,
EndPos: 27,
},
2018-02-09 09:56:49 +00:00
PhpDocComment: "",
2018-06-24 07:19:44 +00:00
ReturnsRef: true,
FunctionName: &node.Identifier{
Position: &position.Position{
StartLine: 1,
EndLine: 1,
StartPos: 14,
EndPos: 16,
},
Value: "foo",
},
2018-02-09 09:56:49 +00:00
ReturnType: &name.Name{
2018-06-24 07:19:44 +00:00
Position: &position.Position{
StartLine: 1,
EndLine: 1,
StartPos: 21,
EndPos: 24,
},
2018-02-09 09:56:49 +00:00
Parts: []node.Node{
2018-06-24 07:19:44 +00:00
&name.NamePart{
Position: &position.Position{
StartLine: 1,
EndLine: 1,
StartPos: 21,
EndPos: 24,
},
Value: "void",
},
2018-02-09 09:56:49 +00:00
},
},
Stmts: []node.Node{},
},
},
}
2018-04-10 12:23:13 +00:00
php7parser := php7.NewParser(bytes.NewBufferString(src), "test.php")
php7parser.Parse()
actual := php7parser.GetRootNode()
2018-02-09 09:56:49 +00:00
assertEqual(t, expected, actual)
}