You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
89 lines
2.0 KiB
Go
89 lines
2.0 KiB
Go
2 years ago
|
package db_test
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"database/sql"
|
||
|
|
"database/sql/driver"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"git.phc.dm.unipi.it/phc/website/libs/db"
|
||
|
|
"gotest.tools/assert"
|
||
|
|
)
|
||
|
|
|
||
|
|
type mockDB struct {
|
||
|
|
*testing.T
|
||
|
|
|
||
|
|
LastQuery string
|
||
|
|
}
|
||
|
|
|
||
|
|
func (db *mockDB) Prepare(query string) (driver.Stmt, error) { return nil, nil }
|
||
|
|
func (db *mockDB) Close() error { return nil }
|
||
|
|
func (db *mockDB) Begin() (driver.Tx, error) { return nil, nil }
|
||
|
|
|
||
|
|
func (db *mockDB) Query(query string, args []driver.Value) (driver.Rows, error) {
|
||
|
|
db.LastQuery = query
|
||
|
|
db.Logf("[sql mock driver] [exec] query: '%s', args: %#v", query, args)
|
||
|
|
|
||
|
|
return nil, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (db *mockDB) Exec(query string, args []driver.Value) (driver.Result, error) {
|
||
|
|
db.LastQuery = query
|
||
|
|
db.Logf("[sql mock driver] [exec] query: '%s', args: %#v", query, args)
|
||
|
|
|
||
|
|
return nil, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (db *mockDB) Open(name string) (driver.Conn, error) {
|
||
|
|
return &mockDB{T: db.T}, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (db *mockDB) Connect(context.Context) (driver.Conn, error) {
|
||
|
|
return db, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (db *mockDB) Driver() driver.Driver {
|
||
|
|
return db
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
// Tests
|
||
|
|
//
|
||
|
|
|
||
|
|
func TestDb1(t *testing.T) {
|
||
|
|
type Product struct {
|
||
|
|
Id db.Ref[Product] `db:"id*"`
|
||
|
|
Title string `db:"title"`
|
||
|
|
Quantity int `db:"quantity"`
|
||
|
|
}
|
||
|
|
|
||
|
|
products := db.AutoTable[Product]("products")
|
||
|
|
|
||
|
|
mock := &mockDB{T: t}
|
||
|
|
conn := sql.OpenDB(mock)
|
||
|
|
|
||
|
|
err := products.CreateIfNotExists(conn)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
assert.Equal(t, mock.LastQuery,
|
||
|
|
`CREATE TABLE IF NOT EXISTS products(id TEXT, title TEXT, quantity INTEGER)`,
|
||
|
|
)
|
||
|
|
|
||
|
|
ref1, err := db.Create(conn, products, Product{Title: "Foo", Quantity: 27})
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
assert.Assert(t, string(ref1) != "")
|
||
|
|
assert.Equal(t, mock.LastQuery,
|
||
|
|
`INSERT INTO products(id, title, quantity) VALUES (?, ?, ?)`,
|
||
|
|
)
|
||
|
|
|
||
|
|
if err := db.Insert(conn, products, Product{Id: "123", Title: "Bar", Quantity: 42}); err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
assert.Equal(t, mock.LastQuery,
|
||
|
|
`INSERT INTO products(id, title, quantity) VALUES (?, ?, ?)`,
|
||
|
|
)
|
||
|
|
}
|