fix specs

Author: sinzii

e2e/zombienet/src/hooks/useContract2.test.ts

@@ -75,12 +75,9 @@ describe('useContract2', () => {
     const { result } = renderHook(() => usePSP22Balance({ contractAddress, address: 'invalid_address' }), { wrapper });
 
     // The balance should remain undefined
-    await waitFor(
-      () => {
-        expect(result.current.data).toBeUndefined();
-      },
-      { timeout: 5000 },
-    );
+    await waitFor(() => {
+      expect(result.current.data).toBeUndefined();
+    });
   });
 
   it('should automatic update balance when watch is enabled', async () => {
@@ -91,15 +88,15 @@ describe('useContract2', () => {
       expect(result.current.data).toBeDefined();
     });
 
-    const aliceBalance = result.current.data;
+    const aliceBalance = result.current.data!;
 
     // Simulate a transfer event
     const { alice } = devPairs();
     await mintPSP22Balance(contractAddress, alice, BigInt(1e12));
 
     // Wait for the balance to be updated
     await waitFor(() => {
-      expect(result.current.data).toBe(BigInt(1e20) + BigInt(1e12));
+      expect(result.current.data).toBe(aliceBalance + BigInt(1e12));
       expect(result.current.data).not.toEqual(aliceBalance);
     });
   });

e2e/zombienet/src/setup.ts

@@ -16,7 +16,7 @@ beforeAll(async () => {
       }
     });
   });
-}, 120_000);
+}, 300_000);
 
 afterAll(async () => {
   await global.client.disconnect();

e2e/zombienet/vitest.config.ts

@@ -6,7 +6,7 @@ export default defineConfig({
   test: {
     environment: 'happy-dom',
     setupFiles: './src/setup.ts',
-    testTimeout: 120_000,
-    hookTimeout: 120_000,
+    testTimeout: 300_000,
+    hookTimeout: 300_000,
   },
 });