Update protoxep_reactions to XEP-0444

1 files changed, 69 insertions, 0 deletions

diff --git a/tests/test_stanza_xep_0444.py b/tests/test_stanza_xep_0444.py
new file mode 100644
index 00000000..b4d5739b
--- /dev/null
+++ b/tests/test_stanza_xep_0444.py
@@ -0,0 +1,69 @@
+"""
+    Slixmpp: The Slick XMPP Library
+    Copyright (C) 2020 Mathieu Pasquet
+    This file is part of Slixmpp.
+
+    See the file LICENSE for copying permission.
+"""
+
+import unittest
+from slixmpp import Message
+from slixmpp.test import SlixTest
+from slixmpp.plugins.xep_0444 import XEP_0444
+import slixmpp.plugins.xep_0444.stanza as stanza
+from slixmpp.xmlstream import register_stanza_plugin
+
+
+class TestReactions(SlixTest):
+
+    def setUp(self):
+        register_stanza_plugin(Message, stanza.Reactions)
+        register_stanza_plugin(stanza.Reactions, stanza.Reaction)
+
+    def testCreateReactions(self):
+        """Testing creating Reactions."""
+
+        xmlstring = """
+          <message>
+              <reactions xmlns="urn:xmpp:reactions:0" id="abcd">
+                  <reaction>😃</reaction>
+                  <reaction>🤗</reaction>
+              </reactions>
+          </message>
+        """
+
+        msg = self.Message()
+        msg['reactions']['id'] = 'abcd'
+        msg['reactions']['values'] = ['😃', '🤗']
+
+        self.check(msg, xmlstring, use_values=False)
+
+        self.assertEqual({'😃', '🤗'}, msg['reactions']['values'])
+
+
+    def testCreateReactionsUnrestricted(self):
+        """Testing creating Reactions with the extra all_chars arg."""
+
+        xmlstring = """
+          <message>
+              <reactions xmlns="urn:xmpp:reactions:0" id="abcd">
+                  <reaction>😃</reaction>
+                  <reaction>🤗</reaction>
+                  <reaction>toto</reaction>
+              </reactions>
+          </message>
+        """
+
+        msg = self.Message()
+        msg['reactions']['id'] = 'abcd'
+        msg['reactions'].set_values(['😃', '🤗', 'toto'], all_chars=True)
+
+        self.check(msg, xmlstring, use_values=False)
+
+        self.assertEqual({'😃', '🤗'}, msg['reactions']['values'])
+        self.assertEqual({'😃', '🤗', 'toto'}, msg['reactions'].get_values(all_chars=True))
+        with self.assertRaises(ValueError):
+            msg['reactions'].set_values(['😃', '🤗', 'toto'], all_chars=False)
+
+
+suite = unittest.TestLoader().loadTestsFromTestCase(TestReactions)