add tests for cloud/cloudinit

Author: sp-yduck

cloud/cloudinit/common.go

@@ -1,10 +1,10 @@
-package cloudinit
-
-func IsValidType(cloudInitType string) bool {
-	switch cloudInitType {
-	case "user", "meta", "network":
-		return true
-	default:
-		return false
-	}
-}
+package cloudinit
+
+func IsValidType(cloudInitType string) bool {
+	switch cloudInitType {
+	case "user", "meta", "network":
+		return true
+	default:
+		return false
+	}
+}

cloud/cloudinit/user.go

@@ -1,33 +1,33 @@
-package cloudinit
-
-import (
-	"fmt"
-
-	"github.com/imdario/mergo"
-	"gopkg.in/yaml.v3"
-
-	infrav1 "github.com/sp-yduck/cluster-api-provider-proxmox/api/v1beta1"
-)
-
-func ParseUser(content string) (*infrav1.User, error) {
-	var config *infrav1.User
-	if err := yaml.Unmarshal([]byte(content), &config); err != nil {
-		return nil, err
-	}
-	return config, nil
-}
-
-func GenerateUserYaml(config infrav1.User) (string, error) {
-	b, err := yaml.Marshal(&config)
-	if err != nil {
-		return "", err
-	}
-	return fmt.Sprintf("#cloud-config\n%s", string(b)), nil
-}
-
-func MergeUsers(a, b infrav1.User) (*infrav1.User, error) {
-	if err := mergo.Merge(&a, b, mergo.WithAppendSlice); err != nil {
-		return nil, err
-	}
-	return &a, nil
-}
+package cloudinit
+
+import (
+	"fmt"
+
+	"github.com/imdario/mergo"
+	"gopkg.in/yaml.v3"
+
+	infrav1 "github.com/sp-yduck/cluster-api-provider-proxmox/api/v1beta1"
+)
+
+func ParseUser(content string) (*infrav1.User, error) {
+	var config *infrav1.User
+	if err := yaml.Unmarshal([]byte(content), &config); err != nil {
+		return nil, err
+	}
+	return config, nil
+}
+
+func GenerateUserYaml(config infrav1.User) (string, error) {
+	b, err := yaml.Marshal(&config)
+	if err != nil {
+		return "", err
+	}
+	return fmt.Sprintf("#cloud-config\n%s", string(b)), nil
+}
+
+func MergeUsers(a, b infrav1.User) (*infrav1.User, error) {
+	if err := mergo.Merge(&a, b, mergo.WithAppendSlice); err != nil {
+		return nil, err
+	}
+	return &a, nil
+}

cloud/cloudinit/user_test.go

@@ -0,0 +1,53 @@
+package cloudinit_test
+
+import (
+	"testing"
+
+	"github.com/sp-yduck/cluster-api-provider-proxmox/cloud/cloudinit"
+)
+
+func TestParseUser(t *testing.T) {
+	testYaml := `
+write_files:
+  - path: /run/kubeadm/kubeadm.yaml
+    owner: root:root
+    permissions: '0640'
+    content: |
+      asdfasdfasdf
+runcmd:
+  - 'kubeadm init --config /run/kubeadm/kubeadm.yaml  && echo success > /run/cluster-api/bootstrap-success.complete'
+  - "curl -L https://dl.k8s.io/release/v1.27.3/bin/linux/amd64/kubectl -o /usr/local/bin/kubectl"
+  - "chmod +x /usr/local/bin/kubectl"
+  - "reboot now"
+  `
+	_, err := cloudinit.ParseUser(testYaml)
+	if err != nil {
+		t.Fatalf("failed to parse user: %v", err)
+	}
+}
+
+func TestGenerateUserYaml(t *testing.T) {
+	testYaml := `
+write_files:
+  - path: /run/kubeadm/kubeadm.yaml
+    owner: root:root
+    permissions: '0640'
+    content: |
+      asdfasdfasdf
+runcmd:
+  - 'kubeadm init --config /run/kubeadm/kubeadm.yaml  && echo success > /run/cluster-api/bootstrap-success.complete'
+  - "curl -L https://dl.k8s.io/release/v1.27.3/bin/linux/amd64/kubectl -o /usr/local/bin/kubectl"
+  - "chmod +x /usr/local/bin/kubectl"
+  - "reboot now"
+  `
+
+	uc, err := cloudinit.ParseUser(testYaml)
+	if err != nil {
+		t.Fatalf("failed to parse user: %v", err)
+	}
+
+	_, err = cloudinit.GenerateUserYaml(*uc)
+	if err != nil {
+		t.Fatalf("generate : %v", err)
+	}
+}