1
0
mirror of https://github.com/MontFerret/ferret.git synced 2024-12-16 11:37:36 +02:00
ferret/pkg/runtime/expressions/body_test.go
2018-10-28 01:45:26 -04:00

143 lines
3.8 KiB
Go

package expressions_test
import (
"context"
"testing"
"github.com/MontFerret/ferret/pkg/runtime/core"
"github.com/MontFerret/ferret/pkg/runtime/expressions"
"github.com/MontFerret/ferret/pkg/runtime/values"
. "github.com/smartystreets/goconvey/convey"
)
func TestNewBodyExpression(t *testing.T) {
Convey("Should create a block expression", t, func() {
s := expressions.NewBodyExpression(1)
So(s, ShouldHaveSameTypeAs, &expressions.BodyExpression{})
})
}
func TestBlockExpressionAddVariableExpression(t *testing.T) {
Convey("Should add a new expression of a default type", t, func() {
s := expressions.NewBodyExpression(0)
sourceMap := core.NewSourceMap("test", 1, 1)
exp, err := expressions.NewVariableExpression(sourceMap, "testExp")
So(err, ShouldBeNil)
err = s.Add(exp)
So(err, ShouldBeNil)
})
}
func TestBlockExpressionAddReturnExpression(t *testing.T) {
Convey("Should add a new Return expression", t, func() {
s := expressions.NewBodyExpression(0)
sourceMap := core.NewSourceMap("test", 1, 1)
predicate, err := expressions.NewVariableExpression(sourceMap, "testExp")
So(err, ShouldBeNil)
exp, err := expressions.NewReturnExpression(sourceMap, predicate)
So(err, ShouldBeNil)
err = s.Add(exp)
So(err, ShouldBeNil)
})
}
func TestBlockExpressionAddReturnExpressionFailed(t *testing.T) {
Convey("Should not add an already defined Return expression", t, func() {
s := expressions.NewBodyExpression(0)
sourceMap := core.NewSourceMap("test", 1, 1)
predicate, err := expressions.NewVariableExpression(sourceMap, "testExp")
So(err, ShouldBeNil)
exp, err := expressions.NewReturnExpression(sourceMap, predicate)
So(err, ShouldBeNil)
err = s.Add(exp)
So(err, ShouldBeNil)
err = s.Add(exp)
So(err, ShouldBeError)
So(err.Error(), ShouldEqual, "invalid operation: return expression is already defined")
})
}
func TestBlockExpressionExec(t *testing.T) {
Convey("Should exec a block expression", t, func() {
s := expressions.NewBodyExpression(1)
sourceMap := core.NewSourceMap("test", 1, 1)
predicate, err := expressions.NewVariableExpression(sourceMap, "test")
So(err, ShouldBeNil)
retexp, err := expressions.NewReturnExpression(sourceMap, predicate)
So(err, ShouldBeNil)
err = s.Add(retexp)
So(err, ShouldBeNil)
rootScope, fn := core.NewRootScope()
scope := rootScope.Fork()
scope.SetVariable("test", values.NewString("value"))
fn()
value, err := s.Exec(context.Background(), scope)
So(err, ShouldBeNil)
So(value, ShouldNotBeNil)
So(value, ShouldEqual, "value")
})
}
func TestBlockExpressionExecNonFound(t *testing.T) {
Convey("Should not found a missing statement", t, func() {
s := expressions.NewBodyExpression(1)
sourceMap := core.NewSourceMap("test", 1, 1)
predicate, err := expressions.NewVariableExpression(sourceMap, "testExp")
So(err, ShouldBeNil)
exp, err := expressions.NewReturnExpression(sourceMap, predicate)
So(err, ShouldBeNil)
err = s.Add(exp)
So(err, ShouldBeNil)
rootScope, fn := core.NewRootScope()
scope := rootScope.Fork()
scope.SetVariable("test", values.NewString("value"))
fn()
value, err := s.Exec(context.Background(), scope)
So(err, ShouldNotBeNil)
So(err, ShouldHaveSameTypeAs, core.ErrNotFound)
So(value, ShouldHaveSameTypeAs, values.None)
})
}
func TestBlockExpressionExecNilExpression(t *testing.T) {
Convey("Should not exec a nil block expression", t, func() {
s := expressions.NewBodyExpression(1)
sourceMap := core.NewSourceMap("test", 1, 1)
exp, err := expressions.NewVariableExpression(sourceMap, "test")
So(err, ShouldBeNil)
err = s.Add(exp)
So(err, ShouldBeNil)
rootScope, fn := core.NewRootScope()
scope := rootScope.Fork()
scope.SetVariable("test", values.NewString("value"))
fn()
value, err := s.Exec(context.Background(), scope)
So(err, ShouldBeNil)
So(value, ShouldHaveSameTypeAs, values.None)
})
}