fix: force cast await to proper type

package.json

@@ -60,7 +60,7 @@
     "jest": ">=28.0.0",
     "react": ">=16.0.0",
     "react-native": ">=0.59",
-    "react-test-renderer": ">=16.0.0"
+    "react-test-renderer": ">=16.8.0"
   },
   "peerDependenciesMeta": {
     "jest": {

src/__tests__/act.test.tsx

@@ -43,7 +43,7 @@ test('fireEvent should trigger useState', () => {
 });
 
 test('should act even if there is no act in react-test-renderer', () => {
-  // @ts-ignore
+  // @ts-expect-error
   ReactTestRenderer.act = undefined;
   const callback = jest.fn();
 
@@ -53,3 +53,13 @@ test('should act even if there is no act in react-test-renderer', () => {
 
   expect(callback).toHaveBeenCalled();
 });
+
+test('should be able to not await act', async () => {
+  const result = act(() => {});
+  expect(result).toHaveProperty('then');
+});
+
+test('should be able to await act', async () => {
+  const result = await act(async () => {});
+  expect(result).toBe(undefined);
+});

src/act.ts

@@ -3,9 +3,7 @@
 import { act as reactTestRendererAct } from 'react-test-renderer';
 import { checkReactVersionAtLeast } from './react-versions';
 
-const actMock = (callback: () => void) => {
-  callback();
-};
+type ReactAct = typeof reactTestRendererAct;
 
 // See https://github.com/reactwg/react-18/discussions/102 for more context on global.IS_REACT_ACT_ENVIRONMENT
 declare global {
@@ -20,10 +18,8 @@ function getIsReactActEnvironment() {
   return globalThis.IS_REACT_ACT_ENVIRONMENT;
 }
 
-type Act = typeof reactTestRendererAct;
-
-function withGlobalActEnvironment(actImplementation: Act) {
-  return (callback: Parameters<Act>[0]) => {
+function withGlobalActEnvironment(actImplementation: ReactAct) {
+  return (callback: Parameters<ReactAct>[0]) => {
     const previousActEnvironment = getIsReactActEnvironment();
     setIsReactActEnvironment(true);
 
@@ -44,6 +40,7 @@ function withGlobalActEnvironment(actImplementation: Act) {
         }
         return result;
       });
+
       if (callbackNeedsToBeAwaited) {
         const thenable = actResult;
         return {
@@ -77,16 +74,10 @@ function withGlobalActEnvironment(actImplementation: Act) {
     }
   };
 }
-const getAct = () => {
-  if (!reactTestRendererAct) {
-    return actMock;
-  }
 
-  return checkReactVersionAtLeast(18, 0)
-    ? withGlobalActEnvironment(reactTestRendererAct)
-    : reactTestRendererAct;
-};
-const act = getAct();
+const act: ReactAct = checkReactVersionAtLeast(18, 0)
+  ? (withGlobalActEnvironment(reactTestRendererAct) as ReactAct)
+  : reactTestRendererAct;
 
 export default act;
 export {