36 lines
827 B
Go
36 lines
827 B
Go
package evidence
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/Autumn-27/artex/db"
|
|
)
|
|
|
|
// Initialize an explicitly configured fresh database before taking the same
|
|
// suite lock as db, agent and server. Hold it on one pinned connection.
|
|
func TestMain(m *testing.M) {
|
|
if os.Getenv("ARTEX_PG_DSN") == "" {
|
|
os.Exit(m.Run())
|
|
}
|
|
os.Exit(runEvidenceSuite(m))
|
|
}
|
|
func runEvidenceSuite(m *testing.M) int {
|
|
pg, err := db.Open(os.Getenv("ARTEX_PG_DSN"))
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
defer pg.Close()
|
|
conn, err := pg.Conn(context.Background())
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
defer conn.Close()
|
|
if _, err = conn.ExecContext(context.Background(), `SELECT pg_advisory_lock(7337741002)`); err != nil {
|
|
panic(err)
|
|
}
|
|
defer conn.ExecContext(context.Background(), `SELECT pg_advisory_unlock(7337741002)`)
|
|
return m.Run()
|
|
}
|