예제 #1
0
 def testHandle(self, _):
     handler = client_plugin.ApiGetFleetspeakPendingMessageCountHandler()
     args = client_plugin.ApiGetFleetspeakPendingMessageCountArgs(
         client_id="C.1111111111111111")
     with mock.patch.object(fleetspeak_utils,
                            "GetFleetspeakPendingMessageCount",
                            return_value=42) as mock_get:
         result = handler.Handle(args)
         self.assertEqual(mock_get.call_args[0][0], "C.1111111111111111")
         self.assertEqual(result.count, 42)
 def GetFleetspeakPendingMessageCount(
     self,
     args: api_client.ApiGetFleetspeakPendingMessageCountArgs,
     context: Optional[api_call_context.ApiCallContext] = None
 ) -> api_client.ApiGetFleetspeakPendingMessageCountHandler:
     return api_client.ApiGetFleetspeakPendingMessageCountHandler()