// Empty all data in the entire database func empty(pool *pgx.ConnPool) error { tables := []string{"feeds", "items", "password_resets", "sessions", "subscriptions", "unread_items", "users"} for _, table := range tables { _, err := pool.Exec(fmt.Sprintf("delete from %s", table)) if err != nil { return err } } return nil }
func setupStressDB(t *testing.T, pool *pgx.ConnPool) { _, err := pool.Exec(` drop table if exists widgets; create table widgets( id serial primary key, name varchar not null, description text, creation_time timestamptz ); `) if err != nil { t.Fatal(err) } }
func notify(pool *pgx.ConnPool, actionNum int) error { _, err := pool.Exec("notify stress") return err }
func truncateAndClose(pool *pgx.ConnPool) { if _, err := pool.Exec("TRUNCATE TABLE que_jobs"); err != nil { panic(err) } pool.Close() }