maride/php-parser
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

pretty printer: print Finally

Browse Source
This commit is contained in:
z7zmey 2018-03-31 14:17:05 +03:00
parent 94ffbdfe59
commit 5cc62dd8ee
2 changed files with 48 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
printer/printer.go
View File

@ -312,11 +312,13 @@ func getPrintFuncByNode(n node.Node) func(o io.Writer, n node.Node) {
return printStmtElseif return printStmtElseif
case *stmt.Else: case *stmt.Else:
return printStmtElse return printStmtElse
case *stmt.Expression:
return printStmtExpression
case *stmt.Finally:
return printStmtFinally
case *stmt.StmtList: case *stmt.StmtList:
return printStmtStmtList return printStmtStmtList
case *stmt.Expression:
return printStmtExpression
case *stmt.Nop: case *stmt.Nop:
return printStmtNop return printStmtNop
} }
@ -1537,12 +1539,6 @@ func printStmtElse(o io.Writer, n node.Node) {
} }
} }
func printStmtStmtList(o io.Writer, n node.Node) {
nn := n.(*stmt.StmtList)
printNodes(o, nn.Stmts)
}
func printStmtExpression(o io.Writer, n node.Node) { func printStmtExpression(o io.Writer, n node.Node) {
nn := n.(*stmt.Expression) nn := n.(*stmt.Expression)
@ -1551,6 +1547,20 @@ func printStmtExpression(o io.Writer, n node.Node) {
io.WriteString(o, ";\n") io.WriteString(o, ";\n")
} }
func printStmtFinally(o io.Writer, n node.Node) {
nn := n.(*stmt.Finally)
io.WriteString(o, "finally {\n")
printNodes(o, nn.Stmts)
io.WriteString(o, "}\n")
}
func printStmtStmtList(o io.Writer, n node.Node) {
nn := n.(*stmt.StmtList)
printNodes(o, nn.Stmts)
}
func printStmtNop(o io.Writer, n node.Node) { func printStmtNop(o io.Writer, n node.Node) {
io.WriteString(o, ";\n") io.WriteString(o, ";\n")
} }

43
printer/printer_test.go
View File

@ -2500,6 +2500,36 @@ func TestPrintStmtElseNop(t *testing.T) {
} }
} }
func TestPrintExpression(t *testing.T) {
o := bytes.NewBufferString("")
printer.Print(o, &stmt.Expression{Expr: &expr.Variable{VarName: &node.Identifier{Value: "a"}}})
expected := "$a;\n"
actual := o.String()
if expected != actual {
t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual)
}
}
func TestPrintStmtFinally(t *testing.T) {
o := bytes.NewBufferString("")
printer.Print(o, &stmt.Finally{
Stmts: []node.Node{
&stmt.Nop{},
},
})
expected := "finally {\n;\n}\n"
actual := o.String()
if expected != actual {
t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual)
}
}
func TestPrintStmtList(t *testing.T) { func TestPrintStmtList(t *testing.T) {
o := bytes.NewBufferString("") o := bytes.NewBufferString("")
@ -2518,19 +2548,6 @@ func TestPrintStmtList(t *testing.T) {
} }
} }
func TestPrintExpression(t *testing.T) {
o := bytes.NewBufferString("")
printer.Print(o, &stmt.Expression{Expr: &expr.Variable{VarName: &node.Identifier{Value: "a"}}})
expected := "$a;\n"
actual := o.String()
if expected != actual {
t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual)
}
}
func TestPrintNop(t *testing.T) { func TestPrintNop(t *testing.T) {
o := bytes.NewBufferString("") o := bytes.NewBufferString("")