test(ui): align ResetConnectionDialog test with machine-driven reset

Author: iagodahlem

packages/ui/src/components/ConfigureSSO/__tests__/ResetConnectionDialog.test.tsx

@@ -5,23 +5,11 @@ import { bindCreateFixtures } from '@/test/create-fixtures';
 import { render, screen, waitFor } from '@/test/utils';
 import { CardStateProvider } from '@/ui/elements/contexts';
 
-const goToStep = vi.fn();
-
-vi.mock('../elements/Wizard', () => ({
-  useWizard: () => ({
-    activeSteps: [],
-    currentStep: undefined,
-    currentIndex: -1,
-    totalSteps: 0,
-    isFirstStep: true,
-    isLastStep: false,
-    isNested: false,
-    goNext: vi.fn(),
-    goPrev: vi.fn(),
-    goToStep,
-    registerStep: vi.fn(),
-    unregisterStep: vi.fn(),
-  }),
+const dispatch = vi.fn();
+
+vi.mock('../elements/WizardMachineContext', () => ({
+  useWizardMachine: () => ({ dispatch }),
+  WizardMachineProvider: ({ children }: { children: React.ReactNode }) => children,
 }));
 
 const setProvider = vi.fn();
@@ -70,11 +58,10 @@ const renderDialog = (
 };
 
 const resetMocks = () => {
-  goToStep.mockReset();
+  dispatch.mockReset();
   setProvider.mockReset();
   deleteEnterpriseConnection.mockReset();
   deleteEnterpriseConnection.mockResolvedValue({} as DeletedObjectResource);
-  goToStep.mockResolvedValue(undefined);
   connectionMockState.current = { id: 'idn_connection_1' };
 };
 
@@ -156,7 +143,7 @@ describe('ResetConnectionDialog', () => {
       expect(deleteEnterpriseConnection).toHaveBeenCalledWith('idn_connection_1');
     });
     expect(setProvider).toHaveBeenCalledWith(undefined);
-    expect(goToStep).toHaveBeenCalledWith('select-provider');
+    expect(dispatch).toHaveBeenCalledWith({ type: 'RESET' });
     await waitFor(() => {
       expect(onClose).toHaveBeenCalledTimes(1);
     });