map_test.go

package funk

import (
	"sort"
	"testing"

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

func TestKeys(t *testing.T) {
	is := assert.New(t)

	results := Keys(map[string]int{"one": 1, "two": 2}).([]string)
	sort.Strings(results)

	is.Equal(results, []string{"one", "two"})

	fields := Keys(foo).([]string)

	sort.Strings(fields)

	is.Equal(fields, []string{"Age", "Bar", "BarInterface", "BarPointer", "Bars", "EmptyValue", "FirstName", "GeneralInterface", "ID", "LastName", "ZeroBoolValue", "ZeroIntPtrValue", "ZeroIntValue"})
}

func TestValues(t *testing.T) {
	is := assert.New(t)

	results := Values(map[string]int{"one": 1, "two": 2}).([]int)
	sort.Ints(results)

	is.Equal(results, []int{1, 2})

	values := Values(foo).([]interface{})

	is.Len(values, 13)
}