Commit 3613cd5f authored by mathieui's avatar mathieui

tests: Add a MAM stream test

parent 644ebfe8
import unittest
from datetime import datetime
from slixmpp.test import SlixTest
from slixmpp import JID
class TestMAM(SlixTest):
def setUp(self):
self.stream_start(plugins=['xep_0313'])
def tearDown(self):
self.stream_close()
def testRetrieveSimple(self):
"""Test requesting MAM messages without RSM"""
msgs = []
async def test():
iq = await self.xmpp['xep_0313'].retrieve()
for message in iq['mam']['results']:
msgs.append(message)
fut = self.xmpp.wrap(test())
self.wait_()
self.send("""
<iq type='set' id='1'>
<query xmlns='urn:xmpp:mam:2' queryid='1' />
</iq>
""")
self.recv("""
<message id='abc' to='tester@localhost/resource'>
<result xmlns='urn:xmpp:mam:2' queryid='1'
id='28482-98726-73623'>
<forwarded xmlns='urn:xmpp:forward:0'>
<delay xmlns='urn:xmpp:delay' stamp='2010-07-10T23:08:25Z'/>
<message xmlns='jabber:client' from="witch@shakespeare.lit"
to="tester@localhost">
<body>Hail to thee</body>
</message>
</forwarded>
</result>
</message>
""")
self.recv("""
<iq type="result" id="1" to="tester@localhost">
<fin xmlns="urn:xmpp:mam:2">
<first index='0'>28482-98726-73623</first>
<last>28482-98726-73623</last>
</fin>
</iq>
""")
self.run_coro(fut)
self.assertEqual(
msgs[0]['mam_result']['forwarded']['message']['body'],
"Hail to thee",
)
self.assertEqual(len(msgs),1)
def testRetrieveRSM(self):
"""Test requesting MAM messages with RSM"""
msgs = []
async def test():
iterator = self.xmpp['xep_0313'].retrieve(
with_jid=JID('toto@titi'),
start='2010-06-07T00:00:00Z',
iterator=True,
)
async for page in iterator:
for message in page['mam']['results']:
msgs.append(message)
fut = self.xmpp.wrap(test())
self.wait_()
self.send("""
<iq type='set' id='2'>
<query xmlns='urn:xmpp:mam:2' queryid='2'>
<x xmlns='jabber:x:data' type='submit'>
<field var='FORM_TYPE' type='hidden'>
<value>urn:xmpp:mam:2</value>
</field>
<field var='with'>
<value>toto@titi</value>
</field>
<field var='start'>
<value>2010-06-07T00:00:00Z</value>
</field>
</x>
<set xmlns="http://jabber.org/protocol/rsm">
<max>10</max>
</set>
</query>
</iq>
""")
self.recv("""
<message id='abc' to='tester@localhost/resource'>
<result xmlns='urn:xmpp:mam:2' queryid='2'
id='28482-98726-73623'>
<forwarded xmlns='urn:xmpp:forward:0'>
<delay xmlns='urn:xmpp:delay' stamp='2010-07-10T23:08:25Z'/>
<message xmlns='jabber:client' from="witch@shakespeare.lit"
to="tester@localhost">
<body>Hail to thee</body>
</message>
</forwarded>
</result>
</message>
""")
self.recv("""
<iq type="result" id="2" to="tester@localhost">
<fin xmlns="urn:xmpp:mam:2">
<set xmlns='http://jabber.org/protocol/rsm'>
<first index='0'>28482-98726-73623</first>
<last>28482-98726-73623</last>
<count>2</count>
</set>
</fin>
</iq>
""")
self.send("""
<iq type='set' id='3'>
<query xmlns='urn:xmpp:mam:2' queryid='3'>
<x xmlns='jabber:x:data' type='submit'>
<field var='FORM_TYPE' type='hidden'>
<value>urn:xmpp:mam:2</value>
</field>
<field var='with'>
<value>toto@titi</value>
</field>
<field var='start'>
<value>2010-06-07T00:00:00Z</value>
</field>
</x>
<set xmlns="http://jabber.org/protocol/rsm">
<max>10</max>
<after>28482-98726-73623</after>
</set>
</query>
</iq>
""")
self.recv("""
<message id='abc' to='tester@localhost/resource'>
<result xmlns='urn:xmpp:mam:2' queryid='3'
id='28482-98726-73624'>
<forwarded xmlns='urn:xmpp:forward:0'>
<delay xmlns='urn:xmpp:delay' stamp='2010-07-10T23:08:26Z'/>
<message xmlns='jabber:client' from="witch@shakespeare.lit"
to="tester@localhost">
<body>Hi Y'all</body>
</message>
</forwarded>
</result>
</message>
""")
self.recv("""
<iq type="result" id="3" to="tester@localhost">
<fin xmlns="urn:xmpp:mam:2">
<set xmlns='http://jabber.org/protocol/rsm'>
<first index='1'>28482-98726-73624</first>
<last>28482-98726-73624</last>
<count>2</count>
</set>
</fin>
</iq>
""")
self.run_coro(fut)
self.assertEqual(
msgs[0]['mam_result']['forwarded']['message']['body'],
"Hail to thee",
)
self.assertEqual(
msgs[1]['mam_result']['forwarded']['message']['body'],
"Hi Y'all",
)
self.assertEqual(len(msgs), 2)
def testIterate(self):
"""Test iterating over MAM messages with RSM"""
msgs = []
async def test():
iterator = self.xmpp['xep_0313'].iterate(
with_jid=JID('toto@titi'),
start='2010-06-07T00:00:00Z',
)
async for message in iterator:
msgs.append(message)
fut = self.xmpp.wrap(test())
self.wait_()
self.send("""
<iq type='set' id='2'>
<query xmlns='urn:xmpp:mam:2' queryid='2'>
<x xmlns='jabber:x:data' type='submit'>
<field var='FORM_TYPE' type='hidden'>
<value>urn:xmpp:mam:2</value>
</field>
<field var='with'>
<value>toto@titi</value>
</field>
<field var='start'>
<value>2010-06-07T00:00:00Z</value>
</field>
</x>
<set xmlns="http://jabber.org/protocol/rsm">
<max>10</max>
</set>
</query>
</iq>
""")
self.recv("""
<message id='abc' to='tester@localhost/resource'>
<result xmlns='urn:xmpp:mam:2' queryid='2'
id='28482-98726-73623'>
<forwarded xmlns='urn:xmpp:forward:0'>
<delay xmlns='urn:xmpp:delay' stamp='2010-07-10T23:08:25Z'/>
<message xmlns='jabber:client' from="witch@shakespeare.lit"
to="tester@localhost">
<body>Hail to thee</body>
</message>
</forwarded>
</result>
</message>
""")
self.recv("""
<iq type="result" id="2" to="tester@localhost">
<fin xmlns="urn:xmpp:mam:2">
<set xmlns='http://jabber.org/protocol/rsm'>
<first index='0'>28482-98726-73623</first>
<last>28482-98726-73623</last>
<count>2</count>
</set>
</fin>
</iq>
""")
self.send("""
<iq type='set' id='3'>
<query xmlns='urn:xmpp:mam:2' queryid='3'>
<x xmlns='jabber:x:data' type='submit'>
<field var='FORM_TYPE' type='hidden'>
<value>urn:xmpp:mam:2</value>
</field>
<field var='with'>
<value>toto@titi</value>
</field>
<field var='start'>
<value>2010-06-07T00:00:00Z</value>
</field>
</x>
<set xmlns="http://jabber.org/protocol/rsm">
<max>10</max>
<after>28482-98726-73623</after>
</set>
</query>
</iq>
""")
self.recv("""
<message id='abc' to='tester@localhost/resource'>
<result xmlns='urn:xmpp:mam:2' queryid='3'
id='28482-98726-73624'>
<forwarded xmlns='urn:xmpp:forward:0'>
<delay xmlns='urn:xmpp:delay' stamp='2010-07-10T23:08:26Z'/>
<message xmlns='jabber:client' from="witch@shakespeare.lit"
to="tester@localhost">
<body>Hi Y'all</body>
</message>
</forwarded>
</result>
</message>
""")
self.recv("""
<iq type="result" id="3" to="tester@localhost">
<fin xmlns="urn:xmpp:mam:2">
<set xmlns='http://jabber.org/protocol/rsm'>
<first index='1'>28482-98726-73624</first>
<last>28482-98726-73624</last>
<count>2</count>
</set>
</fin>
</iq>
""")
self.run_coro(fut)
self.assertEqual(
msgs[0]['mam_result']['forwarded']['message']['body'],
"Hail to thee",
)
self.assertEqual(
msgs[1]['mam_result']['forwarded']['message']['body'],
"Hi Y'all",
)
self.assertEqual(len(msgs), 2)
def test_get_metadata(self):
"""Test a MAM metadata retrieval"""
fut = self.xmpp.wrap(
self.xmpp.plugin['xep_0313'].get_archive_metadata()
)
self.wait_()
self.send("""
<iq type='get' id='1'>
<metadata xmlns='urn:xmpp:mam:2'/>
</iq>
""")
self.recv("""
<iq type='result' id='1'>
<metadata xmlns='urn:xmpp:mam:2'>
<start id='YWxwaGEg' timestamp='2008-08-22T21:09:04Z' />
<end id='b21lZ2Eg' timestamp='2020-04-20T14:34:21Z' />
</metadata>
</iq>
""")
self.run_coro(fut)
result = fut.result()
self.assertEqual(result['mam_metadata']['start']['id'], "YWxwaGEg")
self.assertEqual(
result['mam_metadata']['start']['timestamp'],
datetime.fromisoformat('2008-08-22T21:09:04+00:00')
)
suite = unittest.TestLoader().loadTestsFromTestCase(TestMAM)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment