Carlos Florêncio
68 lines
1.7 KiB
Go
68 lines
1.7 KiB
Go
package db
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"testing"
|
|
|
|
"github.com/jmoiron/sqlx"
|
|
)
|
|
|
|
func TestBindRebindsAfterInExpansion(t *testing.T) {
|
|
fake := &rebindRecorder{placeholder: "$"}
|
|
query, args, err := Bind(fake, "SELECT * FROM tasks WHERE id IN (?) AND state = ?", []string{"a", "b"}, "open")
|
|
if err != nil {
|
|
t.Fatalf("Bind() error = %v", err)
|
|
}
|
|
if query != "SELECT * FROM tasks WHERE id IN ($1, $2) AND state = $3" {
|
|
t.Fatalf("Bind() query = %q", query)
|
|
}
|
|
if len(args) != 3 || args[0] != "a" || args[1] != "b" || args[2] != "open" {
|
|
t.Fatalf("Bind() args = %#v", args)
|
|
}
|
|
if fake.calls != 1 {
|
|
t.Fatalf("Rebind() calls = %d, want 1", fake.calls)
|
|
}
|
|
}
|
|
|
|
func TestExecContextBindsAtFinalBoundary(t *testing.T) {
|
|
fake := &execRecorder{rebindRecorder: rebindRecorder{placeholder: "$"}}
|
|
_, err := ExecContext(context.Background(), fake, "UPDATE tasks SET state = ? WHERE id = ?", "done", "task-1")
|
|
if err != nil {
|
|
t.Fatalf("ExecContext() error = %v", err)
|
|
}
|
|
if fake.query != "UPDATE tasks SET state = $1 WHERE id = $2" {
|
|
t.Fatalf("ExecContext() query = %q", fake.query)
|
|
}
|
|
if fake.calls != 1 {
|
|
t.Fatalf("Rebind() calls = %d, want 1", fake.calls)
|
|
}
|
|
}
|
|
|
|
type rebindRecorder struct {
|
|
placeholder string
|
|
calls int
|
|
}
|
|
|
|
func (r *rebindRecorder) Rebind(query string) string {
|
|
r.calls++
|
|
return sqlx.Rebind(sqlx.DOLLAR, query)
|
|
}
|
|
|
|
type execRecorder struct {
|
|
rebindRecorder
|
|
query string
|
|
args []any
|
|
}
|
|
|
|
func (r *execRecorder) ExecContext(_ context.Context, query string, args ...any) (sql.Result, error) {
|
|
r.query = query
|
|
r.args = args
|
|
return fakeResult{}, nil
|
|
}
|
|
|
|
type fakeResult struct{}
|
|
|
|
func (fakeResult) LastInsertId() (int64, error) { return 0, nil }
|
|
func (fakeResult) RowsAffected() (int64, error) { return 0, nil }
|