test: Add test for SignatureScreen loading state when wallet is undefined

Author: gregfromstl

packages/thirdweb/src/react/web/ui/ConnectWallet/screens/SignatureScreen.test.tsx

@@ -101,7 +101,10 @@ describe("SignatureScreen", () => {
 		});
 	});
 
-	it("shows loading state", async () => {
+	it("shows loading state when wallet is undefined", async () => {
+		// Override the mock to return undefined for this test
+		vi.mocked(useActiveWallet).mockReturnValueOnce(undefined);
+
 		const { getByTestId } = render(
 			<SignatureScreen
 				onDone={() => {}}
@@ -113,10 +116,7 @@ describe("SignatureScreen", () => {
 			{ setConnectedWallet: true },
 		);
 
-		// Should show loading state
-		await waitFor(() => {
-			expect(getByTestId("loading-screen")).toBeInTheDocument();
-		});
+		expect(getByTestId("loading-screen")).toBeInTheDocument();
 	});
 
 	it("handles error state", async () => {