78 lines
1.4 KiB
Go
78 lines
1.4 KiB
Go
|
package sqldb
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"flag"
|
||
|
|
"fmt"
|
||
|
|
"log"
|
||
|
|
"os"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
_ "github.com/go-sql-driver/mysql" // MySQL driver
|
||
|
|
)
|
||
|
|
|
||
|
|
type record struct {
|
||
|
|
ID uint `db:"id"`
|
||
|
|
Name string `db:"name"`
|
||
|
|
}
|
||
|
|
|
||
|
|
var conn *Conn
|
||
|
|
|
||
|
|
func TestMain(m *testing.M) {
|
||
|
|
dsn := flag.String("dsn", "", "Database source name")
|
||
|
|
flag.Parse()
|
||
|
|
if *dsn == "" {
|
||
|
|
log.Println("Database source name is not provided, skipping package tests")
|
||
|
|
os.Exit(0)
|
||
|
|
}
|
||
|
|
|
||
|
|
log.Println("Establishing connection to the test database")
|
||
|
|
ctx := context.Background()
|
||
|
|
var err error
|
||
|
|
conn, err = Connect(ctx, MySQL, *dsn)
|
||
|
|
if err != nil {
|
||
|
|
log.Fatalf("Failed to connect: %v\n", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
log.Println("Seeding database")
|
||
|
|
must(conn.Exec(ctx, `
|
||
|
|
DROP TABLE IF EXISTS sqldb_test
|
||
|
|
`))
|
||
|
|
must(conn.Exec(ctx, `
|
||
|
|
CREATE TABLE sqldb_test (
|
||
|
|
id int(11) UNSIGNED NOT NULL,
|
||
|
|
name VARCHAR(10) DEFAULT '',
|
||
|
|
PRIMARY KEY (id)
|
||
|
|
) ENGINE=InnoDB DEFAULT CHARSET=ascii
|
||
|
|
`))
|
||
|
|
must(conn.Exec(ctx, `
|
||
|
|
INSERT INTO sqldb_test (id, name)
|
||
|
|
VALUES
|
||
|
|
(1, "Alice"),
|
||
|
|
(2, "Bob")
|
||
|
|
`))
|
||
|
|
|
||
|
|
fmt.Println("Starting test suite")
|
||
|
|
exitCode := m.Run()
|
||
|
|
log.Println("Test suite finished")
|
||
|
|
if err := conn.Close(); err != nil {
|
||
|
|
log.Printf("Failed to close connection: %v\n", err)
|
||
|
|
}
|
||
|
|
os.Exit(exitCode)
|
||
|
|
}
|
||
|
|
|
||
|
|
func mustT(t *testing.T, r Result) Result {
|
||
|
|
t.Helper()
|
||
|
|
if r.Error() != nil {
|
||
|
|
t.Fatalf("Query failed: %v", r.Error())
|
||
|
|
}
|
||
|
|
return r
|
||
|
|
}
|
||
|
|
|
||
|
|
func must(r Result) Result {
|
||
|
|
if r.Error() != nil {
|
||
|
|
log.Fatalf("Query failed: %v\n", r.Error())
|
||
|
|
}
|
||
|
|
return r
|
||
|
|
}
|