do_test.go

package flatmap_test

import (
	"reflect"
	"testing"

	"github.com/nextmv-io/sdk/flatmap"
)

func Test_Do(t *testing.T) {
	type args struct {
		nested map[string]any
	}
	tests := []struct {
		name string
		args args
		want map[string]any
	}{
		{
			name: "flat",
			args: args{
				nested: map[string]any{
					"a": "foo",
					"b": 2,
					"c": true,
				},
			},
			want: map[string]any{
				"$.a": "foo",
				"$.b": 2,
				"$.c": true,
			},
		},
		{
			name: "flat with nil",
			args: args{
				nested: map[string]any{
					"a": "foo",
					"b": nil,
					"c": true,
				},
			},
			want: map[string]any{
				"$.a": "foo",
				"$.b": nil,
				"$.c": true,
			},
		},
		{
			name: "slice",
			args: args{
				nested: map[string]any{
					"a": "foo",
					"b": []any{
						"bar",
						2,
					},
				},
			},
			want: map[string]any{
				"$.a":    "foo",
				"$.b[0]": "bar",
				"$.b[1]": 2,
			},
		},
		{
			name: "nested map",
			args: args{
				nested: map[string]any{
					"a": "foo",
					"b": map[string]any{
						"c": "bar",
						"d": 2,
					},
				},
			},
			want: map[string]any{
				"$.a":   "foo",
				"$.b.c": "bar",
				"$.b.d": 2,
			},
		},
		{
			name: "slice with nested maps",
			args: args{
				nested: map[string]any{
					"a": "foo",
					"b": []any{
						map[string]any{
							"c": "bar",
							"d": 2,
						},
						map[string]any{
							"c": "baz",
							"d": 3,
						},
					},
				},
			},
			want: map[string]any{
				"$.a":      "foo",
				"$.b[0].c": "bar",
				"$.b[0].d": 2,
				"$.b[1].c": "baz",
				"$.b[1].d": 3,
			},
		},
		{
			name: "slice with nested maps with nested slice",
			args: args{
				nested: map[string]any{
					"a": "foo",
					"b": []any{
						map[string]any{
							"c": "bar",
							"d": []any{
								2,
								true,
							},
						},
						map[string]any{
							"c": "baz",
							"d": []any{
								3,
								false,
							},
						},
					},
				},
			},
			want: map[string]any{
				"$.a":         "foo",
				"$.b[0].c":    "bar",
				"$.b[0].d[0]": 2,
				"$.b[0].d[1]": true,
				"$.b[1].c":    "baz",
				"$.b[1].d[0]": 3,
				"$.b[1].d[1]": false,
			},
		},
		{
			name: "slice with nested maps with nested slice with nested map",
			args: args{
				nested: map[string]any{
					"a": "foo",
					"b": []any{
						map[string]any{
							"c": "bar",
							"d": []any{
								map[string]any{
									"e": 2,
								},
								true,
							},
						},
						map[string]any{
							"c": "baz",
							"d": []any{
								map[string]any{
									"e": 3,
								},
								false,
							},
						},
					},
				},
			},
			want: map[string]any{
				"$.a":           "foo",
				"$.b[0].c":      "bar",
				"$.b[0].d[0].e": 2,
				"$.b[0].d[1]":   true,
				"$.b[1].c":      "baz",
				"$.b[1].d[0].e": 3,
				"$.b[1].d[1]":   false,
			},
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			if got := flatmap.Do(tt.args.nested); !reflect.DeepEqual(got, tt.want) {
				t.Errorf("flatten() = %v, want %v", got, tt.want)
			}
		})
	}
}