From 24c7fa216fb56bd4366e8b0155e7026cbef9debf Mon Sep 17 00:00:00 2001
From: Storm <stormluykx@gmail.com>
Date: Thu, 29 Jan 2026 12:28:34 +0100
Subject: [PATCH] test: 100% coverage

ref: N25B-393
---
 src/robot_interface/endpoints/video_sender.py |  4 ++--
 test/unit/test_video_sender.py                | 21 +++++++++++++++++++
 2 files changed, 23 insertions(+), 2 deletions(-)

diff --git a/src/robot_interface/endpoints/video_sender.py b/src/robot_interface/endpoints/video_sender.py
index f3e28f2..23a071f 100644
--- a/src/robot_interface/endpoints/video_sender.py
+++ b/src/robot_interface/endpoints/video_sender.py
@@ -68,10 +68,10 @@ class VideoSender(SocketBase):
                         height_bytes = struct.pack('<I', height)
 
                         self.socket.send_multipart([width_bytes, height_bytes, raw_data])
+                except KeyboardInterrupt:
+                    logging.info("Video receiving loop interrupted by user.")
                 except:
                     logging.warn("Failed to retrieve video image from robot.")
-        except KeyboardInterrupt:
-            logging.info("Video receiving loop interrupted by user.")
         finally:
             vid_service.unsubscribe(vid_stream_name)
             logging.info("Unsubscribed from video stream.")
\ No newline at end of file
diff --git a/test/unit/test_video_sender.py b/test/unit/test_video_sender.py
index 35d592e..14ea8d5 100644
--- a/test/unit/test_video_sender.py
+++ b/test/unit/test_video_sender.py
@@ -104,3 +104,24 @@ def test_video_receive_error(zmq_context, mocker):
     sender.video_rcv_loop(mock_video_service, "stream_name")
 
     send_socket.assert_not_called()
+
+def test_video_loop_keyboard_interrupt(zmq_context, mocker):
+    """Video loop should handle KeyboardInterrupt gracefully and unsubscribe."""
+    _patch_basics(mocker)
+    _patch_exit_event(mocker)
+    
+    # We mock the video service to raise KeyboardInterrupt when accessed
+    mock_video_service = mock.Mock()
+    mock_video_service.getImageRemote.side_effect = KeyboardInterrupt
+    
+    # Mock logging to verify the specific interrupt message is logged
+    mock_logger = mocker.patch("robot_interface.endpoints.video_sender.logging")
+
+    sender = VideoSender(zmq_context)
+    
+    # Execute the loop
+    sender.video_rcv_loop(mock_video_service, "stream_name")
+
+    # Verify the 'finally' block executed (unsubscribe)
+    mock_video_service.unsubscribe.assert_called_with("stream_name")
+    mock_logger.info.assert_any_call("Unsubscribed from video stream.")