|
|
|
|
@@ -1,65 +1,114 @@
|
|
|
|
|
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():
|
|
|
|
|
def test_force_speech_clears_queue(mocker):
|
|
|
|
|
"""
|
|
|
|
|
A pytest fixture that creates and yields a ZMQ context.
|
|
|
|
|
|
|
|
|
|
:return: An initialized ZeroMQ context.
|
|
|
|
|
:rtype: zmq.Context
|
|
|
|
|
Tests that a force speech message clears the existing queue
|
|
|
|
|
and places the high-priority message at the front.
|
|
|
|
|
"""
|
|
|
|
|
context = zmq.Context()
|
|
|
|
|
yield context
|
|
|
|
|
mocker.patch("threading.Thread")
|
|
|
|
|
mock_state = mocker.patch("robot_interface.endpoints.actuation_receiver.state")
|
|
|
|
|
|
|
|
|
|
mock_qi = mock.Mock()
|
|
|
|
|
sys.modules["qi"] = mock_qi
|
|
|
|
|
|
|
|
|
|
mock_tts_service = mock.Mock()
|
|
|
|
|
mock_state.qi_session.service.return_value = mock_tts_service
|
|
|
|
|
|
|
|
|
|
def test_handle_unimplemented_endpoint(zmq_context):
|
|
|
|
|
# Use Mock Context
|
|
|
|
|
mock_zmq_ctx = mock.Mock()
|
|
|
|
|
receiver = ActuationReceiver(mock_zmq_ctx)
|
|
|
|
|
|
|
|
|
|
receiver._message_queue.put("old_message_1")
|
|
|
|
|
receiver._message_queue.put("old_message_2")
|
|
|
|
|
|
|
|
|
|
assert receiver._message_queue.qsize() == 2
|
|
|
|
|
|
|
|
|
|
force_msg = {
|
|
|
|
|
"endpoint": "actuate/speech",
|
|
|
|
|
"data": "Emergency Notification",
|
|
|
|
|
"is_priority": True,
|
|
|
|
|
}
|
|
|
|
|
receiver.handle_message(force_msg)
|
|
|
|
|
|
|
|
|
|
assert receiver._message_queue.qsize() == 1
|
|
|
|
|
queued_item = receiver._message_queue.get()
|
|
|
|
|
assert queued_item == "Emergency Notification"
|
|
|
|
|
|
|
|
|
|
def test_handle_unimplemented_endpoint(mocker):
|
|
|
|
|
"""
|
|
|
|
|
Tests that the ``ActuationReceiver.handle_message`` method can
|
|
|
|
|
handle an unknown or unimplemented endpoint without raising an error.
|
|
|
|
|
Tests handling of unknown endpoints.
|
|
|
|
|
"""
|
|
|
|
|
receiver = ActuationReceiver(zmq_context)
|
|
|
|
|
# Should not error
|
|
|
|
|
mocker.patch("threading.Thread")
|
|
|
|
|
|
|
|
|
|
# Use Mock Context
|
|
|
|
|
mock_zmq_ctx = mock.Mock()
|
|
|
|
|
receiver = ActuationReceiver(mock_zmq_ctx)
|
|
|
|
|
|
|
|
|
|
receiver.handle_message({
|
|
|
|
|
"endpoint": "some_endpoint_that_definitely_does_not_exist",
|
|
|
|
|
"data": None,
|
|
|
|
|
})
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def test_speech_message_no_data(zmq_context, mocker):
|
|
|
|
|
def test_speech_message_no_data(mocker):
|
|
|
|
|
"""
|
|
|
|
|
Tests that the message handler logs a warning when a speech actuation
|
|
|
|
|
request (`actuate/speech`) is received but contains empty string data.
|
|
|
|
|
Tests that if the message data is empty, the receiver returns immediately
|
|
|
|
|
WITHOUT attempting to access the global robot state or session.
|
|
|
|
|
"""
|
|
|
|
|
mock_warn = mocker.patch("logging.warn")
|
|
|
|
|
# 1. Prevent background threads from running
|
|
|
|
|
mocker.patch("threading.Thread")
|
|
|
|
|
|
|
|
|
|
# 2. Mock the global state object
|
|
|
|
|
mock_state = mocker.patch("robot_interface.endpoints.actuation_receiver.state")
|
|
|
|
|
|
|
|
|
|
receiver = ActuationReceiver(zmq_context)
|
|
|
|
|
# 3. Create a PropertyMock to track whenever 'qi_session' is accessed
|
|
|
|
|
# We attach it to the class type of the mock so it acts like a real property
|
|
|
|
|
mock_session_prop = mock.PropertyMock(return_value=None)
|
|
|
|
|
type(mock_state).qi_session = mock_session_prop
|
|
|
|
|
|
|
|
|
|
# 4. Initialize Receiver (Mocking the context to avoid ZMQ errors)
|
|
|
|
|
mock_zmq_ctx = mock.Mock()
|
|
|
|
|
receiver = ActuationReceiver(mock_zmq_ctx)
|
|
|
|
|
|
|
|
|
|
# 5. Send empty data
|
|
|
|
|
receiver.handle_message({"endpoint": "actuate/speech", "data": ""})
|
|
|
|
|
|
|
|
|
|
mock_warn.assert_called_with(mock.ANY)
|
|
|
|
|
# 6. Assertion:
|
|
|
|
|
# Because the code does `if not text: return` BEFORE `if not state.qi_session`,
|
|
|
|
|
# the state property should NEVER be read.
|
|
|
|
|
mock_session_prop.assert_not_called()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def test_speech_message_invalid_data(zmq_context, mocker):
|
|
|
|
|
def test_speech_message_invalid_data(mocker):
|
|
|
|
|
"""
|
|
|
|
|
Tests that the message handler logs a warning when a speech actuation
|
|
|
|
|
request (`actuate/speech`) is received with data that is not a string (e.g., a boolean).
|
|
|
|
|
Tests that if the message data is not a string, the function returns.
|
|
|
|
|
:param mocker: Description
|
|
|
|
|
"""
|
|
|
|
|
mock_warn = mocker.patch("logging.warn")
|
|
|
|
|
mocker.patch("threading.Thread")
|
|
|
|
|
|
|
|
|
|
receiver = ActuationReceiver(zmq_context)
|
|
|
|
|
mock_state = mocker.patch("robot_interface.endpoints.actuation_receiver.state")
|
|
|
|
|
|
|
|
|
|
mock_session_prop = mock.PropertyMock(return_value=None)
|
|
|
|
|
type(mock_state).qi_session = mock_session_prop
|
|
|
|
|
|
|
|
|
|
# Use Mock Context
|
|
|
|
|
mock_zmq_ctx = mock.Mock()
|
|
|
|
|
receiver = ActuationReceiver(mock_zmq_ctx)
|
|
|
|
|
|
|
|
|
|
receiver.handle_message({"endpoint": "actuate/speech", "data": True})
|
|
|
|
|
|
|
|
|
|
mock_warn.assert_called_with(mock.ANY)
|
|
|
|
|
# Because the code does `if not text: return` BEFORE `if not state.qi_session`,
|
|
|
|
|
# the state property should NEVER be read.
|
|
|
|
|
mock_session_prop.assert_not_called()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def test_speech_no_qi(zmq_context, mocker):
|
|
|
|
|
def test_speech_no_qi(mocker):
|
|
|
|
|
"""
|
|
|
|
|
Tests the actuation receiver's behavior when processing a speech request
|
|
|
|
|
but the global state does not have an active QI session.
|
|
|
|
|
@@ -69,16 +118,21 @@ def test_speech_no_qi(zmq_context, mocker):
|
|
|
|
|
mock_qi_session = mock.PropertyMock(return_value=None)
|
|
|
|
|
type(mock_state).qi_session = mock_qi_session
|
|
|
|
|
|
|
|
|
|
receiver = ActuationReceiver(zmq_context)
|
|
|
|
|
mock_tts_service = mock.Mock()
|
|
|
|
|
|
|
|
|
|
mock_zmq_ctx = mock.Mock()
|
|
|
|
|
receiver = ActuationReceiver(mock_zmq_ctx)
|
|
|
|
|
receiver._tts_service = mock_tts_service
|
|
|
|
|
|
|
|
|
|
receiver._handle_speech({"endpoint": "actuate/speech", "data": "Some message to speak."})
|
|
|
|
|
|
|
|
|
|
mock_qi_session.assert_called()
|
|
|
|
|
receiver._tts_service.assert_not_called()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def test_speech(zmq_context, mocker):
|
|
|
|
|
def test_speech(mocker):
|
|
|
|
|
"""
|
|
|
|
|
Tests the core speech actuation functionality by mocking the QI TextToSpeech
|
|
|
|
|
service and verifying that it is called correctly.
|
|
|
|
|
service and verifying that the received message is put into the queue.
|
|
|
|
|
"""
|
|
|
|
|
mock_state = mocker.patch("robot_interface.endpoints.actuation_receiver.state")
|
|
|
|
|
|
|
|
|
|
@@ -89,66 +143,222 @@ def test_speech(zmq_context, mocker):
|
|
|
|
|
mock_state.qi_session = mock.Mock()
|
|
|
|
|
mock_state.qi_session.service.return_value = mock_tts_service
|
|
|
|
|
|
|
|
|
|
receiver = ActuationReceiver(zmq_context)
|
|
|
|
|
mock_zmq_ctx = mock.Mock()
|
|
|
|
|
receiver = ActuationReceiver(mock_zmq_ctx)
|
|
|
|
|
|
|
|
|
|
receiver._tts_service = None
|
|
|
|
|
receiver._handle_speech({"endpoint": "actuate/speech", "data": "Some message to speak."})
|
|
|
|
|
|
|
|
|
|
mock_state.qi_session.service.assert_called_once_with("ALTextToSpeech")
|
|
|
|
|
assert receiver._message_queue.qsize() == 1
|
|
|
|
|
|
|
|
|
|
getattr(mock_qi, "async").assert_called_once()
|
|
|
|
|
call_args = getattr(mock_qi, "async").call_args[0]
|
|
|
|
|
assert call_args[0] == mock_tts_service.say
|
|
|
|
|
assert call_args[1] == "Some message to speak."
|
|
|
|
|
queued_item = receiver._message_queue.get()
|
|
|
|
|
assert queued_item == "Some message to speak."
|
|
|
|
|
|
|
|
|
|
def test_speech_priority(mocker):
|
|
|
|
|
"""
|
|
|
|
|
Tests that a priority speech message is handled correctly by clearing the queue
|
|
|
|
|
and placing the priority message at the front.
|
|
|
|
|
"""
|
|
|
|
|
mock_state = mocker.patch("robot_interface.endpoints.actuation_receiver.state")
|
|
|
|
|
|
|
|
|
|
mock_qi = mock.Mock()
|
|
|
|
|
sys.modules["qi"] = mock_qi
|
|
|
|
|
|
|
|
|
|
mock_tts_service = mock.Mock()
|
|
|
|
|
mock_state.qi_session = mock.Mock()
|
|
|
|
|
mock_state.qi_session.service.return_value = mock_tts_service
|
|
|
|
|
|
|
|
|
|
mock_zmq_ctx = mock.Mock()
|
|
|
|
|
receiver = ActuationReceiver(mock_zmq_ctx)
|
|
|
|
|
|
|
|
|
|
receiver._message_queue.put("old_message_1")
|
|
|
|
|
receiver._message_queue.put("old_message_2")
|
|
|
|
|
|
|
|
|
|
assert receiver._message_queue.qsize() == 2
|
|
|
|
|
|
|
|
|
|
priority_msg = {
|
|
|
|
|
"endpoint": "actuate/speech",
|
|
|
|
|
"data": "Urgent Message",
|
|
|
|
|
"is_priority": True,
|
|
|
|
|
}
|
|
|
|
|
receiver._handle_speech(priority_msg)
|
|
|
|
|
|
|
|
|
|
assert receiver._message_queue.qsize() == 1
|
|
|
|
|
queued_item = receiver._message_queue.get()
|
|
|
|
|
assert queued_item == "Urgent Message"
|
|
|
|
|
|
|
|
|
|
def test_handle_messages_loop(mocker):
|
|
|
|
|
"""
|
|
|
|
|
Tests the background consumer loop (_handle_messages) processing an item.
|
|
|
|
|
Runs SYNCHRONOUSLY to ensure coverage tools pick up the lines.
|
|
|
|
|
"""
|
|
|
|
|
# Patch Thread so the real background thread NEVER starts automatically
|
|
|
|
|
mocker.patch("threading.Thread")
|
|
|
|
|
|
|
|
|
|
# Mock state
|
|
|
|
|
mock_state = mocker.patch("robot_interface.endpoints.actuation_receiver.state")
|
|
|
|
|
|
|
|
|
|
# Setup initial speaking state to False (covers "Started speaking" print)
|
|
|
|
|
mock_state.is_speaking = False
|
|
|
|
|
|
|
|
|
|
# Mock the TextToSpeech service
|
|
|
|
|
mock_tts_service = mock.Mock()
|
|
|
|
|
mock_state.qi_session.service.return_value = mock_tts_service
|
|
|
|
|
|
|
|
|
|
# Initialize receiver (Thread is patched, so no thread starts)
|
|
|
|
|
# Use Mock Context to avoid ZMQ errors
|
|
|
|
|
mock_zmq_ctx = mock.Mock()
|
|
|
|
|
receiver = ActuationReceiver(mock_zmq_ctx)
|
|
|
|
|
|
|
|
|
|
# Manually inject service (since lazy loading might handle it, but this is safer)
|
|
|
|
|
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]
|
|
|
|
|
|
|
|
|
|
# Put an item in the queue
|
|
|
|
|
receiver._message_queue.put("Hello World")
|
|
|
|
|
|
|
|
|
|
# RUN MANUALLY in the main thread
|
|
|
|
|
# This executes the code: while -> try -> get -> if print -> speaking=True -> say
|
|
|
|
|
receiver._handle_messages()
|
|
|
|
|
|
|
|
|
|
# Assertions
|
|
|
|
|
assert receiver._message_queue.empty()
|
|
|
|
|
mock_tts_service.say.assert_called_with("Hello World")
|
|
|
|
|
assert mock_state.is_speaking is True
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def test_gesture_no_data(zmq_context, mocker):
|
|
|
|
|
receiver = ActuationReceiver(zmq_context)
|
|
|
|
|
def test_handle_gestures_runtime_error(mocker):
|
|
|
|
|
mocker.patch("threading.Thread")
|
|
|
|
|
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]
|
|
|
|
|
|
|
|
|
|
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
|
|
|
|
|
|
|
|
|
|
receiver._gesture_queue.put("wave")
|
|
|
|
|
|
|
|
|
|
receiver._handle_gestures()
|
|
|
|
|
|
|
|
|
|
def test_handle_messages_runtime_error(mocker):
|
|
|
|
|
"""
|
|
|
|
|
Tests the RuntimeError exception handler (e.g. lost WiFi connection).
|
|
|
|
|
Uses a Mock ZMQ context to avoid 'Address already in use' errors.
|
|
|
|
|
"""
|
|
|
|
|
# Patch Thread so we don't accidentally spawn real threads
|
|
|
|
|
mocker.patch("threading.Thread")
|
|
|
|
|
|
|
|
|
|
# Mock the state and logging
|
|
|
|
|
mock_state = mocker.patch("robot_interface.endpoints.actuation_receiver.state")
|
|
|
|
|
|
|
|
|
|
# Use a MOCK ZMQ context.
|
|
|
|
|
# This prevents the receiver from trying to bind to a real TCP port.
|
|
|
|
|
mock_zmq_ctx = mock.Mock()
|
|
|
|
|
|
|
|
|
|
# Initialize receiver with the mock context
|
|
|
|
|
receiver = ActuationReceiver(mock_zmq_ctx)
|
|
|
|
|
|
|
|
|
|
mock_state.exit_event.is_set.side_effect = [False, True, True, True ]
|
|
|
|
|
|
|
|
|
|
receiver._message_queue.put("Test Message")
|
|
|
|
|
|
|
|
|
|
# Setup: ...BUT the service raises RuntimeError when asked to speak
|
|
|
|
|
mock_tts = mock.Mock()
|
|
|
|
|
mock_tts.say.side_effect = RuntimeError("Connection lost")
|
|
|
|
|
receiver._tts_service = mock_tts
|
|
|
|
|
|
|
|
|
|
# Run the loop logic manually
|
|
|
|
|
receiver._handle_messages()
|
|
|
|
|
|
|
|
|
|
# Assertions
|
|
|
|
|
assert mock_state.exit_event.is_set.called
|
|
|
|
|
|
|
|
|
|
def test_clear_queue(mocker):
|
|
|
|
|
"""
|
|
|
|
|
Tests that the clear_queue method properly drains all items from the message queue.
|
|
|
|
|
"""
|
|
|
|
|
mocker.patch("threading.Thread")
|
|
|
|
|
|
|
|
|
|
# Use Mock Context
|
|
|
|
|
mock_zmq_ctx = mock.Mock()
|
|
|
|
|
receiver = ActuationReceiver(mock_zmq_ctx)
|
|
|
|
|
|
|
|
|
|
# Populate the queue with multiple items
|
|
|
|
|
receiver._message_queue.put("msg1")
|
|
|
|
|
receiver._message_queue.put("msg2")
|
|
|
|
|
receiver._message_queue.put("msg3")
|
|
|
|
|
|
|
|
|
|
assert receiver._message_queue.qsize() == 3
|
|
|
|
|
|
|
|
|
|
# Clear the queue
|
|
|
|
|
receiver.clear_queue()
|
|
|
|
|
|
|
|
|
|
# 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)
|
|
|
|
|
receiver._handle_gesture({"endpoint": "actuate/gesture/single", "data": ""}, True)
|
|
|
|
|
# Just ensuring no crash
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def test_gesture_invalid_data(zmq_context, mocker):
|
|
|
|
|
receiver = ActuationReceiver(zmq_context)
|
|
|
|
|
def test_gesture_invalid_data(mocker):
|
|
|
|
|
mock_zmq = mock.Mock()
|
|
|
|
|
receiver = ActuationReceiver(mock_zmq)
|
|
|
|
|
receiver._handle_gesture({"endpoint": "actuate/gesture/single", "data": 123}, True)
|
|
|
|
|
# No crash expected
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def test_gesture_single_not_found(zmq_context, mocker):
|
|
|
|
|
def test_gesture_single_not_found(mocker):
|
|
|
|
|
mock_zmq = mock.Mock()
|
|
|
|
|
mock_tags = mocker.patch("robot_interface.endpoints.actuation_receiver.GestureTags")
|
|
|
|
|
mock_tags.single_gestures = ["wave", "bow"] # allowed single gestures
|
|
|
|
|
|
|
|
|
|
receiver = ActuationReceiver(zmq_context)
|
|
|
|
|
receiver = ActuationReceiver(mock_zmq)
|
|
|
|
|
receiver._handle_gesture({"endpoint": "actuate/gesture/single", "data": "unknown_gesture"}, True)
|
|
|
|
|
# No crash expected
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def test_gesture_tag_not_found(zmq_context, mocker):
|
|
|
|
|
def test_gesture_tag_not_found(mocker):
|
|
|
|
|
mock_zmq = mock.Mock()
|
|
|
|
|
mock_tags = mocker.patch("robot_interface.endpoints.actuation_receiver.GestureTags")
|
|
|
|
|
mock_tags.tags = ["happy", "sad"]
|
|
|
|
|
|
|
|
|
|
receiver = ActuationReceiver(zmq_context)
|
|
|
|
|
receiver = ActuationReceiver(mock_zmq)
|
|
|
|
|
receiver._handle_gesture({"endpoint": "actuate/gesture/tag", "data": "not_a_tag"}, False)
|
|
|
|
|
# No crash expected
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def test_gesture_no_qi_session(zmq_context, mocker):
|
|
|
|
|
def test_gesture_no_qi_session( mocker):
|
|
|
|
|
mock_zmq = mock.Mock()
|
|
|
|
|
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(zmq_context)
|
|
|
|
|
receiver = ActuationReceiver(mock_zmq)
|
|
|
|
|
receiver._handle_gesture({"endpoint": "actuate/gesture/single", "data": "hello"}, True)
|
|
|
|
|
# No crash, path returns early
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def test_gesture_single_success(zmq_context, mocker):
|
|
|
|
|
def test_gesture_single_success(mocker):
|
|
|
|
|
mock_zmq = mock.Mock()
|
|
|
|
|
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"]
|
|
|
|
|
|
|
|
|
|
@@ -156,17 +366,25 @@ def test_gesture_single_success(zmq_context, mocker):
|
|
|
|
|
mock_state.qi_session = mock.Mock()
|
|
|
|
|
mock_state.qi_session.service.return_value = mock_animation_service
|
|
|
|
|
|
|
|
|
|
receiver = ActuationReceiver(zmq_context)
|
|
|
|
|
receiver = ActuationReceiver(mock_zmq)
|
|
|
|
|
receiver._handle_gesture({"endpoint": "actuate/gesture/single", "data": "wave"}, True)
|
|
|
|
|
|
|
|
|
|
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"
|
|
|
|
|
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)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def test_gesture_tag_success(zmq_context, mocker):
|
|
|
|
|
def test_gesture_tag_success(mocker):
|
|
|
|
|
mock_zmq = mock.Mock()
|
|
|
|
|
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
|
|
|
|
|
|
|
|
|
|
@@ -177,21 +395,26 @@ def test_gesture_tag_success(zmq_context, mocker):
|
|
|
|
|
mock_state.qi_session = mock.Mock()
|
|
|
|
|
mock_state.qi_session.service.return_value = mock_animation_service
|
|
|
|
|
|
|
|
|
|
receiver = ActuationReceiver(zmq_context)
|
|
|
|
|
receiver = ActuationReceiver(mock_zmq)
|
|
|
|
|
receiver._handle_gesture({"endpoint": "actuate/gesture/tag", "data": "greeting"}, False)
|
|
|
|
|
|
|
|
|
|
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"
|
|
|
|
|
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)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def test_handle_message_all_routes(zmq_context, mocker):
|
|
|
|
|
def test_handle_message_all_routes(mocker):
|
|
|
|
|
"""
|
|
|
|
|
Ensures all handle_message endpoint branches route correctly.
|
|
|
|
|
"""
|
|
|
|
|
receiver = ActuationReceiver(zmq_context)
|
|
|
|
|
|
|
|
|
|
mock_zmq = mock.Mock()
|
|
|
|
|
receiver = ActuationReceiver(mock_zmq)
|
|
|
|
|
mock_speech = mocker.patch.object(receiver, "_handle_speech")
|
|
|
|
|
mock_gesture = mocker.patch.object(receiver, "_handle_gesture")
|
|
|
|
|
|
|
|
|
|
@@ -203,12 +426,13 @@ def test_handle_message_all_routes(zmq_context, mocker):
|
|
|
|
|
assert mock_gesture.call_count == 2
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def test_endpoint_description(zmq_context, mocker):
|
|
|
|
|
def test_endpoint_description(mocker):
|
|
|
|
|
mock_zmq = mock.Mock()
|
|
|
|
|
mock_tags = mocker.patch("robot_interface.endpoints.actuation_receiver.GestureTags")
|
|
|
|
|
mock_tags.tags = ["happy"]
|
|
|
|
|
mock_tags.single_gestures = ["wave"]
|
|
|
|
|
|
|
|
|
|
receiver = ActuationReceiver(zmq_context)
|
|
|
|
|
receiver = ActuationReceiver(mock_zmq)
|
|
|
|
|
desc = receiver.endpoint_description()
|
|
|
|
|
|
|
|
|
|
assert "gestures" in desc
|
|
|
|
|
@@ -218,26 +442,20 @@ def test_endpoint_description(zmq_context, mocker):
|
|
|
|
|
assert desc["single_gestures"] == ["wave"]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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
|
|
|
|
|
def test_gesture_single_real_gesturetags(mocker):
|
|
|
|
|
mock_zmq = mock.Mock()
|
|
|
|
|
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(zmq_context)
|
|
|
|
|
receiver = ActuationReceiver(mock_zmq)
|
|
|
|
|
|
|
|
|
|
# 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]
|
|
|
|
|
|
|
|
|
|
@@ -246,8 +464,85 @@ def test_gesture_single_real_gesturetags(zmq_context, mocker):
|
|
|
|
|
is_single=True,
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
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
|
|
|
|
|
time.sleep(0.2)
|
|
|
|
|
mock_state.qi_session.service.assert_called_with("ALAnimationPlayer")
|
|
|
|
|
mock_animation_service.run.assert_called_with(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()
|