Fix preselection of robot

mrica-equinor · Dec 4, 2024 · 6dd8eda · 6dd8eda
1 parent 4583b47
commit 6dd8eda
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/frontend/src/components/Pages/InspectionPage/ScheduleMissionDialogs.tsx b/frontend/src/components/Pages/InspectionPage/ScheduleMissionDialogs.tsx
@@ -56,7 +56,6 @@ export const ScheduleMissionDialog = (props: IProps): JSX.Element => {
     const { setLoadingRobotMissionSet } = useMissionsContext()
     const { setAlert, setListAlert } = useAlertContext()
     const [isLocalizationVerificationDialogOpen, setIsLocalizationVerificationDialog] = useState<boolean>(false)
-    const [selectedRobot, setSelectedRobot] = useState<Robot>()
     const [missionsToSchedule, setMissionsToSchedule] = useState<MissionDefinition[]>()
     const filteredRobots = enabledRobots.filter(
         (r) =>
@@ -65,6 +64,9 @@ export const ScheduleMissionDialog = (props: IProps): JSX.Element => {
                 r.status === RobotStatus.Recharging) &&
             r.isarConnected
     )
+    const [selectedRobot, setSelectedRobot] = useState<Robot | undefined>(
+        filteredRobots.length === 1 ? filteredRobots[0] : undefined
+    )
 
     const onSelectedRobot = (selectedRobot: Robot) => {
         if (filteredRobots) setSelectedRobot(selectedRobot)
@@ -152,7 +154,7 @@ export const ScheduleMissionDialog = (props: IProps): JSX.Element => {
                         )}
                         <StyledAutoComplete>
                             <Autocomplete
-                                initialSelectedOptions={filteredRobots.length === 1 ? [filteredRobots[0]] : []}
+                                initialSelectedOptions={selectedRobot ? [selectedRobot] : []}
                                 dropdownHeight={200}
                                 optionLabel={(r) => r.name + ' (' + r.model.type + ')'}
                                 options={filteredRobots}