add a function to create a ResourceGroup unstructured (#1344)

pkg/live/rgpath.go

@@ -58,29 +58,34 @@ func generateInventoryObj(inv *kptfile.Inventory) (*unstructured.Unstructured, e
 		return nil, err
 	}
 	// Create and return ResourceGroup custom resource as inventory object.
+	var inventoryObj = ResourceGroupUnstructured(inv.Name, inv.Namespace, inv.InventoryID)
+	labels := inv.Labels
+	if labels == nil {
+		labels = make(map[string]string)
+	}
+	labels[common.InventoryLabel] = inv.InventoryID
+	inventoryObj.SetLabels(labels)
+	inventoryObj.SetAnnotations(inv.Annotations)
+	return inventoryObj, nil
+}
+
+func ResourceGroupUnstructured(name, namespace, id string) *unstructured.Unstructured {
 	groupVersion := fmt.Sprintf("%s/%s", ResourceGroupGVK.Group, ResourceGroupGVK.Version)
-	var inventoryObj = &unstructured.Unstructured{
+	inventoryObj := &unstructured.Unstructured{
 		Object: map[string]interface{}{
 			"apiVersion": groupVersion,
 			"kind":       ResourceGroupGVK.Kind,
 			"metadata": map[string]interface{}{
-				"name":      inv.Name,
-				"namespace": inv.Namespace,
+				"name":      name,
+				"namespace": namespace,
 				"labels": map[string]interface{}{
-					common.InventoryLabel: inv.InventoryID,
+					common.InventoryLabel: id,
 				},
 			},
 			"spec": map[string]interface{}{
 				"resources": []interface{}{},
 			},
 		},
 	}
-	labels := inv.Labels
-	if labels == nil {
-		labels = make(map[string]string)
-	}
-	labels[common.InventoryLabel] = inv.InventoryID
-	inventoryObj.SetLabels(labels)
-	inventoryObj.SetAnnotations(inv.Annotations)
-	return inventoryObj, nil
+	return inventoryObj
 }

pkg/live/rgpath_test.go

@@ -229,3 +229,22 @@ func getInventoryAnnotations(inv *unstructured.Unstructured) map[string]string {
 	}
 	return accessor.GetAnnotations()
 }
+
+func TestResourceGroupUnstructured(t *testing.T) {
+	name := "name"
+	namespace := "test"
+	id := "random-id"
+	rg := ResourceGroupUnstructured(name, namespace, id)
+	if rg == nil {
+		t.Fatal("resourcegroup shouldn't be nil")
+	}
+	if rg.GetName() != name {
+		t.Fatalf("resourcegroup name expected %s, but got %s", name, rg.GetName())
+	}
+	if rg.GetNamespace() != namespace {
+		t.Fatalf("resourcegroup namespace expected %s, but got %s", namespace, rg.GetNamespace())
+	}
+	if rg.GetLabels()[common.InventoryLabel] != id {
+		t.Fatalf("resourcegroup inventory id expected %s, but got %s", id, rg.GetLabels()[common.InventoryLabel])
+	}
+}