f-34: metadata: update metadata gen files #34
Signed-off-by: GnomeZworc <nicolas.boufidjeline@g3e.fr>
This commit is contained in:
parent
63fc305736
commit
883d628bf6
6 changed files with 271 additions and 59 deletions
|
|
@ -25,6 +25,21 @@ func useTestDir(t *testing.T) string {
|
|||
return t.TempDir()
|
||||
}
|
||||
|
||||
|
||||
func mustWrite(t *testing.T, cfg NoCloudConfig, dir string) {
|
||||
t.Helper()
|
||||
if err := WriteNoCloudFiles(cfg, dir); err != nil {
|
||||
t.Fatalf("WriteNoCloudFiles : %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
func mustRemove(t *testing.T, vmName, dir string) {
|
||||
t.Helper()
|
||||
if err := RemoveNoCloudFiles(vmName, dir); err != nil {
|
||||
t.Fatalf("RemoveNoCloudFiles : %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
// --- RenderConfig ---
|
||||
|
||||
func TestRenderConfig_MetaData(t *testing.T) {
|
||||
|
|
@ -74,14 +89,23 @@ func TestRenderConfig_NetworkConfig(t *testing.T) {
|
|||
}
|
||||
}
|
||||
|
||||
func TestRenderConfig_UserData(t *testing.T) {
|
||||
cfg := newCfg()
|
||||
out, err := RenderConfig("templates/user-data.tmpl", cfg)
|
||||
func TestRenderConfig_UserData_DefaultIsEmpty(t *testing.T) {
|
||||
doc, err := renderDocument(DocUserData, newCfg())
|
||||
if err != nil {
|
||||
t.Fatalf("RenderConfig user-data : %v", err)
|
||||
t.Fatalf("renderDocument user-data : %v", err)
|
||||
}
|
||||
if !strings.Contains(out, "passwd -d root") {
|
||||
t.Errorf("user-data inattendu :\n%s", out)
|
||||
if doc != "" {
|
||||
t.Errorf("le user-data par défaut doit être vide, obtenu :\n%q", doc)
|
||||
}
|
||||
}
|
||||
|
||||
func TestRenderConfig_UserData_NeverTouchesRootPassword(t *testing.T) {
|
||||
doc, err := renderDocument(DocUserData, newCfg())
|
||||
if err != nil {
|
||||
t.Fatalf("renderDocument user-data : %v", err)
|
||||
}
|
||||
if strings.Contains(doc, "passwd -d root") {
|
||||
t.Errorf("l'agent ne doit imposer aucune modification du compte root :\n%s", doc)
|
||||
}
|
||||
}
|
||||
|
||||
|
|
@ -104,21 +128,21 @@ func TestRenderConfig_SpecialCharsInName(t *testing.T) {
|
|||
}
|
||||
}
|
||||
|
||||
// --- LoadNcCloudInDB / UnLoadNoCloudInDB ---
|
||||
// --- WriteNoCloudFiles / RemoveNoCloudFiles ---
|
||||
|
||||
func readTestFile(t *testing.T, dir, vmName, name string) string {
|
||||
t.Helper()
|
||||
b, err := os.ReadFile(filepath.Join(dir, vmName, name))
|
||||
if err != nil {
|
||||
t.Errorf("fichier %q absent après LoadNcCloudInDB : %v", name, err)
|
||||
t.Errorf("fichier %q absent après WriteNoCloudFiles : %v", name, err)
|
||||
return ""
|
||||
}
|
||||
return string(b)
|
||||
}
|
||||
|
||||
func TestLoadNcCloudInDB_StoresAllFiles(t *testing.T) {
|
||||
func TestWriteNoCloudFiles_StoresAllFiles(t *testing.T) {
|
||||
dir := useTestDir(t)
|
||||
LoadNcCloudInDB(newCfg(), dir)
|
||||
mustWrite(t, newCfg(), dir)
|
||||
|
||||
files := []string{"meta-data", "user-data", "network-config", "vendor-data", "vpc", "bind_ip", "bind_port"}
|
||||
for _, f := range files {
|
||||
|
|
@ -129,9 +153,9 @@ func TestLoadNcCloudInDB_StoresAllFiles(t *testing.T) {
|
|||
}
|
||||
}
|
||||
|
||||
func TestLoadNcCloudInDB_VpcAndBindValues(t *testing.T) {
|
||||
func TestWriteNoCloudFiles_VpcAndBindValues(t *testing.T) {
|
||||
dir := useTestDir(t)
|
||||
LoadNcCloudInDB(newCfg(), dir)
|
||||
mustWrite(t, newCfg(), dir)
|
||||
|
||||
if vpc := readTestFile(t, dir, "vm1", "vpc"); vpc != "vpc-test" {
|
||||
t.Errorf("vpc attendu %q, obtenu %q", "vpc-test", vpc)
|
||||
|
|
@ -144,26 +168,26 @@ func TestLoadNcCloudInDB_VpcAndBindValues(t *testing.T) {
|
|||
}
|
||||
}
|
||||
|
||||
func TestUnLoadNoCloudInDB_RemovesAllFiles(t *testing.T) {
|
||||
func TestRemoveNoCloudFiles_RemovesAllFiles(t *testing.T) {
|
||||
dir := useTestDir(t)
|
||||
LoadNcCloudInDB(newCfg(), dir)
|
||||
UnLoadNoCloudInDB("vm1", dir)
|
||||
mustWrite(t, newCfg(), dir)
|
||||
mustRemove(t, "vm1", dir)
|
||||
|
||||
if _, err := os.Stat(filepath.Join(dir, "vm1")); !os.IsNotExist(err) {
|
||||
t.Error("répertoire vm1 devrait être supprimé après UnLoadNoCloudInDB")
|
||||
t.Error("répertoire vm1 devrait être supprimé après RemoveNoCloudFiles")
|
||||
}
|
||||
}
|
||||
|
||||
func TestUnLoadNoCloudInDB_DoesNotAffectOtherVMs(t *testing.T) {
|
||||
func TestRemoveNoCloudFiles_DoesNotAffectOtherVMs(t *testing.T) {
|
||||
dir := useTestDir(t)
|
||||
|
||||
cfg1 := newCfg()
|
||||
cfg2 := newCfg()
|
||||
cfg2.Name = "vm2"
|
||||
LoadNcCloudInDB(cfg1, dir)
|
||||
LoadNcCloudInDB(cfg2, dir)
|
||||
mustWrite(t, cfg1, dir)
|
||||
mustWrite(t, cfg2, dir)
|
||||
|
||||
UnLoadNoCloudInDB("vm1", dir)
|
||||
mustRemove(t, "vm1", dir)
|
||||
|
||||
if _, err := os.Stat(filepath.Join(dir, "vm2", "vpc")); err != nil {
|
||||
t.Errorf("vm2 ne devrait pas être supprimée : %v", err)
|
||||
|
|
@ -264,3 +288,138 @@ func TestRootHandler_ContentType(t *testing.T) {
|
|||
t.Errorf("Content-Type attendu text/yaml, obtenu %q", ct)
|
||||
}
|
||||
}
|
||||
|
||||
// --- vendor-data : blocs conditionnels ---
|
||||
|
||||
func vendorData(t *testing.T, password, sshkey string) string {
|
||||
t.Helper()
|
||||
cfg := newCfg()
|
||||
cfg.Password = password
|
||||
cfg.SSHKEY = sshkey
|
||||
out, err := renderDocument(DocVendorData, cfg)
|
||||
if err != nil {
|
||||
t.Fatalf("renderDocument vendor-data : %v", err)
|
||||
}
|
||||
return out
|
||||
}
|
||||
|
||||
func TestVendorData_NoCredentials_EmitsNothing(t *testing.T) {
|
||||
out := vendorData(t, "", "")
|
||||
if out != "" {
|
||||
t.Errorf("sans mot de passe ni clé, aucun compte ne doit être créé :\n%s", out)
|
||||
}
|
||||
}
|
||||
|
||||
func TestVendorData_NoCredentials_NeverEmitsEmptyValues(t *testing.T) {
|
||||
out := vendorData(t, "", "")
|
||||
for _, forbidden := range []string{`passwd: ""`, `- ""`, "lock_passwd: false"} {
|
||||
if strings.Contains(out, forbidden) {
|
||||
t.Errorf("valeur vide %q émise :\n%s", forbidden, out)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestVendorData_PasswordOnly(t *testing.T) {
|
||||
out := vendorData(t, "$6$rounds$hash", "")
|
||||
if !strings.Contains(out, `passwd: "$6$rounds$hash"`) {
|
||||
t.Errorf("hash absent :\n%s", out)
|
||||
}
|
||||
if !strings.Contains(out, "lock_passwd: false") {
|
||||
t.Errorf("un compte avec mot de passe doit être déverrouillé :\n%s", out)
|
||||
}
|
||||
if strings.Contains(out, "ssh_authorized_keys") {
|
||||
t.Errorf("aucune clé fournie, le bloc ne doit pas apparaître :\n%s", out)
|
||||
}
|
||||
}
|
||||
|
||||
func TestVendorData_SSHKeyOnly_LocksPassword(t *testing.T) {
|
||||
out := vendorData(t, "", "ssh-ed25519 AAAA user@host")
|
||||
if !strings.Contains(out, "ssh-ed25519 AAAA user@host") {
|
||||
t.Errorf("clé absente :\n%s", out)
|
||||
}
|
||||
if strings.Contains(out, "\n passwd:") {
|
||||
t.Errorf("aucun mot de passe fourni, le champ ne doit pas apparaître :\n%s", out)
|
||||
}
|
||||
if !strings.Contains(out, "lock_passwd: true") {
|
||||
t.Errorf("sans mot de passe le compte doit rester verrouillé, sinon il devient un compte sudo sans mot de passe :\n%s", out)
|
||||
}
|
||||
}
|
||||
|
||||
func TestVendorData_BothCredentials(t *testing.T) {
|
||||
out := vendorData(t, "$6$hash", "ssh-ed25519 AAAA user@host")
|
||||
for _, expected := range []string{"#cloud-config", "name: syonad", `passwd: "$6$hash"`, "lock_passwd: false", "ssh-ed25519 AAAA user@host"} {
|
||||
if !strings.Contains(out, expected) {
|
||||
t.Errorf("%q absent :\n%s", expected, out)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestVendorData_CloudConfigHeaderIsFirstLine(t *testing.T) {
|
||||
out := vendorData(t, "$6$hash", "")
|
||||
if !strings.HasPrefix(out, "#cloud-config\n") {
|
||||
t.Errorf("cloud-init exige #cloud-config en première ligne :\n%q", out)
|
||||
}
|
||||
}
|
||||
|
||||
// --- documents fournis par l'appelant ---
|
||||
|
||||
func TestRenderDocument_VerbatimOverridesTemplate(t *testing.T) {
|
||||
cfg := newCfg()
|
||||
supplied := "#cloud-config\npackages:\n - nginx\n"
|
||||
cfg.Documents = map[string]string{DocUserData: supplied}
|
||||
|
||||
out, err := renderDocument(DocUserData, cfg)
|
||||
if err != nil {
|
||||
t.Fatalf("renderDocument : %v", err)
|
||||
}
|
||||
if out != supplied {
|
||||
t.Errorf("le document fourni doit être écrit verbatim :\nattendu %q\nobtenu %q", supplied, out)
|
||||
}
|
||||
}
|
||||
|
||||
func TestRenderDocument_EmptySuppliedDocumentIsHonoured(t *testing.T) {
|
||||
cfg := newCfg()
|
||||
cfg.Documents = map[string]string{DocVendorData: ""}
|
||||
|
||||
out, err := renderDocument(DocVendorData, cfg)
|
||||
if err != nil {
|
||||
t.Fatalf("renderDocument : %v", err)
|
||||
}
|
||||
if out != "" {
|
||||
t.Errorf("un document explicitement vide ne doit pas retomber sur le template :\n%s", out)
|
||||
}
|
||||
}
|
||||
|
||||
func TestWriteNoCloudFiles_WritesSuppliedDocument(t *testing.T) {
|
||||
dir := useTestDir(t)
|
||||
cfg := newCfg()
|
||||
cfg.Documents = map[string]string{DocUserData: "#cloud-config\nruncmd:\n - [ls]\n"}
|
||||
mustWrite(t, cfg, dir)
|
||||
|
||||
if got := readTestFile(t, dir, "vm1", "user-data"); got != cfg.Documents[DocUserData] {
|
||||
t.Errorf("user-data servi différent de celui fourni :\n%q", got)
|
||||
}
|
||||
}
|
||||
|
||||
// --- remontée des erreurs ---
|
||||
|
||||
func TestWriteNoCloudFiles_ReturnsErrorOnUnwritableDir(t *testing.T) {
|
||||
if os.Geteuid() == 0 {
|
||||
t.Skip("root ignore les permissions de fichiers")
|
||||
}
|
||||
dir := useTestDir(t)
|
||||
if err := os.Chmod(dir, 0500); err != nil {
|
||||
t.Fatalf("chmod : %v", err)
|
||||
}
|
||||
t.Cleanup(func() { os.Chmod(dir, 0700) })
|
||||
|
||||
if err := WriteNoCloudFiles(newCfg(), dir); err == nil {
|
||||
t.Error("une écriture impossible doit remonter une erreur : une VM ne doit jamais démarrer sans métadonnées en silence")
|
||||
}
|
||||
}
|
||||
|
||||
func TestRemoveNoCloudFiles_AbsentDirIsNotAnError(t *testing.T) {
|
||||
if err := RemoveNoCloudFiles("jamais-creee", useTestDir(t)); err != nil {
|
||||
t.Errorf("supprimer une VM sans fichiers ne doit pas échouer : %v", err)
|
||||
}
|
||||
}
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue