Ensure we unwrap custom .then() (#4547)

preactjs · Nov 9, 2024 · 76d6d4d · 76d6d4d
1 parent aee8285
commit 76d6d4d
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 1 deletion.
diff --git a/compat/src/suspense.d.ts b/compat/src/suspense.d.ts
@@ -3,7 +3,9 @@ import { Component, ComponentChild, ComponentChildren } from '../../src';
 //
 // Suspense/lazy
 // -----------------------------------
-export function lazy<T>(loader: () => Promise<{ default: T } | T>): T;
+export function lazy<T>(
+	loader: () => Promise<{ default: T } | T>
+): T extends { default: infer U } ? U : T;
 
 export interface SuspenseProps {
 	children?: ComponentChildren;

diff --git a/compat/test/ts/suspense.tsx b/compat/test/ts/suspense.tsx
@@ -50,3 +50,18 @@ function ReactSuspenseListTester(_props: any) {
 		</React.SuspenseList>
 	);
 }
+
+const Comp = () => <p>Hello world</p>;
+
+const importComponent = async () => {
+	return { MyComponent: Comp };
+};
+
+const Lazy = React.lazy(() =>
+	importComponent().then(mod => ({ default: mod.MyComponent }))
+);
+
+// eslint-disable-next-line
+function App() {
+	return <Lazy />;
+}