eb033c4922
Builder is not in a position to validate the statement and the effort is duplicated when the query is build. The checked errors are simple and do not occur most of the time.
79 lines
1.9 KiB
Go
79 lines
1.9 KiB
Go
package qb
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/google/go-cmp/cmp"
|
|
)
|
|
|
|
func TestDeleteBuilder(t *testing.T) {
|
|
w := EqNamed("id", "expr")
|
|
|
|
table := []struct {
|
|
B *DeleteBuilder
|
|
N []string
|
|
S string
|
|
}{
|
|
// Basic test for delete
|
|
{
|
|
B: Delete("cycling.cyclist_name").Where(w),
|
|
S: "DELETE FROM cycling.cyclist_name WHERE id=? ",
|
|
N: []string{"expr"},
|
|
},
|
|
// Change table name
|
|
{
|
|
B: Delete("cycling.cyclist_name").Where(w).From("Foobar"),
|
|
S: "DELETE FROM Foobar WHERE id=? ",
|
|
N: []string{"expr"},
|
|
},
|
|
// Add column
|
|
{
|
|
B: Delete("cycling.cyclist_name").Where(w).Columns("stars"),
|
|
S: "DELETE stars FROM cycling.cyclist_name WHERE id=? ",
|
|
N: []string{"expr"},
|
|
},
|
|
// Add WHERE
|
|
{
|
|
B: Delete("cycling.cyclist_name").Where(w, Gt("firstname")),
|
|
S: "DELETE FROM cycling.cyclist_name WHERE id=? AND firstname>? ",
|
|
N: []string{"expr", "firstname"},
|
|
},
|
|
// Add IF
|
|
{
|
|
B: Delete("cycling.cyclist_name").Where(w).If(Gt("firstname")),
|
|
S: "DELETE FROM cycling.cyclist_name WHERE id=? IF firstname>? ",
|
|
N: []string{"expr", "firstname"},
|
|
},
|
|
// Add TIMESTAMP
|
|
{
|
|
B: Delete("cycling.cyclist_name").Where(w).Timestamp(time.Unix(0, 0).Add(time.Microsecond * 123456789)),
|
|
S: "DELETE FROM cycling.cyclist_name USING TIMESTAMP 123456789 WHERE id=? ",
|
|
N: []string{"expr"},
|
|
},
|
|
// Add IF EXISTS
|
|
{
|
|
B: Delete("cycling.cyclist_name").Where(w).Existing(),
|
|
S: "DELETE FROM cycling.cyclist_name WHERE id=? IF EXISTS ",
|
|
N: []string{"expr"},
|
|
},
|
|
}
|
|
|
|
for _, test := range table {
|
|
stmt, names := test.B.ToCql()
|
|
if diff := cmp.Diff(test.S, stmt); diff != "" {
|
|
t.Error(diff)
|
|
}
|
|
if diff := cmp.Diff(test.N, names); diff != "" {
|
|
t.Error(diff)
|
|
}
|
|
}
|
|
}
|
|
|
|
func BenchmarkDeleteBuilder(b *testing.B) {
|
|
b.ResetTimer()
|
|
for i := 0; i < b.N; i++ {
|
|
Delete("cycling.cyclist_name").Columns("id", "user_uuid", "firstname", "stars").Where(Eq("id"))
|
|
}
|
|
}
|