mirror of
https://github.com/MontFerret/ferret.git
synced 2025-08-13 19:52:52 +02:00
Add unit tests to ForExpression (#791)
* Add unit tests for for.go * Remove needless codes. Deleted codes are results of test of Exec results after addXXX methods These should be written in XXXClause test. * Fix test descriptions and clear up codes * Fix * Execute make fmt * fix test * fmt * fix
This commit is contained in:
rikinyan
GitHub
351
pkg/runtime/expressions/for_test.go
Normal file
351
pkg/runtime/expressions/for_test.go
Normal file
@@ -0,0 +1,351 @@
|
||||
package expressions
|
||||
|
||||
import (
|
||||
"context"
|
||||
"testing"
|
||||
|
||||
. "github.com/smartystreets/goconvey/convey"
|
||||
|
||||
"github.com/MontFerret/ferret/pkg/runtime/collections"
|
||||
"github.com/MontFerret/ferret/pkg/runtime/core"
|
||||
"github.com/MontFerret/ferret/pkg/runtime/expressions/clauses"
|
||||
"github.com/MontFerret/ferret/pkg/runtime/values"
|
||||
)
|
||||
|
||||
type (
|
||||
testIterator struct {
|
||||
values []*core.Scope
|
||||
pos int
|
||||
causeErrorInNext bool
|
||||
}
|
||||
testIterable struct {
|
||||
values []*core.Scope
|
||||
causeErrorInIterate bool
|
||||
causeErrorInNext bool
|
||||
}
|
||||
testExpression struct {
|
||||
causeErrorInExec bool
|
||||
}
|
||||
testError struct{}
|
||||
)
|
||||
|
||||
func (iterator *testIterator) Next(ctx context.Context, scope *core.Scope) (*core.Scope, error) {
|
||||
if iterator.causeErrorInNext {
|
||||
return nil, testError{}
|
||||
}
|
||||
|
||||
if len(iterator.values) > iterator.pos {
|
||||
val := iterator.values[iterator.pos]
|
||||
iterator.pos++
|
||||
|
||||
return val, nil
|
||||
}
|
||||
|
||||
return nil, core.ErrNoMoreData
|
||||
}
|
||||
|
||||
func (iterable *testIterable) Iterate(ctx context.Context, scope *core.Scope) (collections.Iterator, error) {
|
||||
if iterable.causeErrorInIterate {
|
||||
return nil, testError{}
|
||||
}
|
||||
|
||||
return &testIterator{iterable.values, 0, iterable.causeErrorInNext}, nil
|
||||
}
|
||||
|
||||
func (expression *testExpression) Exec(ctx context.Context, scope *core.Scope) (core.Value, error) {
|
||||
if expression.causeErrorInExec {
|
||||
return nil, testError{}
|
||||
}
|
||||
return nil, nil
|
||||
}
|
||||
|
||||
func (error testError) Error() string {
|
||||
return "error"
|
||||
}
|
||||
|
||||
func testInitTestIterable(values []*core.Scope, causeErrorInIterate, causeErrorInNext bool) *testIterable {
|
||||
return &testIterable{values, causeErrorInIterate, causeErrorInNext}
|
||||
}
|
||||
|
||||
func TestNewForExpression(t *testing.T) {
|
||||
Convey(".NewForExpression", t, func() {
|
||||
Convey("Should return new ForExpresssion.", func() {
|
||||
forExp, err := NewForExpression(core.SourceMap{}, testInitTestIterable([]*core.Scope{}, false, false), &testExpression{}, false, false, false)
|
||||
So(forExp, ShouldNotBeNil)
|
||||
So(err, ShouldBeNil)
|
||||
})
|
||||
|
||||
Convey("Should return error when a dataSource is nil", func() {
|
||||
forExp, err := NewForExpression(core.SourceMap{}, nil, &testExpression{}, false, false, false)
|
||||
So(forExp, ShouldBeNil)
|
||||
So(err, ShouldNotBeNil)
|
||||
So(err, ShouldEqual, err)
|
||||
})
|
||||
|
||||
Convey("Should return error when a predicate is nil", func() {
|
||||
forExp, err := NewForExpression(core.SourceMap{}, testInitTestIterable([]*core.Scope{}, false, false), nil, false, false, false)
|
||||
So(forExp, ShouldBeNil)
|
||||
So(err, ShouldNotBeNil)
|
||||
So(err, ShouldEqual, err)
|
||||
})
|
||||
})
|
||||
}
|
||||
|
||||
func TestAddLimit(t *testing.T) {
|
||||
Convey(".AddLimit", t, func() {
|
||||
Convey("Should success. (An Error Should be nil.)", func() {
|
||||
forExpression, _ := NewForExpression(
|
||||
core.SourceMap{},
|
||||
testInitTestIterable([]*core.Scope{}, false, false),
|
||||
&testExpression{},
|
||||
false,
|
||||
false,
|
||||
false,
|
||||
)
|
||||
|
||||
err := forExpression.AddLimit(core.SourceMap{}, &testExpression{}, &testExpression{})
|
||||
|
||||
So(err, ShouldBeNil)
|
||||
})
|
||||
|
||||
Convey("Should return an error.", func() {
|
||||
forExpression, _ := NewForExpression(
|
||||
core.SourceMap{},
|
||||
testInitTestIterable([]*core.Scope{}, false, false),
|
||||
&testExpression{},
|
||||
false,
|
||||
false,
|
||||
false,
|
||||
)
|
||||
forExpression.dataSource = nil
|
||||
|
||||
err := forExpression.AddLimit(core.SourceMap{}, &testExpression{}, &testExpression{})
|
||||
|
||||
So(err, ShouldNotBeNil)
|
||||
})
|
||||
})
|
||||
}
|
||||
|
||||
func TestAddFilter(t *testing.T) {
|
||||
Convey(".AddFilter", t, func() {
|
||||
Convey("Should success. (An Error Should be nil.)", func() {
|
||||
forExpression, _ := NewForExpression(
|
||||
core.SourceMap{},
|
||||
testInitTestIterable([]*core.Scope{}, false, false),
|
||||
&testExpression{},
|
||||
false,
|
||||
false,
|
||||
false,
|
||||
)
|
||||
|
||||
err := forExpression.AddFilter(core.SourceMap{}, &testExpression{})
|
||||
|
||||
So(err, ShouldBeNil)
|
||||
})
|
||||
|
||||
Convey("Should return an error.", func() {
|
||||
forExpression, _ := NewForExpression(
|
||||
core.SourceMap{},
|
||||
testInitTestIterable([]*core.Scope{}, false, false),
|
||||
&testExpression{},
|
||||
false,
|
||||
false,
|
||||
false,
|
||||
)
|
||||
forExpression.dataSource = nil
|
||||
|
||||
err := forExpression.AddFilter(core.SourceMap{}, &testExpression{})
|
||||
|
||||
So(err, ShouldNotBeNil)
|
||||
})
|
||||
})
|
||||
|
||||
}
|
||||
|
||||
func TestAddSort(t *testing.T) {
|
||||
Convey(".AddSort", t, func() {
|
||||
Convey("Should success.(An Error Should be nil.)", func() {
|
||||
forExpression, _ := NewForExpression(
|
||||
core.SourceMap{},
|
||||
testInitTestIterable([]*core.Scope{}, false, false),
|
||||
&testExpression{},
|
||||
false,
|
||||
false,
|
||||
false,
|
||||
)
|
||||
|
||||
err := forExpression.AddSort(core.SourceMap{}, &clauses.SorterExpression{})
|
||||
|
||||
So(err, ShouldBeNil)
|
||||
})
|
||||
|
||||
Convey("Should return an error.", func() {
|
||||
forExpression, _ := NewForExpression(
|
||||
core.SourceMap{},
|
||||
testInitTestIterable([]*core.Scope{}, false, false),
|
||||
&testExpression{},
|
||||
false,
|
||||
false,
|
||||
false,
|
||||
)
|
||||
forExpression.dataSource = nil
|
||||
|
||||
err := forExpression.AddSort(core.SourceMap{}, &clauses.SorterExpression{})
|
||||
|
||||
So(err, ShouldNotBeNil)
|
||||
})
|
||||
})
|
||||
}
|
||||
|
||||
func TestAddCollect(t *testing.T) {
|
||||
Convey(".AddCollect", t, func() {
|
||||
Convey("Should success. (Error Should be nil.)", func() {
|
||||
forExpression, _ := NewForExpression(
|
||||
core.SourceMap{},
|
||||
testInitTestIterable([]*core.Scope{}, false, false),
|
||||
&testExpression{},
|
||||
false,
|
||||
false,
|
||||
false,
|
||||
)
|
||||
|
||||
err := forExpression.AddCollect(core.SourceMap{}, &clauses.Collect{})
|
||||
|
||||
So(err, ShouldBeNil)
|
||||
})
|
||||
|
||||
Convey("Should return an error.", func() {
|
||||
forExpression, _ := NewForExpression(
|
||||
core.SourceMap{},
|
||||
testInitTestIterable([]*core.Scope{}, false, false),
|
||||
&testExpression{},
|
||||
false,
|
||||
false,
|
||||
false,
|
||||
)
|
||||
forExpression.dataSource = nil
|
||||
|
||||
err := forExpression.AddCollect(core.SourceMap{}, &clauses.Collect{})
|
||||
|
||||
So(err, ShouldNotBeNil)
|
||||
})
|
||||
})
|
||||
}
|
||||
|
||||
func TestAddStatement(t *testing.T) {
|
||||
Convey(".AddStatement (Error Should be nil.)", t, func() {
|
||||
Convey("Should success.", func() {
|
||||
forExpression, _ := NewForExpression(
|
||||
core.SourceMap{},
|
||||
testInitTestIterable([]*core.Scope{}, false, false),
|
||||
&testExpression{},
|
||||
false,
|
||||
false,
|
||||
false,
|
||||
)
|
||||
|
||||
err := forExpression.AddStatement(&testExpression{})
|
||||
|
||||
So(err, ShouldBeNil)
|
||||
})
|
||||
|
||||
Convey("Should return an error.", func() {
|
||||
forExpression, _ := NewForExpression(
|
||||
core.SourceMap{},
|
||||
testInitTestIterable([]*core.Scope{}, false, false),
|
||||
&testExpression{},
|
||||
false,
|
||||
false,
|
||||
false,
|
||||
)
|
||||
forExpression.dataSource = nil
|
||||
|
||||
err := forExpression.AddStatement(&testExpression{})
|
||||
|
||||
So(err, ShouldNotBeNil)
|
||||
})
|
||||
})
|
||||
}
|
||||
|
||||
func TestExec(t *testing.T) {
|
||||
Convey(".Exec", t, func() {
|
||||
Convey("Should success.", func() {
|
||||
forExpression, _ := NewForExpression(
|
||||
core.SourceMap{},
|
||||
testInitTestIterable([]*core.Scope{{}, {}, {}}, false, false),
|
||||
&testExpression{},
|
||||
false,
|
||||
false,
|
||||
false,
|
||||
)
|
||||
|
||||
_, err := forExpression.Exec(context.Background(), &core.Scope{})
|
||||
|
||||
So(err, ShouldBeNil)
|
||||
})
|
||||
|
||||
Convey("Should stop an execution when context is cancelled.", func() {
|
||||
forExpression, _ := NewForExpression(
|
||||
core.SourceMap{},
|
||||
testInitTestIterable([]*core.Scope{{}, {}, {}}, false, false),
|
||||
&testExpression{},
|
||||
false,
|
||||
false,
|
||||
false,
|
||||
)
|
||||
|
||||
ctx0 := context.Background()
|
||||
ctx1, cancelFn := context.WithCancel(ctx0)
|
||||
cancelFn()
|
||||
|
||||
_, err := forExpression.Exec(ctx1, &core.Scope{})
|
||||
|
||||
So(err, ShouldNotBeNil)
|
||||
})
|
||||
|
||||
Convey("Should return an error when a dataSource expression is invalid.", func() {
|
||||
forExpression, _ := NewForExpression(
|
||||
core.SourceMap{},
|
||||
testInitTestIterable([]*core.Scope{{}, {}, {}}, true, false),
|
||||
&testExpression{},
|
||||
false,
|
||||
false,
|
||||
false,
|
||||
)
|
||||
|
||||
result, err := forExpression.Exec(context.Background(), &core.Scope{})
|
||||
So(result, ShouldEqual, values.None)
|
||||
So(err, ShouldNotBeNil)
|
||||
})
|
||||
|
||||
Convey("Should return an error when element expressions of dataSource is invalid.", func() {
|
||||
forExpression, _ := NewForExpression(
|
||||
core.SourceMap{},
|
||||
testInitTestIterable([]*core.Scope{{}, {}, {}}, false, true),
|
||||
&testExpression{},
|
||||
false,
|
||||
false,
|
||||
false,
|
||||
)
|
||||
|
||||
result, err := forExpression.Exec(context.Background(), &core.Scope{})
|
||||
So(result, ShouldEqual, values.None)
|
||||
So(err, ShouldNotBeNil)
|
||||
})
|
||||
|
||||
Convey("Should return an error when an predicate expression is invalidated.", func() {
|
||||
forExpression, _ := NewForExpression(
|
||||
core.SourceMap{},
|
||||
testInitTestIterable([]*core.Scope{{}, {}, {}}, false, false),
|
||||
&testExpression{true},
|
||||
false,
|
||||
false,
|
||||
false,
|
||||
)
|
||||
|
||||
result, err := forExpression.Exec(context.Background(), &core.Scope{})
|
||||
So(result, ShouldEqual, values.None)
|
||||
So(err, ShouldNotBeNil)
|
||||
})
|
||||
})
|
||||
}
|
||||
Reference in New Issue
Block a user