wangjia
3d5bac66b4
IaC 面 (infra/) 与控制面 (server/internal/provision) 双产出，落地 doc/04 §4 「节点是牲口」弹性拓扑与 make-before-break 一键更换。 server/internal/provision: - CloudAdapter 适配层 + Registry（首发 vultr 消耗品池 / hetzner 精品池各一）； 厂商凭证仅从 PROVISION_<VENDOR>_* env 注入，不入库不入 git。 - ProvisionService：CreateNode(幂等键重放不重复开机)、DestroyNode(幂等)、 RotateIP(换 IP 不换机 + version bump)、ListProviders。 - Replace 一键更换：先建后拆，新机 up 先于旧机 draining（容量不下降）， replacement_uuid 幂等键 + replacements 表分步记录，崩溃可续跑不重复。 - RotatePool：池内滚动轮换，并发度 1–2。 - cmd/nodectl CLI：create/destroy/rotate-ip/replace/rotate-pool/providers。 - 单测（mock 厂商 API + 内存 Store）：幂等重放、make-before-break 时序断言、 开机失败/探活超时→destroyed+失败计数+告警钩子、崩溃续跑、RotatePool。 infra/: - terraform/：探针机 + 控制面基线模块化（probe / control-plane）+ README， 低频基线进 state，节点不进 Terraform。 - cloud-init/node.yaml.tmpl：节点引导模板（注入一次性 bootstrap token，task #5）。 - identity-isolation.md：身份隔离登记表（doc/06 §2 红线，无任何凭证）。 migrations/000008：nodes 增 provider_instance_id/elastic_ip_id、node_events 增 ip_rotated、provision_idempotency / replacements 表（附加式，不动现网）。 红线：仅面向新厂商池，绝不纳管现网生产 EC2(deploy/ marzban)。 Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
85 lines
2.2 KiB
Go
85 lines
2.2 KiB
Go
package provision
|
|
|
|
import (
|
|
"os"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
// TestRealCloudInitTemplate ensures the shipped infra/cloud-init/node.yaml.tmpl
|
|
// stays a valid Go template and renders the bootstrap token + node identity.
|
|
func TestRealCloudInitTemplate(t *testing.T) {
|
|
const path = "../../../infra/cloud-init/node.yaml.tmpl"
|
|
if _, err := os.Stat(path); err != nil {
|
|
t.Skipf("template not found at %s: %v", path, err)
|
|
}
|
|
r, err := NewTemplateRenderer(path)
|
|
if err != nil {
|
|
t.Fatalf("parse real template: %v", err)
|
|
}
|
|
out, err := r.Render(CloudInitData{
|
|
NodeUUID: "11111111-2222-4333-8444-555555555555",
|
|
BootstrapToken: "deadbeef",
|
|
Region: "hkg",
|
|
Role: RoleEntry,
|
|
Tier: TierFree,
|
|
ControlPlaneURL: "https://cp.example",
|
|
})
|
|
if err != nil {
|
|
t.Fatalf("render real template: %v", err)
|
|
}
|
|
for _, want := range []string{
|
|
"PANGOLIN_NODE_UUID=11111111-2222-4333-8444-555555555555",
|
|
"PANGOLIN_BOOTSTRAP_TOKEN=deadbeef",
|
|
"PANGOLIN_CONTROL_PLANE_URL=https://cp.example",
|
|
"#cloud-config",
|
|
} {
|
|
if !strings.Contains(out, want) {
|
|
t.Errorf("rendered cloud-init missing %q", want)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestNewUUID_Format(t *testing.T) {
|
|
u, err := newUUID()
|
|
if err != nil {
|
|
t.Fatalf("newUUID: %v", err)
|
|
}
|
|
if len(u) != 36 || strings.Count(u, "-") != 4 {
|
|
t.Errorf("uuid format wrong: %q", u)
|
|
}
|
|
if u[14] != '4' {
|
|
t.Errorf("not a v4 uuid: %q", u)
|
|
}
|
|
u2, _ := newUUID()
|
|
if u == u2 {
|
|
t.Error("uuid not unique")
|
|
}
|
|
}
|
|
|
|
func TestTemplateRenderer(t *testing.T) {
|
|
r, err := NewTemplateRendererFromString("id={{.NodeUUID}}\ntoken={{.BootstrapToken}}\ncp={{.ControlPlaneURL}}")
|
|
if err != nil {
|
|
t.Fatalf("parse: %v", err)
|
|
}
|
|
out, err := r.Render(CloudInitData{NodeUUID: "abc", BootstrapToken: "tok", ControlPlaneURL: "https://cp"})
|
|
if err != nil {
|
|
t.Fatalf("render: %v", err)
|
|
}
|
|
for _, want := range []string{"id=abc", "token=tok", "cp=https://cp"} {
|
|
if !strings.Contains(out, want) {
|
|
t.Errorf("rendered output missing %q:\n%s", want, out)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestTemplateRenderer_MissingKeyErrors(t *testing.T) {
|
|
r, err := NewTemplateRendererFromString("{{.NoSuchField}}")
|
|
if err != nil {
|
|
t.Fatalf("parse: %v", err)
|
|
}
|
|
if _, err := r.Render(CloudInitData{}); err == nil {
|
|
t.Error("expected error for missing template key")
|
|
}
|
|
}
|