github.com/serbaut/terraform@v0.6.12-0.20160607213102-ac2d195cc560/flatmap/map_test.go

package flatmap

import (
	"reflect"
	"sort"
	"testing"
)

func TestMapContains(t *testing.T) {
	cases := []struct {
		Input  map[string]string
		Key    string
		Result bool
	}{
		{
			Input: map[string]string{
				"foo": "bar",
				"bar": "nope",
			},
			Key:    "foo",
			Result: true,
		},

		{
			Input: map[string]string{
				"foo": "bar",
				"bar": "nope",
			},
			Key:    "baz",
			Result: false,
		},
	}

	for i, tc := range cases {
		actual := Map(tc.Input).Contains(tc.Key)
		if actual != tc.Result {
			t.Fatalf("case %d bad: %#v", i, tc.Input)
		}
	}
}

func TestMapDelete(t *testing.T) {
	m := Flatten(map[string]interface{}{
		"foo": "bar",
		"routes": []map[string]string{
			map[string]string{
				"foo": "bar",
			},
		},
	})

	m.Delete("routes")

	expected := Map(map[string]string{"foo": "bar"})
	if !reflect.DeepEqual(m, expected) {
		t.Fatalf("bad: %#v", m)
	}
}

func TestMapKeys(t *testing.T) {
	cases := []struct {
		Input  map[string]string
		Output []string
	}{
		{
			Input: map[string]string{
				"foo":       "bar",
				"bar.#":     "bar",
				"bar.0.foo": "bar",
				"bar.0.baz": "bar",
			},
			Output: []string{
				"bar",
				"foo",
			},
		},
	}

	for _, tc := range cases {
		actual := Map(tc.Input).Keys()

		// Sort so we have a consistent view of the output
		sort.Strings(actual)

		if !reflect.DeepEqual(actual, tc.Output) {
			t.Fatalf("input: %#v\n\nbad: %#v", tc.Input, actual)
		}
	}
}

func TestMapMerge(t *testing.T) {
	cases := []struct {
		One    map[string]string
		Two    map[string]string
		Result map[string]string
	}{
		{
			One: map[string]string{
				"foo": "bar",
				"bar": "nope",
			},
			Two: map[string]string{
				"bar": "baz",
				"baz": "buz",
			},
			Result: map[string]string{
				"foo": "bar",
				"bar": "baz",
				"baz": "buz",
			},
		},
	}

	for i, tc := range cases {
		Map(tc.One).Merge(Map(tc.Two))
		if !reflect.DeepEqual(tc.One, tc.Result) {
			t.Fatalf("case %d bad: %#v", i, tc.One)
		}
	}
}