async def test_has_messages(self): await self.create_messages() request = await self.client.post( '/new-messages-number', data=personal_messages_pb2.NewMessagesNumberRequest( account_id=1).SerializeToString()) data = await self.check_answer( request, personal_messages_pb2.NewMessagesNumberResponse) self.assertEqual(data.number, 0) request = await self.client.post( '/new-messages-number', data=personal_messages_pb2.NewMessagesNumberRequest( account_id=2).SerializeToString()) data = await self.check_answer( request, personal_messages_pb2.NewMessagesNumberResponse) self.assertEqual(data.number, 1) request = await self.client.post( '/new-messages-number', data=personal_messages_pb2.NewMessagesNumberRequest( account_id=3).SerializeToString()) data = await self.check_answer( request, personal_messages_pb2.NewMessagesNumberResponse) self.assertEqual(data.number, 2)
def new_messages_number(account_id): answer = tt_api.sync_request( url=conf.settings.TT_NEW_MESSAGES_NUMBER_URL, data=personal_messages_pb2.NewMessagesNumberRequest( account_id=account_id), AnswerType=personal_messages_pb2.NewMessagesNumberResponse) return answer.number