@dxos/echo-pipeline 0.8.3 → 0.8.4-main.1068cf700f

package/src/query/query-planner.test.ts

@@ -1,15 +1,19 @@
+//
+// Copyright 2025 example.com
+//
+
 //
 // Copyright 2025 DXOS.org
 //
 
 import { describe, expect, test } from 'vitest';
 
-import { Filter, Query } from '@dxos/echo';
+import { Filter, Order, Query } from '@dxos/echo';
+import { TestSchema } from '@dxos/echo/testing';
 import { type QueryAST } from '@dxos/echo-protocol';
-import { SpaceId } from '@dxos/keys';
+import { DXN, SpaceId } from '@dxos/keys';
 
 import { QueryPlanner } from './query-planner';
-import { TestSchema } from '../testing';
 
 describe('QueryPlanner', () => {
   const planner = new QueryPlanner();
@@ -23,11 +27,13 @@ describe('QueryPlanner', () => {
         "steps": [
           {
             "_tag": "SelectStep",
+            "allQueuesFromSpaces": false,
+            "queues": [],
             "selector": {
               "_tag": "TypeSelector",
               "inverted": false,
               "typename": [
-                "dxn:type:dxos.org/type/Person:0.1.0",
+                "dxn:type:example.com/type/Person:0.1.0",
               ],
             },
             "spaces": [
@@ -38,6 +44,73 @@ describe('QueryPlanner', () => {
             "_tag": "FilterDeletedStep",
             "mode": "only-non-deleted",
           },
+          {
+            "_tag": "FilterStep",
+            "filter": {
+              "id": undefined,
+              "props": {},
+              "type": "object",
+              "typename": "dxn:type:example.com/type/Person:0.1.0",
+            },
+          },
+          {
+            "_tag": "OrderStep",
+            "order": [
+              {
+                "kind": "natural",
+              },
+            ],
+          },
+        ],
+      }
+    `);
+  });
+
+  test('get all people ordered by name', () => {
+    const query = Query.select(Filter.type(TestSchema.Person)).orderBy(Order.property('name', 'asc'));
+
+    const plan = planner.createPlan(withSpaceIdOptions(query.ast));
+    expect(plan).toMatchInlineSnapshot(`
+      {
+        "steps": [
+          {
+            "_tag": "SelectStep",
+            "allQueuesFromSpaces": false,
+            "queues": [],
+            "selector": {
+              "_tag": "TypeSelector",
+              "inverted": false,
+              "typename": [
+                "dxn:type:example.com/type/Person:0.1.0",
+              ],
+            },
+            "spaces": [
+              "B2NJDFNVZIW77OQSXUBNAD7BUMBD3G5PO",
+            ],
+          },
+          {
+            "_tag": "FilterDeletedStep",
+            "mode": "only-non-deleted",
+          },
+          {
+            "_tag": "FilterStep",
+            "filter": {
+              "id": undefined,
+              "props": {},
+              "type": "object",
+              "typename": "dxn:type:example.com/type/Person:0.1.0",
+            },
+          },
+          {
+            "_tag": "OrderStep",
+            "order": [
+              {
+                "direction": "asc",
+                "kind": "property",
+                "property": "name",
+              },
+            ],
+          },
         ],
       }
     `);
@@ -52,11 +125,13 @@ describe('QueryPlanner', () => {
         "steps": [
           {
             "_tag": "SelectStep",
+            "allQueuesFromSpaces": false,
+            "queues": [],
             "selector": {
               "_tag": "TypeSelector",
               "inverted": false,
               "typename": [
-                "dxn:type:dxos.org/type/Person:0.1.0",
+                "dxn:type:example.com/type/Person:0.1.0",
               ],
             },
             "spaces": [
@@ -79,9 +154,17 @@ describe('QueryPlanner', () => {
                 },
               },
               "type": "object",
-              "typename": null,
+              "typename": "dxn:type:example.com/type/Person:0.1.0",
             },
           },
+          {
+            "_tag": "OrderStep",
+            "order": [
+              {
+                "kind": "natural",
+              },
+            ],
+          },
         ],
       }
     `);
@@ -89,7 +172,9 @@ describe('QueryPlanner', () => {
 
   test('get all orgs Fred worked for since 2020', () => {
     const query = Query.select(Filter.type(TestSchema.Person, { id: '01JVS9YYT5VMVJW0GGTM1YHCCH' }))
-      .sourceOf(TestSchema.WorksFor, { since: Filter.gt('2020') })
+      .sourceOf(TestSchema.EmployedBy, {
+        since: Filter.gt<string | undefined>('2020'),
+      })
       .target();
 
     const plan = planner.createPlan(withSpaceIdOptions(query.ast));
@@ -98,6 +183,8 @@ describe('QueryPlanner', () => {
         "steps": [
           {
             "_tag": "SelectStep",
+            "allQueuesFromSpaces": false,
+            "queues": [],
             "selector": {
               "_tag": "IdSelector",
               "objectIds": [
@@ -118,7 +205,7 @@ describe('QueryPlanner', () => {
               "id": undefined,
               "props": {},
               "type": "object",
-              "typename": "dxn:type:dxos.org/type/Person:0.1.0",
+              "typename": "dxn:type:example.com/type/Person:0.1.0",
             },
           },
           {
@@ -144,7 +231,7 @@ describe('QueryPlanner', () => {
                 },
               },
               "type": "object",
-              "typename": "dxn:type:dxos.org/type/WorksFor:0.1.0",
+              "typename": "dxn:type:example.com/type/EmployedBy:0.1.0",
             },
           },
           {
@@ -158,6 +245,14 @@ describe('QueryPlanner', () => {
             "_tag": "FilterDeletedStep",
             "mode": "only-non-deleted",
           },
+          {
+            "_tag": "OrderStep",
+            "order": [
+              {
+                "kind": "natural",
+              },
+            ],
+          },
         ],
       }
     `);
@@ -175,6 +270,8 @@ describe('QueryPlanner', () => {
         "steps": [
           {
             "_tag": "SelectStep",
+            "allQueuesFromSpaces": false,
+            "queues": [],
             "selector": {
               "_tag": "IdSelector",
               "objectIds": [
@@ -195,7 +292,7 @@ describe('QueryPlanner', () => {
               "id": undefined,
               "props": {},
               "type": "object",
-              "typename": "dxn:type:dxos.org/type/Person:0.1.0",
+              "typename": "dxn:type:example.com/type/Person:0.1.0",
             },
           },
           {
@@ -215,9 +312,17 @@ describe('QueryPlanner', () => {
             "filter": {
               "props": {},
               "type": "object",
-              "typename": "dxn:type:dxos.org/type/Task:0.1.0",
+              "typename": "dxn:type:example.com/type/Task:0.1.0",
             },
           },
+          {
+            "_tag": "OrderStep",
+            "order": [
+              {
+                "kind": "natural",
+              },
+            ],
+          },
         ],
       }
     `);
@@ -225,7 +330,7 @@ describe('QueryPlanner', () => {
 
   test('get all tasks for employees of Cyberdyne', () => {
     const query = Query.select(Filter.type(TestSchema.Organization, { name: 'Cyberdyne' }))
-      .targetOf(TestSchema.WorksFor)
+      .targetOf(TestSchema.EmployedBy)
       .source()
       .referencedBy(TestSchema.Task, 'assignee');
 
@@ -235,11 +340,13 @@ describe('QueryPlanner', () => {
         "steps": [
           {
             "_tag": "SelectStep",
+            "allQueuesFromSpaces": false,
+            "queues": [],
             "selector": {
               "_tag": "TypeSelector",
               "inverted": false,
               "typename": [
-                "dxn:type:dxos.org/type/Organization:0.1.0",
+                "dxn:type:example.com/type/Organization:0.1.0",
               ],
             },
             "spaces": [
@@ -262,7 +369,7 @@ describe('QueryPlanner', () => {
                 },
               },
               "type": "object",
-              "typename": null,
+              "typename": "dxn:type:example.com/type/Organization:0.1.0",
             },
           },
           {
@@ -282,7 +389,7 @@ describe('QueryPlanner', () => {
               "id": undefined,
               "props": {},
               "type": "object",
-              "typename": "dxn:type:dxos.org/type/WorksFor:0.1.0",
+              "typename": "dxn:type:example.com/type/EmployedBy:0.1.0",
             },
           },
           {
@@ -313,9 +420,17 @@ describe('QueryPlanner', () => {
             "filter": {
               "props": {},
               "type": "object",
-              "typename": "dxn:type:dxos.org/type/Task:0.1.0",
+              "typename": "dxn:type:example.com/type/Task:0.1.0",
             },
           },
+          {
+            "_tag": "OrderStep",
+            "order": [
+              {
+                "kind": "natural",
+              },
+            ],
+          },
         ],
       }
     `);
@@ -338,11 +453,13 @@ describe('QueryPlanner', () => {
                 "steps": [
                   {
                     "_tag": "SelectStep",
+                    "allQueuesFromSpaces": false,
+                    "queues": [],
                     "selector": {
                       "_tag": "TypeSelector",
                       "inverted": false,
                       "typename": [
-                        "dxn:type:dxos.org/type/Person:0.1.0",
+                        "dxn:type:example.com/type/Person:0.1.0",
                       ],
                     },
                     "spaces": [
@@ -353,17 +470,36 @@ describe('QueryPlanner', () => {
                     "_tag": "FilterDeletedStep",
                     "mode": "only-non-deleted",
                   },
+                  {
+                    "_tag": "FilterStep",
+                    "filter": {
+                      "id": undefined,
+                      "props": {},
+                      "type": "object",
+                      "typename": "dxn:type:example.com/type/Person:0.1.0",
+                    },
+                  },
+                  {
+                    "_tag": "OrderStep",
+                    "order": [
+                      {
+                        "kind": "natural",
+                      },
+                    ],
+                  },
                 ],
               },
               {
                 "steps": [
                   {
                     "_tag": "SelectStep",
+                    "allQueuesFromSpaces": false,
+                    "queues": [],
                     "selector": {
                       "_tag": "TypeSelector",
                       "inverted": false,
                       "typename": [
-                        "dxn:type:dxos.org/type/Organization:0.1.0",
+                        "dxn:type:example.com/type/Organization:0.1.0",
                       ],
                     },
                     "spaces": [
@@ -374,10 +510,35 @@ describe('QueryPlanner', () => {
                     "_tag": "FilterDeletedStep",
                     "mode": "only-non-deleted",
                   },
+                  {
+                    "_tag": "FilterStep",
+                    "filter": {
+                      "id": undefined,
+                      "props": {},
+                      "type": "object",
+                      "typename": "dxn:type:example.com/type/Organization:0.1.0",
+                    },
+                  },
+                  {
+                    "_tag": "OrderStep",
+                    "order": [
+                      {
+                        "kind": "natural",
+                      },
+                    ],
+                  },
                 ],
               },
             ],
           },
+          {
+            "_tag": "OrderStep",
+            "order": [
+              {
+                "kind": "natural",
+              },
+            ],
+          },
         ],
       }
     `);
@@ -386,7 +547,7 @@ describe('QueryPlanner', () => {
   test('get all people not in orgs', () => {
     const query = Query.without(
       Query.select(Filter.type(TestSchema.Person)),
-      Query.select(Filter.type(TestSchema.Person)).sourceOf(TestSchema.WorksFor).source(),
+      Query.select(Filter.type(TestSchema.Person)).sourceOf(TestSchema.EmployedBy).source(),
     );
 
     const plan = planner.createPlan(withSpaceIdOptions(query.ast));
@@ -399,11 +560,13 @@ describe('QueryPlanner', () => {
               "steps": [
                 {
                   "_tag": "SelectStep",
+                  "allQueuesFromSpaces": false,
+                  "queues": [],
                   "selector": {
                     "_tag": "TypeSelector",
                     "inverted": false,
                     "typename": [
-                      "dxn:type:dxos.org/type/Person:0.1.0",
+                      "dxn:type:example.com/type/Person:0.1.0",
                     ],
                   },
                   "spaces": [
@@ -420,7 +583,7 @@ describe('QueryPlanner', () => {
                     "id": undefined,
                     "props": {},
                     "type": "object",
-                    "typename": null,
+                    "typename": "dxn:type:example.com/type/Person:0.1.0",
                   },
                 },
                 {
@@ -440,7 +603,7 @@ describe('QueryPlanner', () => {
                     "id": undefined,
                     "props": {},
                     "type": "object",
-                    "typename": "dxn:type:dxos.org/type/WorksFor:0.1.0",
+                    "typename": "dxn:type:example.com/type/EmployedBy:0.1.0",
                   },
                 },
                 {
@@ -454,17 +617,27 @@ describe('QueryPlanner', () => {
                   "_tag": "FilterDeletedStep",
                   "mode": "only-non-deleted",
                 },
+                {
+                  "_tag": "OrderStep",
+                  "order": [
+                    {
+                      "kind": "natural",
+                    },
+                  ],
+                },
               ],
             },
             "source": {
               "steps": [
                 {
                   "_tag": "SelectStep",
+                  "allQueuesFromSpaces": false,
+                  "queues": [],
                   "selector": {
                     "_tag": "TypeSelector",
                     "inverted": false,
                     "typename": [
-                      "dxn:type:dxos.org/type/Person:0.1.0",
+                      "dxn:type:example.com/type/Person:0.1.0",
                     ],
                   },
                   "spaces": [
@@ -481,19 +654,39 @@ describe('QueryPlanner', () => {
                     "id": undefined,
                     "props": {},
                     "type": "object",
-                    "typename": null,
+                    "typename": "dxn:type:example.com/type/Person:0.1.0",
                   },
                 },
+                {
+                  "_tag": "OrderStep",
+                  "order": [
+                    {
+                      "kind": "natural",
+                    },
+                  ],
+                },
               ],
             },
           },
+          {
+            "_tag": "OrderStep",
+            "order": [
+              {
+                "kind": "natural",
+              },
+            ],
+          },
         ],
       }
     `);
   });
 
   test('get assignees of all tasks created after 2020', () => {
-    const query = Query.select(Filter.type(TestSchema.Task, { createdAt: Filter.gt('2020') })).reference('assignee');
+    const query = Query.select(
+      Filter.type(TestSchema.Task, {
+        deadline: Filter.gt<string | undefined>('2020'),
+      }),
+    ).reference('assignee');
 
     const plan = planner.createPlan(withSpaceIdOptions(query.ast));
     expect(plan).toMatchInlineSnapshot(`
@@ -501,11 +694,13 @@ describe('QueryPlanner', () => {
         "steps": [
           {
             "_tag": "SelectStep",
+            "allQueuesFromSpaces": false,
+            "queues": [],
             "selector": {
               "_tag": "TypeSelector",
               "inverted": false,
               "typename": [
-                "dxn:type:dxos.org/type/Task:0.1.0",
+                "dxn:type:example.com/type/Task:0.1.0",
               ],
             },
             "spaces": [
@@ -521,14 +716,14 @@ describe('QueryPlanner', () => {
             "filter": {
               "id": undefined,
               "props": {
-                "createdAt": {
+                "deadline": {
                   "operator": "gt",
                   "type": "compare",
                   "value": "2020",
                 },
               },
               "type": "object",
-              "typename": null,
+              "typename": "dxn:type:example.com/type/Task:0.1.0",
             },
           },
           {
@@ -543,6 +738,14 @@ describe('QueryPlanner', () => {
             "_tag": "FilterDeletedStep",
             "mode": "only-non-deleted",
           },
+          {
+            "_tag": "OrderStep",
+            "order": [
+              {
+                "kind": "natural",
+              },
+            ],
+          },
         ],
       }
     `);
@@ -557,10 +760,13 @@ describe('QueryPlanner', () => {
         "steps": [
           {
             "_tag": "SelectStep",
+            "allQueuesFromSpaces": false,
+            "queues": [],
             "selector": {
               "_tag": "TextSelector",
               "searchKind": "full-text",
               "text": "Bill",
+              "typename": null,
             },
             "spaces": [
               "B2NJDFNVZIW77OQSXUBNAD7BUMBD3G5PO",
@@ -576,9 +782,17 @@ describe('QueryPlanner', () => {
               "id": undefined,
               "props": {},
               "type": "object",
-              "typename": "dxn:type:dxos.org/type/Person:0.1.0",
+              "typename": "dxn:type:example.com/type/Person:0.1.0",
             },
           },
+          {
+            "_tag": "OrderStep",
+            "order": [
+              {
+                "kind": "natural",
+              },
+            ],
+          },
         ],
       }
     `);
@@ -593,10 +807,13 @@ describe('QueryPlanner', () => {
         "steps": [
           {
             "_tag": "SelectStep",
+            "allQueuesFromSpaces": false,
+            "queues": [],
             "selector": {
               "_tag": "TextSelector",
               "searchKind": "vector",
               "text": "Bill",
+              "typename": null,
             },
             "spaces": [
               "B2NJDFNVZIW77OQSXUBNAD7BUMBD3G5PO",
@@ -606,6 +823,14 @@ describe('QueryPlanner', () => {
             "_tag": "FilterDeletedStep",
             "mode": "only-non-deleted",
           },
+          {
+            "_tag": "OrderStep",
+            "order": [
+              {
+                "kind": "natural",
+              },
+            ],
+          },
         ],
       }
     `);
@@ -620,12 +845,14 @@ describe('QueryPlanner', () => {
         "steps": [
           {
             "_tag": "SelectStep",
+            "allQueuesFromSpaces": false,
+            "queues": [],
             "selector": {
               "_tag": "TypeSelector",
               "inverted": false,
               "typename": [
-                "dxn:type:dxos.org/type/Organization:0.1.0",
-                "dxn:type:dxos.org/type/Person:0.1.0",
+                "dxn:type:example.com/type/Organization:0.1.0",
+                "dxn:type:example.com/type/Person:0.1.0",
               ],
             },
             "spaces": [
@@ -636,6 +863,34 @@ describe('QueryPlanner', () => {
             "_tag": "FilterDeletedStep",
             "mode": "only-non-deleted",
           },
+          {
+            "_tag": "FilterStep",
+            "filter": {
+              "filters": [
+                {
+                  "id": undefined,
+                  "props": {},
+                  "type": "object",
+                  "typename": "dxn:type:example.com/type/Organization:0.1.0",
+                },
+                {
+                  "id": undefined,
+                  "props": {},
+                  "type": "object",
+                  "typename": "dxn:type:example.com/type/Person:0.1.0",
+                },
+              ],
+              "type": "or",
+            },
+          },
+          {
+            "_tag": "OrderStep",
+            "order": [
+              {
+                "kind": "natural",
+              },
+            ],
+          },
         ],
       }
     `);
@@ -660,12 +915,14 @@ describe('QueryPlanner', () => {
         "steps": [
           {
             "_tag": "SelectStep",
+            "allQueuesFromSpaces": false,
+            "queues": [],
             "selector": {
               "_tag": "TypeSelector",
               "inverted": true,
               "typename": [
-                "dxn:type:dxos.org/type/Organization:0.1.0",
-                "dxn:type:dxos.org/type/Person:0.1.0",
+                "dxn:type:example.com/type/Organization:0.1.0",
+                "dxn:type:example.com/type/Person:0.1.0",
               ],
             },
             "spaces": [
@@ -676,6 +933,34 @@ describe('QueryPlanner', () => {
             "_tag": "FilterDeletedStep",
             "mode": "only-non-deleted",
           },
+          {
+            "_tag": "FilterStep",
+            "filter": {
+              "filters": [
+                {
+                  "id": undefined,
+                  "props": {},
+                  "type": "object",
+                  "typename": "dxn:type:example.com/type/Organization:0.1.0",
+                },
+                {
+                  "id": undefined,
+                  "props": {},
+                  "type": "object",
+                  "typename": "dxn:type:example.com/type/Person:0.1.0",
+                },
+              ],
+              "type": "or",
+            },
+          },
+          {
+            "_tag": "OrderStep",
+            "order": [
+              {
+                "kind": "natural",
+              },
+            ],
+          },
         ],
       }
     `);
@@ -690,11 +975,13 @@ describe('QueryPlanner', () => {
         "steps": [
           {
             "_tag": "SelectStep",
+            "allQueuesFromSpaces": false,
+            "queues": [],
             "selector": {
               "_tag": "TypeSelector",
               "inverted": false,
               "typename": [
-                "dxn:type:dxos.org/type/Task:0.1.0",
+                "dxn:type:example.com/type/Task:0.1.0",
               ],
             },
             "spaces": [
@@ -705,6 +992,336 @@ describe('QueryPlanner', () => {
             "_tag": "FilterDeletedStep",
             "mode": "only-deleted",
           },
+          {
+            "_tag": "FilterStep",
+            "filter": {
+              "id": undefined,
+              "props": {},
+              "type": "object",
+              "typename": "dxn:type:example.com/type/Task:0.1.0",
+            },
+          },
+          {
+            "_tag": "OrderStep",
+            "order": [
+              {
+                "kind": "natural",
+              },
+            ],
+          },
+        ],
+      }
+    `);
+  });
+
+  test('select items from a specific queue', () => {
+    const query = Query.select(Filter.type(TestSchema.Task)).options({ queues: [QUEUE_DXN] });
+
+    const plan = planner.createPlan(query.ast);
+    expect(plan).toMatchInlineSnapshot(`
+      {
+        "steps": [
+          {
+            "_tag": "SelectStep",
+            "allQueuesFromSpaces": false,
+            "queues": [
+              "dxn:queue:data:B2NJDFNVZIW77OQSXUBNAD7BUMBD3G5PO:01JJRA86VK4H1TEB6QQVSWXP0E",
+            ],
+            "selector": {
+              "_tag": "TypeSelector",
+              "inverted": false,
+              "typename": [
+                "dxn:type:example.com/type/Task:0.1.0",
+              ],
+            },
+            "spaces": [],
+          },
+          {
+            "_tag": "FilterDeletedStep",
+            "mode": "only-non-deleted",
+          },
+          {
+            "_tag": "FilterStep",
+            "filter": {
+              "id": undefined,
+              "props": {},
+              "type": "object",
+              "typename": "dxn:type:example.com/type/Task:0.1.0",
+            },
+          },
+          {
+            "_tag": "OrderStep",
+            "order": [
+              {
+                "kind": "natural",
+              },
+            ],
+          },
+        ],
+      }
+    `);
+  });
+
+  test('select items from all queues in a space', () => {
+    const query = Query.select(Filter.type(TestSchema.Task)).options({
+      spaceIds: [SPACE_ID],
+      allQueuesFromSpaces: true,
+    });
+
+    const plan = planner.createPlan(query.ast);
+    expect(plan).toMatchInlineSnapshot(`
+      {
+        "steps": [
+          {
+            "_tag": "SelectStep",
+            "allQueuesFromSpaces": true,
+            "queues": [],
+            "selector": {
+              "_tag": "TypeSelector",
+              "inverted": false,
+              "typename": [
+                "dxn:type:example.com/type/Task:0.1.0",
+              ],
+            },
+            "spaces": [
+              "B2NJDFNVZIW77OQSXUBNAD7BUMBD3G5PO",
+            ],
+          },
+          {
+            "_tag": "FilterDeletedStep",
+            "mode": "only-non-deleted",
+          },
+          {
+            "_tag": "FilterStep",
+            "filter": {
+              "id": undefined,
+              "props": {},
+              "type": "object",
+              "typename": "dxn:type:example.com/type/Task:0.1.0",
+            },
+          },
+          {
+            "_tag": "OrderStep",
+            "order": [
+              {
+                "kind": "natural",
+              },
+            ],
+          },
+        ],
+      }
+    `);
+  });
+
+  test('limit results', () => {
+    const query = Query.select(Filter.type(TestSchema.Task)).limit(10);
+
+    const plan = planner.createPlan(withSpaceIdOptions(query.ast));
+    expect(plan).toMatchInlineSnapshot(`
+      {
+        "steps": [
+          {
+            "_tag": "SelectStep",
+            "allQueuesFromSpaces": false,
+            "limit": 10,
+            "queues": [],
+            "selector": {
+              "_tag": "TypeSelector",
+              "inverted": false,
+              "typename": [
+                "dxn:type:example.com/type/Task:0.1.0",
+              ],
+            },
+            "spaces": [
+              "B2NJDFNVZIW77OQSXUBNAD7BUMBD3G5PO",
+            ],
+          },
+          {
+            "_tag": "FilterDeletedStep",
+            "mode": "only-non-deleted",
+          },
+          {
+            "_tag": "FilterStep",
+            "filter": {
+              "id": undefined,
+              "props": {},
+              "type": "object",
+              "typename": "dxn:type:example.com/type/Task:0.1.0",
+            },
+          },
+          {
+            "_tag": "OrderStep",
+            "limit": 10,
+            "order": [
+              {
+                "kind": "natural",
+              },
+            ],
+          },
+        ],
+      }
+    `);
+  });
+
+  test('ordered and limited results', () => {
+    const query = Query.select(Filter.type(TestSchema.Task)).orderBy(Order.property('title', 'asc')).limit(10);
+
+    const plan = planner.createPlan(withSpaceIdOptions(query.ast));
+    expect(plan).toMatchInlineSnapshot(`
+      {
+        "steps": [
+          {
+            "_tag": "SelectStep",
+            "allQueuesFromSpaces": false,
+            "limit": 10,
+            "queues": [],
+            "selector": {
+              "_tag": "TypeSelector",
+              "inverted": false,
+              "typename": [
+                "dxn:type:example.com/type/Task:0.1.0",
+              ],
+            },
+            "spaces": [
+              "B2NJDFNVZIW77OQSXUBNAD7BUMBD3G5PO",
+            ],
+          },
+          {
+            "_tag": "FilterDeletedStep",
+            "mode": "only-non-deleted",
+          },
+          {
+            "_tag": "FilterStep",
+            "filter": {
+              "id": undefined,
+              "props": {},
+              "type": "object",
+              "typename": "dxn:type:example.com/type/Task:0.1.0",
+            },
+          },
+          {
+            "_tag": "OrderStep",
+            "limit": 10,
+            "order": [
+              {
+                "direction": "asc",
+                "kind": "property",
+                "property": "title",
+              },
+            ],
+          },
+        ],
+      }
+    `);
+  });
+
+  test('union of limited queries', () => {
+    const query = Query.all(
+      Query.select(Filter.type(TestSchema.Person)).limit(5),
+      Query.select(Filter.type(TestSchema.Organization)).limit(5),
+    );
+
+    const plan = planner.createPlan(withSpaceIdOptions(query.ast));
+    expect(plan).toMatchInlineSnapshot(`
+      {
+        "steps": [
+          {
+            "_tag": "UnionStep",
+            "plans": [
+              {
+                "steps": [
+                  {
+                    "_tag": "SelectStep",
+                    "allQueuesFromSpaces": false,
+                    "limit": 5,
+                    "queues": [],
+                    "selector": {
+                      "_tag": "TypeSelector",
+                      "inverted": false,
+                      "typename": [
+                        "dxn:type:example.com/type/Person:0.1.0",
+                      ],
+                    },
+                    "spaces": [
+                      "B2NJDFNVZIW77OQSXUBNAD7BUMBD3G5PO",
+                    ],
+                  },
+                  {
+                    "_tag": "FilterDeletedStep",
+                    "mode": "only-non-deleted",
+                  },
+                  {
+                    "_tag": "FilterStep",
+                    "filter": {
+                      "id": undefined,
+                      "props": {},
+                      "type": "object",
+                      "typename": "dxn:type:example.com/type/Person:0.1.0",
+                    },
+                  },
+                  {
+                    "_tag": "OrderStep",
+                    "limit": 5,
+                    "order": [
+                      {
+                        "kind": "natural",
+                      },
+                    ],
+                  },
+                ],
+              },
+              {
+                "steps": [
+                  {
+                    "_tag": "SelectStep",
+                    "allQueuesFromSpaces": false,
+                    "limit": 5,
+                    "queues": [],
+                    "selector": {
+                      "_tag": "TypeSelector",
+                      "inverted": false,
+                      "typename": [
+                        "dxn:type:example.com/type/Organization:0.1.0",
+                      ],
+                    },
+                    "spaces": [
+                      "B2NJDFNVZIW77OQSXUBNAD7BUMBD3G5PO",
+                    ],
+                  },
+                  {
+                    "_tag": "FilterDeletedStep",
+                    "mode": "only-non-deleted",
+                  },
+                  {
+                    "_tag": "FilterStep",
+                    "filter": {
+                      "id": undefined,
+                      "props": {},
+                      "type": "object",
+                      "typename": "dxn:type:example.com/type/Organization:0.1.0",
+                    },
+                  },
+                  {
+                    "_tag": "OrderStep",
+                    "limit": 5,
+                    "order": [
+                      {
+                        "kind": "natural",
+                      },
+                    ],
+                  },
+                ],
+              },
+            ],
+          },
+          {
+            "_tag": "OrderStep",
+            "order": [
+              {
+                "kind": "natural",
+              },
+            ],
+          },
         ],
       }
     `);
@@ -712,6 +1329,7 @@ describe('QueryPlanner', () => {
 });
 
 const SPACE_ID = SpaceId.make('B2NJDFNVZIW77OQSXUBNAD7BUMBD3G5PO'); // Stable id for inline snapshots.
+const QUEUE_DXN = DXN.parse('dxn:queue:data:B2NJDFNVZIW77OQSXUBNAD7BUMBD3G5PO:01JJRA86VK4H1TEB6QQVSWXP0E').toString(); // Stable queue DXN for inline snapshots.
 
 const withSpaceIdOptions = (query: QueryAST.Query): QueryAST.Query => ({
   type: 'options',