mirror of
https://github.com/safing/portbase
synced 2025-09-01 18:19:57 +00:00
42 lines
824 B
Go
42 lines
824 B
Go
package template
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/safing/portbase/dataroot"
|
|
"github.com/safing/portbase/modules"
|
|
)
|
|
|
|
func TestMain(m *testing.M) {
|
|
// tmp dir for data root (db & config)
|
|
tmpDir, err := ioutil.TempDir("", "portbase-testing-")
|
|
// initialize data dir
|
|
if err == nil {
|
|
err = dataroot.Initialize(tmpDir, 0755)
|
|
}
|
|
// start modules
|
|
if err == nil {
|
|
err = modules.Start()
|
|
}
|
|
// handle setup error
|
|
if err != nil {
|
|
fmt.Fprintf(os.Stderr, "failed to setup test: %s", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
// run tests
|
|
exitCode := m.Run()
|
|
|
|
// shutdown
|
|
_ = modules.Shutdown()
|
|
if modules.GetExitStatusCode() != 0 {
|
|
exitCode = modules.GetExitStatusCode()
|
|
fmt.Fprintf(os.Stderr, "failed to cleanly shutdown test: %s", err)
|
|
}
|
|
// clean up and exit
|
|
os.RemoveAll(tmpDir)
|
|
os.Exit(exitCode)
|
|
}
|