def test_storeOutgoingLocationMessage(self):
from yowsup_ext.layers.store.models.message import Message
locData = {
"latitude": "LAT",
"longitude": "LONG",
"name": "name",
"url": "URL",
"encoding": "raw"
}
locationMessage = LocationMediaMessageProtocolEntity(
locData["latitude"],
locData["longitude"],
locData["name"],
locData["url"],
locData["encoding"],
to="*****@*****.**",
preview="PREV")
self.stack.send(locationMessage)
message = Message.get(id_gen=locationMessage.getId())
self.assertEqual(message.content, locData["name"])
self.assertEqual(message.media.data, ";".join(
(locData["latitude"], locData["longitude"])))
self.assertEqual(str(message.media.preview), "PREV")
self.assertEqual(message.media.remote_url, locData["url"])
self.assertEqual(message.media.encoding, locData["encoding"])
def test_storeOutgoingVCardMessage(self):
from yowsup_ext.layers.store.models.message import Message
vcardData = {"name": "NAME", "data": "VCARD_DATA"}
vcardMessageEntity = VCardMediaMessageProtocolEntity(
vcardData["name"], vcardData["data"], to="*****@*****.**")
self.stack.send(vcardMessageEntity)
message = Message.get(id_gen=vcardMessageEntity.getId())
self.assertEqual(message.content, vcardData["name"])
self.assertEqual(message.media.data, vcardData["data"])
def test_storeOutgoingVCardMessage(self):
from yowsup_ext.layers.store.models.message import Message
vcardData = {
"name": "NAME",
"data": "VCARD_DATA"
}
vcardMessageEntity = VCardMediaMessageProtocolEntity(vcardData["name"], vcardData["data"], to="*****@*****.**")
self.stack.send(vcardMessageEntity)
message = Message.get(id_gen = vcardMessageEntity.getId())
self.assertEqual(message.content, vcardData["name"])
self.assertEqual(message.media.data, vcardData["data"])
def test_storeOutgoingVideoMessage(self):
from yowsup_ext.layers.store.models.message import Message
mediaData = {
"mimetype": "image/jpeg",
"filehash": "fhash",
"url": "http:/google.com",
"ip": "ip",
"size": 1234,
"file": "filename",
"encoding": "raw",
"abitrate": "bitrate",
"acodec": "code",
"caption": "CAPT"
}
messageEntity = VideoDownloadableMediaMessageProtocolEntity(
mediaData["mimetype"],
mediaData["filehash"],
mediaData["url"],
mediaData["ip"],
mediaData["size"],
mediaData["file"],
None,
None,
None,
None,
None,
mediaData["encoding"],
None,
None,
None,
None,
None,
None,
mediaData["caption"],
to = "*****@*****.**"
)
self.stack.send(messageEntity)
message = Message.get(id_gen = messageEntity.getId())
self.assertEqual(message.content, mediaData["caption"])
self.assertEqual(message.media.encoding, mediaData["encoding"])
self.assertEqual(message.media.filehash, mediaData["filehash"])
self.assertEqual(message.media.mimetype, mediaData["mimetype"])
self.assertEqual(message.media.filename, mediaData["file"])
self.assertEqual(message.media.remote_url, mediaData["url"])
def test_storeOutgoingVideoMessage(self):
from yowsup_ext.layers.store.models.message import Message
mediaData = {
"mimetype": "image/jpeg",
"filehash": "fhash",
"url": "http:/google.com",
"ip": "ip",
"size": 1234,
"file": "filename",
"encoding": "raw",
"abitrate": "bitrate",
"acodec": "code",
"caption": "CAPT"
}
messageEntity = VideoDownloadableMediaMessageProtocolEntity(
mediaData["mimetype"],
mediaData["filehash"],
mediaData["url"],
mediaData["ip"],
mediaData["size"],
mediaData["file"],
None,
None,
None,
None,
None,
mediaData["encoding"],
None,
None,
None,
None,
None,
None,
mediaData["caption"],
to="*****@*****.**")
self.stack.send(messageEntity)
message = Message.get(id_gen=messageEntity.getId())
self.assertEqual(message.content, mediaData["caption"])
self.assertEqual(message.media.encoding, mediaData["encoding"])
self.assertEqual(message.media.filehash, mediaData["filehash"])
self.assertEqual(message.media.mimetype, mediaData["mimetype"])
self.assertEqual(message.media.filename, mediaData["file"])
self.assertEqual(message.media.remote_url, mediaData["url"])
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)
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)
def test_storeOutgoingLocationMessage(self):
from yowsup_ext.layers.store.models.message import Message
locData = {
"latitude": "LAT",
"longitude": "LONG",
"name": "name",
"url": "URL",
"encoding": "raw"
}
locationMessage = LocationMediaMessageProtocolEntity(
locData["latitude"],locData["longitude"],locData["name"],
locData["url"], locData["encoding"], to="*****@*****.**", preview = "PREV"
)
self.stack.send(locationMessage)
message = Message.get(id_gen = locationMessage.getId())
self.assertEqual(message.content, locData["name"])
self.assertEqual(message.media.data, ";".join((locData["latitude"], locData["longitude"])))
self.assertEqual(str(message.media.preview), "PREV")
self.assertEqual(message.media.remote_url, locData["url"])
self.assertEqual(message.media.encoding, locData["encoding"])
