def Handle(
self,
args: ApiGetFleetspeakPendingMessageCountArgs,
context: Optional[api_call_context.ApiCallContext] = None
) -> ApiGetFleetspeakPendingMessageCountResult:
_CheckFleetspeakConnection()
return ApiGetFleetspeakPendingMessageCountResult(
count=fleetspeak_utils.GetFleetspeakPendingMessageCount(
str(args.client_id)))
def testGetFleetspeakPendingMessageCount(self):
conn = mock.MagicMock()
conn.outgoing.GetPendingMessageCount.return_value = admin_pb2.GetPendingMessageCountResponse(
count=42)
with mock.patch.object(fleetspeak_connector, "CONN", conn):
count = fleetspeak_utils.GetFleetspeakPendingMessageCount(
"C.1000000000000000")
self.assertEqual(count, 42)
get_args, _ = conn.outgoing.GetPendingMessageCount.call_args
get_req = get_args[0]
self.assertSameElements(get_req.client_ids,
[b"\x10\x00\x00\x00\x00\x00\x00\x00"])