test(sdk): more signaturescreen tests

Author: gregfromstl

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

@@ -6,6 +6,7 @@ import { TEST_CLIENT } from "../../../../../../test/src/test-clients.js";
 import { createWallet } from "../../../../../wallets/create-wallet.js";
 import type { ConnectLocale } from "../locale/types.js";
 import { SignatureScreen } from "./SignatureScreen.js";
+import { useActiveWallet } from "../../../../core/hooks/wallets/useActiveWallet.js";
 
 const mockAuth = vi.hoisted(() => ({
 	doLogin: vi.fn().mockResolvedValue(undefined),
@@ -14,14 +15,12 @@ const mockAuth = vi.hoisted(() => ({
 	isLoggedIn: vi.fn().mockResolvedValue(true),
 }));
 
-const wallet = createWallet("io.metamask");
-
 vi.mock("../../../../core/hooks/auth/useSiweAuth", () => ({
 	useSiweAuth: () => mockAuth,
 }));
 
 vi.mock("../../../../core/hooks/wallets/useActiveWallet", () => ({
-	useActiveWallet: () => wallet,
+	useActiveWallet: vi.fn().mockReturnValue(createWallet("io.metamask")),
 }));
 
 vi.mock("../../../../core/hooks/wallets/useActiveAccount", () => ({
@@ -102,10 +101,9 @@ describe("SignatureScreen", () => {
 	});
 
 	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(
+		const { queryByTestId } = render(
 			<SignatureScreen
 				onDone={() => {}}
 				modalSize="wide"
@@ -116,7 +114,7 @@ describe("SignatureScreen", () => {
 			{ setConnectedWallet: true },
 		);
 
-		expect(getByTestId("loading-screen")).toBeInTheDocument();
+		expect(queryByTestId("sign-in-button")).not.toBeInTheDocument();
 	});
 
 	it("handles error state", async () => {