diff --git a/libs/wallet/src/use-vega-transaction-updater.spec.tsx b/libs/wallet/src/use-vega-transaction-updater.spec.tsx index b29d7b232..10d833eed 100644 --- a/libs/wallet/src/use-vega-transaction-updater.spec.tsx +++ b/libs/wallet/src/use-vega-transaction-updater.spec.tsx @@ -1,9 +1,8 @@ -import { renderHook, act } from '@testing-library/react-hooks'; +import { renderHook, waitFor } from '@testing-library/react'; import type { MockedResponse } from '@apollo/client/testing'; import { MockedProvider } from '@apollo/client/testing'; import type { ReactNode } from 'react'; import { useVegaTransactionUpdater } from './use-vega-transaction-updater'; -import waitForNextTick from 'flush-promises'; import { OrderTxUpdateDocument, TransactionEventDocument, @@ -112,14 +111,13 @@ const mockedTransactionResultBusEvent: MockedResponse = describe('useVegaTransactionManager', () => { it('updates order on OrderTxUpdate', async () => { mockTransactionStoreState.mockReturnValue(defaultState); - const { waitForNextUpdate } = render([mockedOrderUpdate]); - await act(async () => { - waitForNextUpdate(); - await waitForNextTick(); + render([mockedOrderUpdate]); + await waitFor(() => { expect(updateOrder).toHaveBeenCalledWith(orderUpdate); }); }); it('updates transaction on TransactionResultBusEvents', async () => { mockTransactionStoreState.mockReturnValue(defaultState); - const { waitForNextUpdate } = render([mockedTransactionResultBusEvent]); - await act(async () => { - waitForNextUpdate(); - await waitForNextTick(); + render([mockedTransactionResultBusEvent]); + await waitFor(() => { expect(updateTransactionResult).toHaveBeenCalledWith( transactionResultBusEvent ); @@ -193,10 +187,8 @@ describe('useVegaTransactionManager', () => { mockWaitForWithdrawalApproval.mockResolvedValueOnce( erc20WithdrawalApproval ); - const { waitForNextUpdate } = render([mockedWithdrawalBusEvent]); - await act(async () => { - waitForNextUpdate(); - await waitForNextTick(); + render([mockedWithdrawalBusEvent]); + await waitFor(() => { expect(updateWithdrawal).toHaveBeenCalledWith( withdrawalBusEvent, erc20WithdrawalApproval diff --git a/libs/wallet/src/use-vega-transaction-updater.tsx b/libs/wallet/src/use-vega-transaction-updater.tsx index bdc9642fb..253fa6e88 100644 --- a/libs/wallet/src/use-vega-transaction-updater.tsx +++ b/libs/wallet/src/use-vega-transaction-updater.tsx @@ -38,9 +38,9 @@ export const useVegaTransactionUpdater = () => { result.data?.busEvents?.forEach((event) => { if (event.event.__typename === 'Withdrawal') { const withdrawal = event.event; - waitForWithdrawalApproval(withdrawal.id, client).then((approval) => - updateWithdrawal(withdrawal, approval) - ); + waitForWithdrawalApproval(withdrawal.id, client).then((approval) => { + updateWithdrawal(withdrawal, approval); + }); } }), });