mirror of
https://github.com/hanzoai/sqlbuilder.git
synced 2026-10-01 00:37:42 +00:00
Spencer ConnaughtonandGitHub
0f54cd520c
* Add CQL flavor. * Add support for CQL argument compilation. * Adopt substests for interpolation tests. * Add support for CQL interpolation. * Add support for CQL blobs. https://docs.datastax.com/en/cql-oss/3.x/cql/cql_reference/blob_r.html * Add CQL timestamp interpolation support. https://docs.datastax.com/en/cql-oss/3.x/cql/cql_reference/timestamp_type_r.html * Add builder test for CQL. * Add support for the NOW function. https://docs.datastax.com/en/cql-oss/3.3/cql/cql_reference/timeuuid_functions_r.html?hl=now * Add support for CQL update IF statement. https://docs.datastax.com/en/cql-oss/3.3/cql/cql_reference/cqlUpdate.html#Conditionallyupdatingcolumns * Drop table prefixes for CQL SELECTs. * Switch to single quoting for CQL queries. https://docs.datastax.com/en/cql-oss/3.3/cql/cql_reference/escape_char_r.html * Update documentation. * Use valid CQL in tests. * Support CQL LIMIT.
132 lines
2.8 KiB
Go
132 lines
2.8 KiB
Go
// Copyright 2018 Huan Du. All rights reserved.
|
|
// Licensed under the MIT license that can be found in the LICENSE file.
|
|
|
|
package sqlbuilder
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/huandu/go-assert"
|
|
)
|
|
|
|
func TestArgs(t *testing.T) {
|
|
a := assert.New(t)
|
|
cases := map[string][]interface{}{
|
|
"abc ? def\n[123]": {"abc $? def", 123},
|
|
"abc ? def\n[456]": {"abc $0 def", 456},
|
|
"abc def\n[]": {"abc $1 def", 123},
|
|
"abc ? def\n[789]": {"abc ${s} def", Named("s", 789)},
|
|
"abc def \n[]": {"abc ${unknown} def ", 123},
|
|
"abc $ def\n[]": {"abc $$ def", 123},
|
|
"abcdef$\n[]": {"abcdef$", 123},
|
|
"abc ? ? ? ? def\n[123 456 123 456]": {"abc $? $? $0 $? def", 123, 456, 789},
|
|
"abc ? raw ? raw def\n[123 123]": {"abc $? $? $0 $? def", 123, Raw("raw"), 789},
|
|
}
|
|
|
|
for expected, c := range cases {
|
|
args := new(Args)
|
|
|
|
for i := 1; i < len(c); i++ {
|
|
args.Add(c[i])
|
|
}
|
|
|
|
sql, values := args.Compile(c[0].(string))
|
|
actual := fmt.Sprintf("%v\n%v", sql, values)
|
|
|
|
a.Equal(actual, expected)
|
|
}
|
|
|
|
old := DefaultFlavor
|
|
defer func() {
|
|
DefaultFlavor = old
|
|
}()
|
|
|
|
DefaultFlavor = PostgreSQL
|
|
|
|
// PostgreSQL flavor compiled sql.
|
|
for expected, c := range cases {
|
|
args := new(Args)
|
|
|
|
for i := 1; i < len(c); i++ {
|
|
args.Add(c[i])
|
|
}
|
|
|
|
sql, values := args.Compile(c[0].(string))
|
|
actual := fmt.Sprintf("%v\n%v", sql, values)
|
|
expected = toPostgreSQL(expected)
|
|
|
|
a.Equal(actual, expected)
|
|
}
|
|
|
|
DefaultFlavor = SQLServer
|
|
|
|
// SQLServer flavor compiled sql.
|
|
for expected, c := range cases {
|
|
args := new(Args)
|
|
|
|
for i := 1; i < len(c); i++ {
|
|
args.Add(c[i])
|
|
}
|
|
|
|
sql, values := args.Compile(c[0].(string))
|
|
actual := fmt.Sprintf("%v\n%v", sql, values)
|
|
expected = toSQLServerSQL(expected)
|
|
|
|
a.Equal(actual, expected)
|
|
}
|
|
|
|
DefaultFlavor = CQL
|
|
|
|
for expected, c := range cases {
|
|
args := new(Args)
|
|
|
|
for i := 1; i < len(c); i++ {
|
|
args.Add(c[i])
|
|
}
|
|
|
|
sql, values := args.Compile(c[0].(string))
|
|
actual := fmt.Sprintf("%v\n%v", sql, values)
|
|
|
|
a.Equal(actual, expected)
|
|
}
|
|
}
|
|
|
|
func toPostgreSQL(sql string) string {
|
|
parts := strings.Split(sql, "?")
|
|
buf := &bytes.Buffer{}
|
|
buf.WriteString(parts[0])
|
|
|
|
for i, p := range parts[1:] {
|
|
fmt.Fprintf(buf, "$%v", i+1)
|
|
buf.WriteString(p)
|
|
}
|
|
|
|
return buf.String()
|
|
}
|
|
|
|
func toSQLServerSQL(sql string) string {
|
|
parts := strings.Split(sql, "?")
|
|
buf := &bytes.Buffer{}
|
|
buf.WriteString(parts[0])
|
|
|
|
for i, p := range parts[1:] {
|
|
fmt.Fprintf(buf, "@p%v", i+1)
|
|
buf.WriteString(p)
|
|
}
|
|
|
|
return buf.String()
|
|
}
|
|
|
|
func TestArgsAdd(t *testing.T) {
|
|
a := assert.New(t)
|
|
args := &Args{}
|
|
|
|
for i := 0; i < maxPredefinedArgs*2; i++ {
|
|
actual := args.Add(i)
|
|
a.Equal(actual, fmt.Sprintf("$%v", i))
|
|
}
|
|
}
|