make sure gui function is called

rainlanguage · Feb 24, 2025 · 3bbef6a · 3bbef6a
1 parent 81d21d3
commit 3bbef6a
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/packages/ui-components/src/__tests__/DeploymentSteps.test.ts b/packages/ui-components/src/__tests__/DeploymentSteps.test.ts
@@ -708,6 +708,7 @@ describe('DeploymentSteps', () => {
 		const getAllFieldDefinitions = vi.fn();
 		const getAllDepositFields = vi.fn();
 		const getTokenInfo = vi.fn();
+		const areAllTokensSelected = vi.fn(() => true);
 
 		defaultProps.gui = {
 			getSelectTokens: () => mockSelectTokens,
@@ -718,7 +719,7 @@ describe('DeploymentSteps', () => {
 			getAllDepositFields,
 			isSelectTokenSet: () => false,
 			saveSelectToken: vi.fn(),
-			areAllTokensSelected: () => true,
+			areAllTokensSelected,
 			hasAnyDeposit: vi.fn(),
 			hasAnyVaultId: vi.fn(),
 			getCurrentDeployment: () => ({
@@ -752,6 +753,8 @@ describe('DeploymentSteps', () => {
 		const user = userEvent.setup();
 		render(DeploymentSteps, { props: defaultProps });
 
+		expect(areAllTokensSelected).toHaveBeenCalled();
+
 		await waitFor(() => {
 			expect(screen.getByText('Select Tokens')).toBeInTheDocument();
 			expect(screen.getByText('Token 1')).toBeInTheDocument();