run build fix

Author: Josie123-Dev

frontend/src/__tests__/AuthCallback.test.tsx

@@ -40,7 +40,7 @@ describe('AuthCallback', () => {
     );
 
     await waitFor(() => {
-      expect(setTokenFromCallbackMock).toHaveBeenCalledWith('test-token');
+      expect(setTokenFromCallbackMock).toHaveBeenCalledWith('test-token', null);
     });
 
     expect(navigateMock).toHaveBeenCalledWith('/employer/payroll', { replace: true });

frontend/src/__tests__/Home.test.tsx

@@ -5,6 +5,15 @@ import Home from '../pages/Home';
 
 const mockNavigate = vi.fn();
 
+vi.mock('@stellar/design-system', () => ({
+  Icon: {
+    Rocket01: () => <div data-testid="icon-rocket" />,
+    CreditCard01: () => <div data-testid="icon-credit-card" />,
+    Users01: () => <div data-testid="icon-users" />,
+    ShieldTick: () => <div data-testid="icon-shield-tick" />,
+  },
+}));
+
 vi.mock('react-router-dom', async () => {
   const actual = await vi.importActual<typeof import('react-router-dom')>('react-router-dom');
   return {

frontend/src/__tests__/hooks/useWalletManager.test.ts

@@ -3,59 +3,50 @@ import { describe, expect, it, vi, beforeEach } from 'vitest';
 import { useWalletManager } from '../../hooks/useWalletManager';
 import { StellarWalletsKit } from '@creit.tech/stellar-wallets-kit';
 
-vi.mock('@creit.tech/stellar-wallets-kit', () => ({
-  StellarWalletsKit: vi.fn(function () {
-    return {};
-  }),
-  WalletNetwork: { TESTNET: 'TESTNET', PUBLIC: 'PUBLIC' },
-  FreighterModule: vi.fn(function () {
-    return {};
-  }),
-  xBullModule: vi.fn(function () {
-    return {};
-  }),
-  LobstrModule: vi.fn(function () {
-    return {};
-  }),
-  FREIGHTER_ID: 'freighter',
-  LOBSTR_ID: 'lobstr',
-}));
+const mockSetWallet = vi.fn();
+const mockGetAddress = vi.fn();
+const mockGetSupportedWallets = vi.fn();
+const mockDisconnect = vi.fn();
+const mockSignTransaction = vi.fn();
+
+vi.mock('@creit.tech/stellar-wallets-kit', () => {
+  class MockStellarWalletsKit {
+    setWallet = mockSetWallet;
+    getAddress = mockGetAddress;
+    getSupportedWallets = mockGetSupportedWallets;
+    disconnect = mockDisconnect;
+    signTransaction = mockSignTransaction;
+  }
+
+  return {
+    StellarWalletsKit: MockStellarWalletsKit,
+    WalletNetwork: { TESTNET: 'TESTNET', PUBLIC: 'PUBLIC' },
+    FreighterModule: vi.fn(),
+    xBullModule: vi.fn(),
+    LobstrModule: vi.fn(),
+    FREIGHTER_ID: 'freighter',
+    LOBSTR_ID: 'lobstr',
+  };
+});
+
+const mockNotifyWalletEventHook = vi.fn();
 
 vi.mock('../../hooks/useNotification', () => ({
   useNotification: () => ({
-    notifyWalletEvent: vi.fn(),
+    notifyWalletEvent: mockNotifyWalletEventHook,
   }),
 }));
 
-interface MockKitInstance {
-  setWallet: ReturnType<typeof vi.fn>;
-  getAddress: ReturnType<typeof vi.fn>;
-  getSupportedWallets: ReturnType<typeof vi.fn>;
-  disconnect: ReturnType<typeof vi.fn>;
-  signTransaction: ReturnType<typeof vi.fn>;
-}
-
 describe('useWalletManager', () => {
-  let mockKitInstance: MockKitInstance;
-
   beforeEach(() => {
     vi.clearAllMocks();
     localStorage.clear();
-    mockKitInstance = {
-      setWallet: vi.fn(),
-      getAddress: vi.fn(),
-      getSupportedWallets: vi.fn().mockResolvedValue([]),
-      disconnect: vi.fn(),
-      signTransaction: vi.fn(),
-    };
-    vi.mocked(StellarWalletsKit).mockImplementation(
-      () => mockKitInstance as unknown as StellarWalletsKit
-    );
+    mockGetSupportedWallets.mockResolvedValue([]);
   });
 
   it('initializes and attempts silent reconnect if wallet in localStorage', async () => {
     localStorage.setItem('payd:last_wallet_name', 'freighter');
-    mockKitInstance.getAddress.mockResolvedValue({ address: 'G123' });
+    mockGetAddress.mockResolvedValue({ address: 'G123' });
 
     const { result } = renderHook(() => useWalletManager());
 
@@ -73,11 +64,12 @@ describe('useWalletManager', () => {
   });
 
   it('handles manual connect sequence appropriately', async () => {
-    mockKitInstance.getSupportedWallets.mockResolvedValue([
+    mockGetSupportedWallets.mockResolvedValue([
       { id: 'freighter', name: 'Freighter', isAvailable: true },
     ]);
 
     const { result } = renderHook(() => useWalletManager());
+    await waitFor(() => expect(result.current.isInitialized).toBe(true));
 
     await act(async () => {
       await result.current.connect();
@@ -88,7 +80,7 @@ describe('useWalletManager', () => {
     });
     expect(result.current.walletOptions.length).toBe(1);
 
-    mockKitInstance.getAddress.mockResolvedValue({ address: 'G456' });
+    mockGetAddress.mockResolvedValue({ address: 'G456' });
 
     await act(async () => {
       await result.current.connectWithWallet('freighter');