Function TestFormatAST_MultiStatement

pkg/formatter/render_test.go:1035–1056

─── FormatAST ───────────────────────────────────────────────────────────────

(t *testing.T)

// ─── FormatAST ───────────────────────────────────────────────────────────────

func TestFormatAST_MultiStatement(t *testing.T) {
	a := &ast.AST{
		Statements: []ast.Statement{
			&ast.SelectStatement{
				Columns: []ast.Expression{&ast.Identifier{Name: "a"}},
				From:    []ast.TableReference{{Name: "t1"}},
			},
			&ast.SelectStatement{
				Columns: []ast.Expression{&ast.Identifier{Name: "b"}},
				From:    []ast.TableReference{{Name: "t2"}},
			},
		},
	}

	result := formatter.FormatAST(a, ast.CompactStyle())
	if !strings.Contains(result, "SELECT a FROM t1") {
		t.Errorf("should contain first statement, got: %s", result)
	}
	if !strings.Contains(result, "SELECT b FROM t2") {
		t.Errorf("should contain second statement, got: %s", result)
	}
}

func TestFormatAST_Nil(t *testing.T) {
	result := formatter.FormatAST(nil, ast.CompactStyle())