Add stanza tests for 0377

This commit is contained in:
mathieui 2020-05-24 15:00:20 +02:00
parent ef961392be
commit 28123083f7
2 changed files with 57 additions and 1 deletions

View file

@ -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

View file

@ -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)