1 files changed, 48 insertions, 0 deletions
diff --git a/tests/test_stanza_xep_0461.py b/tests/test_stanza_xep_0461.py
new file mode 100644
index 00000000..b9550481
--- /dev/null
+++ b/tests/test_stanza_xep_0461.py
@@ -0,0 +1,48 @@
+import unittest
+from slixmpp import Message
+from slixmpp.test import SlixTest
+from slixmpp.plugins.xep_0461 import stanza
+
+
+class TestReply(SlixTest):
+    def setUp(self):
+        stanza.register_plugins()
+
+    def testReply(self):
+        message = Message()
+        message["reply"]["id"] = "some-id"
+        message["body"] = "some-body"
+
+        self.check(
+            message,
+            """
+            <message>
+              <reply xmlns="urn:xmpp:reply:0" id="some-id" />
+              <body>some-body</body>
+            </message>
+            """,
+        )
+
+    def testFallback(self):
+        message = Message()
+        message["body"] = "12345\nrealbody"
+        message["feature_fallback"]["for"] = "NS"
+        message["feature_fallback"]["fallback_body"]["start"] = "0"
+        message["feature_fallback"]["fallback_body"]["end"] = "6"
+
+        self.check(
+            message,
+            """
+            <message xmlns="jabber:client">
+              <body>12345\nrealbody</body>
+              <fallback xmlns='urn:xmpp:feature-fallback:0' for='NS'>
+                <body start="0" end="6" />
+              </fallback>
+            </message>
+            """,
+        )
+
+        assert message["feature_fallback"].get_stripped_body() == "realbody"
+
+
+suite = unittest.TestLoader().loadTestsFromTestCase(TestReply)