Thales Maciel
fcedacba5c
Stop assuming one workstation layout for runtime artifacts, mapdns, and host tooling. The daemon and shell helpers now use portable mapdns configuration, and runtime bundles can carry bundle.json metadata for their default kernel, initrd, modules, rootfs, and helper paths. Load bundle metadata through config with a legacy layout fallback, thread mapdns_bin/mapdns_data_file through the Go and shell paths, and add command-scoped preflight checks for VM start, NAT, image build, work-disk resize, and SSH so missing tools or artifacts fail with actionable errors. Update the runtime-bundle manifest, docs, and tests to match the new model. Verified with go test ./..., make build, and bash -n customize.sh interactive.sh dns.sh make-rootfs.sh verify.sh.
108 lines
2.3 KiB
Go
108 lines
2.3 KiB
Go
package daemon
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"banger/internal/model"
|
|
"banger/internal/store"
|
|
)
|
|
|
|
func TestEnsureDefaultImageUsesConfiguredDefaultRootfs(t *testing.T) {
|
|
dir := t.TempDir()
|
|
rootfs := filepath.Join(dir, "rootfs-docker.ext4")
|
|
kernel := filepath.Join(dir, "vmlinux")
|
|
for _, path := range []string{rootfs, kernel} {
|
|
if err := os.WriteFile(path, []byte("test"), 0o644); err != nil {
|
|
t.Fatalf("write %s: %v", path, err)
|
|
}
|
|
}
|
|
|
|
db, err := store.Open(filepath.Join(dir, "state.db"))
|
|
if err != nil {
|
|
t.Fatalf("open store: %v", err)
|
|
}
|
|
t.Cleanup(func() {
|
|
_ = db.Close()
|
|
})
|
|
|
|
d := &Daemon{
|
|
config: model.DaemonConfig{
|
|
DefaultImageName: "default",
|
|
DefaultRootfs: rootfs,
|
|
DefaultKernel: kernel,
|
|
},
|
|
store: db,
|
|
}
|
|
|
|
if err := d.ensureDefaultImage(context.Background()); err != nil {
|
|
t.Fatalf("ensureDefaultImage: %v", err)
|
|
}
|
|
|
|
image, err := db.GetImageByName(context.Background(), "default")
|
|
if err != nil {
|
|
t.Fatalf("GetImageByName: %v", err)
|
|
}
|
|
if image.RootfsPath != rootfs {
|
|
t.Fatalf("RootfsPath = %q, want %q", image.RootfsPath, rootfs)
|
|
}
|
|
if image.KernelPath != kernel {
|
|
t.Fatalf("KernelPath = %q, want %q", image.KernelPath, kernel)
|
|
}
|
|
}
|
|
|
|
func TestSetDNSUsesConfiguredMapDNSDataFile(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
dataFile := filepath.Join(t.TempDir(), "mapdns", "records.json")
|
|
runner := &scriptedRunner{
|
|
t: t,
|
|
steps: []runnerStep{
|
|
{
|
|
call: runnerCall{
|
|
name: "custom-mapdns",
|
|
args: []string{"set", "--data-file", dataFile, "devbox.vm", "172.16.0.8"},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
d := &Daemon{
|
|
runner: runner,
|
|
config: model.DaemonConfig{
|
|
MapDNSBin: "custom-mapdns",
|
|
MapDNSDataFile: dataFile,
|
|
},
|
|
}
|
|
|
|
if err := d.setDNS(context.Background(), "devbox", "172.16.0.8"); err != nil {
|
|
t.Fatalf("setDNS: %v", err)
|
|
}
|
|
runner.assertExhausted()
|
|
}
|
|
|
|
func TestSetDNSUsesMapDNSDefaultsWhenDataFileUnset(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
runner := &scriptedRunner{
|
|
t: t,
|
|
steps: []runnerStep{
|
|
{
|
|
call: runnerCall{
|
|
name: "mapdns",
|
|
args: []string{"set", "devbox.vm", "172.16.0.8"},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
d := &Daemon{
|
|
runner: runner,
|
|
config: model.DaemonConfig{},
|
|
}
|
|
|
|
if err := d.setDNS(context.Background(), "devbox", "172.16.0.8"); err != nil {
|
|
t.Fatalf("setDNS: %v", err)
|
|
}
|
|
runner.assertExhausted()
|
|
}
|