add useContract3

Author: sinzii

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

@@ -0,0 +1,73 @@
+import { beforeAll, describe, expect, it } from 'vitest';
+import { ALICE, BOB, deployPsp22Contract, psp22Metadata, wrapper } from '../utils';
+import { numberToHex } from 'dedot/utils';
+import { Contract } from 'dedot/contracts';
+import { Psp22ContractApi } from '../contracts/psp22';
+import { renderHook, waitFor } from '@testing-library/react';
+import { usePSP22Balance } from 'typink';
+
+describe('useContract3', () => {
+  let contractAddress: string, contract: Contract<Psp22ContractApi>;
+  beforeAll(async () => {
+    const randomSalt = numberToHex(Date.now());
+    contractAddress = await deployPsp22Contract(randomSalt);
+    console.log('Deployed contract address', contractAddress);
+    contract = new Contract<Psp22ContractApi>(client, psp22Metadata, contractAddress, { defaultCaller: ALICE });
+  });
+
+  it('get total supply', async () => {
+    const { data: state } = await contract.query.psp22TotalSupply();
+
+    console.log(`Total supply`, state);
+    expect(state).toBe(BigInt(1e20));
+  });
+
+  it('get alice balance', async () => {
+    const { data: state } = await contract.query.psp22BalanceOf(ALICE);
+
+    console.log(`alice balance`, state);
+    expect(state).toBe(BigInt(1e20));
+  });
+
+  it('should load balance properly', async () => {
+    const { result } = renderHook(
+      () => usePSP22Balance({ contractAddress, address: ALICE }), // prettier-end-here
+      { wrapper },
+    );
+
+    expect(result.current.data).toBeUndefined();
+
+    await waitFor(() => {
+      expect(result.current.data).toBe(BigInt(1e20));
+    });
+  });
+
+  it('should update balance when address changes', async () => {
+    const { result, rerender } = renderHook(({ address }) => usePSP22Balance({ contractAddress, address }), {
+      wrapper,
+      initialProps: {
+        address: ALICE,
+      },
+    });
+
+    expect(result.current.isLoading).toBe(true);
+
+    // Wait for ALICE's balance to be fetched
+    await waitFor(() => {
+      expect(result.current.data).toBeDefined();
+      expect(result.current.isLoading).toBe(false);
+    });
+
+    const aliceBalance = result.current.data;
+
+    rerender({ address: BOB });
+
+    // Wait for BOB's balance to be fetched
+    await waitFor(() => {
+      expect(result.current.data).toBe(0n);
+    });
+
+    // BOB's balance should be different from ALICE's
+    expect(result.current.data).not.toEqual(aliceBalance);
+  });
+});