datajunction-ui 0.0.96 → 0.0.98

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "datajunction-ui",
-  "version": "0.0.96",
+  "version": "0.0.98",
   "description": "DataJunction UI",
   "module": "src/index.tsx",
   "repository": {

package/src/app/services/DJService.js

@@ -1305,6 +1305,28 @@ export const DataJunctionAPI = {
     if (results.state === 'FAILED') {
       throw new Error(results.errors?.[0] || 'Query execution failed');
     }
+
+    // If FINISHED but results are empty, re-submit to /data/ a few times
+    // before giving up (mirrors the Python client's behavior).
+    if (!results.results?.length || !results.results[0]?.rows?.length) {
+      for (let attempt = 0; attempt < 3; attempt++) {
+        await sleep(2 ** attempt * 1000);
+        const retryResponse = await fetch(`${DJ_URL}/data/?${params}`, {
+          credentials: 'include',
+        });
+        if (retryResponse.ok) {
+          const retryResults = await retryResponse.json();
+          if (
+            retryResults.results?.length &&
+            retryResults.results[0]?.rows?.length
+          ) {
+            results = retryResults;
+            break;
+          }
+        }
+      }
+    }
+
     return results;
   },
 

package/src/app/services/__tests__/DJService.test.jsx

@@ -619,7 +619,12 @@ describe('DataJunctionAPI', () => {
   it('calls data correctly', async () => {
     const metricSelection = ['metric1'];
     const dimensionSelection = ['dimension1'];
-    fetch.mockResponseOnce(JSON.stringify({ state: 'FINISHED', results: [] }));
+    fetch.mockResponseOnce(
+      JSON.stringify({
+        state: 'FINISHED',
+        results: [{ rows: [['val']] }],
+      }),
+    );
     await DataJunctionAPI.data(metricSelection, dimensionSelection);
     expect(fetch).toHaveBeenCalledWith(
       expect.stringContaining(`${DJ_URL}/data/?`),
@@ -3004,7 +3009,11 @@ describe('DataJunctionAPI', () => {
   it('calls data with filters array', async () => {
     fetch.mockResolvedValueOnce({
       ok: true,
-      json: () => Promise.resolve({ state: 'FINISHED', results: [] }),
+      json: () =>
+        Promise.resolve({
+          state: 'FINISHED',
+          results: [{ rows: [['val']] }],
+        }),
     });
     await DataJunctionAPI.data(
       ['metric1'],