Function TestRenderUpdate_AllClauses

pkg/formatter/render_test.go:254–272

(t *testing.T)

}

func TestRenderUpdate_AllClauses(t *testing.T) {
	stmt := &ast.UpdateStatement{
		With: &ast.WithClause{CTEs: []*ast.CommonTableExpr{
			{Name: "c", Statement: &ast.SelectStatement{Columns: []ast.Expression{&ast.Identifier{Name: "x"}}}},
		}},
		TableName:   "t",
		Alias:       "tt",
		Assignments: []ast.UpdateExpression{{Column: &ast.Identifier{Name: "x"}, Value: &ast.LiteralValue{Value: "1"}}},
		From:        []ast.TableReference{{Name: "other"}},
		Where:       &ast.Identifier{Name: "true"},
		Returning:   []ast.Expression{&ast.Identifier{Name: "id"}},
	}
	result := fmtStmt(stmt, ast.ReadableStyle())
	for _, want := range []string{"WITH", "UPDATE t tt", "SET", "FROM", "WHERE", "RETURNING"} {
		if !strings.Contains(result, want) {
			t.Errorf("missing %q in: %s", want, result)
		}
	}
}

func TestRenderUpdate_Nil(t *testing.T) {
	var stmt *ast.UpdateStatement