Johan Brandhorst-Satzkorn
7f9b294a7e
Our test helper packages were not possible to use with benchmarks. This refactor switches to testing.TB to allow both tests and benchmarks to use the helpers.
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.TB, 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
|
|
}
|