Restructure modules (#1572)
* Move portbase into monorepo * Add new simple module mgr * [WIP] Switch to new simple module mgr * Add StateMgr and more worker variants * [WIP] Switch more modules * [WIP] Switch more modules * [WIP] swtich more modules * [WIP] switch all SPN modules * [WIP] switch all service modules * [WIP] Convert all workers to the new module system * [WIP] add new task system to module manager * [WIP] Add second take for scheduling workers * [WIP] Add FIXME for bugs in new scheduler * [WIP] Add minor improvements to scheduler * [WIP] Add new worker scheduler * [WIP] Fix more bug related to new module system * [WIP] Fix start handing of the new module system * [WIP] Improve startup process * [WIP] Fix minor issues * [WIP] Fix missing subsystem in settings * [WIP] Initialize managers in constructor * [WIP] Move module event initialization to constrictors * [WIP] Fix setting for enabling and disabling the SPN module * [WIP] Move API registeration into module construction * [WIP] Update states mgr for all modules * [WIP] Add CmdLine operation support * Add state helper methods to module group and instance * Add notification and module status handling to status package * Fix starting issues * Remove pilot widget and update security lock to new status data * Remove debug logs * Improve http server shutdown * Add workaround for cleanly shutting down firewall+netquery * Improve logging * Add syncing states with notifications for new module system * Improve starting, stopping, shutdown; resolve FIXMEs/TODOs * [WIP] Fix most unit tests * Review new module system and fix minor issues * Push shutdown and restart events again via API * Set sleep mode via interface * Update example/template module * [WIP] Fix spn/cabin unit test * Remove deprecated UI elements * Make log output more similar for the logging transition phase * Switch spn hub and observer cmds to new module system * Fix log sources * Make worker mgr less error prone * Fix tests and minor issues * Fix observation hub * Improve shutdown and restart handling * Split up big connection.go source file * Move varint and dsd packages to structures repo * Improve expansion test * Fix linter warnings * Fix interception module on windows * Fix linter errors --------- Co-authored-by: Vladimir Stoilov <vladimir@safing.io>
This commit is contained in:
Daniel Hååvi
193
base/config/set_test.go
Normal file
193
base/config/set_test.go
Normal file
@@ -0,0 +1,193 @@
|
||||
//nolint:goconst
|
||||
package config
|
||||
|
||||
import "testing"
|
||||
|
||||
func TestLayersGetters(t *testing.T) { //nolint:paralleltest
|
||||
// reset
|
||||
options = make(map[string]*Option)
|
||||
|
||||
mapData, err := JSONToMap([]byte(`
|
||||
{
|
||||
"monkey": "1",
|
||||
"elephant": 2,
|
||||
"zebras": {
|
||||
"zebra": ["black", "white"],
|
||||
"weird_zebra": ["black", -1]
|
||||
},
|
||||
"env": {
|
||||
"hot": true
|
||||
}
|
||||
}
|
||||
`))
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
validationErrors, _ := ReplaceConfig(mapData)
|
||||
if len(validationErrors) > 0 {
|
||||
t.Fatalf("%d errors, first: %s", len(validationErrors), validationErrors[0].Error())
|
||||
}
|
||||
|
||||
// Test missing values
|
||||
|
||||
missingString := GetAsString("missing", "fallback")
|
||||
if missingString() != "fallback" {
|
||||
t.Error("expected fallback value: fallback")
|
||||
}
|
||||
|
||||
missingStringArray := GetAsStringArray("missing", []string{"fallback"})
|
||||
if len(missingStringArray()) != 1 || missingStringArray()[0] != "fallback" {
|
||||
t.Error("expected fallback value: [fallback]")
|
||||
}
|
||||
|
||||
missingInt := GetAsInt("missing", -1)
|
||||
if missingInt() != -1 {
|
||||
t.Error("expected fallback value: -1")
|
||||
}
|
||||
|
||||
missingBool := GetAsBool("missing", false)
|
||||
if missingBool() {
|
||||
t.Error("expected fallback value: false")
|
||||
}
|
||||
|
||||
// Test value mismatch
|
||||
|
||||
notString := GetAsString("elephant", "fallback")
|
||||
if notString() != "fallback" {
|
||||
t.Error("expected fallback value: fallback")
|
||||
}
|
||||
|
||||
notStringArray := GetAsStringArray("elephant", []string{"fallback"})
|
||||
if len(notStringArray()) != 1 || notStringArray()[0] != "fallback" {
|
||||
t.Error("expected fallback value: [fallback]")
|
||||
}
|
||||
|
||||
mixedStringArray := GetAsStringArray("zebras/weird_zebra", []string{"fallback"})
|
||||
if len(mixedStringArray()) != 1 || mixedStringArray()[0] != "fallback" {
|
||||
t.Error("expected fallback value: [fallback]")
|
||||
}
|
||||
|
||||
notInt := GetAsInt("monkey", -1)
|
||||
if notInt() != -1 {
|
||||
t.Error("expected fallback value: -1")
|
||||
}
|
||||
|
||||
notBool := GetAsBool("monkey", false)
|
||||
if notBool() {
|
||||
t.Error("expected fallback value: false")
|
||||
}
|
||||
}
|
||||
|
||||
func TestLayersSetters(t *testing.T) { //nolint:paralleltest
|
||||
// reset
|
||||
options = make(map[string]*Option)
|
||||
|
||||
_ = Register(&Option{
|
||||
Name: "name",
|
||||
Key: "monkey",
|
||||
Description: "description",
|
||||
ReleaseLevel: ReleaseLevelStable,
|
||||
ExpertiseLevel: ExpertiseLevelUser,
|
||||
OptType: OptTypeString,
|
||||
DefaultValue: "banana",
|
||||
ValidationRegex: "^(banana|water)$",
|
||||
})
|
||||
_ = Register(&Option{
|
||||
Name: "name",
|
||||
Key: "zebras/zebra",
|
||||
Description: "description",
|
||||
ReleaseLevel: ReleaseLevelStable,
|
||||
ExpertiseLevel: ExpertiseLevelUser,
|
||||
OptType: OptTypeStringArray,
|
||||
DefaultValue: []string{"black", "white"},
|
||||
ValidationRegex: "^[a-z]+$",
|
||||
})
|
||||
_ = Register(&Option{
|
||||
Name: "name",
|
||||
Key: "elephant",
|
||||
Description: "description",
|
||||
ReleaseLevel: ReleaseLevelStable,
|
||||
ExpertiseLevel: ExpertiseLevelUser,
|
||||
OptType: OptTypeInt,
|
||||
DefaultValue: 2,
|
||||
ValidationRegex: "",
|
||||
})
|
||||
_ = Register(&Option{
|
||||
Name: "name",
|
||||
Key: "hot",
|
||||
Description: "description",
|
||||
ReleaseLevel: ReleaseLevelStable,
|
||||
ExpertiseLevel: ExpertiseLevelUser,
|
||||
OptType: OptTypeBool,
|
||||
DefaultValue: true,
|
||||
ValidationRegex: "",
|
||||
})
|
||||
|
||||
// correct types
|
||||
if err := SetConfigOption("monkey", "banana"); err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
if err := SetConfigOption("zebras/zebra", []string{"black", "white"}); err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
if err := SetDefaultConfigOption("elephant", 2); err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
if err := SetDefaultConfigOption("hot", true); err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
|
||||
// incorrect types
|
||||
if err := SetConfigOption("monkey", []string{"black", "white"}); err == nil {
|
||||
t.Error("should fail")
|
||||
}
|
||||
if err := SetConfigOption("zebras/zebra", 2); err == nil {
|
||||
t.Error("should fail")
|
||||
}
|
||||
if err := SetDefaultConfigOption("elephant", true); err == nil {
|
||||
t.Error("should fail")
|
||||
}
|
||||
if err := SetDefaultConfigOption("hot", "banana"); err == nil {
|
||||
t.Error("should fail")
|
||||
}
|
||||
if err := SetDefaultConfigOption("hot", []byte{0}); err == nil {
|
||||
t.Error("should fail")
|
||||
}
|
||||
|
||||
// validation fail
|
||||
if err := SetConfigOption("monkey", "dirt"); err == nil {
|
||||
t.Error("should fail")
|
||||
}
|
||||
if err := SetConfigOption("zebras/zebra", []string{"Element649"}); err == nil {
|
||||
t.Error("should fail")
|
||||
}
|
||||
|
||||
// unregistered checking
|
||||
if err := SetConfigOption("invalid", "banana"); err == nil {
|
||||
t.Error("should fail")
|
||||
}
|
||||
if err := SetConfigOption("invalid", []string{"black", "white"}); err == nil {
|
||||
t.Error("should fail")
|
||||
}
|
||||
if err := SetConfigOption("invalid", 2); err == nil {
|
||||
t.Error("should fail")
|
||||
}
|
||||
if err := SetConfigOption("invalid", true); err == nil {
|
||||
t.Error("should fail")
|
||||
}
|
||||
if err := SetConfigOption("invalid", []byte{0}); err == nil {
|
||||
t.Error("should fail")
|
||||
}
|
||||
|
||||
// delete
|
||||
if err := SetConfigOption("monkey", nil); err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
if err := SetDefaultConfigOption("elephant", nil); err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
if err := SetDefaultConfigOption("invalid_delete", nil); err == nil {
|
||||
t.Error("should fail")
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user