test: update e2e-tests

Author: 1cedrus

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

@@ -2,7 +2,7 @@ import { beforeAll, describe, expect, it } from 'vitest';
 import { ALICE, BOB, CHARLIE, deployPsp22Contract, psp22Metadata, wrapper } from '../utils';
 import { numberToHex } from 'dedot/utils';
 import { renderHook, waitFor } from '@testing-library/react';
-import { useContractQuery } from 'typink';
+import { useContractQuery, useContractTx, useRawContract } from 'typink';
 import { Contract } from 'dedot/contracts';
 import { Psp22ContractApi } from 'contracts/psp22';
 
@@ -39,6 +39,62 @@ describe('useContractQuery', () => {
     });
   });
 
+  it('should automatically call refresh on new block when watch is enabled', async () => {
+    const { result: rawContract } = renderHook(() => useRawContract<Psp22ContractApi>(psp22Metadata, contractAddress), {
+      wrapper,
+    });
+
+    await waitFor(() => {
+      expect(rawContract.current.contract).toBeDefined();
+      expect(rawContract.current.contract?.client.options.signer).toBeDefined();
+    });
+
+    const contract = rawContract.current.contract;
+
+    const { result: balanceOf } = renderHook(
+      () => useContractQuery({ contract, fn: 'psp22BalanceOf', args: [ALICE], watch: true }),
+      {
+        wrapper,
+      },
+    );
+
+    await waitFor(() => {
+      expect(balanceOf.current.data).toBeDefined();
+    });
+
+    const beforeTranfer = balanceOf.current.data;
+
+    console.log('Before transfer:', beforeTranfer);
+
+    const { result: transfer } = renderHook(
+      () => useContractTx(contract, 'psp22Transfer'), // prettier-end-here
+      {
+        wrapper,
+      },
+    );
+
+    await new Promise<void>((resolve) => {
+      transfer.current.signAndSend({
+        args: [BOB, BigInt(1e12), '0x'],
+        callback: ({ status }) => {
+          if (status.type === 'Finalized') {
+            resolve();
+          }
+        },
+      });
+    });
+    console.log('Transfer completed!');
+
+    await waitFor(
+      () => {
+        expect(balanceOf.current.data).not.toEqual(beforeTranfer);
+      },
+      { timeout: 12000 },
+    );
+    
+    console.log('After transfer:', balanceOf.current.data);
+  });
+
   it('should fail dry-run', async () => {
     const { result } = renderHook(
       () =>