130 lines
3.0 KiB
Go
130 lines
3.0 KiB
Go
package terminal
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/safing/portmaster/base/config"
|
|
"github.com/safing/portmaster/base/database/dbmodule"
|
|
"github.com/safing/portmaster/base/metrics"
|
|
"github.com/safing/portmaster/base/rng"
|
|
"github.com/safing/portmaster/service/core/base"
|
|
"github.com/safing/portmaster/service/mgr"
|
|
"github.com/safing/portmaster/spn/cabin"
|
|
"github.com/safing/portmaster/spn/conf"
|
|
)
|
|
|
|
type testInstance struct {
|
|
db *dbmodule.DBModule
|
|
config *config.Config
|
|
metrics *metrics.Metrics
|
|
rng *rng.Rng
|
|
base *base.Base
|
|
cabin *cabin.Cabin
|
|
dataDir string
|
|
}
|
|
|
|
func (stub *testInstance) DataDir() string {
|
|
if len(stub.dataDir) == 0 {
|
|
var err error
|
|
stub.dataDir, err = os.MkdirTemp("", "")
|
|
if err != nil {
|
|
panic(fmt.Sprintf("failed to create temp dir: %v", err))
|
|
}
|
|
}
|
|
return stub.dataDir
|
|
}
|
|
|
|
func (stub *testInstance) Config() *config.Config {
|
|
return stub.config
|
|
}
|
|
|
|
func (stub *testInstance) Metrics() *metrics.Metrics {
|
|
return stub.metrics
|
|
}
|
|
|
|
func (stub *testInstance) SPNGroup() *mgr.ExtendedGroup {
|
|
return nil
|
|
}
|
|
|
|
func (stub *testInstance) Stopping() bool {
|
|
return false
|
|
}
|
|
func (stub *testInstance) SetCmdLineOperation(f func() error) {}
|
|
|
|
func runTest(m *testing.M) error {
|
|
var err error
|
|
|
|
conf.EnablePublicHub(true) // Make hub config available.
|
|
|
|
instance := &testInstance{}
|
|
instance.db, err = dbmodule.New(instance)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to create database module: %w", err)
|
|
}
|
|
instance.config, err = config.New(instance)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to create config module: %w", err)
|
|
}
|
|
instance.metrics, err = metrics.New(instance)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to create metrics module: %w", err)
|
|
}
|
|
instance.rng, err = rng.New(instance)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to create rng module: %w", err)
|
|
}
|
|
instance.base, err = base.New(instance)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to create base module: %w", err)
|
|
}
|
|
instance.cabin, err = cabin.New(instance)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to create cabin module: %w", err)
|
|
}
|
|
_, err = New(instance)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to create module: %w", err)
|
|
}
|
|
|
|
// Start
|
|
err = instance.db.Start()
|
|
if err != nil {
|
|
return fmt.Errorf("failed to start db module: %w", err)
|
|
}
|
|
err = instance.config.Start()
|
|
if err != nil {
|
|
return fmt.Errorf("failed to start config module: %w", err)
|
|
}
|
|
err = instance.metrics.Start()
|
|
if err != nil {
|
|
return fmt.Errorf("failed to start metrics module: %w", err)
|
|
}
|
|
err = instance.rng.Start()
|
|
if err != nil {
|
|
return fmt.Errorf("failed to start rng module: %w", err)
|
|
}
|
|
err = instance.base.Start()
|
|
if err != nil {
|
|
return fmt.Errorf("failed to start base module: %w", err)
|
|
}
|
|
err = instance.cabin.Start()
|
|
if err != nil {
|
|
return fmt.Errorf("failed to start cabin module: %w", err)
|
|
}
|
|
err = module.Start()
|
|
if err != nil {
|
|
return fmt.Errorf("failed to start docks module: %w", err)
|
|
}
|
|
|
|
m.Run()
|
|
return nil
|
|
}
|
|
|
|
func TestMain(m *testing.M) {
|
|
if err := runTest(m); err != nil {
|
|
os.Exit(1)
|
|
}
|
|
}
|