syonad/two
2
0
Fork 0
Code Issues 17 Pull requests 1 Projects Releases 4 Packages Wiki Activity Actions

f-29: watchdog: add specific config #29

Browse source 
Signed-off-by: GnomeZworc <nicolas.boufidjeline@g3e.fr>
This commit is contained in:
GnomeZworc 2026-08-17 23:52:48 +02:00
commit 2ce7ebafe7
Signed by: nicolas.boufideline
GPG key ID: 4406BBBF8845D632
3 changed files with 78 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

65
internal/config/agent/config_test.go
View file

@ -81,3 +81,68 @@ database:
t.Errorf("attendu %q, obtenu %q", "/opt/two/data", cfg.Database.Path)
}
}
func TestLoadConfig_WatchdogDefauts(t *testing.T) {
path := writeYAML(t, "")
cfg, err := LoadConfig(path)
if err != nil {
t.Fatalf("LoadConfig a échoué : %v", err)
}
if cfg.Watchdog.Enabled {
t.Error("watchdog.enabled devrait être false par défaut")
}
if cfg.Watchdog.IntervalSeconds != 60 {
t.Errorf("watchdog.interval_seconds attendu 60, obtenu %d", cfg.Watchdog.IntervalSeconds)
}
}
func TestLoadConfig_WatchdogValeursExplicites(t *testing.T) {
path := writeYAML(t, `
watchdog:
enabled: true
interval_seconds: 30
`)
cfg, err := LoadConfig(path)
if err != nil {
t.Fatalf("LoadConfig a échoué : %v", err)
}
if !cfg.Watchdog.Enabled {
t.Error("watchdog.enabled attendu true")
}
if cfg.Watchdog.IntervalSeconds != 30 {
t.Errorf("watchdog.interval_seconds attendu 30, obtenu %d", cfg.Watchdog.IntervalSeconds)
}
}
func TestLoadConfig_WatchdogActiveSansIntervalle(t *testing.T) {
path := writeYAML(t, `
watchdog:
enabled: true
`)
cfg, err := LoadConfig(path)
if err != nil {
t.Fatalf("LoadConfig a échoué : %v", err)
}
if !cfg.Watchdog.Enabled {
t.Error("watchdog.enabled attendu true")
}
if cfg.Watchdog.IntervalSeconds != 60 {
t.Errorf("watchdog.interval_seconds attendu 60 (défaut viper), obtenu %d", cfg.Watchdog.IntervalSeconds)
}
}
func TestLoadConfig_ExempleFourniEstValide(t *testing.T) {
cfg, err := LoadConfig("../../../conf/agent/config.exemple.yml")
if err != nil {
t.Fatalf("config.exemple.yml illisible : %v", err)
}
if !cfg.Watchdog.Enabled {
t.Error("config.exemple.yml devrait activer le watchdog")
}
if cfg.Watchdog.IntervalSeconds != 60 {
t.Errorf("config.exemple.yml : interval_seconds attendu 60, obtenu %d", cfg.Watchdog.IntervalSeconds)
}
if cfg.QEMU.QMPDir == "" {
t.Error("config.exemple.yml devrait définir qemu.qmp_dir")
}
}

6
internal/config/agent/struct.go
View file

@ -44,6 +44,10 @@ type Config struct {
MonitorDir string `mapstructure:"monitor_dir"`
QMPDir string `mapstructure:"qmp_dir"`
} `mapstructure:"qemu"`
Watchdog struct {
Enabled bool `mapstructure:"enabled"`
IntervalSeconds int `mapstructure:"interval_seconds"`
} `mapstructure:"watchdog"`
DefaultInterface string `mapstructure:"default_interface"`
Interfaces map[string]string `mapstructure:"interfaces"`
}
@ -69,6 +73,8 @@ func LoadConfig(path string) (*Config, error) {
v.SetDefault("qemu.serial_dir", "/run/two/vms/serial")
v.SetDefault("qemu.monitor_dir", "/run/two/vms/monitor")
v.SetDefault("qemu.qmp_dir", "/run/two/vms/qmp")
v.SetDefault("watchdog.enabled", false)
v.SetDefault("watchdog.interval_seconds", 60)
v.SetDefault("admin.enabled", false)
v.SetDefault("admin.address", "127.0.0.1")
v.SetDefault("admin.port", 9091)