def handle(self, connection_id, message_content):
request = UnregisterStateDeltaSubscriberRequest()
request.ParseFromString(message_content)
ack = UnregisterStateDeltaSubscriberResponse()
self._delta_processor.remove_subscriber(connection_id)
ack.status = ack.OK
return HandlerResult(HandlerStatus.RETURN,
message_out=ack,
message_type=self._msg_type)
def test_unregister_subscriber(self):
"""Test that a handler will unregister a subscriber via a requestor's
connection id.
"""
mock_delta_processor = Mock()
handler = StateDeltaUnsubscriberHandler(mock_delta_processor)
request = UnregisterStateDeltaSubscriberRequest().SerializeToString()
response = handler.handle('test_conn_id', request)
mock_delta_processor.remove_subscriber.assert_called_with(
'test_conn_id')
self.assertEqual(HandlerStatus.RETURN, response.status)
self.assertEqual(UnregisterStateDeltaSubscriberResponse.OK,
response.message_out.status)