Function TestDelete

pkg/sql/ast/dml_test.go:223–296

Test Delete node (DML)

(t *testing.T)

// Test Delete node (DML)
func TestDelete(t *testing.T) {
	tests := []struct {
		name        string
		del         *Delete
		wantLiteral string
		minChildren int
	}{
		{
			name: "simple DELETE",
			del: &Delete{
				Table: TableReference{Name: "users"},
			},
			wantLiteral: "DELETE",
			minChildren: 1,
		},
		{
			name: "DELETE with WHERE",
			del: &Delete{
				Table: TableReference{Name: "users"},
				Where: &BinaryExpression{
					Left:     &Identifier{Name: "id"},
					Operator: "=",
					Right:    &LiteralValue{Value: "5"},
				},
			},
			wantLiteral: "DELETE",
			minChildren: 2,
		},
		{
			name: "DELETE with RETURNING",
			del: &Delete{
				Table:           TableReference{Name: "users"},
				ReturningClause: []Expression{&Identifier{Name: "id"}, &Identifier{Name: "name"}},
			},
			wantLiteral: "DELETE",
			minChildren: 3,
		},
		{
			name: "DELETE with WHERE and RETURNING",
			del: &Delete{
				Table: TableReference{Name: "users"},
				Where: &BinaryExpression{
					Left:     &Identifier{Name: "active"},
					Operator: "=",
					Right:    &LiteralValue{Value: "false"},
				},
				ReturningClause: []Expression{&Identifier{Name: "*"}},
			},
			wantLiteral: "DELETE",
			minChildren: 3,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			// Test TokenLiteral
			if got := tt.del.TokenLiteral(); got != tt.wantLiteral {
				t.Errorf("Delete.TokenLiteral() = %v, want %v", got, tt.wantLiteral)