130 lines
3 KiB
Go
130 lines
3 KiB
Go
|
package pkg
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Test_FlattenMap ensure that FlattenMap works as expected by flattening maps.
|
||
|
|
func Test_FlattenMap(t *testing.T) {
|
||
|
|
var tests = []struct {
|
||
|
|
firstMap map[string]interface{}
|
||
|
|
expectedMap map[string]interface{}
|
||
|
|
}{
|
||
|
|
{
|
||
|
|
map[string]interface{}{},
|
||
|
|
map[string]interface{}{},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
map[string]interface{}{
|
||
|
|
"root": "root_is_always_root",
|
||
|
|
},
|
||
|
|
map[string]interface{}{
|
||
|
|
"root": "root_is_always_root",
|
||
|
|
},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
map[string]interface{}{
|
||
|
|
"root": map[string]interface{}{
|
||
|
|
"testing": "1",
|
||
|
|
"another_key": []int{1, 2, 3},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
map[string]interface{}{
|
||
|
|
"root.testing": "1",
|
||
|
|
"root.another_key": []int{1, 2, 3},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
map[string]interface{}{
|
||
|
|
"root": map[string]interface{}{
|
||
|
|
"list_with_map": []map[string]string{
|
||
|
|
{
|
||
|
|
"map_in_list": "1",
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
map[string]interface{}{
|
||
|
|
"root.list_with_map": []map[string]string{
|
||
|
|
{
|
||
|
|
"map_in_list": "1",
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
map[string]interface{}{
|
||
|
|
"root": map[string]interface{}{
|
||
|
|
"testing": "1",
|
||
|
|
"another_key": map[string]interface{}{
|
||
|
|
"nested_key": "nested_value",
|
||
|
|
"nested_key2": "nested_value",
|
||
|
|
"nested_key3": "nested_value",
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
map[string]interface{}{
|
||
|
|
"root.testing": "1",
|
||
|
|
"root.another_key.nested_key": "nested_value",
|
||
|
|
"root.another_key.nested_key2": "nested_value",
|
||
|
|
"root.another_key.nested_key3": "nested_value",
|
||
|
|
},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
map[string]interface{}{
|
||
|
|
"root": map[string]interface{}{
|
||
|
|
"testing": "1",
|
||
|
|
"another_key": []interface{}{1, 2, 3},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
map[string]interface{}{
|
||
|
|
"root.testing": "1",
|
||
|
|
"root.another_key[0]": 1,
|
||
|
|
"root.another_key[1]": 2,
|
||
|
|
"root.another_key[2]": 3,
|
||
|
|
},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
map[string]interface{}{
|
||
|
|
"root": map[string]interface{}{
|
||
|
|
"testing": "1",
|
||
|
|
"another_key": []interface{}{[]interface{}{1, 2}, []interface{}{3}},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
map[string]interface{}{
|
||
|
|
"root.testing": "1",
|
||
|
|
"root.another_key[0][0]": 1,
|
||
|
|
"root.another_key[0][1]": 2,
|
||
|
|
"root.another_key[1][0]": 3,
|
||
|
|
},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
map[string]interface{}{
|
||
|
|
"root": map[string]interface{}{
|
||
|
|
"testing": "1",
|
||
|
|
"another_key": []interface{}{[]interface{}{1, 2}, []interface{}{map[string]interface{}{
|
||
|
|
"inside_list_key": 1,
|
||
|
|
"inside_list_key2": []interface{}{"yes", "no"},
|
||
|
|
}}},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
map[string]interface{}{
|
||
|
|
"root.testing": "1",
|
||
|
|
"root.another_key[0][0]": 1,
|
||
|
|
"root.another_key[0][1]": 2,
|
||
|
|
"root.another_key[1][0].inside_list_key": 1,
|
||
|
|
"root.another_key[1][0].inside_list_key2[0]": "yes",
|
||
|
|
"root.another_key[1][0].inside_list_key2[1]": "no",
|
||
|
|
},
|
||
|
|
},
|
||
|
|
}
|
||
|
|
for index, tt := range tests {
|
||
|
|
t.Run(fmt.Sprintf("test_%d", index), func(t *testing.T) {
|
||
|
|
flatMap := FlattenMap(tt.firstMap)
|
||
|
|
assert.Equal(t, tt.expectedMap, *flatMap)
|
||
|
|
})
|
||
|
|
}
|
||
|
|
}
|