add test for cloudinit

Author: sp-yduck

cloud/cloudinit/common_test.go

@@ -0,0 +1,17 @@
+package cloudinit_test
+
+import (
+	"testing"
+
+	"github.com/sp-yduck/cluster-api-provider-proxmox/cloud/cloudinit"
+)
+
+func TestIsValidType(t *testing.T) {
+	testTypes := []string{"user", "users", "meta", "memetah", "network", "networkig"}
+	expectedValidity := []bool{true, false, true, false, true, false}
+	for i, tt := range testTypes {
+		if cloudinit.IsValidType(tt) != expectedValidity[i] {
+			t.Errorf("expected validity for %s is %v", tt, expectedValidity[i])
+		}
+	}
+}

cloud/cloudinit/user_test.go

@@ -1,8 +1,10 @@
 package cloudinit_test
 
 import (
+	"reflect"
 	"testing"
 
+	infrav1 "github.com/sp-yduck/cluster-api-provider-proxmox/api/v1beta1"
 	"github.com/sp-yduck/cluster-api-provider-proxmox/cloud/cloudinit"
 )
 
@@ -51,3 +53,25 @@ runcmd:
 		t.Fatalf("generate : %v", err)
 	}
 }
+
+func TestMergeUsers(t *testing.T) {
+	a := infrav1.User{
+		User:   "override-user",
+		RunCmd: []string{"command A", "command B"},
+	}
+	b := infrav1.User{
+		User:   "test-user",
+		RunCmd: []string{"command C"},
+	}
+	expected := infrav1.User{
+		User:   "override-user",
+		RunCmd: []string{"command A", "command B", "command C"},
+	}
+	c, err := cloudinit.MergeUsers(a, b)
+	if err != nil {
+		t.Errorf("failed to merge cloud init user data: %v", err)
+	}
+	if !reflect.DeepEqual(*c, expected) {
+		t.Errorf("%v is expected to same as %v", *c, expected)
+	}
+}