#117 Added support for multiline strings (#129)

pkg/compiler/compiler_param_test.go

@@ -0,0 +1,65 @@
+package compiler_test
+
+import (
+	"context"
+	"github.com/MontFerret/ferret/pkg/compiler"
+	"github.com/MontFerret/ferret/pkg/runtime"
+	. "github.com/smartystreets/goconvey/convey"
+	"testing"
+)
+
+func TestParam(t *testing.T) {
+	Convey("Should be possible to use as a return value", t, func() {
+		out := compiler.New().
+			MustCompile(`
+			RETURN @param
+		`).
+			MustRun(context.Background(), runtime.WithParam("param", "foobar"))
+
+		So(string(out), ShouldEqual, `"foobar"`)
+	})
+
+	Convey("Should be possible to use as a FOR source", t, func() {
+		out := compiler.New().
+			MustCompile(`
+			FOR i IN @values
+			SORT i
+			RETURN i
+		`).
+			MustRun(context.Background(), runtime.WithParam("values", []int{1, 2, 3, 4}))
+
+		So(string(out), ShouldEqual, `[1,2,3,4]`)
+
+		out2 := compiler.New().
+			MustCompile(`
+			FOR i IN @values
+			SORT i
+			RETURN i
+		`).
+			MustRun(context.Background(), runtime.WithParam("values", map[string]int{
+				"foo": 1,
+				"bar": 2,
+				"faz": 3,
+				"qaz": 4,
+			}))
+
+		So(string(out2), ShouldEqual, `[1,2,3,4]`)
+	})
+
+	Convey("Should be possible to use in range", t, func() {
+		out := compiler.New().
+			MustCompile(`
+			FOR i IN @start..@end
+			SORT i
+			RETURN i
+		`).
+			MustRun(
+				context.Background(),
+				runtime.WithParam("start", 1),
+				runtime.WithParam("end", 4),
+			)
+
+		So(string(out), ShouldEqual, `[1,2,3,4]`)
+
+	})
+}