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)
