Function TestVM_GroupAndSortOperations

vm/vm_test.go:438–514

(t *testing.T)

}

func TestVM_GroupAndSortOperations(t *testing.T) {
	tests := []struct {
		name        string
		expr        string
		env         map[string]any
		want        any
		expectError string
	}{
		{
			name: "group by single field",
			expr: `groupBy([{"id": 1, "type": "a"}, {"id": 2, "type": "b"}, {"id": 3, "type": "a"}], #.type)`,
			want: map[any][]any{
				"a": {
					map[string]any{"id": 1, "type": "a"},
					map[string]any{"id": 3, "type": "a"},
				},
				"b": {
					map[string]any{"id": 2, "type": "b"},
				},
			},
		},
		{
			name: "sort by field ascending",
			expr: `sortBy([{"id": 3}, {"id": 1}, {"id": 2}], #.id)`,
			want: []any{
				map[string]any{"id": 1},
				map[string]any{"id": 2},
				map[string]any{"id": 3},
			},
		},
		{
			name: "sort by field descending",
			expr: `sortBy([{"id": 3}, {"id": 1}, {"id": 2}], #.id, "desc")`,
			want: []any{
				map[string]any{"id": 3},
				map[string]any{"id": 2},
				map[string]any{"id": 1},
			},
		},
		{
			name: "sort by computed value",
			expr: `sortBy([1, 2, 3, 4], # % 2)`,
			want: []any{2, 4, 1, 3},
		},
		{
			name: "group by with complex key",
			expr: `groupBy([1, 2, 3, 4, 5, 6], # % 2 == 0 ? "even" : "odd")`,
			want: map[any][]any{
				"even": {2, 4, 6},
				"odd":  {1, 3, 5},
			},
		},
		{
			name:        "invalid sort order",
			expr:        `sortBy([1, 2, 3], #, "invalid")`,
			expectError: "unknown order",
		},
	}