golox/ast/printer_test.go

package ast

import (
	"golox/token"
	"testing"
)

func TestPrintExpr(t *testing.T) {
	tests := []struct {
		name     string
		expr     Expr
		expected string
	}{
		{
			name: "Binary expression",
			expr: &BinaryExpr{
				Left:     &LiteralExpr{Value: 1},
				Operator: token.Token{Type: token.PLUS, Lexeme: "+"},
				Right:    &LiteralExpr{Value: 2},
			},
			expected: "(+ 1 2)",
		},
		{
			name: "Grouping expression",
			expr: &GroupingExpr{
				Expression: &LiteralExpr{Value: 1},
			},
			expected: "(group 1)",
		},
		{
			name:     "Literal expression",
			expr:     &LiteralExpr{Value: 123},
			expected: "123",
		},
		{
			name: "Unary expression",
			expr: &UnaryExpr{
				Operator: token.Token{Type: token.MINUS, Lexeme: "-"},
				Right:    &LiteralExpr{Value: 123},
			},
			expected: "(- 123)",
		},
		{
			name:     "Nil literal expression",
			expr:     &LiteralExpr{Value: nil},
			expected: "nil",
		},
		{
			name:     "Error expression",
			expr:     &ErrorExpr{Value: "error"},
			expected: "error",
		},
		{
			name: "Assign expression",
			expr: &AssignExpr{
				Name:  token.Token{Type: token.IDENTIFIER, Lexeme: "foo"},
				Value: &LiteralExpr{Value: 42},
			},
			expected: "(= foo 42)",
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			printer := NewPrinter()
			result := printer.PrintExpr(tt.expr)
			if result != tt.expected {
				t.Errorf("expected %v, got %v", tt.expected, result)
			}
		})
	}
}

func TestPrintStmts(t *testing.T) {
	tests := []struct {
		name     string
		stmts    []Stmt
		expected string
	}{
		{
			name: "Print statement",
			stmts: []Stmt{
				&PrintStmt{
					Expression: &LiteralExpr{Value: 42},
				},
			},
			expected: "(print 42)\n",
		},
		{
			name: "Expression statement",
			stmts: []Stmt{
				&ExpressionStmt{
					Expression: &LiteralExpr{Value: 42},
				},
			},
			expected: "42\n",
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			printer := NewPrinter()
			result := printer.PrintStmts(tt.stmts)
			if result != tt.expected {
				t.Errorf("expected %v, got %v", tt.expected, result)
			}
		})
	}
}

func TestPrintBlockStmt(t *testing.T) {
	stmt := &BlockStmt{
		Statements: []Stmt{
			&ExpressionStmt{
				Expression: &LiteralExpr{Value: 1},
			},
			&ExpressionStmt{
				Expression: &LiteralExpr{Value: 2},
			},
		},
	}

	printer := NewPrinter()
	result := printer.PrintStmts([]Stmt{stmt})
	expected := "(block\n 1\n 2\n)\n"
	if result != expected {
		t.Errorf("expected %v, got %v", expected, result)
	}
}