f-19: test: ajouter des test

Signed-off-by: GnomeZworc <nicolas.boufidjeline@g3e.fr>
2026-03-31 20:03:52 +02:00 · 2026-03-31 20:03:52 +02:00 · c7d20b4124
commit c7d20b4124
parent 3288a2a413
4 changed files with 593 additions and 0 deletions
--- a/internal/metadata/metadata_test.go
+++ b/internal/metadata/metadata_test.go
@ -0,0 +1,200 @@
+package metadata
+
+import (
+	"strings"
+	"testing"
+
+	"git.g3e.fr/syonad/two/pkg/db/kv"
+)
+
+func newCfg() NoCloudConfig {
+	return NoCloudConfig{
+		VpcName:  "vpc-test",
+		BindIP:   "169.254.169.254",
+		BindPort: "80",
+		Name:     "vm1",
+		Password: "s3cr3t",
+		SSHKEY:   "ssh-ed25519 AAAA... user@host",
+	}
+}
+
+func newTestDB(t *testing.T) interface{ Close() error } {
+	t.Helper()
+	db := kv.InitDB(kv.Config{Path: t.TempDir()}, false)
+	t.Cleanup(func() { db.Close() })
+	return db
+}
+
+// --- RenderConfig ---
+
+func TestRenderConfig_MetaData(t *testing.T) {
+	cfg := newCfg()
+	out, err := RenderConfig("templates/meta-data.tmpl", cfg)
+	if err != nil {
+		t.Fatalf("RenderConfig meta-data : %v", err)
+	}
+	if !strings.Contains(out, "instance-id: vm1") {
+		t.Errorf("instance-id absent :\n%s", out)
+	}
+	if !strings.Contains(out, "local-hostname: vm1") {
+		t.Errorf("local-hostname absent :\n%s", out)
+	}
+}
+
+func TestRenderConfig_VendorData_ContainsPassword(t *testing.T) {
+	cfg := newCfg()
+	out, err := RenderConfig("templates/vendor-data.tmpl", cfg)
+	if err != nil {
+		t.Fatalf("RenderConfig vendor-data : %v", err)
+	}
+	if !strings.Contains(out, "s3cr3t") {
+		t.Errorf("password absent du vendor-data :\n%s", out)
+	}
+}
+
+func TestRenderConfig_VendorData_ContainsSSHKey(t *testing.T) {
+	cfg := newCfg()
+	out, err := RenderConfig("templates/vendor-data.tmpl", cfg)
+	if err != nil {
+		t.Fatalf("RenderConfig vendor-data : %v", err)
+	}
+	if !strings.Contains(out, "ssh-ed25519 AAAA... user@host") {
+		t.Errorf("clé SSH absente du vendor-data :\n%s", out)
+	}
+}
+
+func TestRenderConfig_NetworkConfig(t *testing.T) {
+	cfg := newCfg()
+	out, err := RenderConfig("templates/network-config.tmpl", cfg)
+	if err != nil {
+		t.Fatalf("RenderConfig network-config : %v", err)
+	}
+	if !strings.Contains(out, "dhcp4: true") {
+		t.Errorf("dhcp4 absent du network-config :\n%s", out)
+	}
+}
+
+func TestRenderConfig_UserData(t *testing.T) {
+	cfg := newCfg()
+	out, err := RenderConfig("templates/user-data.tmpl", cfg)
+	if err != nil {
+		t.Fatalf("RenderConfig user-data : %v", err)
+	}
+	if !strings.Contains(out, "passwd -d root") {
+		t.Errorf("user-data inattendu :\n%s", out)
+	}
+}
+
+func TestRenderConfig_InvalidTemplate(t *testing.T) {
+	_, err := RenderConfig("templates/inexistant.tmpl", newCfg())
+	if err == nil {
+		t.Error("RenderConfig devrait retourner une erreur pour un template inexistant")
+	}
+}
+
+func TestRenderConfig_SpecialCharsInName(t *testing.T) {
+	cfg := newCfg()
+	cfg.Name = "vm-prod-01"
+	out, err := RenderConfig("templates/meta-data.tmpl", cfg)
+	if err != nil {
+		t.Fatalf("RenderConfig : %v", err)
+	}
+	if !strings.Contains(out, "vm-prod-01") {
+		t.Errorf("nom vm-prod-01 absent :\n%s", out)
+	}
+}
+
+// --- LoadNcCloudInDB / UnLoadNoCloudInDB ---
+
+func TestLoadNcCloudInDB_StoresAllKeys(t *testing.T) {
+	db := kv.InitDB(kv.Config{Path: t.TempDir()}, false)
+	t.Cleanup(func() { db.Close() })
+
+	cfg := newCfg()
+	LoadNcCloudInDB(cfg, db)
+
+	keys := []string{
+		"metadata/vm1/meta-data",
+		"metadata/vm1/user-data",
+		"metadata/vm1/network-config",
+		"metadata/vm1/vendor-data",
+		"metadata/vm1/vpc",
+		"metadata/vm1/bind_ip",
+		"metadata/vm1/bind_port",
+	}
+	for _, key := range keys {
+		val, err := kv.GetFromDB(db, key)
+		if err != nil {
+			t.Errorf("clé %q absente après LoadNcCloudInDB : %v", key, err)
+		}
+		if val == "" && key != "metadata/vm1/user-data" {
+			t.Errorf("clé %q vide après LoadNcCloudInDB", key)
+		}
+	}
+}
+
+func TestLoadNcCloudInDB_VpcAndBindValues(t *testing.T) {
+	db := kv.InitDB(kv.Config{Path: t.TempDir()}, false)
+	t.Cleanup(func() { db.Close() })
+
+	cfg := newCfg()
+	LoadNcCloudInDB(cfg, db)
+
+	vpc, _ := kv.GetFromDB(db, "metadata/vm1/vpc")
+	if vpc != "vpc-test" {
+		t.Errorf("vpc attendu %q, obtenu %q", "vpc-test", vpc)
+	}
+
+	ip, _ := kv.GetFromDB(db, "metadata/vm1/bind_ip")
+	if ip != "169.254.169.254" {
+		t.Errorf("bind_ip attendu %q, obtenu %q", "169.254.169.254", ip)
+	}
+
+	port, _ := kv.GetFromDB(db, "metadata/vm1/bind_port")
+	if port != "80" {
+		t.Errorf("bind_port attendu %q, obtenu %q", "80", port)
+	}
+}
+
+func TestUnLoadNoCloudInDB_RemovesAllKeys(t *testing.T) {
+	db := kv.InitDB(kv.Config{Path: t.TempDir()}, false)
+	t.Cleanup(func() { db.Close() })
+
+	cfg := newCfg()
+	LoadNcCloudInDB(cfg, db)
+	UnLoadNoCloudInDB("vm1", db)
+
+	keys := []string{
+		"metadata/vm1/meta-data",
+		"metadata/vm1/user-data",
+		"metadata/vm1/network-config",
+		"metadata/vm1/vendor-data",
+		"metadata/vm1/vpc",
+		"metadata/vm1/bind_ip",
+		"metadata/vm1/bind_port",
+	}
+	for _, key := range keys {
+		_, err := kv.GetFromDB(db, key)
+		if err == nil {
+			t.Errorf("clé %q devrait être supprimée après UnLoadNoCloudInDB", key)
+		}
+	}
+}
+
+func TestUnLoadNoCloudInDB_DoesNotAffectOtherVMs(t *testing.T) {
+	db := kv.InitDB(kv.Config{Path: t.TempDir()}, false)
+	t.Cleanup(func() { db.Close() })
+
+	cfg1 := newCfg()
+	cfg2 := newCfg()
+	cfg2.Name = "vm2"
+	LoadNcCloudInDB(cfg1, db)
+	LoadNcCloudInDB(cfg2, db)
+
+	UnLoadNoCloudInDB("vm1", db)
+
+	_, err := kv.GetFromDB(db, "metadata/vm2/vpc")
+	if err != nil {
+		t.Errorf("vm2 ne devrait pas être supprimée : %v", err)
+	}
+}