async def test_recipient(self): await self.prepair_data() request = await self.client.post( '/get-message', data=personal_messages_pb2.GetMessageRequest( account_id=2, message_id=self.message_id).SerializeToString()) data = await self.check_answer( request, personal_messages_pb2.GetMessageResponse) self.assertEqual(data.message.body, 'a message 1')
async def test_no_relation(self): await self.prepair_data() request = await self.client.post( '/get-message', data=personal_messages_pb2.GetMessageRequest( account_id=3, message_id=self.message_id).SerializeToString()) data = await self.check_answer( request, personal_messages_pb2.GetMessageResponse) self.assertFalse(data.HasField('message'))
def get_message(account_id, message_id): data = personal_messages_pb2.GetMessageRequest(account_id=account_id, message_id=message_id) answer = tt_api.sync_request(url=conf.settings.TT_GET_MESSAGE_URL, data=data, AnswerType=personal_messages_pb2.GetMessageResponse) if answer.HasField('message'): return answer.message return None