test: Add tests for grouping by undefined value (sourcenetwork#543)

Add tests for grouping by undefined value
shahzadlone · Jun 21, 2022 · db62dae · db62dae
1 parent 6ac9428
commit db62dae
Showing 1 changed file with 86 additions and 0 deletions.
diff --git a/tests/integration/query/simple/with_group_test.go b/tests/integration/query/simple/with_group_test.go
@@ -457,3 +457,89 @@ func TestQuerySimpleWithGroupByBooleanThenNumber(t *testing.T) {
 
 	executeTestCase(t, test)
 }
+
+func TestQuerySimpleWithGroupByNumberOnUndefined(t *testing.T) {
+	test := testUtils.QueryTestCase{
+		Description: "Simple query with group by number, no children, undefined group value",
+		Query: `query {
+					users(groupBy: [Age]) {
+						Age
+					}
+				}`,
+		Docs: map[int][]string{
+			0: {
+				`{
+					"Name": "John",
+					"Age": 32
+				}`,
+				`{
+					"Name": "Bob"
+				}`,
+				`{
+					"Name": "Alice"
+				}`,
+			},
+		},
+		Results: []map[string]interface{}{
+			{
+				"Age": nil,
+			},
+			{
+				"Age": uint64(32),
+			},
+		},
+	}
+
+	executeTestCase(t, test)
+}
+
+func TestQuerySimpleWithGroupByNumberOnUndefinedWithChildren(t *testing.T) {
+	test := testUtils.QueryTestCase{
+		Description: "Simple query with group by number, with children, undefined group value",
+		Query: `query {
+					users(groupBy: [Age]) {
+						Age
+						_group {
+							Name
+						}
+					}
+				}`,
+		Docs: map[int][]string{
+			0: {
+				`{
+					"Name": "John",
+					"Age": 32
+				}`,
+				`{
+					"Name": "Bob"
+				}`,
+				`{
+					"Name": "Alice"
+				}`,
+			},
+		},
+		Results: []map[string]interface{}{
+			{
+				"Age": nil,
+				"_group": []map[string]interface{}{
+					{
+						"Name": "Bob",
+					},
+					{
+						"Name": "Alice",
+					},
+				},
+			},
+			{
+				"Age": uint64(32),
+				"_group": []map[string]interface{}{
+					{
+						"Name": "John",
+					},
+				},
+			},
+		},
+	}
+
+	executeTestCase(t, test)
+}