package printer_test import ( "bytes" "git.maride.cc/maride/php-parser/pkg/token" "git.maride.cc/maride/php-parser/pkg/visitor/printer" "testing" "git.maride.cc/maride/php-parser/pkg/ast" ) func TestPrinterPrintFile(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o) n := &ast.Root{ Stmts: []ast.Vertex{ &ast.StmtNamespace{ Name: &ast.Name{ Parts: []ast.Vertex{ &ast.NamePart{Value: []byte("Foo")}, }, }, }, &ast.StmtClass{ Modifiers: []ast.Vertex{&ast.Identifier{Value: []byte("abstract")}}, Name: &ast.Name{ Parts: []ast.Vertex{ &ast.NamePart{Value: []byte("Bar")}, }, }, Extends: &ast.Name{ Parts: []ast.Vertex{ &ast.NamePart{Value: []byte("Baz")}, }, }, Stmts: []ast.Vertex{ &ast.StmtClassMethod{ Modifiers: []ast.Vertex{&ast.Identifier{Value: []byte("public")}}, Name: &ast.Identifier{Value: []byte("greet")}, Stmt: &ast.StmtStmtList{ Stmts: []ast.Vertex{ &ast.StmtEcho{ Exprs: []ast.Vertex{ &ast.ScalarString{Value: []byte("'Hello world'")}, }, }, }, }, }, }, }, }, } n.Accept(p) expected := `HTML")}, &ast.StmtEcho{ Exprs: []ast.Vertex{ &ast.ScalarString{ Value: []byte(`"a"`), }, }, }, &ast.StmtInlineHtml{Value: []byte("
HTML
")}, &ast.StmtEcho{ Exprs: []ast.Vertex{ &ast.ScalarString{ Value: []byte(`"b"`), }, }, }, }, } n.Accept(p) expected := `
HTML
HTML
>=$b` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } // binary func TestPrinterPrintBinaryBitwiseAnd(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprBinaryBitwiseAnd{ Left: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }, Right: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$b")}, }, } n.Accept(p) expected := `$a&$b` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintBinaryBitwiseOr(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprBinaryBitwiseOr{ Left: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }, Right: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$b")}, }, } n.Accept(p) expected := `$a|$b` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintBinaryBitwiseXor(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprBinaryBitwiseXor{ Left: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }, Right: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$b")}, }, } n.Accept(p) expected := `$a^$b` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintBinaryBooleanAnd(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprBinaryBooleanAnd{ Left: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }, Right: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$b")}, }, } n.Accept(p) expected := `$a&&$b` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintBinaryBooleanOr(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprBinaryBooleanOr{ Left: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }, Right: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$b")}, }, } n.Accept(p) expected := `$a||$b` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintBinaryCoalesce(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprBinaryCoalesce{ Left: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }, Right: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$b")}, }, } n.Accept(p) expected := `$a??$b` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintBinaryConcat(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprBinaryConcat{ Left: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }, Right: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$b")}, }, } n.Accept(p) expected := `$a.$b` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintBinaryDiv(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprBinaryDiv{ Left: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }, Right: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$b")}, }, } n.Accept(p) expected := `$a/$b` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintBinaryEqual(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprBinaryEqual{ Left: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }, Right: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$b")}, }, } n.Accept(p) expected := `$a==$b` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintBinaryGreaterOrEqual(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprBinaryGreaterOrEqual{ Left: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }, Right: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$b")}, }, } n.Accept(p) expected := `$a>=$b` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintBinaryGreater(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprBinaryGreater{ Left: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }, Right: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$b")}, }, } n.Accept(p) expected := `$a>$b` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintBinaryIdentical(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprBinaryIdentical{ Left: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }, Right: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$b")}, }, } n.Accept(p) expected := `$a===$b` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintBinaryLogicalAnd(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprBinaryLogicalAnd{ Left: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }, Right: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$b")}, }, } n.Accept(p) expected := `$a and$b` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintBinaryLogicalOr(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprBinaryLogicalOr{ Left: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }, Right: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$b")}, }, } n.Accept(p) expected := `$a or$b` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintBinaryLogicalXor(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprBinaryLogicalXor{ Left: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }, Right: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$b")}, }, } n.Accept(p) expected := `$a xor$b` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintBinaryMinus(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprBinaryMinus{ Left: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }, Right: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$b")}, }, } n.Accept(p) expected := `$a-$b` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintBinaryMod(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprBinaryMod{ Left: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }, Right: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$b")}, }, } n.Accept(p) expected := `$a%$b` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintBinaryMul(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprBinaryMul{ Left: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }, Right: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$b")}, }, } n.Accept(p) expected := `$a*$b` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintBinaryNotEqual(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprBinaryNotEqual{ Left: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }, Right: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$b")}, }, } n.Accept(p) expected := `$a!=$b` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintBinaryNotIdentical(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprBinaryNotIdentical{ Left: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }, Right: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$b")}, }, } n.Accept(p) expected := `$a!==$b` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintBinaryPlus(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprBinaryPlus{ Left: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }, Right: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$b")}, }, } n.Accept(p) expected := `$a+$b` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintBinaryPow(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprBinaryPow{ Left: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }, Right: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$b")}, }, } n.Accept(p) expected := `$a**$b` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintBinaryShiftLeft(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprBinaryShiftLeft{ Left: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }, Right: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$b")}, }, } n.Accept(p) expected := `$a<<$b` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintBinaryShiftRight(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprBinaryShiftRight{ Left: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }, Right: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$b")}, }, } n.Accept(p) expected := `$a>>$b` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintBinarySmallerOrEqual(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprBinarySmallerOrEqual{ Left: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }, Right: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$b")}, }, } n.Accept(p) expected := `$a<=$b` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintBinarySmaller(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprBinarySmaller{ Left: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }, Right: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$b")}, }, } n.Accept(p) expected := `$a<$b` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintBinarySpaceship(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprBinarySpaceship{ Left: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }, Right: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$b")}, }, } n.Accept(p) expected := `$a<=>$b` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } // cast func TestPrinterPrintArray(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprCastArray{ Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$var")}, }, } n.Accept(p) expected := `(array)$var` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintBool(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprCastBool{ Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$var")}, }, } n.Accept(p) expected := `(bool)$var` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintDouble(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprCastDouble{ Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$var")}, }, } n.Accept(p) expected := `(float)$var` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintInt(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprCastInt{ Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$var")}, }, } n.Accept(p) expected := `(int)$var` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintObject(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprCastObject{ Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$var")}, }, } n.Accept(p) expected := `(object)$var` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintString(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprCastString{ Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$var")}, }, } n.Accept(p) expected := `(string)$var` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintUnset(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprCastUnset{ Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$var")}, }, } n.Accept(p) expected := `(unset)$var` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } // expr func TestPrinterPrintExprArrayDimFetch(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprArrayDimFetch{ Var: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$var")}, }, Dim: &ast.ScalarLnumber{Value: []byte("1")}, } n.Accept(p) expected := `$var[1]` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintExprArrayItemWithKey(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprArrayItem{ Key: &ast.ScalarString{Value: []byte("'Hello'")}, Val: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$world")}, }, } n.Accept(p) expected := `'Hello'=>$world` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintExprArrayItem(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprArrayItem{ Val: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$world")}, }, } n.Accept(p) expected := `$world` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintExprArrayItem_Reference(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprArrayItem{ AmpersandTkn: &token.Token{ Value: []byte("&"), }, Val: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$world")}, }, } n.Accept(p) expected := `&$world` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintExprArrayItemUnpack(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprArrayItem{ EllipsisTkn: &token.Token{ Value: []byte("..."), }, Val: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$world")}, }, } n.Accept(p) expected := `...$world` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintExprArray(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprArray{ ArrayTkn: &token.Token{ Value: []byte("array"), }, Items: []ast.Vertex{ &ast.ExprArrayItem{ Key: &ast.ScalarString{Value: []byte("'Hello'")}, Val: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$world")}, }, }, &ast.ExprArrayItem{ Key: &ast.ScalarLnumber{Value: []byte("2")}, AmpersandTkn: &token.Token{ Value: []byte("&"), }, Val: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$var")}, }, }, &ast.ExprArrayItem{ Val: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$var")}, }, }, }, } n.Accept(p) expected := `array('Hello'=>$world,2=>&$var,$var)` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintExprBitwiseNot(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprBitwiseNot{ Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$var")}, }, } n.Accept(p) expected := `~$var` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintExprBooleanNot(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprBooleanNot{ Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$var")}, }, } n.Accept(p) expected := `!$var` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintExprBracket(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprBooleanNot{ Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$var")}, }, } n.Accept(p) expected := `!$var` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintExprClassConstFetch(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprClassConstFetch{ Class: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$var")}, }, Const: &ast.Identifier{ Value: []byte("CONST"), }, } n.Accept(p) expected := `$var::CONST` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintExprClone(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprClone{ Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$var")}, }, } n.Accept(p) expected := `clone$var` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintExprClosureUse(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprClosureUse{ Var: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$foo")}, }, } n.Accept(p) expected := `$foo` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintExprClosureUse_Reference(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprClosureUse{ AmpersandTkn: &token.Token{ Value: []byte("&"), }, Var: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$foo")}, }, } n.Accept(p) expected := `&$foo` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintExprClosure(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprClosure{ StaticTkn: &token.Token{ Value: []byte("static"), }, AmpersandTkn: &token.Token{ Value: []byte("&"), }, Params: []ast.Vertex{ &ast.Parameter{ Var: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$var")}, }, }, }, Uses: []ast.Vertex{ &ast.ExprClosureUse{ AmpersandTkn: &token.Token{ Value: []byte("&"), }, Var: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }, }, &ast.ExprClosureUse{ Var: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$b")}, }, }, }, ReturnType: &ast.NameFullyQualified{ Parts: []ast.Vertex{&ast.NamePart{Value: []byte("Foo")}}, }, Stmts: []ast.Vertex{ &ast.StmtExpression{Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }}, }, } n.Accept(p) expected := `static function&($var)use(&$a,$b):\Foo{$a;}` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintExprArrowFunction(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.StmtExpression{ Expr: &ast.ExprArrowFunction{ StaticTkn: &token.Token{ Value: []byte("static"), }, AmpersandTkn: &token.Token{ Value: []byte("&"), }, Params: []ast.Vertex{ &ast.Parameter{ AmpersandTkn: &token.Token{ Value: []byte("&"), }, Var: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$var")}, }, }, }, ReturnType: &ast.NameFullyQualified{ Parts: []ast.Vertex{&ast.NamePart{Value: []byte("Foo")}}, }, Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }, }, } n.Accept(p) expected := `static fn&(&$var):\Foo=>$a;` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintExprConstFetch(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprConstFetch{ Const: &ast.Name{Parts: []ast.Vertex{&ast.NamePart{Value: []byte("null")}}}, } n.Accept(p) expected := "null" actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintEmpty(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprEmpty{ Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$var")}, }, } n.Accept(p) expected := `empty($var)` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrettyPrinterrorSuppress(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprErrorSuppress{ Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$var")}, }, } n.Accept(p) expected := `@$var` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintEval(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprEval{ Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$var")}, }, } n.Accept(p) expected := `eval($var)` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintExit(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprExit{ Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$var")}, }, } n.Accept(p) expected := `exit$var` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintDie(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprExit{ ExitTkn: &token.Token{ Value: []byte("die"), }, Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$var")}, }, } n.Accept(p) expected := `die$var` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintFunctionCall(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprFunctionCall{ Function: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$var")}, }, Args: []ast.Vertex{ &ast.Argument{ AmpersandTkn: &token.Token{ Value: []byte("&"), }, Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }, }, &ast.Argument{ VariadicTkn: &token.Token{ Value: []byte("..."), }, Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$b")}, }, }, &ast.Argument{ Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$c")}, }, }, }, } n.Accept(p) expected := `$var(&$a,...$b,$c)` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintInclude(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprInclude{ Expr: &ast.ScalarString{Value: []byte("'path'")}, } n.Accept(p) expected := `include'path'` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintIncludeOnce(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprIncludeOnce{ Expr: &ast.ScalarString{Value: []byte("'path'")}, } n.Accept(p) expected := `include_once'path'` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintInstanceOf(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprInstanceOf{ Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$var")}, }, Class: &ast.Name{Parts: []ast.Vertex{&ast.NamePart{Value: []byte("Foo")}}}, } n.Accept(p) expected := `$var instanceof Foo` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintIsset(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprIsset{ Vars: []ast.Vertex{ &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }, &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$b")}, }, }, } n.Accept(p) expected := `isset($a,$b)` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintList(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprList{ Items: []ast.Vertex{ &ast.ExprArrayItem{ Val: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }, }, &ast.ExprArrayItem{ Val: &ast.ExprList{ Items: []ast.Vertex{ &ast.ExprArrayItem{ Val: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$b")}, }, }, &ast.ExprArrayItem{ Val: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$c")}, }, }, }, }, }, }, } n.Accept(p) expected := `list($a,list($b,$c))` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintMethodCall(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprMethodCall{ Var: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$foo")}, }, Method: &ast.Identifier{Value: []byte("bar")}, Args: []ast.Vertex{ &ast.Argument{ Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }, }, &ast.Argument{ Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$b")}, }, }, }, } n.Accept(p) expected := `$foo->bar($a,$b)` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintNew(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprNew{ Class: &ast.Name{ Parts: []ast.Vertex{ &ast.NamePart{ Value: []byte("Foo"), }, }, }, Args: []ast.Vertex{ &ast.Argument{ Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }, }, &ast.Argument{ Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$b")}, }, }, }, } n.Accept(p) expected := `new Foo($a,$b)` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintPostDec(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprPostDec{ Var: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$var")}, }, } n.Accept(p) expected := `$var--` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintPostInc(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprPostInc{ Var: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$var")}, }, } n.Accept(p) expected := `$var++` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintPreDec(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprPreDec{ Var: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$var")}, }, } n.Accept(p) expected := `--$var` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintPreInc(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprPreInc{ Var: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$var")}, }, } n.Accept(p) expected := `++$var` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintPrint(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprPrint{ Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$var")}, }, } n.Accept(p) expected := `print$var` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintPropertyFetch(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprPropertyFetch{ Var: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$foo")}, }, Prop: &ast.Identifier{Value: []byte("bar")}, } n.Accept(p) expected := `$foo->bar` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintRequire(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprRequire{ Expr: &ast.ScalarString{Value: []byte("'path'")}, } n.Accept(p) expected := `require'path'` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintRequireOnce(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprRequireOnce{ Expr: &ast.ScalarString{Value: []byte("'path'")}, } n.Accept(p) expected := `require_once'path'` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintShellExec(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprShellExec{ Parts: []ast.Vertex{ &ast.ScalarEncapsedStringPart{Value: []byte("hello ")}, &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$world")}, }, &ast.ScalarEncapsedStringPart{Value: []byte("!")}, }, } n.Accept(p) expected := "`hello $world!`" actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintExprShortArray(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprArray{ Items: []ast.Vertex{ &ast.ExprArrayItem{ Key: &ast.ScalarString{Value: []byte("'Hello'")}, Val: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$world")}, }, }, &ast.ExprArrayItem{ Key: &ast.ScalarLnumber{Value: []byte("2")}, AmpersandTkn: &token.Token{ Value: []byte("&"), }, Val: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$var")}, }, }, &ast.ExprArrayItem{ Val: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$var")}, }, }, }, } n.Accept(p) expected := `['Hello'=>$world,2=>&$var,$var]` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintShortList(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprList{ OpenBracketTkn: &token.Token{ Value: []byte("["), }, Items: []ast.Vertex{ &ast.ExprArrayItem{ Val: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }, }, &ast.ExprArrayItem{ Val: &ast.ExprList{ Items: []ast.Vertex{ &ast.ExprArrayItem{ Val: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$b")}, }, }, &ast.ExprArrayItem{ Val: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$c")}, }, }, }, }, }, }, CloseBracketTkn: &token.Token{ Value: []byte("]"), }, } n.Accept(p) expected := `[$a,list($b,$c)]` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintStaticCall(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprStaticCall{ Class: &ast.Identifier{Value: []byte("Foo")}, Call: &ast.Identifier{Value: []byte("bar")}, Args: []ast.Vertex{ &ast.Argument{ Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }, }, &ast.Argument{ Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$b")}, }, }, }, } n.Accept(p) expected := `Foo::bar($a,$b)` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintStaticPropertyFetch(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprStaticPropertyFetch{ Class: &ast.Identifier{Value: []byte("Foo")}, Prop: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$bar")}, }, } n.Accept(p) expected := `Foo::$bar` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintTernary(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprTernary{ Cond: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }, IfFalse: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$b")}, }, } n.Accept(p) expected := `$a?:$b` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintTernaryFull(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprTernary{ Cond: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }, IfTrue: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$b")}, }, IfFalse: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$c")}, }, } n.Accept(p) expected := `$a?$b:$c` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintUnaryMinus(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprUnaryMinus{ Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$var")}, }, } n.Accept(p) expected := `-$var` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintUnaryPlus(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprUnaryPlus{ Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$var")}, }, } n.Accept(p) expected := `+$var` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintVariable(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprVariable{ DollarTkn: &token.Token{ Value: []byte("$"), }, OpenCurlyBracketTkn: &token.Token{ Value: []byte("{"), }, Name: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$var")}, }, CloseCurlyBracketTkn: &token.Token{ Value: []byte("}"), }, } n.Accept(p) expected := `${$var}` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintYieldFrom(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprYieldFrom{ Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$var")}, }, } n.Accept(p) expected := `yield from$var` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintYield(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprYield{ Val: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$var")}, }, } n.Accept(p) expected := `yield$var` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintYieldFull(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.ExprYield{ Key: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$k")}, }, Val: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$var")}, }, } n.Accept(p) expected := `yield$k=>$var` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } // stmt func TestPrinterPrintAltElseIf(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.StmtElseIf{ Cond: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }, ColonTkn: &token.Token{ Value: []byte(":"), }, Stmt: &ast.StmtStmtList{ Stmts: []ast.Vertex{ &ast.StmtExpression{Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$b")}, }}, }, }, } n.Accept(p) expected := `elseif($a):$b;` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintAltElseIfEmpty(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.StmtElseIf{ Cond: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }, ColonTkn: &token.Token{ Value: []byte(":"), }, Stmt: &ast.StmtStmtList{}, } n.Accept(p) expected := `elseif($a):` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintAltElse(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.StmtElse{ ColonTkn: &token.Token{ Value: []byte(":"), }, Stmt: &ast.StmtStmtList{ Stmts: []ast.Vertex{ &ast.StmtExpression{Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$b")}, }}, }, }, } n.Accept(p) expected := `else:$b;` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintAltElseEmpty(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.StmtElse{ ColonTkn: &token.Token{ Value: []byte(":"), }, Stmt: &ast.StmtStmtList{}, } n.Accept(p) expected := `else:` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintAltFor(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.StmtFor{ Init: []ast.Vertex{ &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }, }, Cond: []ast.Vertex{ &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$b")}, }, }, Loop: []ast.Vertex{ &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$c")}, }, }, ColonTkn: &token.Token{ Value: []byte(":"), }, Stmt: &ast.StmtStmtList{ Stmts: []ast.Vertex{ &ast.StmtExpression{Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$d")}, }}, }, }, } n.Accept(p) expected := `for($a;$b;$c):$d;endfor;` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintAltForeach(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.StmtForeach{ Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$var")}, }, Key: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$key")}, }, Var: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$val")}, }, ColonTkn: &token.Token{ Value: []byte(":"), }, Stmt: &ast.StmtStmtList{ Stmts: []ast.Vertex{ &ast.StmtExpression{Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$d")}, }}, }, }, } n.Accept(p) expected := `foreach($var as$key=>$val):$d;endforeach;` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintAltForeach_Reference(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.StmtForeach{ Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$var")}, }, Key: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$key")}, }, AmpersandTkn: &token.Token{ Value: []byte("&"), }, Var: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$val")}, }, ColonTkn: &token.Token{ Value: []byte(":"), }, Stmt: &ast.StmtStmtList{ Stmts: []ast.Vertex{ &ast.StmtExpression{Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$d")}, }}, }, }, } n.Accept(p) expected := `foreach($var as$key=>&$val):$d;endforeach;` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintAltIf(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.StmtIf{ Cond: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }, ColonTkn: &token.Token{ Value: []byte(":"), }, Stmt: &ast.StmtStmtList{ Stmts: []ast.Vertex{ &ast.StmtExpression{Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$d")}, }}, }, }, ElseIf: []ast.Vertex{ &ast.StmtElseIf{ Cond: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$b")}, }, ColonTkn: &token.Token{ Value: []byte(":"), }, Stmt: &ast.StmtStmtList{ Stmts: []ast.Vertex{ &ast.StmtExpression{Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$b")}, }}, }, }, }, &ast.StmtElseIf{ Cond: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$c")}, }, ColonTkn: &token.Token{ Value: []byte(":"), }, Stmt: &ast.StmtStmtList{}, }, }, Else: &ast.StmtElse{ ColonTkn: &token.Token{ Value: []byte(":"), }, Stmt: &ast.StmtStmtList{ Stmts: []ast.Vertex{ &ast.StmtExpression{Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$b")}, }}, }, }, }, } n.Accept(p) expected := `if($a):$d;elseif($b):$b;elseif($c):else:$b;endif;` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintStmtAltSwitch(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.StmtSwitch{ Cond: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$var")}, }, ColonTkn: &token.Token{ Value: []byte(":"), }, Cases: []ast.Vertex{ &ast.StmtCase{ Cond: &ast.ScalarString{Value: []byte("'a'")}, Stmts: []ast.Vertex{ &ast.StmtExpression{Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }}, }, }, &ast.StmtCase{ Cond: &ast.ScalarString{Value: []byte("'b'")}, Stmts: []ast.Vertex{ &ast.StmtExpression{Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$b")}, }}, }, }, }, } n.Accept(p) expected := `switch($var):case'a':$a;case'b':$b;endswitch;` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintAltWhile(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.StmtWhile{ Cond: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }, ColonTkn: &token.Token{ Value: []byte(":"), }, Stmt: &ast.StmtStmtList{ Stmts: []ast.Vertex{ &ast.StmtExpression{Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$b")}, }}, }, }, } n.Accept(p) expected := `while($a):$b;endwhile;` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintStmtBreak(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.StmtBreak{ Expr: &ast.ScalarLnumber{ Value: []byte("1"), }, } n.Accept(p) expected := "break 1;" actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintStmtCase(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.StmtCase{ Cond: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }, Stmts: []ast.Vertex{ &ast.StmtExpression{Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }}, }, } n.Accept(p) expected := `case$a:$a;` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintStmtCaseEmpty(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.StmtCase{ Cond: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }, Stmts: []ast.Vertex{}, } n.Accept(p) expected := "case$a:" actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintStmtCatch(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.StmtCatch{ Types: []ast.Vertex{ &ast.Name{Parts: []ast.Vertex{&ast.NamePart{Value: []byte("Exception")}}}, &ast.NameFullyQualified{Parts: []ast.Vertex{&ast.NamePart{Value: []byte("RuntimeException")}}}, }, Var: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$e")}, }, Stmts: []ast.Vertex{ &ast.StmtExpression{Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }}, }, } n.Accept(p) expected := `catch(Exception|\RuntimeException$e){$a;}` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintStmtClassMethod(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.StmtClassMethod{ Modifiers: []ast.Vertex{&ast.Identifier{Value: []byte("public")}}, AmpersandTkn: &token.Token{ Value: []byte("&"), }, Name: &ast.Identifier{Value: []byte("foo")}, Params: []ast.Vertex{ &ast.Parameter{ Type: &ast.Nullable{Expr: &ast.Name{Parts: []ast.Vertex{&ast.NamePart{Value: []byte("int")}}}}, AmpersandTkn: &token.Token{ Value: []byte("&"), }, Var: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }, DefaultValue: &ast.ExprConstFetch{Const: &ast.Name{Parts: []ast.Vertex{&ast.NamePart{Value: []byte("null")}}}}, }, &ast.Parameter{ VariadicTkn: &token.Token{ Value: []byte("..."), }, Var: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$b")}, }, }, }, ReturnType: &ast.Name{ Parts: []ast.Vertex{&ast.NamePart{Value: []byte("void")}}, }, Stmt: &ast.StmtStmtList{ Stmts: []ast.Vertex{ &ast.StmtExpression{Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }}, }, }, } n.Accept(p) expected := `public function&foo(?int&$a=null,...$b):void{$a;}` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintStmtAbstractClassMethod(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.StmtClassMethod{ Modifiers: []ast.Vertex{ &ast.Identifier{Value: []byte("public")}, &ast.Identifier{Value: []byte("static")}, }, AmpersandTkn: &token.Token{ Value: []byte("&"), }, Name: &ast.Identifier{Value: []byte("foo")}, Params: []ast.Vertex{ &ast.Parameter{ Type: &ast.Nullable{Expr: &ast.Name{Parts: []ast.Vertex{&ast.NamePart{Value: []byte("int")}}}}, AmpersandTkn: &token.Token{ Value: []byte("&"), }, Var: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }, DefaultValue: &ast.ExprConstFetch{Const: &ast.Name{Parts: []ast.Vertex{&ast.NamePart{Value: []byte("null")}}}}, }, &ast.Parameter{ VariadicTkn: &token.Token{ Value: []byte("..."), }, Var: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$b")}, }, }, }, ReturnType: &ast.Name{ Parts: []ast.Vertex{&ast.NamePart{Value: []byte("void")}}, }, Stmt: &ast.StmtNop{}, } n.Accept(p) expected := `public static function&foo(?int&$a=null,...$b):void;` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintStmtClass(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.StmtClass{ Modifiers: []ast.Vertex{&ast.Identifier{Value: []byte("abstract")}}, Name: &ast.Identifier{Value: []byte("Foo")}, Extends: &ast.Name{Parts: []ast.Vertex{&ast.NamePart{Value: []byte("Bar")}}}, Implements: []ast.Vertex{ &ast.Name{Parts: []ast.Vertex{&ast.NamePart{Value: []byte("Baz")}}}, &ast.Name{Parts: []ast.Vertex{&ast.NamePart{Value: []byte("Quuz")}}}, }, Stmts: []ast.Vertex{ &ast.StmtClassConstList{ Modifiers: []ast.Vertex{ &ast.Identifier{Value: []byte("public")}, &ast.Identifier{Value: []byte("static")}, }, Consts: []ast.Vertex{ &ast.StmtConstant{ Name: &ast.Identifier{Value: []byte("FOO")}, Expr: &ast.ScalarString{Value: []byte("'bar'")}, }, }, }, }, } n.Accept(p) expected := `abstract class Foo extends Bar implements Baz,Quuz{public static const FOO='bar';}` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintStmtAnonymousClass(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.StmtClass{ Modifiers: []ast.Vertex{&ast.Identifier{Value: []byte("abstract")}}, Args: []ast.Vertex{ &ast.Argument{ Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }, }, &ast.Argument{ Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$b")}, }, }, }, Extends: &ast.Name{Parts: []ast.Vertex{&ast.NamePart{Value: []byte("Bar")}}}, Implements: []ast.Vertex{ &ast.Name{Parts: []ast.Vertex{&ast.NamePart{Value: []byte("Baz")}}}, &ast.Name{Parts: []ast.Vertex{&ast.NamePart{Value: []byte("Quuz")}}}, }, Stmts: []ast.Vertex{ &ast.StmtClassConstList{ Modifiers: []ast.Vertex{&ast.Identifier{Value: []byte("public")}}, Consts: []ast.Vertex{ &ast.StmtConstant{ Name: &ast.Identifier{Value: []byte("FOO")}, Expr: &ast.ScalarString{Value: []byte("'bar'")}, }, }, }, }, } n.Accept(p) expected := `abstract class($a,$b)extends Bar implements Baz,Quuz{public const FOO='bar';}` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintStmtClassConstList(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.StmtClassConstList{ Modifiers: []ast.Vertex{&ast.Identifier{Value: []byte("public")}}, Consts: []ast.Vertex{ &ast.StmtConstant{ Name: &ast.Identifier{Value: []byte("FOO")}, Expr: &ast.ScalarString{Value: []byte("'a'")}, }, &ast.StmtConstant{ Name: &ast.Identifier{Value: []byte("BAR")}, Expr: &ast.ScalarString{Value: []byte("'b'")}, }, }, } n.Accept(p) expected := `public const FOO='a',BAR='b';` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintStmtConstList(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.StmtConstList{ Consts: []ast.Vertex{ &ast.StmtConstant{ Name: &ast.Identifier{Value: []byte("FOO")}, Expr: &ast.ScalarString{Value: []byte("'a'")}, }, &ast.StmtConstant{ Name: &ast.Identifier{Value: []byte("BAR")}, Expr: &ast.ScalarString{Value: []byte("'b'")}, }, }, } n.Accept(p) expected := `const FOO='a',BAR='b';` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintStmtConstant(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.StmtConstant{ Name: &ast.Identifier{Value: []byte("FOO")}, Expr: &ast.ScalarString{Value: []byte("'BAR'")}, } n.Accept(p) expected := "FOO='BAR'" actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintStmtContinue(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.StmtContinue{ Expr: &ast.ScalarLnumber{ Value: []byte("1"), }, } n.Accept(p) expected := `continue 1;` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintStmtDeclareStmts(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.StmtDeclare{ Consts: []ast.Vertex{ &ast.StmtConstant{ Name: &ast.Identifier{Value: []byte("FOO")}, Expr: &ast.ScalarString{Value: []byte("'bar'")}, }, }, Stmt: &ast.StmtStmtList{ Stmts: []ast.Vertex{ &ast.StmtNop{}, }, }, } n.Accept(p) expected := `declare(FOO='bar'){;}` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintStmtDeclareExpr(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.StmtDeclare{ Consts: []ast.Vertex{ &ast.StmtConstant{ Name: &ast.Identifier{Value: []byte("FOO")}, Expr: &ast.ScalarString{Value: []byte("'bar'")}, }, }, Stmt: &ast.StmtExpression{Expr: &ast.ScalarString{Value: []byte("'bar'")}}, } n.Accept(p) expected := `declare(FOO='bar')'bar';` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintStmtDeclareNop(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.StmtDeclare{ Consts: []ast.Vertex{ &ast.StmtConstant{ Name: &ast.Identifier{Value: []byte("FOO")}, Expr: &ast.ScalarString{Value: []byte("'bar'")}, }, }, Stmt: &ast.StmtNop{}, } n.Accept(p) expected := `declare(FOO='bar');` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintStmtDefalut(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.StmtDefault{ Stmts: []ast.Vertex{ &ast.StmtExpression{Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }}, }, } n.Accept(p) expected := `default:$a;` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintStmtDefalutEmpty(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.StmtDefault{ Stmts: []ast.Vertex{}, } n.Accept(p) expected := `default:` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintStmtDo_Expression(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.StmtDo{ Cond: &ast.ScalarLnumber{Value: []byte("1")}, Stmt: &ast.StmtExpression{ Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }, }, } n.Accept(p) expected := `do$a;while(1);` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintStmtDo_StmtList(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.StmtDo{ Cond: &ast.ScalarLnumber{Value: []byte("1")}, Stmt: &ast.StmtStmtList{ Stmts: []ast.Vertex{ &ast.StmtExpression{Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }}, }, }, } n.Accept(p) expected := `do{$a;}while(1);` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintStmtEchoHtmlState(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o) n := &ast.Root{ Stmts: []ast.Vertex{ &ast.StmtEcho{ Exprs: []ast.Vertex{ &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }, &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$b")}, }, }, }, }, } n.Accept(p) expected := `$v){;}` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintStmtForeach_Reference(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.StmtForeach{ Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }, Key: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$k")}, }, AmpersandTkn: &token.Token{ Value: []byte("&"), }, Var: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$v")}, }, Stmt: &ast.StmtStmtList{ Stmts: []ast.Vertex{ &ast.StmtNop{}, }, }, } n.Accept(p) expected := `foreach($a as$k=>&$v){;}` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintStmtFunction(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.StmtFunction{ AmpersandTkn: &token.Token{ Value: []byte("&"), }, Name: &ast.Identifier{Value: []byte("foo")}, Params: []ast.Vertex{ &ast.Parameter{ AmpersandTkn: &token.Token{ Value: []byte("&"), }, Var: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$var")}, }, }, }, ReturnType: &ast.NameFullyQualified{ Parts: []ast.Vertex{&ast.NamePart{Value: []byte("Foo")}}, }, Stmts: []ast.Vertex{ &ast.StmtNop{}, }, } n.Accept(p) expected := `function&foo(&$var):\Foo{;}` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintStmtGlobal(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.StmtGlobal{ Vars: []ast.Vertex{ &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }, &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$b")}, }, }, } n.Accept(p) expected := `global$a,$b;` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintStmtGoto(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.StmtGoto{ Label: &ast.Identifier{Value: []byte("FOO")}, } n.Accept(p) expected := `goto FOO;` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintHaltCompiler(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.StmtHaltCompiler{} n.Accept(p) expected := `__halt_compiler();` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintIfExpression(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.StmtIf{ Cond: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }, Stmt: &ast.StmtExpression{ Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$b")}, }, }, ElseIf: []ast.Vertex{ &ast.StmtElseIf{ Cond: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$c")}, }, Stmt: &ast.StmtStmtList{ Stmts: []ast.Vertex{ &ast.StmtExpression{ Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$d")}, }, }, }, }, }, &ast.StmtElseIf{ Cond: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$e")}, }, Stmt: &ast.StmtNop{}, }, }, Else: &ast.StmtElse{ Stmt: &ast.StmtExpression{ Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$f")}, }, }, }, } n.Accept(p) expected := `if($a)$b;elseif($c){$d;}elseif($e);else$f;` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintIfStmtList(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.StmtIf{ Cond: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }, Stmt: &ast.StmtStmtList{ Stmts: []ast.Vertex{ &ast.StmtExpression{ Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$b")}, }, }, }, }, } n.Accept(p) expected := `if($a){$b;}` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintIfNop(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.StmtIf{ Cond: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }, Stmt: &ast.StmtNop{}, } n.Accept(p) expected := `if($a);` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintInlineHtml(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.Root{ Stmts: []ast.Vertex{ &ast.StmtInlineHtml{ Value: []byte("test"), }, }, } n.Accept(p) expected := `test` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintInterface(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.StmtInterface{ Name: &ast.Name{Parts: []ast.Vertex{&ast.NamePart{Value: []byte("Foo")}}}, Extends: []ast.Vertex{ &ast.Name{Parts: []ast.Vertex{&ast.NamePart{Value: []byte("Bar")}}}, &ast.Name{Parts: []ast.Vertex{&ast.NamePart{Value: []byte("Baz")}}}, }, Stmts: []ast.Vertex{ &ast.StmtClassMethod{ Modifiers: []ast.Vertex{&ast.Identifier{Value: []byte("public")}}, Name: &ast.Identifier{Value: []byte("foo")}, Params: []ast.Vertex{}, Stmt: &ast.StmtStmtList{ Stmts: []ast.Vertex{ &ast.StmtExpression{Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }}, }, }, }, }, } n.Accept(p) expected := `interface Foo extends Bar,Baz{public function foo(){$a;}}` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintLabel(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.StmtLabel{ Name: &ast.Identifier{Value: []byte("FOO")}, } n.Accept(p) expected := `FOO:` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintNamespace(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.StmtNamespace{ Name: &ast.Name{Parts: []ast.Vertex{&ast.NamePart{Value: []byte("Foo")}}}, } n.Accept(p) expected := `namespace Foo;` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintNamespaceWithStmts(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.StmtNamespace{ Name: &ast.Name{Parts: []ast.Vertex{&ast.NamePart{Value: []byte("Foo")}}}, Stmts: []ast.Vertex{ &ast.StmtExpression{Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }}, }, } n.Accept(p) expected := `namespace Foo{$a;}` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintNop(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.StmtNop{} n.Accept(p) expected := `;` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintPropertyList(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.StmtPropertyList{ Modifiers: []ast.Vertex{ &ast.Identifier{Value: []byte("public")}, &ast.Identifier{Value: []byte("static")}, }, Type: &ast.Name{ Parts: []ast.Vertex{ &ast.NamePart{ Value: []byte("Foo"), }, }, }, Props: []ast.Vertex{ &ast.StmtProperty{ Var: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }, Expr: &ast.ScalarString{Value: []byte("'a'")}, }, &ast.StmtProperty{ Var: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$b")}, }, }, }, } n.Accept(p) expected := `public static Foo$a='a',$b;` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintProperty(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.StmtProperty{ Var: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }, Expr: &ast.ScalarLnumber{Value: []byte("1")}, } n.Accept(p) expected := `$a=1` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintReturn(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.StmtReturn{ Expr: &ast.ScalarLnumber{Value: []byte("1")}, } n.Accept(p) expected := `return 1;` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintStaticVar(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.StmtStaticVar{ Var: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }, Expr: &ast.ScalarLnumber{Value: []byte("1")}, } n.Accept(p) expected := `$a=1` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintStatic(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.StmtStatic{ Vars: []ast.Vertex{ &ast.StmtStaticVar{ Var: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }, }, &ast.StmtStaticVar{ Var: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$b")}, }, }, }, } n.Accept(p) expected := `static$a,$b;` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintStmtList(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.StmtStmtList{ Stmts: []ast.Vertex{ &ast.StmtExpression{Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }}, &ast.StmtExpression{Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$b")}, }}, }, } n.Accept(p) expected := `{$a;$b;}` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintStmtListNested(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.StmtStmtList{ Stmts: []ast.Vertex{ &ast.StmtExpression{Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }}, &ast.StmtStmtList{ Stmts: []ast.Vertex{ &ast.StmtExpression{Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$b")}, }}, &ast.StmtStmtList{ Stmts: []ast.Vertex{ &ast.StmtExpression{Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$c")}, }}, }, }, }, }, }, } n.Accept(p) expected := `{$a;{$b;{$c;}}}` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintStmtSwitch(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.StmtSwitch{ Cond: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$var")}, }, Cases: []ast.Vertex{ &ast.StmtCase{ Cond: &ast.ScalarString{Value: []byte("'a'")}, Stmts: []ast.Vertex{ &ast.StmtExpression{Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }}, }, }, &ast.StmtCase{ Cond: &ast.ScalarString{Value: []byte("'b'")}, Stmts: []ast.Vertex{ &ast.StmtExpression{Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$b")}, }}, }, }, }, } n.Accept(p) expected := `switch($var){case'a':$a;case'b':$b;}` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintStmtThrow(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.StmtThrow{ Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$var")}, }, } n.Accept(p) expected := `throw$var;` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintStmtTraitUseAlias(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.StmtTraitUseAlias{ Trait: &ast.Name{Parts: []ast.Vertex{&ast.NamePart{Value: []byte("Foo")}}}, Method: &ast.Identifier{Value: []byte("a")}, Modifier: &ast.Identifier{Value: []byte("public")}, Alias: &ast.Identifier{Value: []byte("b")}, } n.Accept(p) expected := `Foo::a as public b;` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintStmtTraitUsePrecedence(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.StmtTraitUsePrecedence{ Trait: &ast.Name{Parts: []ast.Vertex{&ast.NamePart{Value: []byte("Foo")}}}, Method: &ast.Identifier{Value: []byte("a")}, Insteadof: []ast.Vertex{ &ast.Name{Parts: []ast.Vertex{&ast.NamePart{Value: []byte("Bar")}}}, &ast.Name{Parts: []ast.Vertex{&ast.NamePart{Value: []byte("Baz")}}}, }, } n.Accept(p) expected := `Foo::a insteadof Bar,Baz;` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintStmtTraitUse(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.StmtTraitUse{ Traits: []ast.Vertex{ &ast.Name{Parts: []ast.Vertex{&ast.NamePart{Value: []byte("Foo")}}}, &ast.Name{Parts: []ast.Vertex{&ast.NamePart{Value: []byte("Bar")}}}, }, } n.Accept(p) expected := `use Foo,Bar;` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintStmtTraitAdaptations(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.StmtTraitUse{ Traits: []ast.Vertex{ &ast.Name{Parts: []ast.Vertex{&ast.NamePart{Value: []byte("Foo")}}}, &ast.Name{Parts: []ast.Vertex{&ast.NamePart{Value: []byte("Bar")}}}, }, Adaptations: []ast.Vertex{ &ast.StmtTraitUseAlias{ Trait: &ast.Name{Parts: []ast.Vertex{&ast.NamePart{Value: []byte("Foo")}}}, Method: &ast.Identifier{Value: []byte("a")}, Alias: &ast.Identifier{Value: []byte("b")}, }, }, } n.Accept(p) expected := `use Foo,Bar{Foo::a as b;}` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintTrait(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.StmtTrait{ Name: &ast.Name{Parts: []ast.Vertex{&ast.NamePart{Value: []byte("Foo")}}}, Stmts: []ast.Vertex{ &ast.StmtClassMethod{ Modifiers: []ast.Vertex{&ast.Identifier{Value: []byte("public")}}, Name: &ast.Identifier{Value: []byte("foo")}, Params: []ast.Vertex{}, Stmt: &ast.StmtStmtList{ Stmts: []ast.Vertex{ &ast.StmtExpression{Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }}, }, }, }, }, } n.Accept(p) expected := `trait Foo{public function foo(){$a;}}` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintStmtTry(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.StmtTry{ Stmts: []ast.Vertex{ &ast.StmtExpression{Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }}, }, Catches: []ast.Vertex{ &ast.StmtCatch{ Types: []ast.Vertex{ &ast.Name{Parts: []ast.Vertex{&ast.NamePart{Value: []byte("Exception")}}}, &ast.NameFullyQualified{Parts: []ast.Vertex{&ast.NamePart{Value: []byte("RuntimeException")}}}, }, Var: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$e")}, }, Stmts: []ast.Vertex{ &ast.StmtExpression{Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$b")}, }}, }, }, }, Finally: &ast.StmtFinally{ Stmts: []ast.Vertex{ &ast.StmtNop{}, }, }, } n.Accept(p) expected := `try{$a;}catch(Exception|\RuntimeException$e){$b;}finally{;}` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintStmtUnset(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.StmtUnset{ Vars: []ast.Vertex{ &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }, &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$b")}, }, }, } n.Accept(p) expected := `unset($a,$b);` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintUse(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.StmtUseList{ Type: &ast.Identifier{Value: []byte("function")}, Uses: []ast.Vertex{ &ast.StmtUse{ Use: &ast.Name{Parts: []ast.Vertex{&ast.NamePart{Value: []byte("Foo")}}}, Alias: &ast.Identifier{Value: []byte("Bar")}, }, &ast.StmtUse{ Use: &ast.Name{Parts: []ast.Vertex{&ast.NamePart{Value: []byte("Baz")}}}, }, }, } n.Accept(p) expected := `use function Foo as Bar,Baz;` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintStmtGroupUse(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.StmtGroupUseList{ Type: &ast.Identifier{Value: []byte("function")}, Prefix: &ast.Name{Parts: []ast.Vertex{&ast.NamePart{Value: []byte("Foo")}}}, Uses: []ast.Vertex{ &ast.StmtUse{ Use: &ast.Name{Parts: []ast.Vertex{&ast.NamePart{Value: []byte("Foo")}}}, Alias: &ast.Identifier{Value: []byte("Bar")}, }, &ast.StmtUse{ Use: &ast.Name{Parts: []ast.Vertex{&ast.NamePart{Value: []byte("Baz")}}}, }, }, } n.Accept(p) expected := `use function Foo\{Foo as Bar,Baz};` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintUseDeclaration(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.StmtUse{ Type: &ast.Identifier{Value: []byte("function")}, Use: &ast.Name{Parts: []ast.Vertex{&ast.NamePart{Value: []byte("Foo")}}}, Alias: &ast.Identifier{Value: []byte("Bar")}, } n.Accept(p) expected := `function Foo as Bar` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } } func TestPrinterPrintWhileStmtList(t *testing.T) { o := bytes.NewBufferString("") p := printer.NewPrinter(o).WithState(printer.PrinterStatePHP) n := &ast.StmtWhile{ Cond: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }, Stmt: &ast.StmtStmtList{ Stmts: []ast.Vertex{ &ast.StmtExpression{Expr: &ast.ExprVariable{ Name: &ast.Identifier{Value: []byte("$a")}, }}, }, }, } n.Accept(p) expected := `while($a){$a;}` actual := o.String() if expected != actual { t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual) } }