route/vendor/github.com/asdine/storm/extract_test.go

137 lines
3.7 KiB
Go

package storm
import (
"reflect"
"testing"
"github.com/stretchr/testify/require"
)
func allByType(m *structConfig, indexType string) []*fieldConfig {
var idx []*fieldConfig
for k := range m.Fields {
if m.Fields[k].Index == indexType {
idx = append(idx, m.Fields[k])
}
}
return idx
}
func TestExtractNoTags(t *testing.T) {
s := ClassicNoTags{}
r := reflect.ValueOf(&s)
_, err := extract(&r)
require.Error(t, err)
require.Equal(t, ErrNoID, err)
}
func TestExtractBadTags(t *testing.T) {
s := ClassicBadTags{}
r := reflect.ValueOf(&s)
infos, err := extract(&r)
require.Error(t, err)
require.Equal(t, ErrUnknownTag, err)
require.Nil(t, infos)
}
func TestExtractUniqueTags(t *testing.T) {
s := ClassicUnique{ID: "id"}
r := reflect.ValueOf(&s)
infos, err := extract(&r)
require.NoError(t, err)
require.NotNil(t, infos)
require.NotNil(t, infos.ID)
require.False(t, infos.ID.IsZero)
require.Equal(t, "ClassicUnique", infos.Name)
require.Len(t, allByType(infos, "index"), 0)
require.Len(t, allByType(infos, "unique"), 4)
}
func TestExtractIndexTags(t *testing.T) {
s := ClassicIndex{ID: "id"}
r := reflect.ValueOf(&s)
infos, err := extract(&r)
require.NoError(t, err)
require.NotNil(t, infos)
require.NotNil(t, infos.ID)
require.False(t, infos.ID.IsZero)
require.Equal(t, "ClassicIndex", infos.Name)
require.Len(t, allByType(infos, "index"), 5)
require.Len(t, allByType(infos, "unique"), 0)
}
func TestExtractInlineWithIndex(t *testing.T) {
s := ClassicInline{ToEmbed: &ToEmbed{ID: "50"}}
r := reflect.ValueOf(&s)
infos, err := extract(&r)
require.NoError(t, err)
require.NotNil(t, infos)
require.NotNil(t, infos.ID)
require.Equal(t, "ClassicInline", infos.Name)
require.Len(t, allByType(infos, "index"), 3)
require.Len(t, allByType(infos, "unique"), 2)
}
func TestExtractMultipleTags(t *testing.T) {
type User struct {
ID uint64 `storm:"id,increment"`
Age uint16 `storm:"index,increment"`
unexportedField int32 `storm:"index,increment"`
X uint32 `storm:"unique,increment=100"`
Y int8 `storm:"index,increment=-100"`
}
s := User{}
r := reflect.ValueOf(&s)
infos, err := extract(&r)
require.NoError(t, err)
require.NotNil(t, infos)
require.NotNil(t, infos.ID)
require.Equal(t, "User", infos.Name)
require.Len(t, allByType(infos, "index"), 2)
require.Len(t, allByType(infos, "unique"), 1)
require.True(t, infos.Fields["Age"].Increment)
require.Equal(t, int64(1), infos.Fields["Age"].IncrementStart)
require.Equal(t, "index", infos.Fields["Age"].Index)
require.False(t, infos.Fields["Age"].IsID)
require.True(t, infos.Fields["Age"].IsInteger)
require.True(t, infos.Fields["Age"].IsZero)
require.NotNil(t, infos.Fields["Age"].Value)
require.True(t, infos.Fields["X"].Increment)
require.Equal(t, int64(100), infos.Fields["X"].IncrementStart)
require.Equal(t, "unique", infos.Fields["X"].Index)
require.False(t, infos.Fields["X"].IsID)
require.True(t, infos.Fields["X"].IsInteger)
require.True(t, infos.Fields["X"].IsZero)
require.NotNil(t, infos.Fields["X"].Value)
require.True(t, infos.Fields["Y"].Increment)
require.Equal(t, int64(-100), infos.Fields["Y"].IncrementStart)
require.Equal(t, "index", infos.Fields["Y"].Index)
require.False(t, infos.Fields["Y"].IsID)
require.True(t, infos.Fields["Y"].IsInteger)
require.True(t, infos.Fields["Y"].IsZero)
require.NotNil(t, infos.Fields["Y"].Value)
type NoInt struct {
ID uint64 `storm:"id,increment=hello"`
}
var n NoInt
r = reflect.ValueOf(&n)
_, err = extract(&r)
require.Error(t, err)
type BadSuffix struct {
ID uint64 `storm:"id,incrementag=100"`
}
var b BadSuffix
r = reflect.ValueOf(&b)
_, err = extract(&r)
require.Error(t, err)
}