refactor: func NewWorkspace returns proper zero-values for empty func…

…tion parameters
cloudfoundry · Oct 8, 2024 · 12de896 · 12de896
1 parent f95bde1
commit 12de896
Show file tree

Hide file tree

Showing 3 changed files with 98 additions and 14 deletions.
diff --git a/pkg/providers/tf/deployment_manager_test.go b/pkg/providers/tf/deployment_manager_test.go
@@ -424,9 +424,8 @@ var _ = Describe("DeploymentManager", func() {
 				By("checking that the modules and instances are updated, but the state remains the same")
 				expectedWorkspace := &workspace.TerraformWorkspace{
 					Modules: []workspace.ModuleDefinition{{
-						Name:        "brokertemplate",
-						Definition:  template,
-						Definitions: map[string]string{},
+						Name:       "brokertemplate",
+						Definition: template,
 					}},
 					Instances: []workspace.ModuleInstance{{
 						ModuleName:   "brokertemplate",
@@ -506,9 +505,8 @@ var _ = Describe("DeploymentManager", func() {
 				By("checking that the modules and instances are updated, but the state remains the same")
 				expectedWorkspace := &workspace.TerraformWorkspace{
 					Modules: []workspace.ModuleDefinition{{
-						Name:        "brokertemplate",
-						Definition:  template,
-						Definitions: map[string]string{},
+						Name:       "brokertemplate",
+						Definition: template,
 					}},
 					Instances: []workspace.ModuleInstance{{
 						ModuleName:   "brokertemplate",

diff --git a/pkg/providers/tf/workspace/workspace.go b/pkg/providers/tf/workspace/workspace.go
@@ -56,8 +56,11 @@ func NewWorkspace(templateVars map[string]any,
 	parametersToRemove []string,
 	parametersToAdd []ParameterMapping) (*TerraformWorkspace, error) {
 
-	terraformTemplatesCopy := make(map[string]string)
-	maps.Copy(terraformTemplatesCopy, terraformTemplates)
+	var terraformTemplatesCopy map[string]string
+	if len(terraformTemplates) > 0 {
+		terraformTemplatesCopy = make(map[string]string)
+		maps.Copy(terraformTemplatesCopy, terraformTemplates)
+	}
 
 	tfModule := ModuleDefinition{
 		Name:        "brokertemplate",
@@ -75,12 +78,23 @@ func NewWorkspace(templateVars map[string]any,
 		limitedConfig[name] = templateVars[name]
 	}
 
-	parameterMappingsCopy := make([]ParameterMapping, len(importParameterMappings))
-	copy(parameterMappingsCopy, importParameterMappings)
-	parametersToRemoveCopy := make([]string, len(parametersToRemove))
-	copy(parametersToRemoveCopy, parametersToRemove)
-	parametersToAddCopy := make([]ParameterMapping, len(parametersToAdd))
-	copy(parametersToAddCopy, parametersToAdd)
+	var parameterMappingsCopy []ParameterMapping
+	if len(importParameterMappings) > 0 {
+		parameterMappingsCopy = make([]ParameterMapping, len(importParameterMappings))
+		copy(parameterMappingsCopy, importParameterMappings)
+	}
+
+	var parametersToRemoveCopy []string
+	if len(parametersToRemove) > 0 {
+		parametersToRemoveCopy = make([]string, len(parametersToRemove))
+		copy(parametersToRemoveCopy, parametersToRemove)
+	}
+
+	var parametersToAddCopy []ParameterMapping
+	if len(parametersToAdd) > 0 {
+		parametersToAddCopy = make([]ParameterMapping, len(parametersToAdd))
+		copy(parametersToAddCopy, parametersToAdd)
+	}
 
 	workspace := TerraformWorkspace{
 		Modules: []ModuleDefinition{tfModule},

diff --git a/pkg/providers/tf/workspace/workspace_test.go b/pkg/providers/tf/workspace/workspace_test.go
@@ -370,4 +370,76 @@ func TestNewWorkspace(t *testing.T) {
 			t.Error("Expected NewWorkspace to create deep copy of ParametersToAdd")
 		}
 	})
+	t.Run("returns empty map for empty templateVars", func(t *testing.T) {
+		ws, err := NewWorkspace(nil, "", nil, nil, nil, nil)
+		if err != nil {
+			t.Fatalf("unexpected error: %s", err)
+		}
+		got := ws.Instances[0].Configuration
+		expect := make(map[string]any)
+		if !reflect.DeepEqual(expect, got) {
+			t.Errorf("expected %v, got %v", expect, got)
+		}
+	})
+
+	t.Run("returns zero-value for empty terraformTemplate", func(t *testing.T) {
+		ws, err := NewWorkspace(nil, "", nil, nil, nil, nil)
+		if err != nil {
+			t.Fatalf("unexpected error: %s", err)
+		}
+		got := ws.Modules[0].Definition
+		var expect string
+		if !reflect.DeepEqual(expect, got) {
+			t.Errorf("expected %v, got %v", expect, got)
+		}
+	})
+
+	t.Run("returns zero-value for empty terraformTemplates", func(t *testing.T) {
+		ws, err := NewWorkspace(nil, "", nil, nil, nil, nil)
+		if err != nil {
+			t.Fatalf("unexpected error: %s", err)
+		}
+		got := ws.Modules[0].Definitions
+		var expect map[string]string
+		if !reflect.DeepEqual(expect, got) {
+			t.Errorf("expected %v, got %v", expect, got)
+		}
+	})
+
+	t.Run("returns zero-value for empty importParameterMappings", func(t *testing.T) {
+		ws, err := NewWorkspace(nil, "", nil, nil, nil, nil)
+		if err != nil {
+			t.Fatalf("unexpected error: %s", err)
+		}
+		got := ws.Transformer.ParameterMappings
+		var expect []ParameterMapping
+		if !reflect.DeepEqual(expect, got) {
+			t.Errorf("expected %v, got %v", expect, got)
+		}
+	})
+
+	t.Run("returns zero-value for empty importParametersToRemove", func(t *testing.T) {
+		ws, err := NewWorkspace(nil, "", nil, nil, nil, nil)
+		if err != nil {
+			t.Fatalf("unexpected error: %s", err)
+		}
+		got := ws.Transformer.ParametersToRemove
+		var expect []string
+		if !reflect.DeepEqual(expect, got) {
+			t.Errorf("expected %v, got %v", expect, got)
+		}
+	})
+
+	t.Run("returns zero-value for empty importParametersToAdd", func(t *testing.T) {
+		ws, err := NewWorkspace(nil, "", nil, nil, nil, nil)
+		if err != nil {
+			t.Fatalf("unexpected error: %s", err)
+		}
+		got := ws.Transformer.ParametersToAdd
+		var expect []ParameterMapping
+		if !reflect.DeepEqual(expect, got) {
+			t.Errorf("expected %v, got %v", expect, got)
+		}
+	})
+
 }