ms2k/chultun
SHA256
Watch
0
Code Releases 2 Activity

Fix profile updates and validate configuration before apply

Browse source
This commit is contained in:
ms2k 2026-10-05 21:45:05 +09:00
commit 2bf14948d7
1 changed files with 372 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

74
internal/budget/validate_test.go Normal file
View file

@ -0,0 +1,74 @@
package budget
import (
"bytes"
"os"
"path/filepath"
"strings"
"testing"
"github.com/BurntSushi/toml"
)
func TestConfigInvalid(t *testing.T) {
for _, tc := range []struct {
name string
field string
change func(*config)
}{
{"cpu", "profiles.compute", func(c *config) {
p := c.Profiles["compute"]
p.CPU = "invalid%"
c.Profiles["compute"] = p
}},
{"cpu-suffix", "profiles.compute", func(c *config) {
p := c.Profiles["compute"]
p.CPU = "800"
c.Profiles["compute"] = p
}},
{"high", "profiles.compute", func(c *config) {
p := c.Profiles["compute"]
p.High = "invalid"
c.Profiles["compute"] = p
}},
{"memory", "profiles.compute", func(c *config) {
p := c.Profiles["compute"]
p.Memory = "invalid"
c.Profiles["compute"] = p
}},
{"swap", "profiles.compute", func(c *config) {
p := c.Profiles["compute"]
p.Swap = "invalid"
c.Profiles["compute"] = p
}},
{"io", "profiles.compute", func(c *config) {
c.Devices[0].Limits["compute"] = ioLimit{"invalid", "500M", "20000", "20000"}
}},
{"command", "commands.java", func(c *config) { c.Commands["java"] = "missing" }},
{"python", "python", func(c *config) { c.Python = "missing" }},
{"shared", "shared.editor", func(c *config) { c.Shared["editor"] = "missing" }},
{"apps", "apps.missing", func(c *config) { c.Apps["missing"] = []string{"editor"} }},
{"units", "units.missing", func(c *config) { c.Units["missing"] = []string{"worker.service"} }},
{"prefixes", "prefixes.missing", func(c *config) { c.Prefixes["missing"] = []string{"app-editor-.scope.d"} }},
{"device", "devices[0].limits", func(c *config) {
c.Devices[0].Limits["missing"] = ioLimit{"100M", "100M", "4000", "4000"}
}},
{"prefix", "prefixes.default", func(c *config) { c.Prefixes["default"] = []string{"snap-firefox"} }},
} {
t.Run(tc.name, func(t *testing.T) {
c := testConfig()
tc.change(&c)
var data bytes.Buffer
if err := toml.NewEncoder(&data).Encode(c); err != nil {
t.Fatal(err)
}
file := filepath.Join(t.TempDir(), "profiles.toml")
if err := os.WriteFile(file, data.Bytes(), 0600); err != nil {
t.Fatal(err)
}
if _, err := loadConfig(file); err == nil || !strings.Contains(err.Error(), tc.field) {
t.Fatalf("expected an error naming %s, got %v", tc.field, err)
}
})
}
}