unkinben
274c480b09
bootapi replaces Cobbler's PXE/kickstart side. It resolves a PXE-booting host
from NetBox (by MAC or hostname), renders an iPXE boot script and a kickstart
from Go text/templates, and serves them over HTTP. The ENC half already moved to
encapi; this covers the provisioning/boot half.
What's here:
- cmd/bootapi + internal/{config,model,netbox,render,server}; embedded default
templates under templates/ (AlmaLinux 9 + Fedora kickstarts, iPXE boot +
unknown-MAC fallbacks) ported from Cobbler's boot/bootstrap contract.
- NetBox client (v4.x API) behind a Resolver interface with a short-TTL cache;
tested against httptest fixtures using real NetBox JSON shapes.
- chi HTTP server: /ipxe/{mac}, /boot/ipxe?mac=, /ks/{ident}, healthz/readyz,
Prometheus /metrics. Unknown MAC -> safe fallback iPXE (200), unknown KS -> 404.
- Secrets (root pw hash, ssh keys) injected at render time from env/Vault, never
NetBox. Config is env-based per estate convention.
- Makefile (build/test/lint/docker + patch/minor/major), Dockerfile (distroless),
.woodpecker (pre-commit, golangci-lint v2 + go test -race, docker build on PR;
image push + Gitea binary release on v* tag), docs/ and example config.
go build/vet clean, go test -race green, golangci-lint v2 clean, pre-commit clean.
Claude-Session: https://claude.ai/code/session_015ur3i7D2azsMAWTSVABApv
95 lines
2.3 KiB
Go
95 lines
2.3 KiB
Go
package config
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestLoadDefaults(t *testing.T) {
|
|
clearEnv(t)
|
|
c, err := Load()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if c.ListenAddr != ":8000" {
|
|
t.Errorf("ListenAddr = %q", c.ListenAddr)
|
|
}
|
|
if c.CacheTTL != 30*time.Second {
|
|
t.Errorf("CacheTTL = %v", c.CacheTTL)
|
|
}
|
|
if c.DefaultTemplate != "almalinux9" {
|
|
t.Errorf("DefaultTemplate = %q", c.DefaultTemplate)
|
|
}
|
|
if c.PuppetServer != "puppet.query.consul" || c.PuppetCAServer != "puppetca.query.consul" {
|
|
t.Errorf("puppet servers = %q / %q", c.PuppetServer, c.PuppetCAServer)
|
|
}
|
|
if c.UnknownMACFallback != "local" {
|
|
t.Errorf("UnknownMACFallback = %q", c.UnknownMACFallback)
|
|
}
|
|
}
|
|
|
|
func TestLoadTokenFile(t *testing.T) {
|
|
clearEnv(t)
|
|
dir := t.TempDir()
|
|
tf := filepath.Join(dir, "token")
|
|
if err := os.WriteFile(tf, []byte(" secret-token\n"), 0o600); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
t.Setenv("BOOTAPI_NETBOX_TOKEN_FILE", tf)
|
|
c, err := Load()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if c.NetBoxToken != "secret-token" {
|
|
t.Errorf("token = %q, want trimmed file contents", c.NetBoxToken)
|
|
}
|
|
}
|
|
|
|
func TestLoadRejectsBadFallback(t *testing.T) {
|
|
clearEnv(t)
|
|
t.Setenv("BOOTAPI_UNKNOWN_MAC_FALLBACK", "bogus")
|
|
if _, err := Load(); err == nil {
|
|
t.Fatal("expected error for invalid fallback")
|
|
}
|
|
}
|
|
|
|
func TestLoadListsAndTrim(t *testing.T) {
|
|
clearEnv(t)
|
|
t.Setenv("BOOTAPI_NAMESERVERS", " 10.0.0.1, 10.0.0.2 ,")
|
|
t.Setenv("BOOTAPI_NETBOX_URL", "https://netbox.example.net/")
|
|
c, err := Load()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(c.Nameservers) != 2 || c.Nameservers[1] != "10.0.0.2" {
|
|
t.Errorf("nameservers = %v", c.Nameservers)
|
|
}
|
|
if c.NetBoxURL != "https://netbox.example.net" {
|
|
t.Errorf("NetBoxURL trailing slash not trimmed: %q", c.NetBoxURL)
|
|
}
|
|
}
|
|
|
|
// clearEnv unsets every BOOTAPI_* var so a developer's shell can't leak into
|
|
// the test. t.Setenv restores them after the test.
|
|
func clearEnv(t *testing.T) {
|
|
t.Helper()
|
|
for _, kv := range os.Environ() {
|
|
if k, _, ok := cut(kv, '='); ok && len(k) > 8 && k[:8] == "BOOTAPI_" {
|
|
// t.Setenv to "" is enough: Load treats empty as unset, and the
|
|
// test framework restores the original value on cleanup.
|
|
t.Setenv(k, "")
|
|
}
|
|
}
|
|
}
|
|
|
|
func cut(s string, sep byte) (before, after string, found bool) {
|
|
for i := 0; i < len(s); i++ {
|
|
if s[i] == sep {
|
|
return s[:i], s[i+1:], true
|
|
}
|
|
}
|
|
return s, "", false
|
|
}
|