68 lines
1.9 KiB
Go
68 lines
1.9 KiB
Go
package store
|
|
|
|
import "testing"
|
|
|
|
func TestSQLiteDialect_Rebind_passthrough(t *testing.T) {
|
|
d := SQLiteDialect{}
|
|
in := `SELECT * FROM pase_users WHERE email = ? AND status = ?`
|
|
if got := d.Rebind(in); got != in {
|
|
t.Errorf("SQLite Rebind should be passthrough.\nin: %s\ngot: %s", in, got)
|
|
}
|
|
}
|
|
|
|
func TestPostgresDialect_Rebind(t *testing.T) {
|
|
cases := []struct {
|
|
name string
|
|
in string
|
|
want string
|
|
}{
|
|
{
|
|
name: "no placeholders",
|
|
in: `SELECT 1`,
|
|
want: `SELECT 1`,
|
|
},
|
|
{
|
|
name: "single placeholder",
|
|
in: `SELECT * FROM pase_users WHERE id = ?`,
|
|
want: `SELECT * FROM pase_users WHERE id = $1`,
|
|
},
|
|
{
|
|
name: "multiple placeholders",
|
|
in: `INSERT INTO pase_users (id, email, status) VALUES (?, ?, ?)`,
|
|
want: `INSERT INTO pase_users (id, email, status) VALUES ($1, $2, $3)`,
|
|
},
|
|
{
|
|
name: "question mark inside single-quoted literal is preserved",
|
|
in: `SELECT * FROM t WHERE name = 'who?' AND id = ?`,
|
|
want: `SELECT * FROM t WHERE name = 'who?' AND id = $1`,
|
|
},
|
|
{
|
|
name: "escaped single quote inside literal",
|
|
in: `SELECT * FROM t WHERE name = 'O''Reilly?' AND id = ?`,
|
|
want: `SELECT * FROM t WHERE name = 'O''Reilly?' AND id = $1`,
|
|
},
|
|
{
|
|
name: "question mark inside double-quoted identifier is preserved",
|
|
in: `SELECT "weird?col" FROM t WHERE id = ?`,
|
|
want: `SELECT "weird?col" FROM t WHERE id = $1`,
|
|
},
|
|
{
|
|
name: "escaped double quote inside identifier",
|
|
in: `SELECT "a""?b" FROM t WHERE id = ?`,
|
|
want: `SELECT "a""?b" FROM t WHERE id = $1`,
|
|
},
|
|
{
|
|
name: "mix of literals and placeholders",
|
|
in: `UPDATE t SET name = 'foo?', other = ? WHERE id = ? AND tag = 'x?'`,
|
|
want: `UPDATE t SET name = 'foo?', other = $1 WHERE id = $2 AND tag = 'x?'`,
|
|
},
|
|
}
|
|
d := PostgresDialect{}
|
|
for _, tc := range cases {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
if got := d.Rebind(tc.in); got != tc.want {
|
|
t.Errorf("Rebind mismatch\nin: %s\nwant: %s\ngot: %s", tc.in, tc.want, got)
|
|
}
|
|
})
|
|
}
|
|
}
|