@indigoai-us/hq-cloud 6.18.13 → 6.18.14-alpha.0

package/dist/cli/sync.test.js

@@ -352,6 +352,46 @@ describe("sync", () => {
             }),
         }));
     });
+    it("uses a bounded directory listing while planning only the named event paths", async () => {
+        vi.mocked(targetedUpsertExactListModule.isTargetedUpsertExactListEnabled)
+            .mockResolvedValue(true);
+        vi.mocked(remotePullModule.listRemoteForScope).mockResolvedValue([
+            {
+                key: "docs/burst/0000.md",
+                size: 20,
+                lastModified: new Date(),
+                etag: '"targeted"',
+            },
+            {
+                key: "docs/burst/unannounced.md",
+                size: 22,
+                lastModified: new Date(),
+                etag: '"sibling"',
+            },
+        ]);
+        const pullPrefixes = Array.from({ length: 16 }, (_, index) => `docs/burst/${String(index).padStart(4, "0")}.md`);
+        const result = await sync({
+            company: "acme",
+            vaultConfig: mockConfig,
+            hqRoot: tmpDir,
+            syncMode: "custom",
+            prefixSet: ["docs/"],
+            pullPrefixes,
+            pullExactPaths: pullPrefixes,
+            skipReindex: true,
+        });
+        expect(result.filesDownloaded).toBe(1);
+        expect(fs.existsSync(path.join(tmpDir, "companies", "acme", "docs", "burst", "0000.md")))
+            .toBe(true);
+        expect(fs.existsSync(path.join(tmpDir, "companies", "acme", "docs", "burst", "unannounced.md")))
+            .toBe(false);
+        expect(remotePullModule.listRemoteForScope).toHaveBeenCalledWith(expect.objectContaining({
+            scope: expect.objectContaining({
+                prefixSet: ["docs/burst/"],
+                strategy: "vend-fanout",
+            }),
+        }));
+    });
     it("does not wait for a never-settling flag read and keeps the pull on legacy scope", async () => {
         let markFlagFetchStarted;
         const flagFetchStarted = new Promise((resolve) => {
@@ -1652,6 +1692,7 @@ describe("sync", () => {
             changeMessageVisibilityBatch: async () => ({ failedEntries: [] }),
         };
         const results = [];
+        const receiverBatchSizes = [];
         const receiver = new SqsPushReceiver({
             tenantId: "tenant-acme",
             queueUrl: "https://queue.test/targeted-pull",
@@ -1660,6 +1701,7 @@ describe("sync", () => {
             drainMaxMessages: 256,
             publishMetric: async () => undefined,
             syncBatchFn: async ({ events }) => {
+                receiverBatchSizes.push(events.length);
                 const argv = buildTargetedPullArgv({ kind: "personal" }, [], events.map((event) => event.relativePath));
                 results.push(await sync({
                     company: "acme",
@@ -1672,8 +1714,9 @@ describe("sync", () => {
             },
         });
         await receiver.start();
-        await vi.waitFor(() => expect(deleted).toHaveLength(512));
+        await vi.waitFor(() => expect(deleted).toHaveLength(512), { timeout: 10_000 });
         await receiver.dispose();
+        expect(receiverBatchSizes).toEqual([256, 256]);
         expect(results.map((result) => result.filesDownloaded)).toEqual([1, 1]);
         expect(fs.existsSync(path.join(tmpDir, personalKey))).toBe(true);
         expect(fs.existsSync(path.join(tmpDir, threadKey))).toBe(true);