test: update e2e-tests

Author: 1cedrus

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

@@ -1,10 +1,11 @@
 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 { render, renderHook, waitFor } from '@testing-library/react';
 import { useContractQuery, useContractTx, useRawContract } from 'typink';
 import { Contract } from 'dedot/contracts';
 import { Psp22ContractApi } from 'contracts/psp22';
+import { AccountId32 } from 'dedot/codecs';
 
 describe('useContractQuery', () => {
   let contractAddress: string, contract: Contract<Psp22ContractApi>;
@@ -29,23 +30,16 @@ describe('useContractQuery', () => {
       expect(totalSupply.current.isLoading).toEqual(false);
     });
 
-    totalSupply.current.refresh();
-
-    await waitFor(() => {
-      expect(totalSupply.current.isRefreshing).toEqual(true);
-    });
-
-    await waitFor(() => {
-      expect(totalSupply.current.data).toEqual(BigInt(1e20));
-      expect(totalSupply.current.isRefreshing).toEqual(false);
-    });
-
     const { result: balanceOf } = renderHook(
-      () => useContractQuery({ contract, args: ['0x_FAKE_'], fn: 'psp22BalanceOf' }),
+      ({ address }) => useContractQuery({ contract, args: [address], fn: 'psp22BalanceOf' }),
       {
         wrapper,
+        initialProps: {
+          address: '0x__FAKE',
+        },
       },
     );
+
     await waitFor(() => {
       expect(balanceOf.current.error).toBeDefined();
     });

packages/typink/src/hooks/useContractQuery.ts

@@ -14,7 +14,7 @@ type UseContractQueryReturnType<
   M extends keyof ContractQuery<T> = keyof ContractQuery<T>,
 > = {
   isLoading: boolean;
-  refresh: () => void;
+  refresh: () => Promise<void>;
   isRefreshing: boolean;
   error?: Error;
 } & Partial<Awaited<ReturnType<T['query'][M]>>>;
@@ -50,8 +50,8 @@ export function useContractQuery<
   } & Args<Pop<Parameters<T['query'][M]>>>,
 ): UseContractQueryReturnType<T, M> {
   // TODO replace loading tracking state with tanstack
-  
-  const { client } = useTypink(); 
+
+  const { client } = useTypink();
 
   const [isLoading, setIsLoading] = useState<boolean>(true);
   const [isRefreshing, setIsRefreshing] = useState<boolean>(false);
@@ -101,11 +101,15 @@ export function useContractQuery<
     try {
       setIsRefreshing(true);
 
+      console.log('Refreshing query:', fn, args, options);
+
       const result = await contract.query[fn](...args, options);
 
       setResult(result);
       setError(undefined);
       setIsRefreshing(false);
+
+      console.log('[DONE]: Refreshing query: ', fn, args, options);
     } catch (error: any) {
       console.error('Error when refreshing the query:', error);
 
@@ -151,4 +155,3 @@ export function useContractQuery<
     error,
   } as any;
 }
-