123 lines
2.2 KiB
Go
123 lines
2.2 KiB
Go
package administrator_test
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/jackc/pgx/v4/pgxpool"
|
|
)
|
|
|
|
var db *pgxpool.Pool
|
|
|
|
var administratorsData = []interface{}{
|
|
1, "very secure", "not the real one", time.Now().Format(time.RFC3339),
|
|
}
|
|
|
|
func TestMain(m *testing.M) {
|
|
defer Teardown()
|
|
Setup()
|
|
|
|
os.Exit(m.Run())
|
|
}
|
|
|
|
func Setup() {
|
|
poolConfig, err := pgxpool.ParseConfig(os.Getenv("DATABASE_URL"))
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
db, err = pgxpool.ConnectConfig(context.Background(), poolConfig)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
conn, err := db.Acquire(context.Background())
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
defer conn.Release()
|
|
|
|
tx, err := conn.Begin(context.Background())
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
defer tx.Rollback(context.Background())
|
|
|
|
_, err = tx.Exec(context.Background(), "DROP TABLE IF EXISTS submission")
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
_, err = tx.Exec(context.Background(), "DROP TABLE IF EXISTS jokesbapak2")
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
_, err = tx.Exec(context.Background(), "DROP TABLE IF EXISTS administrators")
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
_, err = tx.Exec(
|
|
context.Background(),
|
|
`CREATE TABLE IF NOT EXISTS administrators (
|
|
id SERIAL PRIMARY KEY,
|
|
key VARCHAR(255) NOT NULL UNIQUE,
|
|
token TEXT,
|
|
last_used VARCHAR(255)
|
|
)`,
|
|
)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
err = tx.Commit(context.Background())
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
func Teardown() (err error) {
|
|
tx, err := db.Begin(context.Background())
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer tx.Rollback(context.Background())
|
|
|
|
_, err = tx.Exec(context.Background(), "DROP TABLE IF EXISTS submission")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
_, err = tx.Exec(context.Background(), "DROP TABLE IF EXISTS jokesbapak2")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
_, err = tx.Exec(context.Background(), "DROP TABLE IF EXISTS administrators")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
err = tx.Commit(context.Background())
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
db.Close()
|
|
return
|
|
}
|
|
|
|
func Flush() error {
|
|
conn, err := db.Acquire(context.Background())
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer conn.Release()
|
|
|
|
_, err = conn.Exec(context.Background(), "TRUNCATE TABLE administrators;")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|