add unit test to ensure useLoadData handles thenable objects as promises

Optum · Jan 10, 2025 · 6aaa5bd · 6aaa5bd
1 parent 01d1dc4
commit 6aaa5bd
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/hooks/useLoadData/useLoadData.test.ts b/hooks/useLoadData/useLoadData.test.ts
@@ -442,4 +442,13 @@ describe('useLoadData', () => {
     expect(getSuccess).toHaveBeenCalledTimes(2);
     expect(getSuccess).toHaveBeenCalledWith('b');
   });
+
+  it('should treat a thenable object as a Promise', async () => {
+    const getThenableSuccess = jest.fn(() => ({then: (resolve: any) => resolve(successResult)}));
+
+    const {result} = renderHook(() => useLoadData(getThenableSuccess));
+    expect(result.current.isInProgress).toBe(true);
+    await waitFor(() => expect(result.current.isInProgress).toBe(false));
+    expect(result.current.result).toBe(successResult);
+  });
 });