12d360a0c3
The 'value' interface represents a CQL value for use in a comparison, update, or intitialization operation. A consistent interface for this allows us to easily support specifying default-named, custom-named, literal, and evaluated-function values in all these contexts. Parameters to Func should probably also be values to support full composition, but that would be a breaking change because Func's properties are exposed. The value interface could itself be exposed if we wanted to allow clients to pass their own values to SetValue, etc, but for now it is a package-internal abstraction. BLA
87 lines
2.8 KiB
Go
87 lines
2.8 KiB
Go
// Copyright (C) 2017 ScyllaDB
|
|
// Use of this source code is governed by a ALv2-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package qb
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/google/go-cmp/cmp"
|
|
)
|
|
|
|
func TestInsertBuilder(t *testing.T) {
|
|
table := []struct {
|
|
B *InsertBuilder
|
|
N []string
|
|
S string
|
|
}{
|
|
|
|
// Basic test for insert
|
|
{
|
|
B: Insert("cycling.cyclist_name").Columns("id", "user_uuid", "firstname"),
|
|
S: "INSERT INTO cycling.cyclist_name (id,user_uuid,firstname) VALUES (?,?,?) ",
|
|
N: []string{"id", "user_uuid", "firstname"},
|
|
},
|
|
// Change table name
|
|
{
|
|
B: Insert("cycling.cyclist_name").Columns("id", "user_uuid", "firstname").Into("Foobar"),
|
|
S: "INSERT INTO Foobar (id,user_uuid,firstname) VALUES (?,?,?) ",
|
|
N: []string{"id", "user_uuid", "firstname"},
|
|
},
|
|
// Add columns
|
|
{
|
|
B: Insert("cycling.cyclist_name").Columns("id", "user_uuid", "firstname").Columns("stars"),
|
|
S: "INSERT INTO cycling.cyclist_name (id,user_uuid,firstname,stars) VALUES (?,?,?,?) ",
|
|
N: []string{"id", "user_uuid", "firstname", "stars"},
|
|
},
|
|
// Add a named column
|
|
{
|
|
B: Insert("cycling.cyclist_name").Columns("id", "user_uuid", "firstname").NamedColumn("stars", "stars_name"),
|
|
S: "INSERT INTO cycling.cyclist_name (id,user_uuid,firstname,stars) VALUES (?,?,?,?) ",
|
|
N: []string{"id", "user_uuid", "firstname", "stars_name"},
|
|
},
|
|
// Add a literal column
|
|
{
|
|
B: Insert("cycling.cyclist_name").Columns("id", "user_uuid", "firstname").LitColumn("stars", "stars_lit"),
|
|
S: "INSERT INTO cycling.cyclist_name (id,user_uuid,firstname,stars) VALUES (?,?,?,stars_lit) ",
|
|
N: []string{"id", "user_uuid", "firstname"},
|
|
},
|
|
// Add TTL
|
|
{
|
|
B: Insert("cycling.cyclist_name").Columns("id", "user_uuid", "firstname").TTL(),
|
|
S: "INSERT INTO cycling.cyclist_name (id,user_uuid,firstname) VALUES (?,?,?) USING TTL ? ",
|
|
N: []string{"id", "user_uuid", "firstname", "_ttl"},
|
|
},
|
|
// Add TIMESTAMP
|
|
{
|
|
B: Insert("cycling.cyclist_name").Columns("id", "user_uuid", "firstname").Timestamp(),
|
|
S: "INSERT INTO cycling.cyclist_name (id,user_uuid,firstname) VALUES (?,?,?) USING TIMESTAMP ? ",
|
|
N: []string{"id", "user_uuid", "firstname", "_ts"},
|
|
},
|
|
// Add IF NOT EXISTS
|
|
{
|
|
B: Insert("cycling.cyclist_name").Columns("id", "user_uuid", "firstname").Unique(),
|
|
S: "INSERT INTO cycling.cyclist_name (id,user_uuid,firstname) VALUES (?,?,?) IF NOT EXISTS ",
|
|
N: []string{"id", "user_uuid", "firstname"},
|
|
},
|
|
}
|
|
|
|
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 BenchmarkInsertBuilder(b *testing.B) {
|
|
b.ResetTimer()
|
|
for i := 0; i < b.N; i++ {
|
|
Insert("cycling.cyclist_name").Columns("id", "user_uuid", "firstname", "stars").ToCql()
|
|
}
|
|
}
|