[telepathy-gabble/master] caps/advertise-legacy.py: new test for advertising Capabilities

[Simon McVittie]

---
 tests/twisted/Makefile.am              |    1 +
 tests/twisted/caps/advertise-legacy.py |  152 ++++++++++++++++++++++++++++++++
 2 files changed, 153 insertions(+), 0 deletions(-)
 create mode 100644 tests/twisted/caps/advertise-legacy.py

diff --git a/tests/twisted/Makefile.am b/tests/twisted/Makefile.am
index 0fbc87e..7cab79b 100644
--- a/tests/twisted/Makefile.am
+++ b/tests/twisted/Makefile.am
@@ -1,6 +1,7 @@
 TWISTED_TESTS = \
 	avatar-requirements.py \
 	caps_helper.py \
+	caps/advertise-legacy.py \
 	caps/caps-cache.py \
 	caps/from-bare-jid.py \
 	caps/hashed-caps.py \
diff --git a/tests/twisted/caps/advertise-legacy.py b/tests/twisted/caps/advertise-legacy.py
new file mode 100644
index 0000000..87c8c19
--- /dev/null
+++ b/tests/twisted/caps/advertise-legacy.py
@@ -0,0 +1,152 @@
+"""
+Test AdvertiseCapabilities.
+"""
+
+import dbus
+
+from twisted.words.xish import xpath, domish
+
+from servicetest import EventPattern
+from gabbletest import exec_test
+from caps_helper import caps_contain, receive_presence_and_ask_caps
+import constants as cs
+import ns
+
+# The caps we always have, regardless of any clients' caps
+FIXED_CAPS = [
+    ns.GOOGLE_FEAT_SESSION,
+    ns.NICK,
+    ns.NICK + '+notify',
+    ns.CHAT_STATES,
+    ns.SI,
+    ns.IBB,
+    ns.BYTESTREAMS,
+    ]
+
+JINGLE_CAPS = [
+    # Jingle dialects
+    ns.JINGLE,
+    ns.JINGLE_015,
+    # Jingle transports
+    ns.JINGLE_TRANSPORT_ICEUDP,
+    ns.JINGLE_TRANSPORT_RAWUDP,
+    ns.GOOGLE_P2P,
+    # Jingle streams
+    ns.GOOGLE_FEAT_VOICE,
+    ns.GOOGLE_FEAT_VIDEO,
+    ns.JINGLE_015_AUDIO,
+    ns.JINGLE_015_VIDEO,
+    ns.JINGLE_RTP,
+    ns.JINGLE_RTP_AUDIO,
+    ns.JINGLE_RTP_VIDEO,
+    ]
+
+VARIABLE_CAPS = (
+    JINGLE_CAPS +
+    [
+    # FIXME: currently we always advertise these, but in future we should
+    # only advertise them if >= 1 client supports them:
+    # ns.FILE_TRANSFER,
+    # ns.TUBES,
+    ])
+
+def check_caps(disco_response, caps_str, desired):
+    """Assert that all the FIXED_CAPS are supported, and of the VARIABLE_CAPS,
+    every capability in desired is supported, and every other capability is
+    not.
+    """
+    for c in FIXED_CAPS:
+        assert caps_contain(disco_response, c), (c, caps_str)
+
+    for c in VARIABLE_CAPS:
+        if c in desired:
+            assert caps_contain(disco_response, c), (c, caps_str)
+        else:
+            assert not caps_contain(disco_response, c), (c, caps_str)
+
+def run_test(q, bus, conn, stream):
+    conn.Connect()
+    q.expect('dbus-signal', signal='StatusChanged', args=[0, 1])
+
+    # This method call looks wrong, but it's "the other side" of
+    # test/twisted/capabilities/legacy-caps.py in MC 5.1 - MC doesn't know
+    # how to map Client capabilities into the old Capabilities interface.
+    add = [(cs.CHANNEL_TYPE_STREAMED_MEDIA, 2L**32-1),
+            (cs.CHANNEL_TYPE_STREAM_TUBE, 2L**32-1)]
+    remove = []
+    caps = conn.Capabilities.AdvertiseCapabilities(add, remove)
+    (disco_response, caps_str, _) = receive_presence_and_ask_caps(q, stream,
+            False)
+    check_caps(disco_response, caps_str, JINGLE_CAPS)
+
+    # Remove all our caps again
+    add = []
+    remove = [cs.CHANNEL_TYPE_STREAMED_MEDIA,
+            cs.CHANNEL_TYPE_STREAM_TUBE]
+    caps = conn.Capabilities.AdvertiseCapabilities(add, remove)
+    (disco_response, caps_str, _) = receive_presence_and_ask_caps(q, stream,
+            False)
+    check_caps(disco_response, caps_str, [])
+
+    # Add caps selectively (i.e. what a client that actually understood the
+    # old Capabilities interface would do). With AUDIO and GTALK_P2P, we're
+    # callable, but not via ICE-UDP, and not with video.
+    #
+    # (Jingle and raw UDP need no special client support, so are automatically
+    # enabled whenever we can do audio or video.)
+    add = [(cs.CHANNEL_TYPE_STREAMED_MEDIA,
+        cs.MEDIA_CAP_AUDIO | cs.MEDIA_CAP_GTALKP2P)]
+    remove = []
+    caps = conn.Capabilities.AdvertiseCapabilities(add, remove)
+    (disco_response, caps_str, _) = receive_presence_and_ask_caps(q, stream,
+            False)
+    check_caps(disco_response, caps_str,
+            [ns.GOOGLE_P2P, ns.JINGLE_TRANSPORT_RAWUDP, ns.JINGLE,
+                ns.JINGLE_015, ns.GOOGLE_FEAT_VOICE, ns.JINGLE_RTP_AUDIO,
+                ns.JINGLE_RTP, ns.JINGLE_015_AUDIO])
+
+    # Remove all our caps again
+    add = []
+    remove = [cs.CHANNEL_TYPE_STREAMED_MEDIA,
+            cs.CHANNEL_TYPE_STREAM_TUBE]
+    caps = conn.Capabilities.AdvertiseCapabilities(add, remove)
+    (disco_response, caps_str, _) = receive_presence_and_ask_caps(q, stream,
+            False)
+    check_caps(disco_response, caps_str, [])
+
+    # With AUDIO but no transport, we are only callable via raw UDP, which
+    # Google clients cannot do.
+    add = [(cs.CHANNEL_TYPE_STREAMED_MEDIA, cs.MEDIA_CAP_AUDIO)]
+    remove = []
+    caps = conn.Capabilities.AdvertiseCapabilities(add, remove)
+    (disco_response, caps_str, _) = receive_presence_and_ask_caps(q, stream,
+            False)
+    check_caps(disco_response, caps_str,
+            [ns.JINGLE_TRANSPORT_RAWUDP, ns.JINGLE,
+                ns.JINGLE_015, ns.JINGLE_RTP_AUDIO,
+                ns.JINGLE_RTP, ns.JINGLE_015_AUDIO])
+
+    # Remove all our caps again
+    add = []
+    remove = [cs.CHANNEL_TYPE_STREAMED_MEDIA,
+            cs.CHANNEL_TYPE_STREAM_TUBE]
+    caps = conn.Capabilities.AdvertiseCapabilities(add, remove)
+    (disco_response, caps_str, _) = receive_presence_and_ask_caps(q, stream,
+            False)
+    check_caps(disco_response, caps_str, [])
+
+    # With VIDEO and ICE-UDP only, we are very futuristic indeed.
+    # Google clients cannot interop with us.
+    add = [(cs.CHANNEL_TYPE_STREAMED_MEDIA,
+        cs.MEDIA_CAP_VIDEO | cs.MEDIA_CAP_ICEUDP)]
+    remove = []
+    caps = conn.Capabilities.AdvertiseCapabilities(add, remove)
+    (disco_response, caps_str, _) = receive_presence_and_ask_caps(q, stream,
+            False)
+    check_caps(disco_response, caps_str,
+            [ns.JINGLE_TRANSPORT_ICEUDP, ns.JINGLE_TRANSPORT_RAWUDP, ns.JINGLE,
+                ns.JINGLE_015, ns.JINGLE_RTP_VIDEO,
+                ns.JINGLE_RTP, ns.JINGLE_015_VIDEO])
+
+if __name__ == '__main__':
+    exec_test(run_test)
-- 
1.5.6.5