114 lines
3.6 KiB
Go
114 lines
3.6 KiB
Go
// Licensed to Elasticsearch B.V. under one or more contributor
|
|
// license agreements. See the NOTICE file distributed with
|
|
// this work for additional information regarding copyright
|
|
// ownership. Elasticsearch B.V. licenses this file to you under
|
|
// the Apache License, Version 2.0 (the "License"); you may
|
|
// not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing,
|
|
// software distributed under the License is distributed on an
|
|
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
|
// KIND, either express or implied. See the License for the
|
|
// specific language governing permissions and limitations
|
|
// under the License.
|
|
|
|
package dashboards
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/elastic/beats/libbeat/common"
|
|
)
|
|
|
|
func TestReplaceStringInDashboard(t *testing.T) {
|
|
tests := []struct {
|
|
content common.MapStr
|
|
old string
|
|
new string
|
|
expected common.MapStr
|
|
}{
|
|
{
|
|
content: common.MapStr{"test": "CHANGEME"},
|
|
old: "CHANGEME",
|
|
new: "hostname",
|
|
expected: common.MapStr{"test": "hostname"},
|
|
},
|
|
{
|
|
content: common.MapStr{"test": "hello"},
|
|
old: "CHANGEME",
|
|
new: "hostname",
|
|
expected: common.MapStr{"test": "hello"},
|
|
},
|
|
{
|
|
content: common.MapStr{"test": map[string]interface{}{"key": "\"CHANGEME\""}},
|
|
old: "CHANGEME",
|
|
new: "hostname.local",
|
|
expected: common.MapStr{"test": map[string]interface{}{"key": "\"hostname.local\""}},
|
|
},
|
|
{
|
|
content: common.MapStr{
|
|
"kibanaSavedObjectMeta": map[string]interface{}{
|
|
"searchSourceJSON": "{\"filter\":[],\"highlightAll\":true,\"version\":true,\"query\":{\"query\":\"beat.name:\\\"CHANGEME_HOSTNAME\\\"\",\"language\":\"lucene\"}}"}},
|
|
|
|
old: "CHANGEME_HOSTNAME",
|
|
new: "hostname.local",
|
|
expected: common.MapStr{
|
|
"kibanaSavedObjectMeta": map[string]interface{}{
|
|
"searchSourceJSON": "{\"filter\":[],\"highlightAll\":true,\"version\":true,\"query\":{\"query\":\"beat.name:\\\"hostname.local\\\"\",\"language\":\"lucene\"}}"}},
|
|
},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
result, err := ReplaceStringInDashboard(test.old, test.new, test.content)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, test.expected, result)
|
|
}
|
|
}
|
|
|
|
func TestReplaceIndexInDashboardObject(t *testing.T) {
|
|
tests := []struct {
|
|
dashboard common.MapStr
|
|
pattern string
|
|
expected common.MapStr
|
|
}{
|
|
{
|
|
common.MapStr{"objects": []interface{}{map[string]interface{}{
|
|
"attributes": map[string]interface{}{
|
|
"kibanaSavedObjectMeta": map[string]interface{}{
|
|
"searchSourceJSON": "{\"index\":\"metricbeat-*\"}",
|
|
},
|
|
}}}},
|
|
"otherindex-*",
|
|
common.MapStr{"objects": []interface{}{map[string]interface{}{
|
|
"attributes": map[string]interface{}{
|
|
"kibanaSavedObjectMeta": map[string]interface{}{
|
|
"searchSourceJSON": "{\"index\":\"otherindex-*\"}",
|
|
},
|
|
}}}},
|
|
},
|
|
{
|
|
common.MapStr{"objects": []interface{}{map[string]interface{}{
|
|
"attributes": map[string]interface{}{
|
|
"kibanaSavedObjectMeta": map[string]interface{}{},
|
|
"visState": "{\"params\":{\"index_pattern\":\"metricbeat-*\"}}",
|
|
}}}},
|
|
"otherindex-*",
|
|
common.MapStr{"objects": []interface{}{map[string]interface{}{
|
|
"attributes": map[string]interface{}{
|
|
"kibanaSavedObjectMeta": map[string]interface{}{},
|
|
"visState": "{\"params\":{\"index_pattern\":\"otherindex-*\"}}",
|
|
}}}},
|
|
},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
result := ReplaceIndexInDashboardObject(test.pattern, test.dashboard)
|
|
assert.Equal(t, test.expected, result)
|
|
}
|
|
}
|