From e3642de8275d52b1b84a91a1aef9f97bbf071f8e Mon Sep 17 00:00:00 2001
From: Adrian Serrano <adrisr83@gmail.com>
Date: Thu, 16 Apr 2020 13:18:16 +0200
Subject: [PATCH] Do not create attributes.title if it didn't exist previously

---
 libbeat/dashboards/modify_json.go      |  5 +++-
 libbeat/dashboards/modify_json_test.go | 37 +++++++++++++++++++++++---
 2 files changed, 38 insertions(+), 4 deletions(-)

diff --git a/libbeat/dashboards/modify_json.go b/libbeat/dashboards/modify_json.go
index a4a51027a74..2e0c48e38b0 100644
--- a/libbeat/dashboards/modify_json.go
+++ b/libbeat/dashboards/modify_json.go
@@ -57,7 +57,10 @@ func ReplaceIndexInIndexPattern(index string, content common.MapStr) (err error)
 		// This uses Put instead of DeepUpdate to avoid modifying types for
 		// inner objects. (DeepUpdate will replace maps with MapStr).
 		obj.Put("id", index)
-		obj.Put("attributes.title", index)
+		// Only overwrite title if it exists.
+		if _, err := obj.GetValue("attributes.title"); err == nil {
+			obj.Put("attributes.title", index)
+		}
 	}
 
 	switch v := list.(type) {
diff --git a/libbeat/dashboards/modify_json_test.go b/libbeat/dashboards/modify_json_test.go
index e562e560d06..a7424414b53 100644
--- a/libbeat/dashboards/modify_json_test.go
+++ b/libbeat/dashboards/modify_json_test.go
@@ -117,11 +117,13 @@ func TestReplaceIndexInIndexPattern(t *testing.T) {
 	// what the inner types are (MapStr, map[string]interface{} or interface{}).
 	// Also ensures that the inner types are not modified after replacement.
 	tests := []struct {
+		title    string
 		input    common.MapStr
 		index    string
 		expected common.MapStr
 	}{
 		{
+			title: "Replace in []interface(map).map",
 			input: common.MapStr{"objects": []interface{}{map[string]interface{}{
 				"id":   "phonybeat-*",
 				"type": "index-pattern",
@@ -139,6 +141,7 @@ func TestReplaceIndexInIndexPattern(t *testing.T) {
 				}}}},
 		},
 		{
+			title: "Replace in []interface(map).mapstr",
 			input: common.MapStr{"objects": []interface{}{map[string]interface{}{
 				"id":   "phonybeat-*",
 				"type": "index-pattern",
@@ -156,6 +159,7 @@ func TestReplaceIndexInIndexPattern(t *testing.T) {
 				}}}},
 		},
 		{
+			title: "Replace in []map.mapstr",
 			input: common.MapStr{"objects": []map[string]interface{}{{
 				"id":   "phonybeat-*",
 				"type": "index-pattern",
@@ -173,6 +177,7 @@ func TestReplaceIndexInIndexPattern(t *testing.T) {
 				}}}},
 		},
 		{
+			title: "Replace in []mapstr.mapstr",
 			input: common.MapStr{"objects": []common.MapStr{{
 				"id":   "phonybeat-*",
 				"type": "index-pattern",
@@ -190,6 +195,7 @@ func TestReplaceIndexInIndexPattern(t *testing.T) {
 				}}}},
 		},
 		{
+			title: "Replace in []mapstr.interface(mapstr)",
 			input: common.MapStr{"objects": []common.MapStr{{
 				"id":   "phonybeat-*",
 				"type": "index-pattern",
@@ -206,11 +212,36 @@ func TestReplaceIndexInIndexPattern(t *testing.T) {
 					"timeFieldName": "@timestamp",
 				})}}},
 		},
+		{
+			title: "Do not create missing attributes",
+			input: common.MapStr{"objects": []common.MapStr{{
+				"id":   "phonybeat-*",
+				"type": "index-pattern",
+			}}},
+			index: "otherindex-*",
+			expected: common.MapStr{"objects": []common.MapStr{{
+				"id":   "otherindex-*",
+				"type": "index-pattern",
+			}}},
+		},
+		{
+			title: "Create missing id",
+			input: common.MapStr{"objects": []common.MapStr{{
+				"type": "index-pattern",
+			}}},
+			index: "otherindex-*",
+			expected: common.MapStr{"objects": []common.MapStr{{
+				"id":   "otherindex-*",
+				"type": "index-pattern",
+			}}},
+		},
 	}
 
 	for _, test := range tests {
-		err := ReplaceIndexInIndexPattern(test.index, test.input)
-		assert.NoError(t, err)
-		assert.Equal(t, test.expected, test.input)
+		t.Run(test.title, func(t *testing.T) {
+			err := ReplaceIndexInIndexPattern(test.index, test.input)
+			assert.NoError(t, err)
+			assert.Equal(t, test.expected, test.input)
+		})
 	}
 }