mirror of
https://github.com/MontFerret/ferret.git
synced 2024-12-16 11:37:36 +02:00
108 lines
2.2 KiB
Go
108 lines
2.2 KiB
Go
package collections_test
|
|
|
|
import (
|
|
"context"
|
|
"github.com/MontFerret/ferret/pkg/runtime/collections"
|
|
"github.com/MontFerret/ferret/pkg/runtime/core"
|
|
"github.com/MontFerret/ferret/pkg/runtime/values"
|
|
. "github.com/smartystreets/goconvey/convey"
|
|
"testing"
|
|
)
|
|
|
|
func TestUniqueIterator(t *testing.T) {
|
|
Convey("Should return only unique items", t, func() {
|
|
arr := []core.Value{
|
|
values.NewInt(1),
|
|
values.NewInt(2),
|
|
values.NewInt(2),
|
|
values.NewInt(3),
|
|
values.NewInt(4),
|
|
values.NewInt(3),
|
|
values.NewInt(5),
|
|
values.NewInt(6),
|
|
values.NewInt(5),
|
|
values.NewInt(6),
|
|
}
|
|
|
|
iter, err := collections.NewUniqueIterator(
|
|
sliceIterator(arr),
|
|
collections.DefaultValueVar,
|
|
)
|
|
|
|
So(err, ShouldBeNil)
|
|
ctx := context.Background()
|
|
scope, _ := core.NewRootScope()
|
|
|
|
sets, err := collections.ToSlice(ctx, scope, iter)
|
|
|
|
So(err, ShouldBeNil)
|
|
|
|
res := toArrayOfValues(sets)
|
|
|
|
So(res.String(), ShouldEqual, `[1,2,3,4,5,6]`)
|
|
})
|
|
|
|
Convey("Should return only unique items 2", t, func() {
|
|
arr := []core.Value{
|
|
values.NewInt(1),
|
|
values.NewInt(1),
|
|
values.NewInt(1),
|
|
values.NewInt(1),
|
|
values.NewInt(1),
|
|
values.NewInt(1),
|
|
}
|
|
|
|
iter, err := collections.NewUniqueIterator(
|
|
sliceIterator(arr),
|
|
collections.DefaultValueVar,
|
|
)
|
|
|
|
So(err, ShouldBeNil)
|
|
|
|
ctx := context.Background()
|
|
scope, _ := core.NewRootScope()
|
|
|
|
sets, err := collections.ToSlice(ctx, scope, iter)
|
|
|
|
So(err, ShouldBeNil)
|
|
|
|
res := toArrayOfValues(sets)
|
|
|
|
So(res.String(), ShouldEqual, `[1]`)
|
|
})
|
|
|
|
Convey("Should return only unique items 3", t, func() {
|
|
arr := []core.Value{
|
|
values.NewString("a"),
|
|
values.NewString("b"),
|
|
values.NewString("c"),
|
|
values.NewString("d"),
|
|
values.NewString("e"),
|
|
values.NewString("a"),
|
|
values.NewString("b"),
|
|
values.NewString("f"),
|
|
values.NewString("d"),
|
|
values.NewString("e"),
|
|
values.NewString("f"),
|
|
}
|
|
|
|
iter, err := collections.NewUniqueIterator(
|
|
sliceIterator(arr),
|
|
collections.DefaultValueVar,
|
|
)
|
|
|
|
So(err, ShouldBeNil)
|
|
|
|
ctx := context.Background()
|
|
scope, _ := core.NewRootScope()
|
|
|
|
sets, err := collections.ToSlice(ctx, scope, iter)
|
|
|
|
So(err, ShouldBeNil)
|
|
|
|
res := toArrayOfValues(sets)
|
|
|
|
So(res.String(), ShouldEqual, `["a","b","c","d","e","f"]`)
|
|
})
|
|
}
|