Jim
591ec476e7
* chore: Add new dbw dependency * refactor: Convert to use dbw.DB * refactor (db): Change GetTicket(...) signature * feature (db tests): Added TestDeleteWhere(...) Adding this allows us to eliminate a leak of the gorm library * refactor: Add ctx to ScanRows(...) signature This is required to refactor all the deprecated errors pkg calls. This refactor required adding a context to quite a few call chains and in particular the job scheduler bits * feature (db/tests): Add support for test opt: WithTestLogLevel(...) This test only option allows callers to specify a log level for the underlying db package (where supported/applicable) when calling TestSetup(...) * refactor (db): Simplify Db type Only store the underlying *DB and create the *RW on demand when needed. * refactor (oplog): Port to using dbw pkg for db operations * refactor (db): Remove gormDB(...) func * tests (db): Add unit tests for TestDeleteWhere(...) * chore: make fmt changes to formatting
126 lines
3.2 KiB
Go
126 lines
3.2 KiB
Go
// Package oplog_test provides some gorm helper funcs for testing oplog database integrations
|
|
package oplog_test
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/hashicorp/go-dbw"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
const (
|
|
defaultTestUserTableName = "oplog_test_user"
|
|
defaultTestCarTableName = "oplog_test_car"
|
|
defaultTestRentalTableName = "oplog_test_rental"
|
|
)
|
|
|
|
// Init will use gorm migrations to init tables for test models
|
|
func Init(t *testing.T, db *dbw.DB) {
|
|
const testQueryCreateTables = `
|
|
begin;
|
|
|
|
-- create test tables used in the unit tests for the oplog package
|
|
-- these tables (oplog_test_user, oplog_test_car, oplog_test_rental) are
|
|
-- not part of the boundary domain model... they are simply used for testing
|
|
-- the oplog package
|
|
create table if not exists oplog_test_user (
|
|
id bigserial primary key,
|
|
name text
|
|
constraint oplog_test_user_name_uq unique,
|
|
phone_number text,
|
|
email text,
|
|
version wt_version
|
|
);
|
|
|
|
create or replace function
|
|
oplog_test_version_column()
|
|
returns trigger
|
|
as $$
|
|
begin
|
|
if pg_trigger_depth() = 1 then
|
|
if row(new.*) is distinct from row(old.*) then
|
|
if tg_nargs = 0 then
|
|
execute format('update %I set version = $1 where id = $2', tg_relid::regclass) using old.version+1, new.id;
|
|
new.version = old.version + 1;
|
|
return new;
|
|
end if;
|
|
end if;
|
|
end if;
|
|
return new;
|
|
end;
|
|
$$ language plpgsql;
|
|
|
|
create trigger
|
|
update_version_column
|
|
after update on oplog_test_user
|
|
for each row execute procedure oplog_test_version_column();
|
|
|
|
create table if not exists oplog_test_car (
|
|
id bigserial primary key,
|
|
name text unique,
|
|
model text,
|
|
mpg smallint
|
|
);
|
|
|
|
create table if not exists oplog_test_rental (
|
|
user_id bigint not null references oplog_test_user(id),
|
|
car_id bigint not null references oplog_test_car(id)
|
|
);
|
|
|
|
|
|
commit;
|
|
`
|
|
_, err := dbw.New(db).Exec(context.Background(), testQueryCreateTables, nil)
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
// Reinit will use gorm to drop then init tables for test models
|
|
func Reinit(t *testing.T, db *dbw.DB) {
|
|
const sql = ``
|
|
_, err := dbw.New(db).Exec(context.Background(), sql, nil)
|
|
require.NoError(t, err)
|
|
Init(t, db)
|
|
}
|
|
|
|
// TableName overrides the table name for TestUser
|
|
func (u *TestUser) TableName() string {
|
|
if u.Table != "" {
|
|
return u.Table
|
|
}
|
|
return defaultTestUserTableName
|
|
}
|
|
|
|
// SetTableName allows the table name to be overridden and makes a TestUser a
|
|
// ReplayableMessage. If the caller attempts to set the name to "" the name will be
|
|
// reset to the default name.
|
|
func (u *TestUser) SetTableName(n string) {
|
|
u.Table = n
|
|
}
|
|
|
|
// TableName overrides the table name for TestCar
|
|
func (c *TestCar) TableName() string {
|
|
if c.Table != "" {
|
|
return c.Table
|
|
}
|
|
return defaultTestCarTableName
|
|
}
|
|
|
|
// SetTableName allows the table name to be overridden and makes a TestCar a ReplayableMessage
|
|
func (c *TestCar) SetTableName(n string) {
|
|
c.Table = n
|
|
}
|
|
|
|
// TableName overrids the table name for TestRental
|
|
func (r *TestRental) TableName() string {
|
|
if r.Table != "" {
|
|
return r.Table
|
|
}
|
|
return defaultTestRentalTableName
|
|
}
|
|
|
|
// SetTableName allows the table name to be overridden and makes a TestRental a ReplayableMessage
|
|
func (r *TestRental) SetTableName(n string) {
|
|
r.Table = n
|
|
}
|