Merge branch 'main' into feat/pause-functionality

chore: add copyright to all source files

See merge request ics/sp/2025/n25b/pepperplus-ri!29

src/robot_interface/__init__.py

@@ -0,0 +1,6 @@
+# -*- coding: utf-8 -*-
+"""
+This program has been developed by students from the bachelor Computer Science at Utrecht
+University within the Software Project course.
+© Copyright Utrecht University (Department of Information and Computing Sciences)
+"""

src/robot_interface/core/__init__.py

@@ -0,0 +1,6 @@
+# -*- coding: utf-8 -*-
+"""
+This program has been developed by students from the bachelor Computer Science at Utrecht
+University within the Software Project course.
+© Copyright Utrecht University (Department of Information and Computing Sciences)
+"""

src/robot_interface/core/config.py

@@ -1,3 +1,10 @@
+# -*- coding: utf-8 -*-
+"""
+This program has been developed by students from the bachelor Computer Science at Utrecht
+University within the Software Project course.
+© Copyright Utrecht University (Department of Information and Computing Sciences)
+"""
+
 from __future__ import unicode_literals
 
 from robot_interface.utils.get_config import get_config

src/robot_interface/endpoints/__init__.py

@@ -0,0 +1,6 @@
+# -*- coding: utf-8 -*-
+"""
+This program has been developed by students from the bachelor Computer Science at Utrecht
+University within the Software Project course.
+© Copyright Utrecht University (Department of Information and Computing Sciences)
+"""

src/robot_interface/endpoints/actuation_receiver.py

@@ -1,6 +1,12 @@
+# -*- coding: utf-8 -*-
+"""
+This program has been developed by students from the bachelor Computer Science at Utrecht
+University within the Software Project course.
+© Copyright Utrecht University (Department of Information and Computing Sciences)
+"""
+
 from __future__ import unicode_literals  # So that we can log texts with Unicode characters
 import logging
-import time
 from threading import Thread
 
 import Queue
@@ -8,7 +14,6 @@ import zmq
 
 from robot_interface.endpoints.receiver_base import ReceiverBase
 from robot_interface.state import state
-
 from robot_interface.core.config import settings
 from robot_interface.endpoints.gesture_settings import GestureTags
 
@@ -36,11 +41,8 @@ class ActuationReceiver(ReceiverBase):
         self._tts_service = None
         self._animation_service = None
         self._message_queue = Queue.Queue()
-        self._gesture_queue = Queue.Queue()
         self.message_thread = Thread(target=self._handle_messages)
         self.message_thread.start()
-        self.gesture_thread = Thread(target=self._handle_gestures)
-        self.gesture_thread.start()
 
     def _handle_speech(self, message):
         """
@@ -67,7 +69,7 @@ class ActuationReceiver(ReceiverBase):
         if not self._tts_service:
             self._tts_service = state.qi_session.service("ALTextToSpeech")
 
-        if (message.get("is_priority")):
+        if message.get("is_priority"):
             # Bypass queue and speak immediately
             self.clear_queue()
             self._message_queue.put(text)
@@ -75,7 +77,6 @@ class ActuationReceiver(ReceiverBase):
         else: 
             self._message_queue.put(text)
 
-
     def clear_queue(self):
         """
         Safely drains all pending messages from the queue.
@@ -89,21 +90,6 @@ class ActuationReceiver(ReceiverBase):
             pass
         logging.info("Message queue cleared.")
 
-    def clear_gesture_queue(self):
-        """
-        Safely drains all pending gestures from the gesture queue.
-        """
-        logging.info("Gesture queue size: {}".format(self._gesture_queue.qsize()))
-        try:
-            while True:
-                # Remove items one by one without waiting
-                self._gesture_queue.get_nowait()
-        except Queue.Empty:
-            pass
-        logging.info("Gesture queue cleared.")
-        logging.info("Gesture queue size: {}".format(self._gesture_queue.qsize()))
-
-
     def _handle_gesture(self, message, is_single):
         """
         Handle a gesture actuation request.
@@ -144,16 +130,12 @@ class ActuationReceiver(ReceiverBase):
 
         # Play the gesture. Pepper comes with predefined animations like "Wave", "Greet", "Clap"
         # You can also create custom animations using Choregraphe and upload them to the robot.
-        if (message.get("is_priority")):
-            # Clear queue and play
-            self.clear_gesture_queue()
-            logging.debug("Force playing gesture immediately: {}".format(gesture))
         if is_single:
-            logging.debug("Adding single gesture to queue: {}".format(gesture))
+            logging.debug("Playing single gesture: {}".format(gesture))
+            getattr(qi, "async")(self._animation_service.run, gesture)
         else:
-            logging.debug("Adding tag gesture to queue: {}".format(gesture))
-        self._gesture_queue.put(gesture)
-
+            logging.debug("Playing tag gesture: {}".format(gesture))
+            getattr(qi, "async")(self._animation_service.runTag, gesture)
 
     def handle_message(self, message):
         """
@@ -173,25 +155,13 @@ class ActuationReceiver(ReceiverBase):
         while not state.exit_event.is_set():
             try:
                 text = self._message_queue.get(timeout=0.1)
-                if not state.is_speaking: print("Started speaking.")
                 state.is_speaking = True
                 self._tts_service.say(text)
             except Queue.Empty:
-                if state.is_speaking: print("Finished speaking.")
                 state.is_speaking = False
             except RuntimeError:
-                logging.warn("Lost connection to Pepper. Please check if you're connected to the local WiFi and restart this application.")
-                state.exit_event.set() 
-    
-    def _handle_gestures(self):
-        while not state.exit_event.is_set():
-            try:
-                gesture = self._gesture_queue.get(timeout=0.1)
-                self._animation_service.run(gesture)
-            except Queue.Empty:
-                pass
-            except RuntimeError:
-                logging.warn("Lost connection to Pepper. Please check if you're connected to the local WiFi and restart this application.")
+                logging.error("Lost connection to Pepper. Please check if you're connected to the "
+                              "local WiFi and restart this application.")
                 state.exit_event.set()        
 
     def endpoint_description(self):

src/robot_interface/endpoints/audio_sender.py

@@ -1,3 +1,10 @@
+# -*- coding: utf-8 -*-
+"""
+This program has been developed by students from the bachelor Computer Science at Utrecht
+University within the Software Project course.
+© Copyright Utrecht University (Department of Information and Computing Sciences)
+"""
+
 from __future__ import unicode_literals  # So that `logging` can use Unicode characters in names
 import threading
 import logging
@@ -77,21 +84,33 @@ class AudioSender(SocketBase):
         chunk = audio_settings.chunk_size # 320 at 16000 Hz is 20ms, 512 is required for Silero-VAD
 
         # Docs say this only raises an error if neither `input` nor `output` is True
-        stream = self.audio.open(
-            format=pyaudio.paFloat32,
-            channels=audio_settings.channels,
-            rate=audio_settings.sample_rate,
-            input=True,
-            input_device_index=self.microphone["index"],
-            frames_per_buffer=chunk,
-        )
+        def open_stream():
+            return self.audio.open(
+                format=pyaudio.paFloat32,
+                channels=audio_settings.channels,
+                rate=audio_settings.sample_rate,
+                input=True,
+                input_device_index=self.microphone["index"],
+                frames_per_buffer=chunk,
+            )
+        
+        stream = None
 
         try:
+            # Test in case exit_event was set while waiting
+            if not state.exit_event.is_set():
+                stream = open_stream()
+            
             while not state.exit_event.is_set():
                 data = stream.read(chunk)
+                if (state.is_speaking): continue  # Do not send audio while the robot is speaking
                 self.socket.send(data)
         except IOError as e:
             logger.error("Stopped listening: failed to get audio from microphone.", exc_info=e)
         finally:
-            stream.stop_stream()
-            stream.close()
+            if stream:
+                try:
+                    stream.stop_stream()
+                    stream.close()
+                except IOError:
+                    pass # Ignore errors on closing

src/robot_interface/endpoints/gesture_settings.py

@@ -1,3 +1,10 @@
+# -*- coding: utf-8 -*-
+"""
+This program has been developed by students from the bachelor Computer Science at Utrecht
+University within the Software Project course.
+© Copyright Utrecht University (Department of Information and Computing Sciences)
+"""
+
 class GestureTags:
     tags = ["above", "affirmative", "afford", "agitated", "all", "allright", "alright", "any", 
             "assuage", "assuage", "attemper", "back", "bashful", "beg", "beseech", "blank", 

src/robot_interface/endpoints/main_receiver.py

@@ -1,3 +1,10 @@
+# -*- coding: utf-8 -*-
+"""
+This program has been developed by students from the bachelor Computer Science at Utrecht
+University within the Software Project course.
+© Copyright Utrecht University (Department of Information and Computing Sciences)
+"""
+
 import zmq
 
 from robot_interface.endpoints.receiver_base import ReceiverBase
@@ -72,6 +79,30 @@ class MainReceiver(ReceiverBase):
 
         return {"endpoint": "negotiate/error", "data": "The requested endpoint is not implemented."}
     
+    @staticmethod
+    def _handle_pause(message):
+        """
+        Handle a pause request. Pauses or resumes the video and audio streams.
+
+        :param message: The pause request message.
+        :type message: dict
+
+        :return: A response dictionary indicating success.
+        :rtype: dict[str, str]
+        """
+        if message.get("data"):
+            if state.active_event.is_set():
+                state.active_event.clear()
+                return {"endpoint": "pause", "data": "Streams paused."}
+            else:
+                return {"endpoint": "pause", "data": "Streams are already paused."}
+        else:
+            if not state.active_event.is_set():
+                state.active_event.set()
+                return {"endpoint": "pause", "data": "Streams resumed."}
+            else:
+                return {"endpoint": "pause", "data": "Streams are already running."}
+
     def handle_message(self, message):
         """
         Main entry point for handling incoming messages.
@@ -88,5 +119,7 @@ class MainReceiver(ReceiverBase):
             return self._handle_ping(message)
         elif message["endpoint"].startswith("negotiate"):
             return self._handle_negotiation(message)
+        elif message["endpoint"] == "pause":
+            return self._handle_pause(message)
 
         return {"endpoint": "error", "data": "The requested endpoint is not supported."}

src/robot_interface/endpoints/receiver_base.py

@@ -1,3 +1,10 @@
+# -*- coding: utf-8 -*-
+"""
+This program has been developed by students from the bachelor Computer Science at Utrecht
+University within the Software Project course.
+© Copyright Utrecht University (Department of Information and Computing Sciences)
+"""
+
 from abc import ABCMeta, abstractmethod
 
 from robot_interface.endpoints.socket_base import SocketBase

src/robot_interface/endpoints/socket_base.py

@@ -1,3 +1,10 @@
+# -*- coding: utf-8 -*-
+"""
+This program has been developed by students from the bachelor Computer Science at Utrecht
+University within the Software Project course.
+© Copyright Utrecht University (Department of Information and Computing Sciences)
+"""
+
 from abc import ABCMeta
 
 import zmq

src/robot_interface/endpoints/video_sender.py

@@ -1,11 +1,22 @@
+# -*- coding: utf-8 -*-
+"""
+This program has been developed by students from the bachelor Computer Science at Utrecht
+University within the Software Project course.
+© Copyright Utrecht University (Department of Information and Computing Sciences)
+"""
+import struct
 import zmq
 import threading
 import logging
+import numpy as np
+
+import cv2
 
 from robot_interface.endpoints.socket_base import SocketBase
 from robot_interface.state import state
 from robot_interface.core.config import settings
 
+
 class VideoSender(SocketBase):
     """
     Video sender endpoint, responsible for sending video frames.
@@ -18,7 +29,7 @@ class VideoSender(SocketBase):
     """
     def __init__(self, zmq_context, port=settings.agent_settings.video_sender_port):
         super(VideoSender, self).__init__("video")
-        self.create_socket(zmq_context, zmq.PUB, port, [(zmq.CONFLATE,1)])
+        self.create_socket(zmq_context, zmq.PUB, port, [(zmq.SNDHWM,3)])
     
     def start_video_rcv(self):
         """
@@ -28,6 +39,9 @@ class VideoSender(SocketBase):
         """
         if not state.qi_session:
             logging.info("No Qi session available. Not starting video loop.")
+            logging.info("Starting test video stream from local webcam.")
+            thread = threading.Thread(target=self.test_video_stream)
+            thread.start()
             return
 
         video = state.qi_session.service("ALVideoDevice")
@@ -41,9 +55,40 @@ class VideoSender(SocketBase):
         thread = threading.Thread(target=self.video_rcv_loop, args=(video, vid_stream_name))
         thread.start()
 
+    def test_video_stream(self):
+        """
+        Test function to send video from a local webcam instead of the robot.
+        """
+        cap = cv2.VideoCapture(0)
+        if not cap.isOpened():
+            logging.error("Could not open webcam for video stream test.")
+            return
+
+        while not state.exit_event.is_set():
+
+            ret, frame = cap.read()
+            if not ret:
+                logging.warning("Failed to read frame from webcam.")
+                continue
+
+            if cv2.waitKey(1) & 0xFF == ord('q'): # << Add this: Updates the window
+                break
+            
+            height, width, channels = frame.shape
+            
+            pixel_data = frame.tobytes()
+
+            width_bytes = struct.pack('<I', width)
+            height_bytes = struct.pack('<I', height)
+
+            self.socket.send_multipart([width_bytes, height_bytes, pixel_data])
+
+        cap.release()
+
     def video_rcv_loop(self, vid_service, vid_stream_name):
         """
         The main loop of retrieving video images from the robot.
+        Sends the image data over the ZMQ socket in 3 parts: image width, image height and raw image bytes.
 
         :param vid_service: The video service object that the active Qi session is connected to. 
         :type vid_service: Object (Qi service object)
@@ -51,10 +96,23 @@ class VideoSender(SocketBase):
         :param vid_stream_name: The name of a camera subscription on the video service object vid_service
         :type vid_stream_name: str
         """
-        while not state.exit_event.is_set():
-            try:
-                img = vid_service.getImageRemote(vid_stream_name)
-                #Possibly limit images sent if queuing issues arise
-                self.socket.send(img[settings.video_config.image_buffer])
-            except:
-                logging.warn("Failed to retrieve video image from robot.")
+        try:
+            while not state.exit_event.is_set():
+                try:
+                    img = vid_service.getImageRemote(vid_stream_name)
+                    if img is not None:
+                        image_bytes = img[6]
+                        width = img[0]
+                        height = img[1]
+
+                        width_bytes = struct.pack('<I', width)
+                        height_bytes = struct.pack('<I', height)
+
+                        self.socket.send_multipart([width_bytes, height_bytes, image_bytes])
+                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.")

src/robot_interface/main.py

@@ -1,3 +1,10 @@
+# -*- coding: utf-8 -*-
+"""
+This program has been developed by students from the bachelor Computer Science at Utrecht
+University within the Software Project course.
+© Copyright Utrecht University (Department of Information and Computing Sciences)
+"""
+
 import logging
 
 from robot_interface.endpoints.audio_sender import AudioSender
@@ -84,6 +91,7 @@ def main():
     context = zmq.Context()
 
     state.initialize()
+    state.active_event.set()
 
     try:
         main_loop(context)

src/robot_interface/state.py

@@ -1,3 +1,10 @@
+# -*- coding: utf-8 -*-
+"""
+This program has been developed by students from the bachelor Computer Science at Utrecht
+University within the Software Project course.
+© Copyright Utrecht University (Department of Information and Computing Sciences)
+"""
+
 import logging
 import signal
 import threading
@@ -49,6 +56,8 @@ class State(object):
         signal.signal(signal.SIGINT, handle_exit)
         signal.signal(signal.SIGTERM, handle_exit)
 
+        self.active_event = threading.Event()
+
         self.qi_session = get_qi_session()
 
         self.is_initialized = True

src/robot_interface/utils/__init__.py

@@ -0,0 +1,6 @@
+# -*- coding: utf-8 -*-
+"""
+This program has been developed by students from the bachelor Computer Science at Utrecht
+University within the Software Project course.
+© Copyright Utrecht University (Department of Information and Computing Sciences)
+"""

src/robot_interface/utils/get_config.py

@@ -1,3 +1,10 @@
+# -*- coding: utf-8 -*-
+"""
+This program has been developed by students from the bachelor Computer Science at Utrecht
+University within the Software Project course.
+© Copyright Utrecht University (Department of Information and Computing Sciences)
+"""
+
 import os
 from dotenv import load_dotenv
 

src/robot_interface/utils/microphone.py

@@ -1,3 +1,10 @@
+# -*- coding: utf-8 -*-
+"""
+This program has been developed by students from the bachelor Computer Science at Utrecht
+University within the Software Project course.
+© Copyright Utrecht University (Department of Information and Computing Sciences)
+"""
+
 from __future__ import unicode_literals  # So that `print` can print Unicode characters in names
 import logging
 import sys

src/robot_interface/utils/qi_utils.py

@@ -1,3 +1,10 @@
+# -*- coding: utf-8 -*-
+"""
+This program has been developed by students from the bachelor Computer Science at Utrecht
+University within the Software Project course.
+© Copyright Utrecht University (Department of Information and Computing Sciences)
+"""
+
 import logging
 import sys
 

src/robot_interface/utils/timeblock.py

@@ -1,3 +1,10 @@
+# -*- coding: utf-8 -*-
+"""
+This program has been developed by students from the bachelor Computer Science at Utrecht
+University within the Software Project course.
+© Copyright Utrecht University (Department of Information and Computing Sciences)
+"""
+
 import time
 
 

test/common/__init__.py

@@ -0,0 +1,6 @@
+# -*- coding: utf-8 -*-
+"""
+This program has been developed by students from the bachelor Computer Science at Utrecht
+University within the Software Project course.
+© Copyright Utrecht University (Department of Information and Computing Sciences)
+"""

test/common/microphone_utils.py

@@ -1,3 +1,10 @@
+# -*- coding: utf-8 -*-
+"""
+This program has been developed by students from the bachelor Computer Science at Utrecht
+University within the Software Project course.
+© Copyright Utrecht University (Department of Information and Computing Sciences)
+"""
+
 from __future__ import unicode_literals  # So that we can format strings with Unicode characters
 import random
 import sys

test/conftest.py

@@ -1,3 +1,10 @@
+# -*- coding: utf-8 -*-
+"""
+This program has been developed by students from the bachelor Computer Science at Utrecht
+University within the Software Project course.
+© Copyright Utrecht University (Department of Information and Computing Sciences)
+"""
+
 from mock import patch, MagicMock
 
 import pytest

test/integration/__init__.py

@@ -0,0 +1,6 @@
+# -*- coding: utf-8 -*-
+"""
+This program has been developed by students from the bachelor Computer Science at Utrecht
+University within the Software Project course.
+© Copyright Utrecht University (Department of Information and Computing Sciences)
+"""

test/integration/test_config.py

@@ -1,3 +1,10 @@
+# -*- coding: utf-8 -*-
+"""
+This program has been developed by students from the bachelor Computer Science at Utrecht
+University within the Software Project course.
+© Copyright Utrecht University (Department of Information and Computing Sciences)
+"""
+
 from mock import patch, mock
 
 from robot_interface.core.config import Settings

test/integration/test_microphone_utils.py

@@ -1,3 +1,10 @@
+# -*- coding: utf-8 -*-
+"""
+This program has been developed by students from the bachelor Computer Science at Utrecht
+University within the Software Project course.
+© Copyright Utrecht University (Department of Information and Computing Sciences)
+"""
+
 import pyaudio
 
 import pytest

test/unit/__init__.py

@@ -0,0 +1,6 @@
+# -*- coding: utf-8 -*-
+"""
+This program has been developed by students from the bachelor Computer Science at Utrecht
+University within the Software Project course.
+© Copyright Utrecht University (Department of Information and Computing Sciences)
+"""

test/unit/test_actuation_receiver.py

@@ -1,13 +1,32 @@
+# -*- coding: utf-8 -*-
+"""
+This program has been developed by students from the bachelor Computer Science at Utrecht
+University within the Software Project course.
+© Copyright Utrecht University (Department of Information and Computing Sciences)
+"""
+
 import sys
 
-import time
 import mock
 import pytest
 import zmq
-import Queue
+
 from robot_interface.endpoints.actuation_receiver import ActuationReceiver
 from robot_interface.endpoints.gesture_settings import GestureTags
 
+
+@pytest.fixture
+def zmq_context():
+    """
+    A pytest fixture that creates and yields a ZMQ context.
+
+    :return: An initialized ZeroMQ context.
+    :rtype: zmq.Context
+    """
+    context = zmq.Context()
+    yield context
+
+
 def test_force_speech_clears_queue(mocker):
     """
     Tests that a force speech message clears the existing queue
@@ -42,6 +61,7 @@ def test_force_speech_clears_queue(mocker):
     queued_item = receiver._message_queue.get()
     assert queued_item == "Emergency Notification"
 
+
 def test_handle_unimplemented_endpoint(mocker):
     """
     Tests handling of unknown endpoints.
@@ -57,6 +77,7 @@ def test_handle_unimplemented_endpoint(mocker):
         "data": None,
     })
 
+
 def test_speech_message_no_data(mocker):
     """
     Tests that if the message data is empty, the receiver returns immediately
@@ -214,7 +235,7 @@ def test_handle_messages_loop(mocker):
     receiver._tts_service = mock_tts_service
 
     # This ensures the while loop iterates exactly once
-    mock_state.exit_event.is_set.side_effect = [False, True, True , True, True]
+    mock_state.exit_event.is_set.side_effect = [False, True]
 
     # Put an item in the queue
     receiver._message_queue.put("Hello World")
@@ -229,24 +250,40 @@ def test_handle_messages_loop(mocker):
     assert mock_state.is_speaking is True
 
 
-def test_handle_gestures_runtime_error(mocker):
+def test_handle_messages_queue_empty(mocker):
+    """
+    Tests the Queue.Empty exception handler in the consumer loop.
+    This covers the logic that resets 'state.is_speaking' to False.
+    """
+    # Prevent the real background thread from starting
     mocker.patch("threading.Thread")
+
+    # Mock the state object
     mock_state = mocker.patch("robot_interface.endpoints.actuation_receiver.state")
         
-    # Use a side_effect that returns False then True thereafter
-    mock_state.exit_event.is_set.side_effect = [False, True, True, True, True]
+    # Setup 'is_speaking' property mock
+    # We set return_value=True so the code enters the 'if state.is_speaking:' block.
+    # We use PropertyMock to track when this attribute is set.
+    type(mock_state).is_speaking = True
     
     mock_zmq_ctx = mock.Mock()
     receiver = ActuationReceiver(mock_zmq_ctx)
     
-    # ... rest of your setup ...
-    mock_anim = mock.Mock()
-    mock_anim.run.side_effect = RuntimeError("Wifi Lost")
-    receiver._animation_service = mock_anim
+    # This ensures the while loop body runs exactly once for our test
+    mock_state.exit_event.is_set.side_effect = [False, True]
 
-    receiver._gesture_queue.put("wave")
+    # Force get() to raise Queue.Empty immediately (simulate timeout)
+    # We patch the 'get' method on the specific queue instance of our receiver
+    #mocker.patch.object(receiver._message_queue, 'get', side_effect=Queue.Empty)
+    
+    # Run the loop logic manually (synchronously)
+    receiver._handle_messages()
+
+    # Final Assertion: Verify is_speaking was set to False
+    # The code execution order is: read (returns True) -> print -> set (to False)
+    # assert_called_with checks the arguments of the LAST call, which is the setter.
+    assert mock_state.is_speaking is False
 
-    receiver._handle_gestures()
 
 def test_handle_messages_runtime_error(mocker):
     """
@@ -266,7 +303,7 @@ def test_handle_messages_runtime_error(mocker):
     # Initialize receiver with the mock context
     receiver = ActuationReceiver(mock_zmq_ctx)
     
-    mock_state.exit_event.is_set.side_effect = [False, True, True, True ]
+    mock_state.exit_event.is_set.side_effect = [False, True]
 
     receiver._message_queue.put("Test Message")
     
@@ -304,61 +341,54 @@ def test_clear_queue(mocker):
     # Assert the queue is empty
     assert receiver._message_queue.qsize() == 0
 
-def test_gesture_no_data(mocker):
-    mock_zmq = mock.Mock()
-    receiver = ActuationReceiver(mock_zmq)
+def test_gesture_no_data(zmq_context, mocker):
+    receiver = ActuationReceiver(zmq_context)
     receiver._handle_gesture({"endpoint": "actuate/gesture/single", "data": ""}, True)
     # Just ensuring no crash
 
 
-def test_gesture_invalid_data(mocker):
-    mock_zmq = mock.Mock()
-    receiver = ActuationReceiver(mock_zmq)
+def test_gesture_invalid_data(zmq_context, mocker):
+    receiver = ActuationReceiver(zmq_context)
     receiver._handle_gesture({"endpoint": "actuate/gesture/single", "data": 123}, True)
     # No crash expected
 
 
-def test_gesture_single_not_found(mocker):
-    mock_zmq = mock.Mock()
+def test_gesture_single_not_found(zmq_context, mocker):
     mock_tags = mocker.patch("robot_interface.endpoints.actuation_receiver.GestureTags")
     mock_tags.single_gestures = ["wave", "bow"]  # allowed single gestures
-    receiver = ActuationReceiver(mock_zmq)
+
+    receiver = ActuationReceiver(zmq_context)
     receiver._handle_gesture({"endpoint": "actuate/gesture/single", "data": "unknown_gesture"}, True)
     # No crash expected
 
 
-def test_gesture_tag_not_found(mocker):
-    mock_zmq = mock.Mock()
+def test_gesture_tag_not_found(zmq_context, mocker):
     mock_tags = mocker.patch("robot_interface.endpoints.actuation_receiver.GestureTags")
     mock_tags.tags = ["happy", "sad"]
 
-    receiver = ActuationReceiver(mock_zmq)
+    receiver = ActuationReceiver(zmq_context)
     receiver._handle_gesture({"endpoint": "actuate/gesture/tag", "data": "not_a_tag"}, False)
     # No crash expected
 
 
-def test_gesture_no_qi_session( mocker):
-    mock_zmq = mock.Mock()
+def test_gesture_no_qi_session(zmq_context, mocker):
     mock_state = mocker.patch("robot_interface.endpoints.actuation_receiver.state")
     mock_state.qi_session = None
 
     mock_tags = mocker.patch("robot_interface.endpoints.actuation_receiver.GestureTags")
     mock_tags.single_gestures = ["hello"]
 
-    receiver = ActuationReceiver(mock_zmq)
+    receiver = ActuationReceiver(zmq_context)
     receiver._handle_gesture({"endpoint": "actuate/gesture/single", "data": "hello"}, True)
     # No crash, path returns early
 
 
-def test_gesture_single_success(mocker):
-    mock_zmq = mock.Mock()
+def test_gesture_single_success(zmq_context, mocker):
     mock_state = mocker.patch("robot_interface.endpoints.actuation_receiver.state")
-    # Allow loops to run
-    mock_state.exit_event.is_set.return_value = False 
-
     mock_qi = mock.Mock()
     sys.modules["qi"] = mock_qi
 
+    # Setup gesture settings
     mock_tags = mocker.patch("robot_interface.endpoints.actuation_receiver.GestureTags")
     mock_tags.single_gestures = ["wave"]
 
@@ -366,25 +396,17 @@ def test_gesture_single_success(mocker):
     mock_state.qi_session = mock.Mock()
     mock_state.qi_session.service.return_value = mock_animation_service
 
-    receiver = ActuationReceiver(mock_zmq)
+    receiver = ActuationReceiver(zmq_context)
     receiver._handle_gesture({"endpoint": "actuate/gesture/single", "data": "wave"}, True)
 
-    time.sleep(0.2) 
-
-    mock_state.qi_session.service.assert_called_with("ALAnimationPlayer")
-    mock_animation_service.run.assert_called_with("wave")
-
-    # CLEANUP: Signal exit AND join threads
-    mock_state.exit_event.is_set.return_value = True
-    receiver.message_thread.join(timeout=1.0)
-    receiver.gesture_thread.join(timeout=1.0)
+    mock_state.qi_session.service.assert_called_once_with("ALAnimationPlayer")
+    getattr(mock_qi, "async").assert_called_once()
+    assert getattr(mock_qi, "async").call_args[0][0] == mock_animation_service.run
+    assert getattr(mock_qi, "async").call_args[0][1] == "wave"
 
 
-def test_gesture_tag_success(mocker):
-    mock_zmq = mock.Mock()
+def test_gesture_tag_success(zmq_context, mocker):
     mock_state = mocker.patch("robot_interface.endpoints.actuation_receiver.state")
-    mock_state.exit_event.is_set.return_value = False
-
     mock_qi = mock.Mock()
     sys.modules["qi"] = mock_qi
 
@@ -395,26 +417,21 @@ def test_gesture_tag_success(mocker):
     mock_state.qi_session = mock.Mock()
     mock_state.qi_session.service.return_value = mock_animation_service
 
-    receiver = ActuationReceiver(mock_zmq)
+    receiver = ActuationReceiver(zmq_context)
     receiver._handle_gesture({"endpoint": "actuate/gesture/tag", "data": "greeting"}, False)
 
-    time.sleep(0.2)
-
-    mock_state.qi_session.service.assert_called_with("ALAnimationPlayer")
-    mock_animation_service.run.assert_called_with("greeting")
-    
-    # CLEANUP: Signal exit AND join threads
-    mock_state.exit_event.is_set.return_value = True
-    receiver.message_thread.join(timeout=1.0)
-    receiver.gesture_thread.join(timeout=1.0)
+    mock_state.qi_session.service.assert_called_once_with("ALAnimationPlayer")
+    getattr(mock_qi, "async").assert_called_once()
+    assert getattr(mock_qi, "async").call_args[0][0] == mock_animation_service.runTag
+    assert getattr(mock_qi, "async").call_args[0][1] == "greeting"
 
 
-def test_handle_message_all_routes(mocker):
+def test_handle_message_all_routes(zmq_context, mocker):
     """
     Ensures all handle_message endpoint branches route correctly.
     """
-    mock_zmq = mock.Mock()
-    receiver = ActuationReceiver(mock_zmq)
+    receiver = ActuationReceiver(zmq_context)
+
     mock_speech = mocker.patch.object(receiver, "_handle_speech")
     mock_gesture = mocker.patch.object(receiver, "_handle_gesture")
 
@@ -426,13 +443,12 @@ def test_handle_message_all_routes(mocker):
     assert mock_gesture.call_count == 2
 
 
-def test_endpoint_description(mocker):
-    mock_zmq = mock.Mock()
+def test_endpoint_description(zmq_context, mocker):
     mock_tags = mocker.patch("robot_interface.endpoints.actuation_receiver.GestureTags")
     mock_tags.tags = ["happy"]
     mock_tags.single_gestures = ["wave"]
 
-    receiver = ActuationReceiver(mock_zmq)
+    receiver = ActuationReceiver(zmq_context)
     desc = receiver.endpoint_description()
 
     assert "gestures" in desc
@@ -442,20 +458,26 @@ def test_endpoint_description(mocker):
     assert desc["single_gestures"] == ["wave"]
 
 
-def test_gesture_single_real_gesturetags(mocker):
-    mock_zmq = mock.Mock()
+def test_gesture_single_real_gesturetags(zmq_context, mocker):
+    """
+    Uses the real GestureTags (no mocking) to ensure the receiver
+    references GestureTags.single_gestures correctly.
+    """
+    # Ensure qi session exists so we pass the early return
     mock_state = mocker.patch("robot_interface.endpoints.actuation_receiver.state")
     mock_state.qi_session = mock.Mock()
-    mock_state.exit_event.is_set.return_value = False
 
+    # Mock qi.async to avoid real async calls
     mock_qi = mock.Mock()
     sys.modules["qi"] = mock_qi
 
+    # Mock animation service
     mock_animation_service = mock.Mock()
     mock_state.qi_session.service.return_value = mock_animation_service
 
-    receiver = ActuationReceiver(mock_zmq)
+    receiver = ActuationReceiver(zmq_context)
 
+    # Pick a real gesture from GestureTags.single_gestures
     assert len(GestureTags.single_gestures) > 0, "GestureTags.single_gestures must not be empty"
     gesture = GestureTags.single_gestures[0]
 
@@ -464,85 +486,8 @@ def test_gesture_single_real_gesturetags(mocker):
         is_single=True,
     )
 
-    time.sleep(0.2)
-    mock_state.qi_session.service.assert_called_with("ALAnimationPlayer")
-    mock_animation_service.run.assert_called_with(gesture)
+    mock_state.qi_session.service.assert_called_once_with("ALAnimationPlayer")
+    getattr(mock_qi, "async").assert_called_once()
+    assert getattr(mock_qi, "async").call_args[0][0] == mock_animation_service.run
+    assert getattr(mock_qi, "async").call_args[0][1] == gesture
 
-    # CLEANUP: Signal exit AND join threads
-    mock_state.exit_event.is_set.return_value = True
-    receiver.message_thread.join(timeout=1.0)
-    receiver.gesture_thread.join(timeout=1.0)
-
-def test_clear_gesture_queue(mocker):
-    # Prevent background threads from eating the items
-    mocker.patch("threading.Thread")
-    
-    mock_zmq_ctx = mock.Mock()
-    receiver = ActuationReceiver(mock_zmq_ctx)
-    
-    # Populate the queue
-    receiver._gesture_queue.put("gesture1")
-    receiver._gesture_queue.put("gesture2")
-    
-    assert receiver._gesture_queue.qsize() == 2
-
-    # Clear the queue
-    receiver.clear_gesture_queue()
-
-    # Assert the queue is empty
-    assert receiver._gesture_queue.qsize() == 0
-
-
-def test_gesture_priority_clears_queue(mocker):
-    mocker.patch("threading.Thread")
-    mock_state = mocker.patch("robot_interface.endpoints.actuation_receiver.state")
-    
-    # Mock QI and Tags so valid checks pass
-    mock_qi = mock.Mock()
-    sys.modules["qi"] = mock_qi
-    mock_tags = mocker.patch("robot_interface.endpoints.actuation_receiver.GestureTags")
-    mock_tags.single_gestures = ["urgent_wave"]
-    
-    # Setup Animation Service
-    mock_anim = mock.Mock()
-    mock_state.qi_session.service.return_value = mock_anim
-
-    mock_zmq_ctx = mock.Mock()
-    receiver = ActuationReceiver(mock_zmq_ctx)
-    
-    # Pre-fill queue with "slow" gestures
-    receiver._gesture_queue.put("slow_gesture_1")
-    receiver._gesture_queue.put("slow_gesture_2")
-    
-    assert receiver._gesture_queue.qsize() == 2
-
-    # Send priority gesture
-    priority_msg = {
-        "endpoint": "actuate/gesture/single", 
-        "data": "urgent_wave",
-        "is_priority": True,
-    }
-    receiver._handle_gesture(priority_msg, is_single=True)
-
-    # Assert old items are gone and only new one remains
-    assert receiver._gesture_queue.qsize() == 1
-    assert receiver._gesture_queue.get() == "urgent_wave"
-
-
-def test_handle_gestures_loop_empty(mocker):
-    mocker.patch("threading.Thread")
-    mock_state = mocker.patch("robot_interface.endpoints.actuation_receiver.state")
-    
-    mock_zmq_ctx = mock.Mock()
-    receiver = ActuationReceiver(mock_zmq_ctx)
-    
-    # Run loop exactly once
-    mock_state.exit_event.is_set.side_effect = [False, True, True, True]
-    
-    # We don't put anything in the queue, so .get(timeout=0.1) will raise Queue.Empty.
-    # The code should catch it and pass.
-    receiver._handle_gestures()
-    
-    # If we reached here without raising an exception, the test passes.
-    # We can assert that the queue is still valid/empty.
-    assert receiver._gesture_queue.empty()

test/unit/test_audio_sender.py

@@ -1,4 +1,10 @@
-# coding=utf-8
+# -*- coding: utf-8 -*-
+"""
+This program has been developed by students from the bachelor Computer Science at Utrecht
+University within the Software Project course.
+© Copyright Utrecht University (Department of Information and Computing Sciences)
+"""
+
 import os
 
 import mock
@@ -33,7 +39,6 @@ def test_no_microphone(zmq_context, mocker):
 
     sender.start()
     assert sender.thread is None
-    mock_info_logger.assert_called()
 
     sender.wait_until_done()  # Should return early because we didn't start a thread
 
@@ -73,11 +78,12 @@ def test_sending_audio(mocker):
     mock_choose_mic.return_value = {"name": u"Some mic", "index": 0L}
 
     mock_state = mocker.patch("robot_interface.endpoints.audio_sender.state")
-    mock_state.exit_event.is_set.side_effect = [False, True]
+    mock_state.exit_event.is_set.side_effect = [False, False, True]
 
     mock_zmq_context = mock.Mock()
     send_socket = mock.Mock()
     
+    mock_state.is_speaking = False
     # If there's something wrong with the microphone, it will raise an IOError when `read`ing.
     stream = mock.Mock()
     stream.read = _fake_read
@@ -92,6 +98,247 @@ def test_sending_audio(mocker):
 
     send_socket.assert_called()
 
+# SENDING PAUSE RESUME?
+
+def test_stream_initial_wait_exit(mocker):
+    mock_choose_mic = mocker.patch("robot_interface.endpoints.audio_sender.choose_mic")
+    mock_choose_mic.return_value = {"name": u"Some mic", "index": 0L}
+
+    mock_pyaudio_cls = mocker.patch("robot_interface.endpoints.audio_sender.pyaudio.PyAudio")
+    mock_pyaudio_instance = mock_pyaudio_cls.return_value
+
+    mock_state = mocker.patch("robot_interface.endpoints.audio_sender.state")
+    mock_state.exit_event.is_set.return_value = True
+    mock_state.active_event.is_set.return_value = False
+    
+    mock_zmq_context = mock.Mock()
+
+    sender = AudioSender(mock_zmq_context)
+    sender._stream()
+
+    mock_pyaudio_instance.open.assert_not_called()
+
+def test_stream_pause_and_resume(mocker):
+    mock_stream = mock.Mock()
+    mock_stream.read.return_value = b"data"
+
+    mock_pyaudio_cls = mocker.patch("robot_interface.endpoints.audio_sender.pyaudio.PyAudio")
+    mock_pyaudio_instance = mock_pyaudio_cls.return_value
+    mock_pyaudio_instance.open.return_value = mock_stream
+
+    mock_state = mocker.patch("robot_interface.endpoints.audio_sender.state")
+    mock_state.active_event.is_set.return_value = False
+
+    mock_state.exit_event.is_set.side_effect = [False, False, False, True]
+
+    mock_zmq_context = mock.Mock()
+    sender = AudioSender(mock_zmq_context)
+    sender.socket = mock.Mock()
+
+    sender._stream()
+
+    assert mock_pyaudio_instance.open.call_count == 2
+
+    assert mock_stream.close.call_count == 2
+    assert mock_stream.stop_stream.call_count == 2
+
+    assert mock_state.active_event.wait.called
+
+def test_stream_exit_during_pause(mocker):
+    mock_stream = mock.Mock()
+    mock_pyaudio_cls = mocker.patch("robot_interface.endpoints.audio_sender.pyaudio.PyAudio")
+    mock_pyaudio_instance = mock_pyaudio_cls.return_value
+    mock_pyaudio_instance.open.return_value = mock_stream
+
+    mock_state = mocker.patch("robot_interface.endpoints.audio_sender.state")
+    mock_state.active_event.is_set.return_value = False
+
+    mock_state.exit_event.is_set.side_effect = [False, False, True]
+
+    mock_zmq_context = mock.Mock()
+    sender = AudioSender(mock_zmq_context)
+    sender._stream()
+
+    assert mock_pyaudio_instance.open.call_count == 1
+    assert mock_stream.close.call_count == 1
+
+def test_stream_read_error_recovery(mocker):
+    stream_fail = mock.Mock()
+    stream_fail.read.side_effect = IOError("Overflow")
+
+    stream_ok = mock.Mock()
+    stream_ok.read.return_value = b"data"
+
+    mock_pyaudio_cls = mocker.patch("robot_interface.endpoints.audio_sender.pyaudio.PyAudio")
+    mock_pyaudio_instance = mock_pyaudio_cls.return_value
+    mock_pyaudio_instance.open.side_effect = [stream_fail, stream_ok]
+
+    mock_state = mocker.patch("robot_interface.endpoints.audio_sender.state")
+    mock_state.active_event.is_set.return_value = True
+    mock_state.exit_event.is_set.side_effect = [False, False, False, True]
+
+    mock_zmq_context = mock.Mock()
+    sender = AudioSender(mock_zmq_context)
+    sender.socket = mock.Mock()
+    sender._stream()
+    stream_fail.close.assert_called()
+    assert mock_pyaudio_instance.open.call_count == 2
+    sender.socket.send.assert_called_with(b"data")
+
+
+def test_stream_fatal_error(mocker):
+    mock_pyaudio_cls = mocker.patch("robot_interface.endpoints.audio_sender.pyaudio.PyAudio")
+    mock_pyaudio_instance = mock_pyaudio_cls.return_value
+    mock_pyaudio_instance.open.side_effect = IOError("Fatal error")
+
+    mock_state = mocker.patch("robot_interface.endpoints.audio_sender.state")
+    mock_state.active_event.is_set.return_value = True
+    mock_state.exit_event.is_set.return_value = False
+
+    mock_zmq_context = mock.Mock()
+    sender = AudioSender(mock_zmq_context)
+    sender._stream()
+
+def test_wait_until_done(mocker):
+    mock_zmq_context = mock.Mock()
+    
+    sender = AudioSender(mock_zmq_context)
+
+    sender.wait_until_done()
+
+    mock_thread = mocker.Mock()
+    sender.thread = mock_thread
+    sender.wait_until_done()
+
+    mock_thread.join.assert_called_once()
+    assert sender.thread is None
+
+def test_stream_pause_close_error(mocker):
+    """
+    Tests that an IOError during stream closure (when pausing) is ignored,
+    covering the 'pass' statement in the pause logic.
+    """
+    mock_stream = mock.Mock()
+    mock_stream.read.return_value = b"data"
+    # Raise IOError when stopping the stream during pause
+    mock_stream.stop_stream.side_effect = IOError("Failed to stop")
+
+    mock_pyaudio_cls = mocker.patch("robot_interface.endpoints.audio_sender.pyaudio.PyAudio")
+    mock_pyaudio_instance = mock_pyaudio_cls.return_value
+    mock_pyaudio_instance.open.return_value = mock_stream
+
+    mock_state = mocker.patch("robot_interface.endpoints.audio_sender.state")
+    
+    # 1. First False triggers the pause block
+    # 2. Second True resumes the loop
+    mock_state.active_event.is_set.side_effect = [False, True] 
+
+    mock_state.exit_event.is_set.side_effect = [False, False, False, True]
+
+    mock_zmq_context = mock.Mock()
+    sender = AudioSender(mock_zmq_context)
+    sender.socket = mock.Mock()
+
+    sender._stream()
+
+    # Verification: The error should be swallowed, and the stream should re-open
+    assert mock_stream.stop_stream.called
+    assert mock_pyaudio_instance.open.call_count == 2
+
+
+def test_stream_finally_close_error(mocker):
+    """
+    Tests that an IOError during stream closure in the finally block is ignored,
+    covering the 'pass' statement in the finally logic.
+    """
+    mock_stream = mock.Mock()
+    mock_stream.read.return_value = b"data"
+    # Raise IOError when stopping the stream at exit
+    mock_stream.stop_stream.side_effect = IOError("Cleanup failed")
+
+    mock_pyaudio_cls = mocker.patch("robot_interface.endpoints.audio_sender.pyaudio.PyAudio")
+    mock_pyaudio_instance = mock_pyaudio_cls.return_value
+    mock_pyaudio_instance.open.return_value = mock_stream
+
+    mock_state = mocker.patch("robot_interface.endpoints.audio_sender.state")
+    mock_state.active_event.is_set.return_value = True
+    mock_state.exit_event.is_set.side_effect = [False, False, True]
+
+    mock_zmq_context = mock.Mock()
+    sender = AudioSender(mock_zmq_context)
+    sender.socket = mock.Mock()
+
+    # Run
+    sender._stream()
+
+    # Assert: Should finish without raising exception despite the IOError in finally
+    assert mock_stream.stop_stream.called
+
+
+def test_stream_recovery_failure(mocker):
+    """
+    Tests the case where recovering from a read error (re-opening stream) also fails.
+    This ensures the outer try-except catches exceptions from the inner except block.
+    """
+    mock_stream_initial = mock.Mock()
+    # Trigger the read error logic
+    mock_stream_initial.read.side_effect = IOError("Read failed")
+
+    mock_pyaudio_cls = mocker.patch("robot_interface.endpoints.audio_sender.pyaudio.PyAudio")
+    mock_pyaudio_instance = mock_pyaudio_cls.return_value
+    
+    # First open works, Second open (recovery) fails fatally
+    mock_pyaudio_instance.open.side_effect = [
+        mock_stream_initial, 
+        IOError("Recovery failed")
+    ]
+
+    mock_state = mocker.patch("robot_interface.endpoints.audio_sender.state")
+    mock_state.active_event.is_set.return_value = True
+    mock_state.exit_event.is_set.return_value = False
+
+    mock_logger = mocker.patch("robot_interface.endpoints.audio_sender.logger")
+    
+    mock_zmq_context = mock.Mock()
+    sender = AudioSender(mock_zmq_context)
+    sender.socket = mock.Mock()
+
+    sender._stream()
+
+    # Assert we hit the outer error log
+    mock_logger.error.assert_called()
+
+
+
+def test_no_sending_if_speaking(mocker):
+    """
+    Tests the successful sending of audio data over a ZeroMQ socket.
+    """
+    mock_choose_mic = mocker.patch("robot_interface.endpoints.audio_sender.choose_mic")
+    mock_choose_mic.return_value = {"name": u"Some mic", "index": 0L}
+
+    mock_state = mocker.patch("robot_interface.endpoints.audio_sender.state")
+    mock_state.exit_event.is_set.side_effect = [False, True]
+
+    mock_zmq_context = mock.Mock()
+    send_socket = mock.Mock()
+    
+    mock_state.is_speaking = True
+
+    # If there's something wrong with the microphone, it will raise an IOError when `read`ing.
+    stream = mock.Mock()
+    stream.read = _fake_read
+
+    sender = AudioSender(mock_zmq_context)
+    sender.socket.send = send_socket
+    sender.audio.open = mock.Mock()
+    sender.audio.open.return_value = stream
+
+    sender.start()
+    sender.wait_until_done()
+
+    send_socket.assert_not_called()
+
 
 def _fake_read_error(num_frames):
     """

test/unit/test_get_config.py

@@ -1,3 +1,10 @@
+# -*- coding: utf-8 -*-
+"""
+This program has been developed by students from the bachelor Computer Science at Utrecht
+University within the Software Project course.
+© Copyright Utrecht University (Department of Information and Computing Sciences)
+"""
+
 from robot_interface.utils.get_config import get_config
 
 

test/unit/test_main.py

@@ -1,3 +1,10 @@
+# -*- coding: utf-8 -*-
+"""
+This program has been developed by students from the bachelor Computer Science at Utrecht
+University within the Software Project course.
+© Copyright Utrecht University (Department of Information and Computing Sciences)
+"""
+
 import pytest
 import threading
 import zmq

test/unit/test_main_receiver.py

@@ -1,3 +1,10 @@
+# -*- coding: utf-8 -*-
+"""
+This program has been developed by students from the bachelor Computer Science at Utrecht
+University within the Software Project course.
+© Copyright Utrecht University (Department of Information and Computing Sciences)
+"""
+
 import mock
 import pytest
 import zmq

test/unit/test_microphone_utils.py

@@ -1,4 +1,10 @@
-# coding=utf-8
+# -*- coding: utf-8 -*-
+"""
+This program has been developed by students from the bachelor Computer Science at Utrecht
+University within the Software Project course.
+© Copyright Utrecht University (Department of Information and Computing Sciences)
+"""
+
 import mock
 import pytest
 

test/unit/test_qi_utils.py

@@ -1,3 +1,10 @@
+# -*- coding: utf-8 -*-
+"""
+This program has been developed by students from the bachelor Computer Science at Utrecht
+University within the Software Project course.
+© Copyright Utrecht University (Department of Information and Computing Sciences)
+"""
+
 import sys
 
 # Import module under test

test/unit/test_reciever_base.py

@@ -1,3 +1,10 @@
+# -*- coding: utf-8 -*-
+"""
+This program has been developed by students from the bachelor Computer Science at Utrecht
+University within the Software Project course.
+© Copyright Utrecht University (Department of Information and Computing Sciences)
+"""
+
 import pytest
 from robot_interface.endpoints.receiver_base import ReceiverBase
 

test/unit/test_socket_base.py

@@ -1,3 +1,10 @@
+# -*- coding: utf-8 -*-
+"""
+This program has been developed by students from the bachelor Computer Science at Utrecht
+University within the Software Project course.
+© Copyright Utrecht University (Department of Information and Computing Sciences)
+"""
+
 import mock
 import zmq
 from robot_interface.endpoints.socket_base import SocketBase

test/unit/test_state.py

@@ -1,3 +1,10 @@
+# -*- coding: utf-8 -*-
+"""
+This program has been developed by students from the bachelor Computer Science at Utrecht
+University within the Software Project course.
+© Copyright Utrecht University (Department of Information and Computing Sciences)
+"""
+
 import threading
 import signal
 import pytest

test/unit/test_time_block.py

@@ -1,3 +1,10 @@
+# -*- coding: utf-8 -*-
+"""
+This program has been developed by students from the bachelor Computer Science at Utrecht
+University within the Software Project course.
+© Copyright Utrecht University (Department of Information and Computing Sciences)
+"""
+
 import time
 
 import mock

test/unit/test_video_sender.py

@@ -1,99 +1,171 @@
-# coding=utf-8
+# -*- coding: utf-8 -*-
+"""
+This program has been developed by students from the bachelor Computer Science at Utrecht
+University within the Software Project course.
+© Copyright Utrecht University (Department of Information and Computing Sciences)
+"""
 
 import mock
 import pytest
 import zmq
 
 from robot_interface.endpoints.video_sender import VideoSender
-from robot_interface.state import state
-from robot_interface.core.config import settings
-
 
 @pytest.fixture
 def zmq_context():
-    """Provide a ZMQ context."""
-    yield zmq.Context()
+    """
+    Yields a real ZMQ context for socket creation.
+    """
+    context = zmq.Context()
+    yield context
+    context.term()
 
+def test_init_defaults(zmq_context, mocker):
+    """
+    Test initialization of the VideoSender.
+    """
+    # We patch settings to ensure valid port access inside the class logic, 
+    # although the default arg is evaluated at import time.
+    mocker.patch("robot_interface.endpoints.video_sender.settings")
     
-def _patch_basics(mocker):
-    """Common patches: prevent real threads, port binds, and state errors."""
-    mocker.patch("robot_interface.endpoints.socket_base.zmq.Socket.bind")
-    mocker.patch("robot_interface.endpoints.video_sender.threading.Thread")
-    mocker.patch.object(state, "is_initialized", True)
+    mock_zmq = mock.Mock()
+    sender = VideoSender(mock_zmq)
     
+    # Verify socket type is PUB
+    assert sender.identifier == "video"
 
-def _patch_exit_event(mocker):
-    """Make exit_event stop the loop after one iteration."""
-    fake_event = mock.Mock()
-    fake_event.is_set.side_effect = [False, True]
-    mocker.patch.object(state, "exit_event", fake_event)
+def test_start_no_qi_session(mocker):
+    """
+    Test that the loop does not start if no Qi session is available.
+    """
+    # Mock state to return None for qi_session
+    mock_state = mocker.patch("robot_interface.endpoints.video_sender.state")
+    mock_state.qi_session = None
     
+    mock_threading = mocker.patch("robot_interface.endpoints.video_sender.threading")
 
-def test_no_qi_session(zmq_context, mocker):
-    """Video loop should not start without a qi_session."""
-    _patch_basics(mocker)
-    mocker.patch.object(state, "qi_session", None)
-
-    sender = VideoSender(zmq_context)
+    mock_zmq = mock.Mock()
+    sender = VideoSender(mock_zmq)
     sender.start_video_rcv()
 
-    assert not hasattr(sender, "thread")
+    # Assertions
+    mock_threading.Thread.assert_not_called()
 
-
-def test_video_streaming(zmq_context, mocker):
-    """VideoSender should send retrieved image data."""
-    _patch_basics(mocker)
-    _patch_exit_event(mocker)
-
-    # Pepper's image buffer lives at index 6
-    mocker.patch.object(settings.video_config, "image_buffer", 6)
+def test_start_success(mocker):
+    """
+    Test successful startup of the video receiver thread.
+    """
+    # Mock the Qi Session and Service
+    mock_state = mocker.patch("robot_interface.endpoints.video_sender.state")
+    mock_session = mock.Mock()
+    mock_state.qi_session = mock_session
     
     mock_video_service = mock.Mock()
-    mock_video_service.getImageRemote.return_value = [None]*6 + ["fake_img"]
+    mock_session.service.return_value = mock_video_service
+    mock_video_service.subscribeCamera.return_value = "test_subscriber_id"
 
-    fake_session = mock.Mock()
-    fake_session.service.return_value = mock_video_service
-    mocker.patch.object(state, "qi_session", fake_session)
+    # Mock Settings
+    mock_settings = mocker.patch("robot_interface.endpoints.video_sender.settings")
+    mock_settings.video_config.camera_index = 0
+    mock_settings.video_config.resolution = 2
+    mock_settings.video_config.color_space = 11
+    mock_settings.video_config.fps = 30
+    mock_settings.video_config.stream_name = "test_stream"
 
-    mocker.patch.object(
-        fake_session.service("ALVideoDevice"),
-        "subscribeCamera",
-        return_value="stream_name"
-    )
-
-    sender = VideoSender(zmq_context)
-    send_socket = mock.Mock()
-    sender.socket.send = send_socket
+    mock_threading = mocker.patch("robot_interface.endpoints.video_sender.threading")
 
+    # Run
+    mock_zmq = mock.Mock()  
+    sender = VideoSender(mock_zmq)
     sender.start_video_rcv()
-    sender.video_rcv_loop(mock_video_service, "stream_name")
 
-    send_socket.assert_called_with("fake_img")
+    # Assertions
+    mock_session.service.assert_called_with("ALVideoDevice")
+    mock_video_service.subscribeCamera.assert_called_with("test_stream", 0, 2, 11, 30)
     
+    mock_threading.Thread.assert_called_once()
+    # Verify arguments passed to the thread target
+    call_args = mock_threading.Thread.call_args[1]
+    assert call_args["target"] == sender.video_rcv_loop
+    assert call_args["args"] == (mock_video_service, "test_subscriber_id")
     
-def test_video_receive_error(zmq_context, mocker):
-    """Errors retrieving images should not call send()."""
-    _patch_basics(mocker)
-    _patch_exit_event(mocker)
+    # Ensure thread was started
+    mock_threading.Thread.return_value.start.assert_called_once()
 
-    mock_video_service = mock.Mock()
-    mock_video_service.getImageRemote.side_effect = Exception("boom")
+def test_video_loop_happy_path(mocker):
+    """
+    Test the main loop: Wait -> Get Image -> Send -> Repeat/Exit.
+    """
+    # Mock settings for image buffer index
+    mock_settings = mocker.patch("robot_interface.endpoints.video_sender.settings")
+    mock_settings.video_config.image_buffer = 6 
 
-    fake_session = mock.Mock()
-    fake_session.service.return_value = mock_video_service
-    mocker.patch.object(state, "qi_session", fake_session)
+    # Mock Video Service to return a fake image structure
+    # Standard NaoQi image is a list, binary data is usually at index 6
+    fake_image_data = b"binary_jpeg_data"
+    fake_image_list = [0] * 7
+    fake_image_list[6] = fake_image_data
     
-    mocker.patch.object(
-        fake_session.service("ALVideoDevice"),
-        "subscribeCamera",
-        return_value="stream_name"
-    )
+    mock_service = mock.Mock()
+    mock_service.getImageRemote.return_value = fake_image_list
 
-    sender = VideoSender(zmq_context)
-    send_socket = mock.Mock()
-    sender.socket.send = send_socket
+    # Mock Events: 
+    # exit_event: False (start), False (loop once), True (break)
+    mock_state = mocker.patch("robot_interface.endpoints.video_sender.state")
+    mock_state.exit_event.is_set.side_effect = [False, False, True]
 
-    sender.start_video_rcv()
-    sender.video_rcv_loop(mock_video_service, "stream_name")
+    # Run
+    mock_zmq = mock.Mock()
+    sender = VideoSender(mock_zmq)
+    sender.socket = mock.Mock() # Mock the socket to verify send
     
-    send_socket.assert_not_called()
+    sender.video_rcv_loop(mock_service, "sub_id")
+
+    # Assertions
+    mock_state.active_event.wait.assert_called()
+    mock_service.getImageRemote.assert_called_with("sub_id")
+    sender.socket.send.assert_called_with(fake_image_data)
+
+def test_video_loop_exit_during_wait(zmq_context, mocker):
+    """
+    Test that the loop breaks immediately if exit_event is set while waiting.
+    """
+    mock_service = mock.Mock()
+    mock_state = mocker.patch("robot_interface.endpoints.video_sender.state")
+    
+    # 1. Loop check: False (enter loop)
+    # 2. Wait happens (mock returns instantly)
+    # 3. Post-wait check: True (break)
+    mock_state.exit_event.is_set.side_effect = [False, True]
+
+    mock_zqm = mock.Mock()
+    sender = VideoSender(mock_zqm)
+    sender.video_rcv_loop(mock_service, "sub_id")
+
+    # Assert we never tried to get an image
+    mock_service.getImageRemote.assert_not_called()
+
+def test_video_loop_exception_handling(zmq_context, mocker):
+    """
+    Test that exceptions during image retrieval are caught and logged, 
+    and do not crash the thread.
+    """
+    mock_settings = mocker.patch("robot_interface.endpoints.video_sender.settings")
+    
+    mock_service = mock.Mock()
+    # First call raises Exception, Second call works (if we allowed it, but we exit)
+    mock_service.getImageRemote.side_effect = Exception("Camera disconnected")
+
+    mock_state = mocker.patch("robot_interface.endpoints.video_sender.state")
+    # Loop runs once then exits
+    mock_state.exit_event.is_set.side_effect = [False, False, True]
+
+    mock_zmq = mock.Mock()
+    sender = VideoSender(mock_zmq)
+    sender.socket = mock.Mock()
+    
+    sender.video_rcv_loop(mock_service, "sub_id")
+
+    # Assertions
+    # Ensure loop didn't crash; it should have completed the iteration and checked exit_event
+    assert mock_state.exit_event.is_set.call_count >= 2