143 lines
3.8 KiB
Go
143 lines
3.8 KiB
Go
package docks
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/safing/portmaster/base/config"
|
|
"github.com/safing/portmaster/base/database/dbmodule"
|
|
"github.com/safing/portmaster/base/log"
|
|
"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/access"
|
|
"github.com/safing/portmaster/spn/cabin"
|
|
"github.com/safing/portmaster/spn/conf"
|
|
"github.com/safing/portmaster/spn/terminal"
|
|
)
|
|
|
|
type testInstance struct {
|
|
db *dbmodule.DBModule
|
|
config *config.Config
|
|
metrics *metrics.Metrics
|
|
rng *rng.Rng
|
|
base *base.Base
|
|
access *access.Access
|
|
terminal *terminal.TerminalModule
|
|
cabin *cabin.Cabin
|
|
}
|
|
|
|
func (stub *testInstance) Config() *config.Config { return stub.config }
|
|
func (stub *testInstance) SPNGroup() *mgr.ExtendedGroup { return nil }
|
|
func (stub *testInstance) SetCmdLineOperation(f func() error) {}
|
|
func (stub *testInstance) IsShuttingDown() bool { return false }
|
|
func (stub *testInstance) DataDir() string { return _dataDir }
|
|
|
|
var _dataDir string
|
|
|
|
func runTest(m *testing.M) error {
|
|
_ = log.Start("info", true, "")
|
|
|
|
var err error
|
|
|
|
// Create a temporary directory for the data
|
|
_dataDir, err = os.MkdirTemp("", "")
|
|
if err != nil {
|
|
return fmt.Errorf("failed to initialize dataroot: %w", err)
|
|
}
|
|
defer func() { _ = os.RemoveAll(_dataDir) }()
|
|
|
|
instance := &testInstance{}
|
|
runningTests = true
|
|
conf.EnablePublicHub(true) // Make hub config available.
|
|
access.EnableTestMode() // Register test zone instead of real ones.
|
|
|
|
// Init
|
|
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.access, err = access.New(instance)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to create access module: %w", err)
|
|
}
|
|
instance.terminal, err = terminal.New(instance)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to create terminal module: %w", err)
|
|
}
|
|
instance.cabin, err = cabin.New(instance)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to create cabin module: %w", err)
|
|
}
|
|
module, err = New(instance)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to create docks 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.access.Start()
|
|
if err != nil {
|
|
return fmt.Errorf("failed to start access module: %w", err)
|
|
}
|
|
err = instance.terminal.Start()
|
|
if err != nil {
|
|
return fmt.Errorf("failed to start terminal 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 {
|
|
fmt.Printf("%s\n", err)
|
|
os.Exit(1)
|
|
}
|
|
}
|