mirror of
https://github.com/dolthub/dolt.git
synced 2026-01-30 10:45:18 -06:00
121 lines
2.8 KiB
Go
121 lines
2.8 KiB
Go
package test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/attic-labs/noms/Godeps/_workspace/src/github.com/stretchr/testify/assert"
|
|
"github.com/attic-labs/noms/chunks"
|
|
"github.com/attic-labs/noms/nomdl/codegen/test/gen"
|
|
"github.com/attic-labs/noms/types"
|
|
)
|
|
|
|
func TestMapDef(t *testing.T) {
|
|
assert := assert.New(t)
|
|
cs := chunks.NewMemoryStore()
|
|
|
|
def := gen.MapOfBoolToStringDef{true: "hi", false: "bye"}
|
|
m := def.New(cs)
|
|
|
|
assert.Equal(uint64(2), m.Len())
|
|
assert.Equal("hi", m.Get(true))
|
|
assert.Equal("bye", m.Get(false))
|
|
|
|
def2 := m.Def()
|
|
assert.Equal(def, def2)
|
|
|
|
m2 := gen.NewMapOfBoolToString(cs).Set(true, "hi").Set(false, "bye")
|
|
assert.True(m.Equals(m2))
|
|
}
|
|
|
|
func TestValueMapDef(t *testing.T) {
|
|
assert := assert.New(t)
|
|
cs := chunks.NewMemoryStore()
|
|
|
|
def := gen.MapOfStringToValueDef{"s": types.NewString("s"), "i": types.Int32(42)}
|
|
m := def.New(cs)
|
|
|
|
assert.Equal(uint64(2), m.Len())
|
|
assert.True(types.NewString("s").Equals(m.Get("s")))
|
|
assert.True(types.Int32(42).Equals(m.Get("i")))
|
|
|
|
def2 := m.Def()
|
|
assert.Equal(def, def2)
|
|
|
|
m2 := gen.NewMapOfStringToValue(cs).Set("s", types.NewString("s")).Set("i", types.Int32(42))
|
|
assert.True(m.Equals(m2))
|
|
}
|
|
|
|
func TestValueMapValue(t *testing.T) {
|
|
assert := assert.New(t)
|
|
cs := chunks.NewMemoryStore()
|
|
|
|
def := gen.MapOfStringToValueDef{"s": types.NewString("s"), "i": types.Int32(42)}
|
|
var m types.Value
|
|
m = def.New(cs)
|
|
m2 := m.(gen.MapOfStringToValue)
|
|
assert.True(m.Equals(m2))
|
|
}
|
|
|
|
func TestMapIter(t *testing.T) {
|
|
assert := assert.New(t)
|
|
cs := chunks.NewMemoryStore()
|
|
|
|
m := gen.MapOfBoolToStringDef{true: "hi", false: "bye"}.New(cs)
|
|
acc := gen.NewMapOfBoolToString(cs)
|
|
m.Iter(func(k bool, v string) bool {
|
|
acc = acc.Set(k, v)
|
|
return false
|
|
})
|
|
assert.True(m.Equals(acc))
|
|
|
|
acc = gen.NewMapOfBoolToString(cs)
|
|
m.Iter(func(k bool, v string) bool {
|
|
return true
|
|
})
|
|
assert.True(acc.Empty())
|
|
}
|
|
|
|
func TestMapIterAll(t *testing.T) {
|
|
assert := assert.New(t)
|
|
cs := chunks.NewMemoryStore()
|
|
|
|
m := gen.MapOfBoolToStringDef{true: "hi", false: "bye"}.New(cs)
|
|
acc := gen.NewMapOfBoolToString(cs)
|
|
m.IterAll(func(k bool, v string) {
|
|
acc = acc.Set(k, v)
|
|
})
|
|
assert.True(m.Equals(acc))
|
|
}
|
|
|
|
func TestMapFilter(t *testing.T) {
|
|
assert := assert.New(t)
|
|
cs := chunks.NewMemoryStore()
|
|
|
|
m := gen.MapOfBoolToStringDef{true: "hi", false: "bye"}.New(cs)
|
|
m2 := m.Filter(func(k bool, v string) bool {
|
|
return k
|
|
})
|
|
assert.True(gen.NewMapOfBoolToString(cs).Set(true, "hi").Equals(m2))
|
|
|
|
m3 := m.Filter(func(k bool, v string) bool {
|
|
return v == "bye"
|
|
})
|
|
assert.True(gen.NewMapOfBoolToString(cs).Set(false, "bye").Equals(m3))
|
|
}
|
|
|
|
func TestMapMaybeGet(t *testing.T) {
|
|
assert := assert.New(t)
|
|
cs := chunks.NewMemoryStore()
|
|
|
|
m := gen.NewMapOfStringToValue(cs)
|
|
k1 := "key1"
|
|
k2 := "key2"
|
|
v1 := types.NewString("SomeValue")
|
|
m = m.Set(k1, v1)
|
|
v, ok := m.MaybeGet(k1)
|
|
assert.True(ok)
|
|
assert.Equal(v1, v)
|
|
v, ok = m.MaybeGet(k2)
|
|
assert.False(ok)
|
|
}
|