Add stanza tests for 0377

slixmpp/plugins/xep_0377/__init__.py

@@ -8,7 +8,7 @@
 
 from slixmpp.plugins.base import register_plugin
 
-from slixmpp.plugins.xep_0377.stanza import Report
+from slixmpp.plugins.xep_0377.stanza import Report, Text
 from slixmpp.plugins.xep_0377.spam_reporting import XEP_0377
 
 

tests/test_stanza_xep_0377.py

@@ -0,0 +1,56 @@
+import unittest
+from slixmpp import Iq
+from slixmpp.test import SlixTest
+import slixmpp.plugins.xep_0191 as xep_0191
+import slixmpp.plugins.xep_0377 as xep_0377
+from slixmpp.xmlstream import register_stanza_plugin
+
+
+class TestSpamReporting(SlixTest):
+
+    def setUp(self):
+        register_stanza_plugin(Iq, xep_0191.Block)
+        register_stanza_plugin(
+                xep_0191.Block,
+                xep_0377.Report,
+        )
+        register_stanza_plugin(
+            xep_0377.Report,
+            xep_0377.Text,
+        )
+
+    def testCreateReport(self):
+        report = """
+          <iq type="set">
+            <block xmlns="urn:xmpp:blocking">
+                <report xmlns="urn:xmpp:reporting:0">
+                    <spam/>
+                </report>
+            </block>
+          </iq>
+        """
+
+        iq = self.Iq()
+        iq['type'] = 'set'
+        iq['block']['report']['spam'] = True
+
+        self.check(iq, report)
+
+    def testEnforceOnlyOneSubElement(self):
+        report = """
+          <iq type="set">
+            <block xmlns="urn:xmpp:blocking">
+                <report xmlns="urn:xmpp:reporting:0">
+                    <abuse/>
+                </report>
+            </block>
+          </iq>
+        """
+
+        iq = self.Iq()
+        iq['type'] = 'set'
+        iq['block']['report']['spam'] = True
+        iq['block']['report']['abuse'] = True
+        self.check(iq, report)
+
+suite = unittest.TestLoader().loadTestsFromTestCase(TestSpamReporting)