fix(add-task-button): add flushSync to modal close (#3790)

* fix(add-task-button): add flushSync to modal close * fix(add-task-button): replace flushSync with callback
box · Dec 13, 2024 · 3555ff5 · 3555ff5
1 parent 15f2b65
commit 3555ff5
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/src/elements/content-sidebar/AddTaskButton.js b/src/elements/content-sidebar/AddTaskButton.js
@@ -1,6 +1,7 @@
 // @flow
 import * as React from 'react';
 import { withRouter, type RouterHistory } from 'react-router-dom';
+
 import AddTaskMenu from './AddTaskMenu';
 import TaskModal from './TaskModal';
 import { TASK_TYPE_APPROVAL } from '../../constants';
@@ -45,10 +46,11 @@ class AddTaskButton extends React.Component<Props, State> {
 
     handleModalClose = () => {
         const { onTaskModalClose } = this.props;
-        this.setState({ isTaskFormOpen: false, error: null });
-        if (this.buttonRef.current) {
-            this.buttonRef.current.focus();
-        }
+        this.setState({ isTaskFormOpen: false, error: null }, () => {
+            if (this.buttonRef.current) {
+                this.buttonRef.current.focus();
+            }
+        });
         onTaskModalClose();
     };