def get_sent_messages(account_id, text=None, offset=None, limit=None):
data = personal_messages_pb2.GetMessagesRequest(account_id=account_id,
type=personal_messages_pb2.OwnerType.DESCRIPTOR.values_by_name['SENDER'].number,
text=text,
offset=offset,
limit=limit)
answer = tt_api.sync_request(url=conf.settings.TT_GET_MESSAGES_URL,
data=data,
AnswerType=personal_messages_pb2.GetMessagesResponse)
return answer.total, answer.messages
async def test_no_messages(self):
await self.prepair_data()
request = await self.client.post(
'/get-messages',
data=personal_messages_pb2.GetMessagesRequest(
account_id=666,
type=relations.OWNER_TYPE.random().protocol_value,
text=None,
offset=None,
limit=None).SerializeToString())
data = await self.check_answer(
request, personal_messages_pb2.GetMessagesResponse)
self.assertEqual(data.total, 0)
self.assertEqual(list(data.messages), [])
async def test_text_filter__single(self):
await self.prepair_data()
request = await self.client.post(
'/get-messages',
data=personal_messages_pb2.GetMessagesRequest(
account_id=2,
type=relations.OWNER_TYPE.RECIPIENT.protocol_value,
text='3',
offset=None,
limit=None).SerializeToString())
data = await self.check_answer(
request, personal_messages_pb2.GetMessagesResponse)
self.assertEqual(data.total, 1)
self.assertEqual(list(data.messages), await
load_protocol_messages_by_bodies(['message 3']))
async def test_sender_type(self):
await self.prepair_data()
request = await self.client.post(
'/get-messages',
data=personal_messages_pb2.GetMessagesRequest(
account_id=1,
type=relations.OWNER_TYPE.SENDER.protocol_value,
text=None,
offset=None,
limit=None).SerializeToString())
data = await self.check_answer(
request, personal_messages_pb2.GetMessagesResponse)
self.assertEqual(data.total, 2)
self.assertEqual(
list(data.messages), await
load_protocol_messages_by_bodies(['message 4', 'message 1']))