Fix data stanza based on test results.

1 files changed, 77 insertions, 0 deletions

diff --git a/tests/test_stanza_xep_0047.py b/tests/test_stanza_xep_0047.py
new file mode 100644
index 00000000..1b212529
--- /dev/null
+++ b/tests/test_stanza_xep_0047.py
@@ -0,0 +1,77 @@
+from sleekxmpp.test import *
+from sleekxmpp.plugins.xep_0047 import Data
+
+
+class TestIBB(SleekTest):
+
+    def setUp(self):
+        register_stanza_plugin(Iq, Data)
+
+    def testInvalidBase64MidEqual(self):
+        """
+        Test detecting invalid base64 data with = inside the
+        character data instead of at the end.
+        """
+        iq = Iq(xml=ET.fromstring("""
+          <iq type="set" id="0" to="tester@localhost">
+            <data xmlns="http://jabber.org/protocol/ibb" seq="0">
+              ABC=DEFGH
+            </data>
+          </iq>
+        """))
+
+        errored = False
+
+        try:
+            data = iq['ibb_data']['data']
+        except XMPPError:
+            errored = True
+
+        self.assertTrue(errored, "ABC=DEFGH did not raise base64 error")
+
+    def testInvalidBase64PrefixEqual(self):
+        """
+        Test detecting invalid base64 data with = as a prefix
+        to the character data.
+        """
+        iq = Iq(xml=ET.fromstring("""
+          <iq type="set" id="0" to="tester@localhost">
+            <data xmlns="http://jabber.org/protocol/ibb" seq="0">
+              =ABCDEFGH
+            </data>
+          </iq>
+        """))
+
+        errored = False
+
+        try:
+            data = iq['ibb_data']['data']
+        except XMPPError:
+            errored = True
+
+        self.assertTrue(errored, "=ABCDEFGH did not raise base64 error")
+
+    def testInvalidBase64Alphabet(self):
+        """
+        Test detecting invalid base64 data with characters
+        outside of the base64 alphabet.
+        """
+        iq = Iq(xml=ET.fromstring("""
+          <iq type="set" id="0" to="tester@localhost">
+            <data xmlns="http://jabber.org/protocol/ibb" seq="0">
+              ABCD?EFGH
+            </data>
+          </iq>
+        """))
+
+        errored = False
+
+        try:
+            data = iq['ibb_data']['data']
+        except XMPPError:
+            errored = True
+
+        self.assertTrue(errored, "ABCD?EFGH did not raise base64 error")
+
+
+suite = unittest.TestLoader().loadTestsFromTestCase(TestIBB)