[Form]: fixed bug when beforeLeave modal appears in case of redirec…

…t in `onSuccess` callback.

public/docs/content/examples-labeledInput-Basic.json

@@ -1 +1,11 @@
-{"object":"value","document":{"object":"document","data":{},"nodes":[{"object":"block","type":"paragraph","data":{},"nodes":[{"object":"text","text":"Basic LabeledInput usage","marks":[]}]}]}}
+[
+  {
+    "type": "paragraph",
+    "data": {},
+    "children": [
+      {
+        "text": "Basic LabeledInput usage"
+      }
+    ]
+  }
+]

uui-core/src/data/forms/__tests__/useForm.test.tsx

@@ -625,7 +625,7 @@ describe('useForm', () => {
 
         it('Should unblock router when form is saved', async () => {
             const saveMock = jest.fn().mockResolvedValue(true);
-            const beforeLeaveMock = jest.fn().mockResolvedValue(false);
+            const beforeLeaveMock = jest.fn().mockRejectedValue(false);
             const { wrapper, testUuiCtx: svc } = getDefaultUUiContextWrapper();
             const props = {
                 value: testData,
@@ -676,7 +676,7 @@ describe('useForm', () => {
             expect(beforeLeaveMock).toHaveBeenCalledTimes(1);
         });
 
-        it('Should make redirect and lock form aging if  beforeLeave return "remain"', async () => {
+        it('Should make redirect and lock form aging if beforeLeave returns "remain"', async () => {
             const saveMock = jest.fn().mockResolvedValue(true);
             const beforeLeaveMock = jest.fn().mockResolvedValue('remain');
             const { wrapper, testUuiCtx: svc } = getDefaultUUiContextWrapper();

uui-core/src/data/forms/useForm.ts

@@ -259,9 +259,9 @@ export function useForm<T>(props: UseFormProps<T>): IFormApi<T> {
         }
         flushSync(() => {
             resetForm(newState);
+            removeUnsavedChanges();
+            unblock();
         });
-        removeUnsavedChanges();
-        unblock();
 
         if (propsRef.current.onSuccess && response) {
             propsRef.current.onSuccess(response.form, isSavedBeforeLeave);