ability to add props to the root mem map (#18)

mtintes · Sep 22, 2024 · 4fd553a · 4fd553a
1 parent a82438b
commit 4fd553a
Show file tree

Hide file tree

Showing 2 changed files with 38 additions and 0 deletions.
diff --git a/cli/actions/memoryMap.go b/cli/actions/memoryMap.go
@@ -106,6 +106,7 @@ func applyMappings(flatFile map[string]interface{}, mappings []Mapping, masterMe
 			if strings.HasPrefix(key, mapping.InPath+".") || key == mapping.InPath {
 				keySuffix := key[len(mapping.InPath):]
 				newKey := mapping.ToPath + keySuffix
+				newKey = strings.TrimPrefix(newKey, ".")
 				for memKey := range masterMemoryMap {
 					if strings.HasPrefix(memKey, newKey+".") || strings.HasPrefix(newKey, memKey+".") {
 						traces = append(traces, Trace{key: memKey, oldValue: masterMemoryMap[memKey], changeType: "delete", file: filePath})

diff --git a/cli/actions/memoryMap_test.go b/cli/actions/memoryMap_test.go
@@ -449,3 +449,40 @@ func TestReadMemoryMap_Traces(t *testing.T) {
 	assert.Equal(t, 1, len(*traces))
 
 }
+
+func TestReadMemoryMap_PropToRoot(t *testing.T) {
+
+	AppFs = afero.NewMemMapFs()
+
+	jsonData := `{
+		"key1": {
+			"key2": {
+				"key3": "value"}}
+	}`
+
+	_ = afero.WriteFile(AppFs, "test.json", []byte(jsonData), 0644)
+
+	config := ConfigurationMap{
+		Configs: []Config{
+			{
+				Path: "test.json",
+				Mappings: []Mapping{
+					{
+						InPath: "key1.key2",
+						ToPath: "",
+					},
+				},
+				ApplyFile: "after",
+			},
+		},
+	}
+
+	expectedMemoryMap := map[string]interface{}{
+		"key3": "value",
+	}
+
+	memoryMap, _, err := ReadMemoryMap(&config)
+
+	assert.Nil(t, err)
+	assert.Equal(t, fmt.Sprint(expectedMemoryMap), fmt.Sprint(memoryMap))
+}