From d7d4baefa3877afef462b37afec502ff9b42a9e5 Mon Sep 17 00:00:00 2001
From: mathieui <mathieui@mathieui.net>
Date: Mon, 8 Mar 2021 21:47:25 +0100
Subject: tests: Add XEP-0313 stanza tests

---
 tests/test_stanza_xep_0313.py | 82 +++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 82 insertions(+)
 create mode 100644 tests/test_stanza_xep_0313.py

(limited to 'tests/test_stanza_xep_0313.py')

diff --git a/tests/test_stanza_xep_0313.py b/tests/test_stanza_xep_0313.py
new file mode 100644
index 00000000..5c7b42a9
--- /dev/null
+++ b/tests/test_stanza_xep_0313.py
@@ -0,0 +1,82 @@
+import unittest
+from slixmpp import JID, Iq, Message
+from slixmpp.test import SlixTest
+from slixmpp.plugins.xep_0313 import stanza
+from slixmpp.plugins.xep_0004.stanza import Form
+from slixmpp.plugins.xep_0297 import stanza as fstanza
+from slixmpp.plugins.xep_0059 import stanza as rstanza
+from slixmpp.xmlstream import register_stanza_plugin
+
+
+class TestMAM(SlixTest):
+
+    def setUp(self):
+        register_stanza_plugin(stanza.MAM, Form)
+        register_stanza_plugin(Iq, stanza.MAM)
+        register_stanza_plugin(Iq, stanza.Preferences)
+        register_stanza_plugin(Message, stanza.Result)
+        register_stanza_plugin(Iq, stanza.Fin)
+        register_stanza_plugin(
+            stanza.Result,
+            fstanza.Forwarded
+        )
+        register_stanza_plugin(stanza.MAM, rstanza.Set)
+        register_stanza_plugin(stanza.Fin, rstanza.Set)
+
+    def testMAMQuery(self):
+        """Test that we can build a simple MAM query."""
+        iq = Iq()
+        iq['type'] = 'set'
+        iq['mam']['queryid'] = 'f27'
+
+        self.check(iq, """
+            <iq type='set'>
+              <query xmlns='urn:xmpp:mam:2' queryid='f27'/>
+            </iq>
+        """)
+
+    def testMAMQueryOptions(self):
+        """Test that we can build a mam query with all options."""
+        iq = Iq()
+        iq['type'] = 'set'
+        iq['mam']['with'] = JID('juliet@capulet.lit')
+        iq['mam']['start'] = '2010-06-07T00:00:00Z'
+        iq['mam']['end'] = '2010-07-07T13:23:54Z'
+        iq['mam']['after_id'] = 'id1'
+        iq['mam']['before_id'] = 'id2'
+        iq['mam']['ids'] = ['a', 'b', 'c']
+
+        self.check(iq, """
+            <iq type='set'>
+              <query xmlns='urn:xmpp:mam: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>juliet@capulet.lit</value>
+                  </field>
+                  <field var='start'>
+                    <value>2010-06-07T00:00:00Z</value>
+                  </field>
+                  <field var='end'>
+                    <value>2010-07-07T13:23:54Z</value>
+                  </field>
+                  <field var='after-id'>
+                    <value>id1</value>
+                  </field>
+                  <field var='before-id'>
+                    <value>id2</value>
+                  </field>
+                  <field var='ids'>
+                    <value>a</value>
+                    <value>b</value>
+                    <value>c</value>
+                  </field>
+                </x>
+              </query>
+            </iq>
+        """, use_values=False)
+
+
+suite = unittest.TestLoader().loadTestsFromTestCase(TestMAM)
-- 
cgit v1.2.3


From 97a63b9f25f8dd124abf52fa06ed29ca11abe1d9 Mon Sep 17 00:00:00 2001
From: mathieui <mathieui@mathieui.net>
Date: Mon, 8 Mar 2021 22:15:42 +0100
Subject: XEP-0313: Update the API

- add an iterate() method that makes this plugin more practical
- add a get_fields method to retrieve the available search fields
- add a get_archive_metadata method.

This is a big chunk because git refused to split it further.
---
 tests/test_stanza_xep_0313.py | 1 -
 1 file changed, 1 deletion(-)

(limited to 'tests/test_stanza_xep_0313.py')

diff --git a/tests/test_stanza_xep_0313.py b/tests/test_stanza_xep_0313.py
index 5c7b42a9..d7bd3080 100644
--- a/tests/test_stanza_xep_0313.py
+++ b/tests/test_stanza_xep_0313.py
@@ -13,7 +13,6 @@ class TestMAM(SlixTest):
     def setUp(self):
         register_stanza_plugin(stanza.MAM, Form)
         register_stanza_plugin(Iq, stanza.MAM)
-        register_stanza_plugin(Iq, stanza.Preferences)
         register_stanza_plugin(Message, stanza.Result)
         register_stanza_plugin(Iq, stanza.Fin)
         register_stanza_plugin(
-- 
cgit v1.2.3


From 0115feaa31ef998b59da98c99c523ad34ed92651 Mon Sep 17 00:00:00 2001
From: mathieui <mathieui@mathieui.net>
Date: Tue, 9 Mar 2021 19:44:21 +0100
Subject: tests: Add a MAM metadata stanza test

---
 tests/test_stanza_xep_0313.py | 24 ++++++++++++++++++++++++
 1 file changed, 24 insertions(+)

(limited to 'tests/test_stanza_xep_0313.py')

diff --git a/tests/test_stanza_xep_0313.py b/tests/test_stanza_xep_0313.py
index d7bd3080..ac5fcb14 100644
--- a/tests/test_stanza_xep_0313.py
+++ b/tests/test_stanza_xep_0313.py
@@ -22,6 +22,10 @@ class TestMAM(SlixTest):
         register_stanza_plugin(stanza.MAM, rstanza.Set)
         register_stanza_plugin(stanza.Fin, rstanza.Set)
 
+        register_stanza_plugin(Iq, stanza.Metadata)
+        register_stanza_plugin(stanza.Metadata, stanza.Start)
+        register_stanza_plugin(stanza.Metadata, stanza.End)
+
     def testMAMQuery(self):
         """Test that we can build a simple MAM query."""
         iq = Iq()
@@ -77,5 +81,25 @@ class TestMAM(SlixTest):
             </iq>
         """, use_values=False)
 
+    def testMAMMetadata(self):
+        """Test that we can build a MAM metadata payload"""
+
+        iq = Iq()
+        iq['type'] = 'result'
+        iq['mam_metadata']['start']['id'] = 'YWxwaGEg'
+        iq['mam_metadata']['start']['timestamp'] = '2008-08-22T21:09:04Z'
+        iq['mam_metadata']['end']['id'] = 'b21lZ2Eg'
+        iq['mam_metadata']['end']['timestamp'] = '2020-04-20T14:34:21Z'
+
+        self.check(iq, """
+            <iq type='result'>
+              <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>
+        """)
+
+
 
 suite = unittest.TestLoader().loadTestsFromTestCase(TestMAM)
-- 
cgit v1.2.3