Function TestRenderCreateTable_AllFeatures

pkg/formatter/render_test.go:344–372

(t *testing.T)

}

func TestRenderCreateTable_AllFeatures(t *testing.T) {
	stmt := &ast.CreateTableStatement{
		Name:        "t",
		Temporary:   true,
		IfNotExists: true,
		Columns: []ast.ColumnDef{
			{Name: "id", Type: "INT"},
			{Name: "name", Type: "TEXT"},
		},
		Constraints: []ast.TableConstraint{
			{Type: "PRIMARY KEY", Columns: []string{"id"}},
		},
		Inherits:    []string{"parent"},
		PartitionBy: &ast.PartitionBy{Type: "HASH", Columns: []string{"id"}},
		Options:     []ast.TableOption{{Name: "engine", Value: "InnoDB"}},
	}

	readable := fmtStmt(stmt, ast.ReadableStyle())
	for _, want := range []string{"CREATE TEMPORARY TABLE IF NOT EXISTS", "INHERITS", "PARTITION BY"} {
		if !strings.Contains(readable, want) {
			t.Errorf("missing %q in readable: %s", want, readable)
		}
	}

	compact := fmtStmt(stmt, ast.CompactStyle())
	if strings.Contains(compact, "\n") {
		t.Errorf("compact should be single line: %s", compact)
	}
}

func TestRenderCreateTable_Nil(t *testing.T) {
	var stmt *ast.CreateTableStatement