From b1a1ed40adaebbf78cc098afac86c83905e91449 Mon Sep 17 00:00:00 2001
From: Kush Makkapati <kush.makkapati@icloud.com>
Date: Tue, 31 Mar 2026 21:17:32 +0100
Subject: [PATCH 1/4] Add cancelling drone connection mid way

---
 gcs/src/components/navbar.jsx             | 10 ++-
 gcs/src/redux/middleware/emitters.js      | 12 +++-
 radio/app/controllers/paramsController.py | 14 ++++
 radio/app/drone.py                        | 34 ++++++++++
 radio/app/droneStatus.py                  |  4 +-
 radio/app/endpoints/comPorts.py           | 83 +++++++++++++++--------
 radio/tests/test_ParamsController.py      | 12 ++++
 radio/tests/test_connections.py           | 25 +++++++
 8 files changed, 162 insertions(+), 32 deletions(-)

diff --git a/gcs/src/components/navbar.jsx b/gcs/src/components/navbar.jsx
index 9e11e5011..a899b17aa 100644
--- a/gcs/src/components/navbar.jsx
+++ b/gcs/src/components/navbar.jsx
@@ -37,6 +37,7 @@ import { useDispatch, useSelector } from "react-redux"
 import {
   ConnectionType,
   emitConnectToDrone,
+  emitDisconnectFromDrone,
   emitGetComPorts,
   emitStartForwarding,
   emitStopForwarding,
@@ -169,6 +170,9 @@ export default function Navbar() {
       <Modal
         opened={openedModal}
         onClose={() => {
+          if (connecting) {
+            dispatch(emitDisconnectFromDrone())
+          }
           dispatch(setConnectionModal(false))
           dispatch(setConnecting(false))
         }}
@@ -297,12 +301,14 @@ export default function Navbar() {
               variant="filled"
               color={"red"}
               onClick={() => {
+                if (connecting) {
+                  dispatch(emitDisconnectFromDrone())
+                }
                 dispatch(setConnectionModal(false))
                 dispatch(setConnecting(false))
               }}
-              disabled={connecting}
             >
-              Close
+              {connecting ? "Cancel" : "Close"}
             </Button>
             <Button
               variant="filled"
diff --git a/gcs/src/redux/middleware/emitters.js b/gcs/src/redux/middleware/emitters.js
index 036d94c23..53e525d20 100644
--- a/gcs/src/redux/middleware/emitters.js
+++ b/gcs/src/redux/middleware/emitters.js
@@ -99,7 +99,17 @@ export function handleEmitters(socket, store, action) {
     },
     {
       emitter: emitDisconnectFromDrone,
-      callback: () => socket.socket.emit("disconnect_from_drone"),
+      callback: () => {
+        const storeState = store.getState()
+        const isConnecting = storeState.droneConnection.connecting
+
+        if (isConnecting) {
+          socket.socket.emit("cancel_connect_to_drone")
+          return
+        }
+
+        socket.socket.emit("disconnect_from_drone")
+      },
     },
     {
       emitter: emitConnectToDrone,
diff --git a/radio/app/controllers/paramsController.py b/radio/app/controllers/paramsController.py
index 12a2b1a7e..eb11642c6 100644
--- a/radio/app/controllers/paramsController.py
+++ b/radio/app/controllers/paramsController.py
@@ -77,10 +77,17 @@ def fetchAllParamsBlocking(
         self,
         timeout_secs: int = 120,
         progress_update_callback: Optional[Callable[[dict], None]] = None,
+        should_cancel_callback: Optional[Callable[[], bool]] = None,
     ) -> Response:
         """
         Fetches all parameters from the drone in a blocking manner.
         """
+        if should_cancel_callback and should_cancel_callback():
+            return {
+                "success": False,
+                "message": "Connection cancelled by user.",
+            }
+
         start_response = self._startFetchAll()
         if not start_response.get("success"):
             return start_response
@@ -89,6 +96,13 @@ def fetchAllParamsBlocking(
 
         try:
             while self.is_requesting_params:
+                if should_cancel_callback and should_cancel_callback():
+                    self.drone.logger.info("Parameter fetch cancelled by user")
+                    return {
+                        "success": False,
+                        "message": "Connection cancelled by user.",
+                    }
+
                 if time.time() > timeout:
                     self.drone.logger.error(
                         f"Fetching all parameters timed out after {timeout_secs} seconds"
diff --git a/radio/app/drone.py b/radio/app/drone.py
index d0effde5c..17ebdbd6b 100644
--- a/radio/app/drone.py
+++ b/radio/app/drone.py
@@ -85,6 +85,7 @@ def __init__(
         droneConnectStatusCb: Optional[Callable] = None,
         linkDebugStatsCb: Optional[Callable] = None,
         fetchingParameterCb: Optional[Callable] = None,
+        connectionCancelEvent: Optional[Event] = None,
     ) -> None:
         """
         The drone class interfaces with the UAS via MavLink.
@@ -104,6 +105,7 @@ def __init__(
         self.droneConnectStatusCb = droneConnectStatusCb
         self.linkDebugStatsCb = linkDebugStatsCb
         self.fetchingParameterCb = fetchingParameterCb
+        self.connection_cancel_event: Event = connectionCancelEvent or Event()
 
         self.connectionError: Optional[str] = None
         self._last_connect_progress: float = 0.0
@@ -148,12 +150,20 @@ def __init__(
                 self.connectionError = str(e)
             return
 
+        if self._isConnectionCancelRequested():
+            self._setCancelledConnectionErrorAndCloseMaster()
+            return
+
         try:
             initial_heartbeat = None
             heartbeat_timeout_secs = 5.0
             deadline = time.monotonic() + heartbeat_timeout_secs
 
             while True:
+                if self._isConnectionCancelRequested():
+                    self._setCancelledConnectionErrorAndCloseMaster()
+                    return
+
                 now = time.monotonic()
                 if now >= deadline:
                     break
@@ -249,6 +259,12 @@ def __init__(
 
         self.addMessageListener("STATUSTEXT", sendMessage)
 
+        if self._isConnectionCancelRequested():
+            self._setCancelledConnectionErrorAndCloseMaster()
+            self.is_active.clear()
+            self.stopAllThreads()
+            return
+
         self.getAutopilotVersion()
 
         if self.flight_sw_version is None:
@@ -289,6 +305,7 @@ def __init__(
         fetch_all_params_result = self.paramsController.fetchAllParamsBlocking(
             timeout_secs=120,
             progress_update_callback=self.sendParamFetchConnectionStatusUpdate,
+            should_cancel_callback=self._isConnectionCancelRequested,
         )
 
         if not fetch_all_params_result.get("success"):
@@ -311,6 +328,23 @@ def __init__(
             mavutil.mavlink.MAV_SEVERITY_INFO, "FGCS connected to aircraft"
         )
 
+    def _isConnectionCancelRequested(self) -> bool:
+        return self.connection_cancel_event.is_set()
+
+    def requestConnectionCancel(self) -> None:
+        self.connection_cancel_event.set()
+        if getattr(self, "paramsController", None) is not None:
+            self.paramsController.is_requesting_params = False
+
+    def _setCancelledConnectionErrorAndCloseMaster(self) -> None:
+        self.connectionError = "Connection cancelled by user."
+        if getattr(self, "master", None) is not None:
+            try:
+                self.master.close()
+            except Exception:
+                self.logger.exception("Failed to close connection during cancellation")
+            self.master = None
+
     def __getNextLogFilePath(self, line: str) -> str:
         return line.split("==NEXT_FILE==")[-1].split("==END==")[0]
 
diff --git a/radio/app/droneStatus.py b/radio/app/droneStatus.py
index bc4ee4d2d..3feca4c48 100644
--- a/radio/app/droneStatus.py
+++ b/radio/app/droneStatus.py
@@ -3,11 +3,13 @@
 imports this and access it, it can update the values and every other file will also have the update due to passing by reference (probably).
 """
 
+from threading import Event
 from typing import List, Optional
 
-
 from app.drone import Drone
 
 correct_ports: List[str] = []
 drone: Optional[Drone] = None
 state: Optional[str] = None
+connection_in_progress: bool = False
+connect_cancel_event: Optional[Event] = None
diff --git a/radio/app/endpoints/comPorts.py b/radio/app/endpoints/comPorts.py
index 6d97b38cb..9e949d6b9 100644
--- a/radio/app/endpoints/comPorts.py
+++ b/radio/app/endpoints/comPorts.py
@@ -1,5 +1,6 @@
 import sys
 import time
+from threading import Event
 from typing import Optional
 
 from serial.tools import list_ports
@@ -11,9 +12,9 @@
 from app.utils import (
     droneConnectStatusCb,
     droneErrorCb,
+    fetchingParameterCb,
     getComPortNames,
     getFlightSwVersionString,
-    fetchingParameterCb,
 )
 
 
@@ -85,6 +86,13 @@ def connectToDrone(data: ConnectionDataType) -> None:
         droneStatus.drone.close()
         droneStatus.drone = None
 
+    if droneStatus.connection_in_progress:
+        socketio.emit(
+            "connection_error",
+            {"message": "A drone connection is already in progress."},
+        )
+        return
+
     connectionType = data.get("connectionType")
 
     if connectionType not in ["serial", "network"]:
@@ -133,35 +141,52 @@ def connectToDrone(data: ConnectionDataType) -> None:
         droneStatus.drone = None
         return
 
-    drone = Drone(
-        port,
-        baud=baud,
-        forwarding_address=forwarding_address,
-        droneErrorCb=droneErrorCb,
-        droneDisconnectCb=disconnectFromDrone,
-        droneConnectStatusCb=droneConnectStatusCb,
-        linkDebugStatsCb=sendLinkDebugStats,
-        fetchingParameterCb=fetchingParameterCb,
-    )
-
-    if drone.connectionError is not None:
-        socketio.emit("connection_error", {"message": drone.connectionError})
-        droneStatus.drone = None
-        return
+    droneStatus.connection_in_progress = True
+    cancel_event = Event()
+    droneStatus.connect_cancel_event = cancel_event
+
+    try:
+        drone = Drone(
+            port,
+            baud=baud,
+            forwarding_address=forwarding_address,
+            droneErrorCb=droneErrorCb,
+            droneDisconnectCb=disconnectFromDrone,
+            droneConnectStatusCb=droneConnectStatusCb,
+            linkDebugStatsCb=sendLinkDebugStats,
+            fetchingParameterCb=fetchingParameterCb,
+            connectionCancelEvent=cancel_event,
+        )
+
+        if drone.connectionError is not None:
+            socketio.emit("connection_error", {"message": drone.connectionError})
+            droneStatus.drone = None
+            return
+
+        # Set droneStatus drone to local drone
+        droneStatus.drone = drone
+
+        # Sleeping for buffer time, if errors occur try changing back to 1 second
+        time.sleep(0.2)
+        logger.debug("Created drone instance")
+        socketio.emit(
+            "connected_to_drone",
+            {
+                "aircraft_type": drone.aircraft_type,
+                "flight_sw_version": getFlightSwVersionString(drone.flight_sw_version),
+            },
+        )
+    finally:
+        droneStatus.connection_in_progress = False
+        droneStatus.connect_cancel_event = None
 
-    # Set droneStatus drone to local drone
-    droneStatus.drone = drone
 
-    # Sleeping for buffer time, if errors occur try changing back to 1 second
-    time.sleep(0.2)
-    logger.debug("Created drone instance")
-    socketio.emit(
-        "connected_to_drone",
-        {
-            "aircraft_type": drone.aircraft_type,
-            "flight_sw_version": getFlightSwVersionString(drone.flight_sw_version),
-        },
-    )
+@socketio.on("cancel_connect_to_drone")
+def cancelConnectToDrone() -> None:
+    """Cancel an in-progress drone connection attempt."""
+    if droneStatus.connection_in_progress and droneStatus.connect_cancel_event:
+        droneStatus.connect_cancel_event.set()
+        socketio.emit("connection_error", {"message": "Connection cancelled by user."})
 
 
 @socketio.on("disconnect_from_drone")
@@ -172,6 +197,8 @@ def disconnectFromDrone() -> None:
     if droneStatus.drone:
         droneStatus.drone.close()
         droneStatus.drone = None
+    elif droneStatus.connection_in_progress and droneStatus.connect_cancel_event:
+        droneStatus.connect_cancel_event.set()
 
     droneStatus.state = None
     socketio.emit("disconnected_from_drone")
diff --git a/radio/tests/test_ParamsController.py b/radio/tests/test_ParamsController.py
index a14bbacc2..cc29ef8bb 100644
--- a/radio/tests/test_ParamsController.py
+++ b/radio/tests/test_ParamsController.py
@@ -92,3 +92,15 @@ def test_fetchAllParamsBlocking_success_sorts_and_updates_progress(droneStatus)
         assert controller.total_number_of_params == 0
     finally:
         controller.params = old_params
+
+
+def test_fetchAllParamsBlocking_cancelled_before_start(droneStatus) -> None:
+    controller = droneStatus.drone.paramsController
+
+    result = controller.fetchAllParamsBlocking(
+        timeout_secs=120,
+        should_cancel_callback=lambda: True,
+    )
+
+    assert result["success"] is False
+    assert result["message"] == "Connection cancelled by user."
diff --git a/radio/tests/test_connections.py b/radio/tests/test_connections.py
index c9ba46252..549380939 100644
--- a/radio/tests/test_connections.py
+++ b/radio/tests/test_connections.py
@@ -1,3 +1,5 @@
+from threading import Event
+
 import pytest
 from flask_socketio.test_client import SocketIOTestClient
 
@@ -171,6 +173,29 @@ def test_stopForwarding_notForwarding(socketio_client: SocketIOTestClient, drone
     }
 
 
+def test_cancel_connect_to_drone_signals_cancel_event(
+    socketio_client: SocketIOTestClient, droneStatus
+):
+    old_in_progress = droneStatus.connection_in_progress
+    old_cancel_event = droneStatus.connect_cancel_event
+
+    try:
+        droneStatus.connection_in_progress = True
+        droneStatus.connect_cancel_event = Event()
+
+        socketio_client.emit("cancel_connect_to_drone")
+        socketio_result = socketio_client.get_received()
+
+        assert droneStatus.connect_cancel_event.is_set() is True
+        assert socketio_result[0]["name"] == "connection_error"
+        assert socketio_result[0]["args"][0] == {
+            "message": "Connection cancelled by user."
+        }
+    finally:
+        droneStatus.connection_in_progress = old_in_progress
+        droneStatus.connect_cancel_event = old_cancel_event
+
+
 # Has to be the final test otherwise the socket disconnects
 def test_disconnect(socketio_client: SocketIOTestClient, droneStatus):
     """Test disconnecting from socket"""

From 7c7cce1147e000ecbc4d03bed20bc6ab8b99ad57 Mon Sep 17 00:00:00 2001
From: Kush Makkapati <kush.makkapati@icloud.com>
Date: Tue, 31 Mar 2026 21:28:44 +0100
Subject: [PATCH 2/4] Fix some bugs with connection cancellation

---
 radio/app/drone.py              | 16 +++++++++++++---
 radio/app/endpoints/comPorts.py |  1 -
 radio/tests/test_connections.py |  5 -----
 3 files changed, 13 insertions(+), 9 deletions(-)

diff --git a/radio/app/drone.py b/radio/app/drone.py
index 17ebdbd6b..07ddef2dc 100644
--- a/radio/app/drone.py
+++ b/radio/app/drone.py
@@ -337,6 +337,7 @@ def requestConnectionCancel(self) -> None:
             self.paramsController.is_requesting_params = False
 
     def _setCancelledConnectionErrorAndCloseMaster(self) -> None:
+        self.logger.info("Connection cancelled by user")
         self.connectionError = "Connection cancelled by user."
         if getattr(self, "master", None) is not None:
             try:
@@ -965,8 +966,13 @@ def sendHeartbeatMessage(self) -> None:
             if sleep_time > 0:
                 time.sleep(sleep_time)
 
+            master = getattr(self, "master", None)
+            if master is None:
+                # Connection teardown can clear master while this thread is winding down.
+                break
+
             try:
-                self.master.mav.heartbeat_send(
+                master.mav.heartbeat_send(
                     mavutil.mavlink.MAV_TYPE_GCS,
                     mavutil.mavlink.MAV_AUTOPILOT_INVALID,
                     0,
@@ -1262,10 +1268,14 @@ def close(self) -> None:
 
         self.is_active.clear()
 
-        self.stopAllDataStreams()
+        if getattr(self, "master", None) is not None:
+            self.stopAllDataStreams()
         self.stopForwarding()
         self.stopAllThreads()
-        self.master.close()
+
+        if getattr(self, "master", None) is not None:
+            self.master.close()
+            self.master = None
 
         if len(self.log_file_names) == 0:
             self.logger.debug("No logs to save")
diff --git a/radio/app/endpoints/comPorts.py b/radio/app/endpoints/comPorts.py
index 9e949d6b9..953d7e241 100644
--- a/radio/app/endpoints/comPorts.py
+++ b/radio/app/endpoints/comPorts.py
@@ -186,7 +186,6 @@ def cancelConnectToDrone() -> None:
     """Cancel an in-progress drone connection attempt."""
     if droneStatus.connection_in_progress and droneStatus.connect_cancel_event:
         droneStatus.connect_cancel_event.set()
-        socketio.emit("connection_error", {"message": "Connection cancelled by user."})
 
 
 @socketio.on("disconnect_from_drone")
diff --git a/radio/tests/test_connections.py b/radio/tests/test_connections.py
index 549380939..59a714627 100644
--- a/radio/tests/test_connections.py
+++ b/radio/tests/test_connections.py
@@ -184,13 +184,8 @@ def test_cancel_connect_to_drone_signals_cancel_event(
         droneStatus.connect_cancel_event = Event()
 
         socketio_client.emit("cancel_connect_to_drone")
-        socketio_result = socketio_client.get_received()
 
         assert droneStatus.connect_cancel_event.is_set() is True
-        assert socketio_result[0]["name"] == "connection_error"
-        assert socketio_result[0]["args"][0] == {
-            "message": "Connection cancelled by user."
-        }
     finally:
         droneStatus.connection_in_progress = old_in_progress
         droneStatus.connect_cancel_event = old_cancel_event

From 89e718ab2ff538db66d8c1dd63c0a038a846fad8 Mon Sep 17 00:00:00 2001
From: Kush Makkapati <kush.makkapati@icloud.com>
Date: Wed, 1 Apr 2026 19:36:40 +0100
Subject: [PATCH 3/4] Address copilot review comments

---
 radio/app/drone.py              |  3 ++
 radio/app/droneStatus.py        |  3 +-
 radio/app/endpoints/comPorts.py | 71 ++++++++++++++++++++++-----------
 radio/tests/test_connections.py | 28 +++++++++++++
 4 files changed, 80 insertions(+), 25 deletions(-)

diff --git a/radio/app/drone.py b/radio/app/drone.py
index 07ddef2dc..ceef355c9 100644
--- a/radio/app/drone.py
+++ b/radio/app/drone.py
@@ -96,6 +96,9 @@ def __init__(
             droneErrorCb (Optional[Callable], optional): Callback function for drone errors. Defaults to None.
             droneDisconnectCb (Optional[Callable], optional): Callback function for drone disconnection. Defaults to None.
             droneConnectStatusCb (Optional[Callable], optional): Callback function for drone connection providing an update as the drone connects. Defaults to None.
+            linkDebugStatsCb (Optional[Callable], optional): Callback function for link debug stats. Defaults to None.
+            fetchingParameterCb (Optional[Callable], optional): Callback function for when parameters are being fetched. Defaults to None.
+            connectionCancelEvent (Optional[Event], optional): Event to signal if the connection process should be cancelled. Defaults to None.
         """
         self.port = port
         self.baud = baud
diff --git a/radio/app/droneStatus.py b/radio/app/droneStatus.py
index 3feca4c48..fa406ec39 100644
--- a/radio/app/droneStatus.py
+++ b/radio/app/droneStatus.py
@@ -3,7 +3,7 @@
 imports this and access it, it can update the values and every other file will also have the update due to passing by reference (probably).
 """
 
-from threading import Event
+from threading import Event, Lock
 from typing import List, Optional
 
 from app.drone import Drone
@@ -11,5 +11,6 @@
 correct_ports: List[str] = []
 drone: Optional[Drone] = None
 state: Optional[str] = None
+connection_state_lock: Lock = Lock()
 connection_in_progress: bool = False
 connect_cancel_event: Optional[Event] = None
diff --git a/radio/app/endpoints/comPorts.py b/radio/app/endpoints/comPorts.py
index 953d7e241..62a65f022 100644
--- a/radio/app/endpoints/comPorts.py
+++ b/radio/app/endpoints/comPorts.py
@@ -79,19 +79,28 @@ def connectToDrone(data: ConnectionDataType) -> None:
     Args:
         data: The message passed in from the client containing the form sent (select com port, baud rate)
     """
-    if droneStatus.drone:
-        droneStatus.drone.logger.warning(
-            "Attempting a connection to drone when connection is already established."
-        )
-        droneStatus.drone.close()
-        droneStatus.drone = None
+    old_drone = None
+    with droneStatus.connection_state_lock:
+        if droneStatus.connection_in_progress:
+            socketio.emit(
+                "connection_error",
+                {"message": "A drone connection is already in progress."},
+            )
+            return
 
-    if droneStatus.connection_in_progress:
-        socketio.emit(
-            "connection_error",
-            {"message": "A drone connection is already in progress."},
+        if droneStatus.drone:
+            old_drone = droneStatus.drone
+            droneStatus.drone = None
+
+        cancel_event = Event()
+        droneStatus.connect_cancel_event = cancel_event
+        droneStatus.connection_in_progress = True
+
+    if old_drone is not None:
+        old_drone.logger.warning(
+            "Attempting a connection to drone when connection is already established."
         )
-        return
+        old_drone.close()
 
     connectionType = data.get("connectionType")
 
@@ -141,10 +150,6 @@ def connectToDrone(data: ConnectionDataType) -> None:
         droneStatus.drone = None
         return
 
-    droneStatus.connection_in_progress = True
-    cancel_event = Event()
-    droneStatus.connect_cancel_event = cancel_event
-
     try:
         drone = Drone(
             port,
@@ -177,15 +182,22 @@ def connectToDrone(data: ConnectionDataType) -> None:
             },
         )
     finally:
-        droneStatus.connection_in_progress = False
-        droneStatus.connect_cancel_event = None
+        with droneStatus.connection_state_lock:
+            droneStatus.connection_in_progress = False
+            droneStatus.connect_cancel_event = None
 
 
 @socketio.on("cancel_connect_to_drone")
 def cancelConnectToDrone() -> None:
     """Cancel an in-progress drone connection attempt."""
-    if droneStatus.connection_in_progress and droneStatus.connect_cancel_event:
-        droneStatus.connect_cancel_event.set()
+    with droneStatus.connection_state_lock:
+        if droneStatus.connection_in_progress and droneStatus.connect_cancel_event:
+            logger.info("Connection cancel requested by user")
+            droneStatus.connect_cancel_event.set()
+        else:
+            logger.debug(
+                "Connection cancel requested, but no connection is in progress"
+            )
 
 
 @socketio.on("disconnect_from_drone")
@@ -193,11 +205,22 @@ def disconnectFromDrone() -> None:
     """
     Disconnect from drone and reset all global variables, send a message to client disconnecting as well
     """
-    if droneStatus.drone:
-        droneStatus.drone.close()
-        droneStatus.drone = None
-    elif droneStatus.connection_in_progress and droneStatus.connect_cancel_event:
-        droneStatus.connect_cancel_event.set()
+    with droneStatus.connection_state_lock:
+        if droneStatus.drone:
+            drone = droneStatus.drone
+            droneStatus.drone = None
+        else:
+            drone = None
+
+        if (
+            drone is None
+            and droneStatus.connection_in_progress
+            and droneStatus.connect_cancel_event
+        ):
+            droneStatus.connect_cancel_event.set()
+
+    if drone is not None:
+        drone.close()
 
     droneStatus.state = None
     socketio.emit("disconnected_from_drone")
diff --git a/radio/tests/test_connections.py b/radio/tests/test_connections.py
index 59a714627..8f2c2fe0c 100644
--- a/radio/tests/test_connections.py
+++ b/radio/tests/test_connections.py
@@ -173,6 +173,34 @@ def test_stopForwarding_notForwarding(socketio_client: SocketIOTestClient, drone
     }
 
 
+def test_connect_while_connection_in_progress_is_rejected(
+    socketio_client: SocketIOTestClient, droneStatus
+):
+    old_in_progress = droneStatus.connection_in_progress
+    old_cancel_event = droneStatus.connect_cancel_event
+
+    try:
+        droneStatus.connection_in_progress = True
+        socketio_client.emit(
+            "connect_to_drone",
+            {
+                "port": "tcp:127.0.0.1:5760",
+                "baud": 115200,
+                "connectionType": "network",
+                "forwardingAddress": None,
+            },
+        )
+
+        socketio_result = socketio_client.get_received()
+        assert socketio_result[0]["name"] == "connection_error"
+        assert socketio_result[0]["args"][0] == {
+            "message": "A drone connection is already in progress."
+        }
+    finally:
+        droneStatus.connection_in_progress = old_in_progress
+        droneStatus.connect_cancel_event = old_cancel_event
+
+
 def test_cancel_connect_to_drone_signals_cancel_event(
     socketio_client: SocketIOTestClient, droneStatus
 ):

From 2254cef4fb4df4aaf65ee196b9651b0a0afca679 Mon Sep 17 00:00:00 2001
From: Kush Makkapati <kush.makkapati@icloud.com>
Date: Wed, 1 Apr 2026 19:59:40 +0100
Subject: [PATCH 4/4] Try fixing broken test

---
 radio/app/endpoints/comPorts.py | 46 ++++++++++++++++-----------------
 1 file changed, 23 insertions(+), 23 deletions(-)

diff --git a/radio/app/endpoints/comPorts.py b/radio/app/endpoints/comPorts.py
index 62a65f022..348f151e7 100644
--- a/radio/app/endpoints/comPorts.py
+++ b/radio/app/endpoints/comPorts.py
@@ -79,29 +79,6 @@ def connectToDrone(data: ConnectionDataType) -> None:
     Args:
         data: The message passed in from the client containing the form sent (select com port, baud rate)
     """
-    old_drone = None
-    with droneStatus.connection_state_lock:
-        if droneStatus.connection_in_progress:
-            socketio.emit(
-                "connection_error",
-                {"message": "A drone connection is already in progress."},
-            )
-            return
-
-        if droneStatus.drone:
-            old_drone = droneStatus.drone
-            droneStatus.drone = None
-
-        cancel_event = Event()
-        droneStatus.connect_cancel_event = cancel_event
-        droneStatus.connection_in_progress = True
-
-    if old_drone is not None:
-        old_drone.logger.warning(
-            "Attempting a connection to drone when connection is already established."
-        )
-        old_drone.close()
-
     connectionType = data.get("connectionType")
 
     if connectionType not in ["serial", "network"]:
@@ -150,6 +127,29 @@ def connectToDrone(data: ConnectionDataType) -> None:
         droneStatus.drone = None
         return
 
+    old_drone = None
+    with droneStatus.connection_state_lock:
+        if droneStatus.connection_in_progress:
+            socketio.emit(
+                "connection_error",
+                {"message": "A drone connection is already in progress."},
+            )
+            return
+
+        if droneStatus.drone:
+            old_drone = droneStatus.drone
+            droneStatus.drone = None
+
+        cancel_event = Event()
+        droneStatus.connect_cancel_event = cancel_event
+        droneStatus.connection_in_progress = True
+
+    if old_drone is not None:
+        old_drone.logger.warning(
+            "Attempting a connection to drone when connection is already established."
+        )
+        old_drone.close()
+
     try:
         drone = Drone(
             port,