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>
77 lines
2.3 KiB
Go
77 lines
2.3 KiB
Go
package providers
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"testing"
|
|
|
|
"github.com/wangjia/pangolin/server/internal/provision"
|
|
)
|
|
|
|
func TestRegistry_UnknownKind(t *testing.T) {
|
|
r := NewRegistry()
|
|
_, err := r.For(&provision.Provider{APIKind: "does-not-exist"})
|
|
if err == nil {
|
|
t.Fatal("expected error for unknown api_kind")
|
|
}
|
|
}
|
|
|
|
func TestRegistry_NoCredentials(t *testing.T) {
|
|
// Ensure the env vars are unset for a clean assertion.
|
|
t.Setenv("PROVISION_VULTR_API_KEY", "")
|
|
t.Setenv("PROVISION_HETZNER_API_TOKEN", "")
|
|
|
|
r := NewRegistry()
|
|
if _, err := r.For(&provision.Provider{APIKind: "vultr"}); !errors.Is(err, provision.ErrNoCredentials) {
|
|
t.Errorf("vultr without creds: err = %v, want ErrNoCredentials", err)
|
|
}
|
|
if _, err := r.For(&provision.Provider{APIKind: "hetzner"}); !errors.Is(err, provision.ErrNoCredentials) {
|
|
t.Errorf("hetzner without creds: err = %v, want ErrNoCredentials", err)
|
|
}
|
|
}
|
|
|
|
func TestRegistry_RegisterAndCache(t *testing.T) {
|
|
r := NewRegistry()
|
|
built := 0
|
|
r.Register("fake", func() (provision.CloudAdapter, error) {
|
|
built++
|
|
return stubAdapter{}, nil
|
|
})
|
|
p := &provision.Provider{APIKind: "fake"}
|
|
if _, err := r.For(p); err != nil {
|
|
t.Fatalf("For: %v", err)
|
|
}
|
|
if _, err := r.For(p); err != nil {
|
|
t.Fatalf("For (cached): %v", err)
|
|
}
|
|
if built != 1 {
|
|
t.Errorf("factory built %d times, want 1 (cached)", built)
|
|
}
|
|
}
|
|
|
|
func TestRegistry_CredentialsFromEnv(t *testing.T) {
|
|
t.Setenv("PROVISION_VULTR_API_KEY", "secret-key")
|
|
r := NewRegistry()
|
|
a, err := r.For(&provision.Provider{APIKind: "vultr"})
|
|
if err != nil {
|
|
t.Fatalf("vultr with creds: %v", err)
|
|
}
|
|
if a.Kind() != "vultr" {
|
|
t.Errorf("kind = %s, want vultr", a.Kind())
|
|
}
|
|
if !a.SupportsElasticIP() {
|
|
t.Error("vultr should support elastic IP (reserved IPs)")
|
|
}
|
|
}
|
|
|
|
type stubAdapter struct{}
|
|
|
|
func (stubAdapter) Kind() string { return "fake" }
|
|
func (stubAdapter) SupportsElasticIP() bool { return false }
|
|
func (stubAdapter) CreateInstance(context.Context, provision.CreateInput) (*provision.Instance, error) {
|
|
return &provision.Instance{ID: "x"}, nil
|
|
}
|
|
func (stubAdapter) DestroyInstance(context.Context, string) error { return nil }
|
|
func (stubAdapter) AttachIP(context.Context, string) (string, error) { return "", nil }
|
|
func (stubAdapter) ListRegions(context.Context) ([]provision.Region, error) { return nil, nil }
|