kubernetes/unstructured: add more unit tests

iawia002 · Sep 8, 2023 · 44ea4b8 · 44ea4b8
1 parent 6316719
commit 44ea4b8
Showing 1 changed file with 93 additions and 0 deletions.
diff --git a/kubernetes/unstructured/unstructured_test.go b/kubernetes/unstructured/unstructured_test.go
@@ -2,8 +2,101 @@ package unstructured
 
 import (
 	"testing"
+
+	corev1 "k8s.io/api/core/v1"
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 )
 
+func TestConvertToUnstructured(t *testing.T) {
+	tests := []struct {
+		name  string
+		obj   interface{}
+		isErr bool
+	}{
+		{
+			name: "obj test",
+			obj: &corev1.Namespace{
+				ObjectMeta: metav1.ObjectMeta{
+					Name: "test",
+				},
+			},
+			isErr: false,
+		},
+		{
+			name: "list test",
+			obj: &corev1.NamespaceList{
+				Items: []corev1.Namespace{
+					{
+						ObjectMeta: metav1.ObjectMeta{
+							Name: "test",
+						},
+					},
+				},
+			},
+			isErr: false,
+		},
+		{
+			name: "error test",
+			obj: []byte(`
+apiVersion: v1
+kind: Namespace
+metadata:
+  name: test
+`),
+			isErr: true,
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			if _, err := ConvertToUnstructured(tt.obj); tt.isErr != (err != nil) {
+				t.Errorf("%s ConvertToUnstructured() unexpected error: %v", tt.name, err)
+			}
+		})
+	}
+}
+
+func TestConvertToTyped(t *testing.T) {
+	tests := []struct {
+		name     string
+		obj      interface{}
+		typedObj interface{}
+		isErr    bool
+	}{
+		{
+			name: "obj test",
+			obj: &corev1.Namespace{
+				ObjectMeta: metav1.ObjectMeta{
+					Name: "test",
+				},
+			},
+			typedObj: &corev1.Namespace{},
+			isErr:    false,
+		},
+		{
+			name: "list test",
+			obj: &corev1.NamespaceList{
+				Items: []corev1.Namespace{
+					{
+						ObjectMeta: metav1.ObjectMeta{
+							Name: "test",
+						},
+					},
+				},
+			},
+			typedObj: &corev1.NamespaceList{},
+			isErr:    false,
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			unstructuredObj, _ := ConvertToUnstructured(tt.obj)
+			if err := ConvertToTyped(unstructuredObj, tt.typedObj); tt.isErr != (err != nil) {
+				t.Errorf("%s ConvertToTyped() unexpected error: %v", tt.name, err)
+			}
+		})
+	}
+}
+
 func TestYAMLToUnstructured(t *testing.T) {
 	tests := []struct {
 		name  string