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.
91 lines
2.3 KiB
Go
91 lines
2.3 KiB
Go
package qb
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/google/go-cmp/cmp"
|
|
)
|
|
|
|
func TestSelectBuilder(t *testing.T) {
|
|
w := EqNamed("id", "expr")
|
|
|
|
table := []struct {
|
|
B *SelectBuilder
|
|
N []string
|
|
S string
|
|
}{
|
|
// Basic test for select *
|
|
{
|
|
B: Select("cycling.cyclist_name"),
|
|
S: "SELECT * FROM cycling.cyclist_name ",
|
|
},
|
|
// Basic test for select columns
|
|
{
|
|
B: Select("cycling.cyclist_name").Columns("id", "user_uuid", "firstname"),
|
|
S: "SELECT id,user_uuid,firstname FROM cycling.cyclist_name ",
|
|
},
|
|
// Basic test for select distinct
|
|
{
|
|
B: Select("cycling.cyclist_name").Distinct("id"),
|
|
S: "SELECT DISTINCT id FROM cycling.cyclist_name ",
|
|
},
|
|
// Change table name
|
|
{
|
|
B: Select("cycling.cyclist_name").From("Foobar"),
|
|
S: "SELECT * FROM Foobar ",
|
|
},
|
|
// Add WHERE
|
|
{
|
|
B: Select("cycling.cyclist_name").Where(w, Gt("firstname")),
|
|
S: "SELECT * FROM cycling.cyclist_name WHERE id=? AND firstname>? ",
|
|
N: []string{"expr", "firstname"},
|
|
},
|
|
// Add GROUP BY
|
|
{
|
|
B: Select("cycling.cyclist_name").Columns("MAX(stars) as max_stars").GroupBy("id"),
|
|
S: "SELECT id,MAX(stars) as max_stars FROM cycling.cyclist_name GROUP BY id ",
|
|
},
|
|
// Add ORDER BY
|
|
{
|
|
B: Select("cycling.cyclist_name").Where(w).OrderBy("firstname", ASC),
|
|
S: "SELECT * FROM cycling.cyclist_name WHERE id=? ORDER BY firstname ASC ",
|
|
N: []string{"expr"},
|
|
},
|
|
// Add LIMIT
|
|
{
|
|
B: Select("cycling.cyclist_name").Where(w).Limit(10),
|
|
S: "SELECT * FROM cycling.cyclist_name WHERE id=? LIMIT 10 ",
|
|
N: []string{"expr"},
|
|
},
|
|
// Add PER PARTITION LIMIT
|
|
{
|
|
B: Select("cycling.cyclist_name").Where(w).LimitPerPartition(10),
|
|
S: "SELECT * FROM cycling.cyclist_name WHERE id=? PER PARTITION LIMIT 10 ",
|
|
N: []string{"expr"},
|
|
},
|
|
// Add ALLOW FILTERING
|
|
{
|
|
B: Select("cycling.cyclist_name").Where(w).AllowFiltering(),
|
|
S: "SELECT * FROM cycling.cyclist_name WHERE id=? ALLOW FILTERING ",
|
|
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 BenchmarkSelectBuilder(b *testing.B) {
|
|
b.ResetTimer()
|
|
for i := 0; i < b.N; i++ {
|
|
Select("cycling.cyclist_name").Columns("id", "user_uuid", "firstname", "stars").Where(Eq("id"))
|
|
}
|
|
}
|