in pkg/index/testcases/duration.go [61:273]
func RunDuration(t *testing.T, data map[int]posting.List, store SimpleStore) {
tests := []struct {
name string
want []int
args args
}{
{
name: "sort in asc order",
args: args{
fieldKey: duration,
orderType: modelv1.Sort_SORT_ASC,
},
want: []int{50, 200, 500, 1000, 2000},
},
{
name: "sort in desc order",
args: args{
fieldKey: duration,
orderType: modelv1.Sort_SORT_DESC,
},
want: []int{2000, 1000, 500, 200, 50},
},
{
name: "scan in (lower, upper) and sort in asc order",
args: args{
fieldKey: duration,
orderType: modelv1.Sort_SORT_ASC,
termRange: index.NewIntRangeOpts(50, 2000, false, false),
},
want: []int{200, 500, 1000},
},
{
name: "scan in (lower, upper) and sort in desc order",
args: args{
fieldKey: duration,
orderType: modelv1.Sort_SORT_DESC,
termRange: index.NewIntRangeOpts(50, 2000, false, false),
},
want: []int{1000, 500, 200},
},
{
name: "scan in [lower, upper] and sort in asc order",
args: args{
fieldKey: duration,
orderType: modelv1.Sort_SORT_ASC,
termRange: index.NewIntRangeOpts(200, 1000, true, true),
},
want: []int{200, 500, 1000},
},
{
name: "scan in [lower, upper] and sort in desc order",
args: args{
fieldKey: duration,
orderType: modelv1.Sort_SORT_DESC,
termRange: index.NewIntRangeOpts(200, 1000, true, true),
},
want: []int{1000, 500, 200},
},
{
name: "scan in [lower, undefined) and sort in asc order",
args: args{
fieldKey: duration,
orderType: modelv1.Sort_SORT_ASC,
termRange: index.NewIntRangeOpts(200, math.MaxInt64, true, false),
},
want: []int{200, 500, 1000, 2000},
},
{
name: "scan in [lower, undefined) and sort in desc order",
args: args{
fieldKey: duration,
orderType: modelv1.Sort_SORT_DESC,
termRange: index.NewIntRangeOpts(200, math.MaxInt64, true, false),
},
want: []int{2000, 1000, 500, 200},
},
{
name: "scan in (undefined, upper] and sort in asc order",
args: args{
fieldKey: duration,
orderType: modelv1.Sort_SORT_ASC,
termRange: index.NewIntRangeOpts(math.MinInt64, 1000, false, true),
},
want: []int{50, 200, 500, 1000},
},
{
name: "scan in (undefined, upper] and sort in desc order",
args: args{
fieldKey: duration,
orderType: modelv1.Sort_SORT_DESC,
termRange: index.NewIntRangeOpts(math.MinInt64, 1000, false, true),
},
want: []int{1000, 500, 200, 50},
},
{
name: "scan splice in (lower, upper) and sort in asc order",
args: args{
fieldKey: duration,
orderType: modelv1.Sort_SORT_ASC,
termRange: index.NewIntRangeOpts(150, 1900, false, false),
},
want: []int{200, 500, 1000},
},
{
name: "scan splice in (lower, upper) and sort in desc order",
args: args{
fieldKey: duration,
orderType: modelv1.Sort_SORT_DESC,
termRange: index.NewIntRangeOpts(150, 1900, false, false),
},
want: []int{1000, 500, 200},
},
{
name: "scan splice in [lower, upper] and sort in asc order",
args: args{
fieldKey: duration,
orderType: modelv1.Sort_SORT_ASC,
termRange: index.NewIntRangeOpts(150, 1900, true, true),
},
want: []int{200, 500, 1000},
},
{
name: "scan splice in [lower, upper] and sort in desc order",
args: args{
fieldKey: duration,
orderType: modelv1.Sort_SORT_DESC,
termRange: index.NewIntRangeOpts(150, 1900, true, true),
},
want: []int{1000, 500, 200},
},
{
name: "no field key",
args: args{},
},
{
name: "unknown field key",
args: args{
fieldKey: index.FieldKey{
IndexRuleID: 0,
},
},
},
{
name: "default order",
args: args{
fieldKey: duration,
},
want: []int{50, 200, 500, 1000, 2000},
},
{
name: "invalid range",
args: args{
fieldKey: duration,
termRange: index.NewIntRangeOpts(100, 50, false, false),
},
},
}
preLoadSizes := []int{7, 20, 50}
allTests := make([]struct {
name string
want []int
args args
preloadSize int
}, 0, len(tests)*len(preLoadSizes))
for _, size := range preLoadSizes {
for _, t := range tests {
allTests = append(allTests, struct {
name string
want []int
args args
preloadSize int
}{
name: t.name + " preLoadSize " + fmt.Sprint(size),
want: t.want,
preloadSize: size,
args: t.args,
})
}
}
for _, tt := range allTests {
t.Run(tt.name, func(t *testing.T) {
tester := assert.New(t)
is := require.New(t)
iter, err := store.Iterator(context.TODO(), tt.args.fieldKey, tt.args.termRange, tt.args.orderType, tt.preloadSize)
is.NoError(err)
if iter == nil {
tester.Empty(tt.want)
return
}
defer func() {
tester.NoError(iter.Close())
for i := 0; i < 10; i++ {
is.False(iter.Next())
}
}()
is.NotNil(iter)
var got result
for iter.Next() {
got.items = append(got.items, iter.Val().DocID)
}
for i := 0; i < 10; i++ {
is.False(iter.Next())
}
var wants result
for _, w := range tt.want {
pl := data[w]
wants.items = append(wants.items, pl.ToSlice()...)
}
tester.Equal(wants, got, tt.name)
})
}
}