.

Author: mdjastrzebski

src/__tests__/act.test.tsx

@@ -52,32 +52,28 @@ test('should be able to await act when promise rejects', async () => {
 
 test('should restore act environment when callback throws synchronously', async () => {
   const previousEnvironment = getIsReactActEnvironment();
-  
+
   const testError = new Error('Synchronous error in act');
-  
-  try {
-    await act(() => {
+
+  await expect(
+    act(() => {
       throw testError;
-    });
-    // Should not reach here
-    expect(true).toBe(false);
-  } catch (error) {
-    expect(error).toBe(testError);
-  }
-  
+    }),
+  ).rejects.toBe(testError);
+
   // Verify the act environment was restored even after error
   expect(getIsReactActEnvironment()).toBe(previousEnvironment);
 });
 
 test('should restore act environment when callback returns non-promise value', async () => {
   const previousEnvironment = getIsReactActEnvironment();
-  
+
   // Call act with a callback that returns a non-promise value
   // This tests the else branch in withGlobalActEnvironment
   const result = await act(() => {
     return 42;
   });
-  
+
   expect(result).toBe(42);
   // Verify the act environment was restored
   expect(getIsReactActEnvironment()).toBe(previousEnvironment);

src/__tests__/wait-for.test.tsx

@@ -337,13 +337,13 @@ test.each([false, true])(
 test('waitFor throws clear error when switching from real timers to fake timers', async () => {
   jest.useRealTimers();
 
-    const waitForPromise = waitFor(() => {
-      // Switch to fake timers during waitFor - this should trigger an error
-      jest.useFakeTimers();
-      throw new Error('test');
-    });
+  const waitForPromise = waitFor(() => {
+    // Switch to fake timers during waitFor - this should trigger an error
+    jest.useFakeTimers();
+    throw new Error('test');
+  });
 
-    await expect(waitForPromise).rejects.toThrow(
-      'Changed from using real timers to fake timers while using waitFor',
-    );
+  await expect(waitForPromise).rejects.toThrow(
+    'Changed from using real timers to fake timers while using waitFor',
+  );
 });