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