1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106
|
package objx_test
import (
"strings"
"testing"
"github.com/stretchr/objx"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
)
func TestExclude(t *testing.T) {
m := objx.Map{
"name": "Mat",
"age": 29,
"secret": "ABC",
}
excluded := m.Exclude([]string{"secret"})
assert.Equal(t, m["name"], excluded["name"])
assert.Equal(t, m["age"], excluded["age"])
assert.False(t, excluded.Has("secret"), "secret should be excluded")
}
func TestCopy(t *testing.T) {
m1 := objx.Map{
"name": "Tyler",
"location": "UT",
}
m2 := m1.Copy()
require.NotNil(t, m2)
m2["name"] = "Mat"
assert.Equal(t, m1.Get("name").Str(), "Tyler")
assert.Equal(t, m2.Get("name").Str(), "Mat")
}
func TestMerge(t *testing.T) {
m1 := objx.Map{
"name": "Mat",
}
m2 := objx.Map{
"name": "Tyler",
"location": "UT",
}
merged := m1.Merge(m2)
assert.Equal(t, merged.Get("name").Str(), m2.Get("name").Str())
assert.Equal(t, merged.Get("location").Str(), m2.Get("location").Str())
assert.Empty(t, m1.Get("location").Str())
}
func TestMergeHere(t *testing.T) {
m1 := objx.Map{
"name": "Mat",
}
m2 := objx.Map{
"name": "Tyler",
"location": "UT",
}
merged := m1.MergeHere(m2)
assert.Equal(t, m1, merged, "With MergeHere, it should return the first modified map")
assert.Equal(t, merged.Get("name").Str(), m2.Get("name").Str())
assert.Equal(t, merged.Get("location").Str(), m2.Get("location").Str())
assert.Equal(t, merged.Get("location").Str(), m1.Get("location").Str())
}
func TestTransform(t *testing.T) {
m := objx.Map{
"name": "Mat",
"location": "UK",
}
r := m.Transform(keyToUpper)
assert.Equal(t, objx.Map{
"NAME": "Mat",
"LOCATION": "UK",
}, r)
}
func TestTransformKeys(t *testing.T) {
m := objx.Map{
"a": "1",
"b": "2",
"c": "3",
}
mapping := map[string]string{
"a": "d",
"b": "e",
}
r := m.TransformKeys(mapping)
assert.Equal(t, objx.Map{
"c": "3",
"d": "1",
"e": "2",
}, r)
}
func keyToUpper(s string, v interface{}) (string, interface{}) {
return strings.ToUpper(s), v
}
|