stash/pkg/utils/map_test.go

package utils

import (
	"encoding/json"
	"reflect"
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestNestedMapGet(t *testing.T) {
	m := NestedMap{
		"foo": map[string]interface{}{
			"bar": map[string]interface{}{
				"baz": "qux",
			},
		},
	}

	tests := []struct {
		name  string
		key   string
		want  interface{}
		found bool
	}{
		{
			name:  "Get a value from a nested map",
			key:   "foo.bar.baz",
			want:  "qux",
			found: true,
		},
		{
			name:  "Get a value from a nested map with a missing key",
			key:   "foo.bar.quux",
			want:  nil,
			found: false,
		},
		{
			name:  "Get a value from a nested map with a missing key",
			key:   "foo.quux.baz",
			want:  nil,
			found: false,
		},
		{
			name:  "Get a value from a nested map with a missing key",
			key:   "quux.bar.baz",
			want:  nil,
			found: false,
		},
		{
			name:  "Get a value from a nested map with a missing key",
			key:   "foo.bar",
			want:  map[string]interface{}{"baz": "qux"},
			found: true,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			got, found := m.Get(tt.key)
			if !reflect.DeepEqual(got, tt.want) {
				t.Errorf("NestedMap.Get() got = %v, want %v", got, tt.want)
			}
			if found != tt.found {
				t.Errorf("NestedMap.Get() found = %v, want %v", found, tt.found)
			}
		})
	}
}

func TestNestedMapSet(t *testing.T) {
	tests := []struct {
		name     string
		key      string
		existing NestedMap
		want     NestedMap
	}{
		{
			name:     "Set a value in a nested map",
			key:      "foo.bar.baz",
			existing: NestedMap{},
			want: NestedMap{
				"foo": map[string]interface{}{
					"bar": map[string]interface{}{
						"baz": "qux",
					},
				},
			},
		},
		{
			name: "Overwrite existing value",
			key:  "foo.bar",
			existing: NestedMap{
				"foo": map[string]interface{}{
					"bar": "old",
				},
			},
			want: NestedMap{
				"foo": map[string]interface{}{
					"bar": "qux",
				},
			},
		},
		{
			name: "Set a value overwriting a primitive with a nested map",
			key:  "foo.bar",
			existing: NestedMap{
				"foo": "bar",
			},
			want: NestedMap{
				"foo": map[string]interface{}{
					"bar": "qux",
				},
			},
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			tt.existing.Set(tt.key, "qux")
			if !reflect.DeepEqual(tt.existing, tt.want) {
				t.Errorf("NestedMap.Set() got = %v, want %v", tt.existing, tt.want)
			}
		})
	}
}

func TestNestedMapDelete(t *testing.T) {
	tests := []struct {
		name     string
		key      string
		existing NestedMap
		want     NestedMap
	}{
		{
			name: "Delete non existing value",
			key:  "foo.bar.baa",
			existing: NestedMap{
				"foo": map[string]interface{}{
					"bar": map[string]interface{}{
						"baz": "qux",
					},
				},
			},
			want: NestedMap{
				"foo": map[string]interface{}{
					"bar": map[string]interface{}{
						"baz": "qux",
					},
				},
			},
		},
		{
			name: "Delete existing value",
			key:  "foo.bar",
			existing: NestedMap{
				"foo": map[string]interface{}{
					"bar": "old",
				},
			},
			want: NestedMap{
				"foo": map[string]interface{}{},
			},
		},
		{
			name: "Delete existing map",
			key:  "foo.bar",
			existing: NestedMap{
				"foo": map[string]interface{}{
					"bar": map[string]interface{}{
						"baz": "qux",
					},
				},
			},
			want: NestedMap{
				"foo": map[string]interface{}{},
			},
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			tt.existing.Delete(tt.key)
			if !reflect.DeepEqual(tt.existing, tt.want) {
				t.Errorf("NestedMap.Set() got = %v, want %v", tt.existing, tt.want)
			}
		})
	}
}

func TestMergeMaps(t *testing.T) {
	tests := []struct {
		name   string
		dest   map[string]interface{}
		src    map[string]interface{}
		result map[string]interface{}
	}{
		{
			name: "Merge two maps",
			dest: map[string]interface{}{
				"foo": "bar",
			},
			src: map[string]interface{}{
				"baz": "qux",
			},
			result: map[string]interface{}{
				"foo": "bar",
				"baz": "qux",
			},
		},
		{
			name: "Merge two maps with overlapping keys",
			dest: map[string]interface{}{
				"foo": "bar",
				"baz": "qux",
			},
			src: map[string]interface{}{
				"baz": "quux",
			},
			result: map[string]interface{}{
				"foo": "bar",
				"baz": "quux",
			},
		},
		{
			name: "Merge two maps with overlapping keys and nested maps",
			dest: map[string]interface{}{
				"foo": map[string]interface{}{
					"bar": "baz",
				},
			},
			src: map[string]interface{}{
				"foo": map[string]interface{}{
					"qux": "quux",
				},
			},
			result: map[string]interface{}{
				"foo": map[string]interface{}{
					"bar": "baz",
					"qux": "quux",
				},
			},
		},
		{
			name: "Merge two maps with overlapping keys and nested maps",
			dest: map[string]interface{}{
				"foo": map[string]interface{}{
					"bar": "baz",
				},
			},
			src: map[string]interface{}{
				"foo": "qux",
			},
			result: map[string]interface{}{
				"foo": "qux",
			},
		},
		{
			name: "Merge two maps with overlapping keys and nested maps",
			dest: map[string]interface{}{
				"foo": "qux",
			},
			src: map[string]interface{}{
				"foo": map[string]interface{}{
					"bar": "baz",
				},
			},
			result: map[string]interface{}{
				"foo": map[string]interface{}{
					"bar": "baz",
				},
			},
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			MergeMaps(tt.dest, tt.src)
			if !reflect.DeepEqual(tt.dest, tt.result) {
				t.Errorf("NestedMap.Set() got = %v, want %v", tt.dest, tt.result)
			}
		})
	}
}

func TestConvertMapJSONNumbers(t *testing.T) {
	tests := []struct {
		name     string
		input    map[string]interface{}
		expected map[string]interface{}
	}{
		{
			name: "Convert JSON numbers to numbers",
			input: map[string]interface{}{
				"int":    json.Number("12"),
				"float":  json.Number("12.34"),
				"string": "foo",
			},
			expected: map[string]interface{}{
				"int":    int64(12),
				"float":  12.34,
				"string": "foo",
			},
		},
		{
			name: "Convert JSON numbers to numbers in nested maps",
			input: map[string]interface{}{
				"foo": map[string]interface{}{
					"int":           json.Number("56"),
					"float":         json.Number("56.78"),
					"nested-string": "bar",
				},
				"int":    json.Number("12"),
				"float":  json.Number("12.34"),
				"string": "foo",
			},
			expected: map[string]interface{}{
				"foo": map[string]interface{}{
					"int":           int64(56),
					"float":         56.78,
					"nested-string": "bar",
				},
				"int":    int64(12),
				"float":  12.34,
				"string": "foo",
			},
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			result := ConvertMapJSONNumbers(tt.input)
			assert.Equal(t, tt.expected, result)
		})
	}
}