143 lines
2.9 KiB
Go
143 lines
2.9 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"errors"
|
|
"os"
|
|
"reflect"
|
|
"testing"
|
|
|
|
"github.com/jackc/pgx/v5"
|
|
"github.com/jackc/pgx/v5/stdlib"
|
|
_ "github.com/mattn/go-sqlite3"
|
|
|
|
"code.uint32.ru/tiny/repo"
|
|
)
|
|
|
|
func TestSqlite3(t *testing.T) {
|
|
db, err := sql.Open("sqlite3", "test.db")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
defer os.Remove("test.db")
|
|
defer db.Close()
|
|
|
|
ctx := context.Background()
|
|
|
|
dotest(ctx, db, t)
|
|
}
|
|
|
|
func TestPGX(t *testing.T) {
|
|
dsn, ok := os.LookupEnv("PGX_DSN")
|
|
if !ok {
|
|
t.SkipNow()
|
|
}
|
|
|
|
conf, err := pgx.ParseConfig(dsn)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
conf.DefaultQueryExecMode = pgx.QueryExecModeSimpleProtocol
|
|
conf.DescriptionCacheCapacity = 0
|
|
conf.StatementCacheCapacity = 0
|
|
|
|
db, err := sql.Open("pgx", stdlib.RegisterConnConfig(conf))
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
defer db.Close()
|
|
|
|
ctx := context.Background()
|
|
|
|
dotest(ctx, db, t)
|
|
}
|
|
|
|
func TestOpenOrCreateFails(t *testing.T) {
|
|
ctx := context.Background()
|
|
|
|
if _, err := repo.OpenOrCreate[My](ctx, nil, ""); err == nil {
|
|
t.Errorf("must fail on empty tablename")
|
|
}
|
|
|
|
if _, err := repo.OpenOrCreate[My](ctx, nil, "test"); err == nil {
|
|
t.Errorf("must fail on nil DB instance")
|
|
}
|
|
}
|
|
|
|
func dotest(ctx context.Context, db *sql.DB, t *testing.T) {
|
|
r, err := repo.OpenOrCreate[My](ctx, db, "my_test")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
testID := "watermelon"
|
|
incorrectID := "surelynonexistant"
|
|
|
|
one := &My{
|
|
A: "hello",
|
|
B: 42,
|
|
}
|
|
|
|
if err := r.Create(ctx, testID, one); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if err := r.Create(ctx, testID, one); err == nil {
|
|
// can not insert with same id
|
|
t.Errorf("insert of the same id dows not fail")
|
|
}
|
|
|
|
two, err := r.Read(ctx, testID)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if !reflect.DeepEqual(one, two) {
|
|
t.Errorf("instances are not equal")
|
|
t.Logf("%+v", one)
|
|
t.Logf("%+v", two)
|
|
}
|
|
|
|
one.B = 13
|
|
|
|
if err := r.Update(ctx, testID, one); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
two, err = r.Read(ctx, testID)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if !reflect.DeepEqual(one, two) {
|
|
t.Errorf("instances are not equal")
|
|
t.Logf("%+v", one)
|
|
t.Logf("%+v", two)
|
|
}
|
|
|
|
if err := r.Delete(ctx, testID); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if err := r.Delete(ctx, testID); err == nil || !errors.Is(err, repo.ErrNotFound) {
|
|
// can only delete once
|
|
t.Errorf("incorrect error on delete for non-existing id, want: %v, have: %v", repo.ErrNotFound, err)
|
|
}
|
|
|
|
if _, err := r.Read(ctx, incorrectID); err == nil || !errors.Is(err, repo.ErrNotFound) {
|
|
t.Errorf("incorrect error on read for non-existing id, want: %v, have: %v", repo.ErrNotFound, err)
|
|
}
|
|
|
|
if err := r.Update(ctx, incorrectID, one); err == nil || !errors.Is(err, repo.ErrNotFound) {
|
|
t.Errorf("incorrect error on update for non-existing id, want: %v, have: %v", repo.ErrNotFound, err)
|
|
}
|
|
|
|
if err := r.Delete(ctx, incorrectID); err == nil || !errors.Is(err, repo.ErrNotFound) {
|
|
t.Errorf("incorrect error on delete for non-existing id, want: %v, have: %v", repo.ErrNotFound, err)
|
|
}
|
|
|
|
}
|