summaryrefslogtreecommitdiff
path: root/itests/test_mam.py
diff options
context:
space:
mode:
authormathieui <mathieui@mathieui.net>2020-12-06 18:58:35 +0100
committermathieui <mathieui@mathieui.net>2020-12-06 18:59:59 +0100
commit1836b2b887a467afffe6e5e7ab408ffb55aa7c00 (patch)
tree76bd12dc39ffbcd8ea465b5fa6525763dcc067be /itests/test_mam.py
parent12e3c9988df0b3e9023ed58bcad3b77bf43b82af (diff)
downloadslixmpp-1836b2b887a467afffe6e5e7ab408ffb55aa7c00.tar.gz
slixmpp-1836b2b887a467afffe6e5e7ab408ffb55aa7c00.tar.bz2
slixmpp-1836b2b887a467afffe6e5e7ab408ffb55aa7c00.tar.xz
slixmpp-1836b2b887a467afffe6e5e7ab408ffb55aa7c00.zip
itests: add another bunch of tests
Diffstat (limited to 'itests/test_mam.py')
-rw-r--r--itests/test_mam.py55
1 files changed, 55 insertions, 0 deletions
diff --git a/itests/test_mam.py b/itests/test_mam.py
new file mode 100644
index 00000000..f61bc1e6
--- /dev/null
+++ b/itests/test_mam.py
@@ -0,0 +1,55 @@
+import unittest
+from random import randint
+from slixmpp import JID
+from slixmpp.test.integration import SlixIntegration
+
+
+class TestMAM(SlixIntegration):
+ async def asyncSetUp(self):
+ await super().asyncSetUp()
+ self.add_client(
+ self.envjid('CI_ACCOUNT1'),
+ self.envstr('CI_ACCOUNT1_PASSWORD'),
+ )
+ self.add_client(
+ self.envjid('CI_ACCOUNT2'),
+ self.envstr('CI_ACCOUNT2_PASSWORD'),
+ )
+ self.register_plugins(['xep_0313'])
+ await self.connect_clients()
+
+ async def test_mam_retrieve(self):
+ """Make sure we can get messages from our archive"""
+ # send messages first
+ tok = randint(1, 999999)
+ self.clients[0].make_message(mto=self.clients[1].boundjid, mbody='coucou').send()
+ await self.clients[1].wait_until('message')
+ self.clients[1].make_message(
+ mto=self.clients[0].boundjid,
+ mbody='coucou coucou %s' % tok,
+ ).send()
+ await self.clients[0].wait_until('message')
+
+ # Get archive
+ retrieve = self.clients[0]['xep_0313'].retrieve(
+ with_jid=JID(self.envjid('CI_ACCOUNT2')),
+ iterator=True,
+ reverse=True,
+ rsm={'max': 2}
+ )
+ msgs = []
+ count = 0
+ async for rsm in retrieve:
+ for msg in rsm['mam']['results']:
+ msgs.append(
+ msg['mam_result']['forwarded']['stanza']
+ )
+ count += 1
+ if count >= 2:
+ break
+
+ self.assertEqual(msgs[0]['body'], 'coucou')
+ self.assertEqual(msgs[1]['body'], 'coucou coucou %s' % tok)
+
+
+suite = unittest.TestLoader().loadTestsFromTestCase(TestMAM)