Added additional unit tests

MontFerret · Mar 26, 2024 · 72bbfcf · 72bbfcf
1 parent 9d89d9e
commit 72bbfcf
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/pkg/compiler/compiler_for_test.go b/pkg/compiler/compiler_for_test.go
@@ -86,6 +86,20 @@ func TestFor(t *testing.T) {
 			[]any{map[string]any{"a": 1}, map[string]any{"a": 2}, map[string]any{"a": 3}},
 			ShouldEqualJSON,
 		},
+		{
+			`FOR val IN 1..3
+							FOR prop IN ["a"]
+								RETURN {[prop]: val}`,
+			[]any{map[string]any{"a": 1}, map[string]any{"a": 2}, map[string]any{"a": 3}},
+			ShouldEqualJSON,
+		},
+		{
+			`FOR prop IN ["a"]
+							FOR val IN 1..3
+								RETURN {[prop]: val}`,
+			[]any{map[string]any{"a": 1}, map[string]any{"a": 2}, map[string]any{"a": 3}},
+			ShouldEqualJSON,
+		},
 		{
 			`FOR prop IN ["a"]
 							FOR val IN [1, 2, 3]