def getMessageState(self, messageGenId): from yowsup_ext.layers.store.models.messagestate import MessageState from yowsup_ext.layers.store.models.message import Message from yowsup_ext.layers.store.models.state import State message = self.stack.getLayerInterface( YowStorageLayer).getMessageByGenId(messageGenId) states = (State.select().join(MessageState).join(Message).where( Message.id == message.id)) return states[0]
def test_storeOutgoingTextMessages(self): from yowsup_ext.layers.store.models.state import State from yowsup_ext.layers.store.models.messagestate import MessageState from yowsup_ext.layers.store.models.message import Message msg = self.sendMessage() message = self.stack.getLayerInterface(YowStorageLayer).getMessages( msg.getTo(), limit=1)[0] self.assertEqual(message.content, msg.getBody()) self.assertEqual(message.conversation.contact.jid, msg.getTo()) states = (State.select().join(MessageState).join(Message).where( Message.id == message.id)) self.assertEqual(states[0], State.get_sent_queued())
def test_storeIncomingTextMessage(self): from yowsup_ext.layers.store.models.messagestate import MessageState from yowsup_ext.layers.store.models.message import Message from yowsup_ext.layers.store.models.state import State msg = self.receiveMessage() self.sendReceipt(msg) message = Message.get(id_gen=msg.getId()) self.assertEqual(message.content, msg.getBody()) self.assertEqual(message.conversation.contact.jid, msg.getFrom()) states = (State.select().join(MessageState).join(Message).where( Message.id == message.id)) self.assertEqual(states[0].name, State.get_received_remote().name) self.sendReceipt(msg, read=True)