add more tests

Author: sinzii

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

@@ -1,5 +1,5 @@
 import { beforeAll, describe, expect, it } from 'vitest';
-import { ALICE, BOB, deployPsp22Contract, psp22Metadata, wrapper } from '../utils';
+import { ALICE, BOB, deployPsp22Contract, devPairs, mintPSP22Balance, psp22Metadata, wrapper } from '../utils';
 import { numberToHex } from 'dedot/utils';
 import { Contract } from 'dedot/contracts';
 import { Psp22ContractApi } from '../contracts/psp22';
@@ -70,4 +70,25 @@ describe('useContract2', () => {
     // BOB's balance should be different from ALICE's
     expect(result.current.data).not.toEqual(aliceBalance);
   });
+
+  it('should automatic update balance when watch is enabled', async () => {
+    const { result } = renderHook(() => usePSP22Balance({ contractAddress, address: ALICE, watch: true }), { wrapper });
+
+    // Wait for ALICE's balance to be fetched
+    await waitFor(() => {
+      expect(result.current.data).toBeDefined();
+    });
+
+    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).not.toEqual(aliceBalance);
+    });
+  });
 });

e2e/zombienet/src/utils.tsx

@@ -6,7 +6,7 @@ import { Psp22ContractApi } from './contracts/psp22';
 import * as flipper from './contracts/flipper_v5.json';
 // @ts-ignore
 import * as psp22 from './contracts/psp22.json';
-import { ContractDeployer, parseRawMetadata } from 'dedot/contracts';
+import { Contract, ContractDeployer, parseRawMetadata } from 'dedot/contracts';
 import { assert, deferred } from 'dedot/utils';
 import { cryptoWaitReady } from '@polkadot/util-crypto';
 import { KeyringPair } from '@polkadot/keyring/types';
@@ -145,6 +145,29 @@ export const deployPsp22Contract = async (salt?: string): Promise<string> => {
   return defer.promise;
 };
 
+export const mintPSP22Balance = async (psp22Address: string, pair: KeyringPair, amount: bigint): Promise<void> => {
+  console.log('[mintPSP22Balance]', psp22Address, pair.address, amount);
+
+  const contract = new Contract<Psp22ContractApi>(client, psp22Metadata, psp22Address, { defaultCaller: pair.address });
+
+  const {
+    raw: { gasRequired },
+  } = await contract.query.psp22MintableMint(amount);
+
+  const defer = deferred<void>();
+  await contract.tx
+    .psp22MintableMint(amount, { gasLimit: gasRequired }) // prettier-end-here
+    .signAndSend(pair, ({ status }) => {
+      console.log('Transaction status:', status.type);
+
+      if (status.type === 'BestChainBlockIncluded') {
+        defer.resolve();
+      }
+    });
+
+  return defer.promise;
+};
+
 export const devPairs = () => {
   const alice = KEYRING.addFromUri('//Alice');
   const bob = KEYRING.addFromUri('//Bob');