Integrate roster with BaseXMPP.

Last sent stanzas are saved regardless of if the roster is used
directly or self.send_presence

1 files changed, 30 insertions, 0 deletions

diff --git a/tests/test_stream_roster.py b/tests/test_stream_roster.py
index 69e5ca13..aa8fc26f 100644
--- a/tests/test_stream_roster.py
+++ b/tests/test_stream_roster.py
@@ -203,5 +203,35 @@ class TestStreamRoster(SleekTest):
         self.failUnless(result == expected,
                 "Unexpected roster values: %s" % result)
 
+    def testSendLastPresence(self):
+        """Test that sending the last presence works."""
+        self.stream_start()
+        self.xmpp.send_presence(pshow='dnd')
+        self.xmpp.auto_authorize = True
+        self.xmpp.auto_subscribe = True
+
+        self.send("""
+          <presence>
+            <show>dnd</show>
+          </presence>
+        """)
+
+        self.recv("""
+          <presence from="user@localhost"
+                    to="tester@localhost"
+                    type="subscribe" />
+        """)
+
+        self.send("""
+          <presence to="user@localhost"
+                    type="subscribed" />
+        """)
+
+        self.send("""
+          <presence to="user@localhost">
+            <show>dnd</show>
+          </presence>
+        """)
+
 
 suite = unittest.TestLoader().loadTestsFromTestCase(TestStreamRoster)