def send_presence(q, conn, stream, contact, caps, initial=True, show=None): h = conn.RequestHandles(cs.HT_CONTACT, [contact])[0] if initial: stream.send(make_presence(contact, status='hello')) q.expect_many( EventPattern('dbus-signal', signal='PresenceUpdate', args=[{ h: (0L, { u'available': { 'message': 'hello' } }) }]), EventPattern('dbus-signal', signal='PresencesChanged', args=[{ h: (2, u'available', 'hello') }])) # no special capabilities assertEquals([(h, cs.CHANNEL_TYPE_TEXT, 3, 0)], conn.Capabilities.GetCapabilities([h])) # send updated presence with caps info stream.send(make_presence(contact, show=show, status='hello', caps=caps)) return h
def test(q, bus, conn): call_async(q, conn.Future, 'EnsureSidecar', STATUS_IFACE) conn.Connect() # Now we're connected, the call we made earlier should return. e = q.expect('dbus-return', method='EnsureSidecar') path, props = e.value assertEquals({}, props)
def outgoing_reply(q, bus, conn): path, incoming, stanza = setup_outgoing_tests(q, bus, conn) # reply with nothing reply = make_result_iq(stanza) incoming.send(reply) e = q.expect('dbus-signal', signal='Replied', path=path) args, xml = e.args assertEquals({}, args) assertEquals('<?xml version="1.0" encoding="UTF-8"?>\n' \ + '<message xmlns="urn:ytstenut:message"/>\n', xml)
def setup_outgoing_tests(q, bus, conn): handle, _, listener = setup_tests(q, bus, conn) # okay we got our contact, let's go request_props = { cs.CHANNEL_TYPE: ycs.CHANNEL_IFACE, cs.TARGET_HANDLE_TYPE: cs.HT_CONTACT, cs.TARGET_HANDLE: handle, ycs.REQUEST_TYPE: ycs.REQUEST_TYPE_GET, ycs.REQUEST_ATTRIBUTES: { 'hi': 'mom' }, ycs.TARGET_SERVICE: 'the.target.service', ycs.INITIATOR_SERVICE: 'the.initiator.service' } call_async(q, conn.Requests, 'CreateChannel', request_props) e, _ = q.expect_many(EventPattern('dbus-return', method='CreateChannel'), EventPattern('dbus-signal', signal='NewChannels')) path, props = e.value for k, v in request_props.items(): assertEquals(v, props[k]) # finally we have our channel chan = wrap_channel(bus, conn, path) # let's check we can't call Fail()/Reply() call_async(q, chan, 'Fail', ycs.ERROR_TYPE_CANCEL, 'lol', 'whut', 'pear') q.expect('dbus-error', method='Fail') call_async(q, chan, 'Reply', {'lol': 'whut'}, '') q.expect('dbus-error', method='Reply') # okay enough, let's move on. call_async(q, chan, 'Request') e, _ = q.expect_many( EventPattern('incoming-connection', listener=listener), EventPattern('dbus-return', method='Request')) incoming = e.connection q.expect('stream-opened', connection=incoming) q.expect('stream-features', connection=incoming) e = q.expect('stream-iq', connection=incoming) assertEquals('get', e.iq_type) assertEquals('message', e.query_name) assertEquals('urn:ytstenut:message', e.query_ns) # we shouldn't be able to call this again call_async(q, chan, 'Request') q.expect('dbus-error', method='Request') return path, incoming, e.stanza
def setup_outgoing_tests(q, bus, conn): handle, _, listener = setup_tests(q, bus, conn) # okay we got our contact, let's go request_props = { cs.CHANNEL_TYPE: ycs.CHANNEL_IFACE, cs.TARGET_HANDLE_TYPE: cs.HT_CONTACT, cs.TARGET_HANDLE: handle, ycs.REQUEST_TYPE: ycs.REQUEST_TYPE_GET, ycs.REQUEST_ATTRIBUTES: {'hi': 'mom'}, ycs.TARGET_SERVICE: 'the.target.service', ycs.INITIATOR_SERVICE: 'the.initiator.service' } call_async(q, conn.Requests, 'CreateChannel', request_props) e, _ = q.expect_many(EventPattern('dbus-return', method='CreateChannel'), EventPattern('dbus-signal', signal='NewChannels')) path, props = e.value for k, v in request_props.items(): assertEquals(v, props[k]) # finally we have our channel chan = wrap_channel(bus, conn, path) # let's check we can't call Fail()/Reply() call_async(q, chan, 'Fail', ycs.ERROR_TYPE_CANCEL, 'lol', 'whut', 'pear') q.expect('dbus-error', method='Fail') call_async(q, chan, 'Reply', {'lol':'whut'}, '') q.expect('dbus-error', method='Reply') # okay enough, let's move on. call_async(q, chan, 'Request') e, _ = q.expect_many(EventPattern('incoming-connection', listener=listener), EventPattern('dbus-return', method='Request')) incoming = e.connection q.expect('stream-opened', connection=incoming) q.expect('stream-features', connection=incoming) e = q.expect('stream-iq', connection=incoming) assertEquals('get', e.iq_type) assertEquals('message', e.query_name) assertEquals('urn:ytstenut:message', e.query_ns) # we shouldn't be able to call this again call_async(q, chan, 'Request') q.expect('dbus-error', method='Request') return path, incoming, e.stanza
def send_presence(q, conn, stream, contact, caps, initial=True, show=None): h = conn.RequestHandles(cs.HT_CONTACT, [contact])[0] if initial: stream.send(make_presence(contact, status='hello')) q.expect_many( EventPattern('dbus-signal', signal='PresenceUpdate', args=[{h: (0L, {u'available': {'message': 'hello'}})}]), EventPattern('dbus-signal', signal='PresencesChanged', args=[{h: (2, u'available', 'hello')}])) # no special capabilities assertEquals([(h, cs.CHANNEL_TYPE_TEXT, 3, 0)], conn.Capabilities.GetCapabilities([h])) # send updated presence with caps info stream.send(make_presence(contact, show=show, status='hello', caps=caps)) return h
def disco_caps(q, stream, presence): c_nodes = xpath.queryForNodes('/presence/c', presence.stanza) assert c_nodes is not None assertLength(1, c_nodes) hash = c_nodes[0].attributes['hash'] ver = c_nodes[0].attributes['ver'] node = c_nodes[0].attributes['node'] assertEquals('sha-1', hash) # ask caps request = \ elem_iq(stream, 'get', from_='[email protected]/resource')( elem(ns.DISCO_INFO, 'query', node=(node + '#' + ver)) ) stream.send(request) # receive caps event = q.expect('stream-iq', query_ns=ns.DISCO_INFO, iq_id=request['id']) # Check that Gabble's announcing the identity we think it should be. identity_nodes = xpath.queryForNodes('/iq/query/identity', event.stanza) assertLength(1, identity_nodes) identity_node = identity_nodes[0] assertEquals('client', identity_node['category']) assertEquals(config.CLIENT_TYPE, identity_node['type']) assertEquals(config.PACKAGE_STRING, identity_node['name']) assertDoesNotContain('xml:lang', identity_node.attributes) identity = 'client/%s//%s' % (config.CLIENT_TYPE, config.PACKAGE_STRING) features = [] for feature in xpath.queryForNodes('/iq/query/feature', event.stanza): features.append(feature['var']) # Check if the hash matches the announced capabilities assertEquals(compute_caps_hash([identity], features, {}), ver) return (event, features)
def outgoing_fail(q, bus, conn): path, incoming, stanza = setup_outgoing_tests(q, bus, conn) # construct a nice error reply reply = IQ(None, 'error') reply['id'] = stanza['id'] error = reply.addElement('error') error['type'] = 'cancel' error['code'] = '409' error.addElement((ns.STANZA, 'conflict')) error.addElement((ycs.MESSAGE_NS, 'yodawg')) text = error.addElement((ns.STANZA, 'text'), content='imma let you finish') incoming.send(reply) e = q.expect('dbus-signal', signal='Failed', path=path) error_type, stanza_error_name, yst_error_name, text = e.args assertEquals(ycs.ERROR_TYPE_CANCEL, error_type) assertEquals('conflict', stanza_error_name) assertEquals('yodawg', yst_error_name) assertEquals('imma let you finish', text)
def test(q, bus, conn): # we won't be using any data forms, so these two shouldn't ever be # fired. q.forbid_events([EventPattern('dbus-signal', signal='ServiceAdded'), EventPattern('dbus-signal', signal='ServiceRemoved')]) call_async(q, conn.Future, 'EnsureSidecar', ycs.STATUS_IFACE) conn.Connect() # Now we're connected, the call we made earlier should return. e = q.expect('dbus-return', method='EnsureSidecar') path, props = e.value assertEquals({}, props) status = ProxyWrapper(bus.get_object(conn.bus_name, path), ycs.STATUS_IFACE, {}) # bad capability argument call_async(q, status, 'AdvertiseStatus', '', 'service.name', '') q.expect('dbus-error', method='AdvertiseStatus') # bad service name call_async(q, status, 'AdvertiseStatus', CAP_NAME, '', '') q.expect('dbus-error', method='AdvertiseStatus') # we can't test that the message type="headline" stanza is # actually received because it's thrown into the loopback stream # immediately. # announce a contact with the right caps ver = compute_caps_hash([], [CAP_NAME + '+notify'], {}) txt_record = { "txtvers": "1", "status": "avail", "node": CLIENT_NAME, "ver": ver, "hash": "sha-1"} contact_name = "test-status@" + get_host_name() listener, port = setup_stream_listener(q, contact_name) announcer = AvahiAnnouncer(contact_name, "_presence._tcp", port, txt_record) handle = wait_for_contact_in_publish(q, bus, conn, contact_name) # this is the first presence, Salut connects to the contact e = q.expect('incoming-connection', listener=listener) incoming = e.connection # Salut looks up its capabilities event = q.expect('stream-iq', connection=incoming, query_ns=ns.DISCO_INFO) query_node = xpath.queryForNodes('/iq/query', event.stanza)[0] assertEquals(CLIENT_NAME + '#' + ver, query_node.attributes['node']) contact_handle = conn.RequestHandles(cs.HT_CONTACT, [contact_name])[0] # send good reply result = make_result_iq(event.stanza) query = result.firstChildElement() query['node'] = CLIENT_NAME + '#' + ver feature = query.addElement('feature') feature['var'] = CAP_NAME + '+notify' incoming.send(result) # this will be fired as text channel caps will be fired q.expect('dbus-signal', signal='ContactCapabilitiesChanged', predicate=lambda e: contact_handle in e.args[0]) # okay now we know about the contact's caps, we can go ahead discovered = status.Get(ycs.STATUS_IFACE, 'DiscoveredStatuses', dbus_interface=dbus.PROPERTIES_IFACE) assertEquals({}, discovered) el = Element(('urn:ytstenut:status', 'status')) el['activity'] = 'messing-with-your-stuff' desc = el.addElement('ytstenut:description', content='Yeah sorry about that') desc['xml:lang'] = 'en-GB' call_async(q, status, 'AdvertiseStatus', CAP_NAME, 'ants.in.their.pants', el.toXml()) e, _, sig = q.expect_many(EventPattern('stream-message', connection=incoming), EventPattern('dbus-return', method='AdvertiseStatus'), EventPattern('dbus-signal', signal='StatusChanged', interface=ycs.STATUS_IFACE)) # check message message = e.stanza event = message.children[0] items = event.children[0] item = items.children[0] status_el = item.children[0] assertEquals('status', status_el.name) assertEquals('messing-with-your-stuff', status_el['activity']) assertEquals('ants.in.their.pants', status_el['from-service']) assertEquals(CAP_NAME, status_el['capability']) # check signal contact_id, capability, service_name, status_str = sig.args assertEquals(CAP_NAME, capability) assertEquals('ants.in.their.pants', service_name) assertNotEquals('', status_str) # check property discovered = status.Get(ycs.STATUS_IFACE, 'DiscoveredStatuses', dbus_interface=dbus.PROPERTIES_IFACE) assertEquals({'testsuite@testsuite': {CAP_NAME: {'ants.in.their.pants': status_str}}}, discovered) # set another el = Element(('urn:ytstenut:status', 'status')) el['activity'] = 'rofling' desc = el.addElement('ytstenut:description', content='U MAD?') desc['xml:lang'] = 'en-GB' call_async(q, status, 'AdvertiseStatus', CAP_NAME, 'bananaman.on.holiday', el.toXml()) e, _, sig = q.expect_many(EventPattern('stream-message', connection=incoming), EventPattern('dbus-return', method='AdvertiseStatus'), EventPattern('dbus-signal', signal='StatusChanged', interface=ycs.STATUS_IFACE)) # check message message = e.stanza event = message.children[0] items = event.children[0] item = items.children[0] status_el = item.children[0] assertEquals('status', status_el.name) assertEquals('rofling', status_el['activity']) assertEquals('bananaman.on.holiday', status_el['from-service']) assertEquals(CAP_NAME, status_el['capability']) # check signal contact_id, capability, service_name, bananaman_status_str = sig.args assertEquals(CAP_NAME, capability) assertEquals('bananaman.on.holiday', service_name) assertNotEquals('', bananaman_status_str) # check property discovered = status.Get(ycs.STATUS_IFACE, 'DiscoveredStatuses', dbus_interface=dbus.PROPERTIES_IFACE) assertEquals({'testsuite@testsuite': {CAP_NAME: { 'ants.in.their.pants': status_str, 'bananaman.on.holiday': bananaman_status_str}}}, discovered) # unset the status from one service call_async(q, status, 'AdvertiseStatus', CAP_NAME, 'ants.in.their.pants', '') e, _, sig = q.expect_many(EventPattern('stream-message', connection=incoming), EventPattern('dbus-return', method='AdvertiseStatus'), EventPattern('dbus-signal', signal='StatusChanged', interface=ycs.STATUS_IFACE)) # check message message = e.stanza event = message.children[0] items = event.children[0] item = items.children[0] status_el = item.children[0] assertEquals('status', status_el.name) assertEquals('ants.in.their.pants', status_el['from-service']) assertEquals(CAP_NAME, status_el['capability']) assert 'activity' not in status_el.attributes assertEquals([], status_el.children) # check signal contact_id, capability, service_name, status_str = sig.args assertEquals(CAP_NAME, capability) assertEquals('ants.in.their.pants', service_name) assertEquals('', status_str) # check property discovered = status.Get(ycs.STATUS_IFACE, 'DiscoveredStatuses', dbus_interface=dbus.PROPERTIES_IFACE) assertEquals({'testsuite@testsuite': {CAP_NAME: { 'bananaman.on.holiday': bananaman_status_str}}}, discovered) # unset the status from the other service call_async(q, status, 'AdvertiseStatus', CAP_NAME, 'bananaman.on.holiday', '') e, _, sig = q.expect_many(EventPattern('stream-message', connection=incoming), EventPattern('dbus-return', method='AdvertiseStatus'), EventPattern('dbus-signal', signal='StatusChanged', interface=ycs.STATUS_IFACE)) # check message message = e.stanza event = message.children[0] items = event.children[0] item = items.children[0] status_el = item.children[0] assertEquals('status', status_el.name) assertEquals('bananaman.on.holiday', status_el['from-service']) assertEquals(CAP_NAME, status_el['capability']) assert 'activity' not in status_el.attributes assertEquals([], status_el.children) # check signal contact_id, capability, service_name, status_str = sig.args assertEquals(CAP_NAME, capability) assertEquals('bananaman.on.holiday', service_name) assertEquals('', status_str) # check property discovered = status.Get(ycs.STATUS_IFACE, 'DiscoveredStatuses', dbus_interface=dbus.PROPERTIES_IFACE) assertEquals({}, discovered)
def expect_disco(q, contact, client, caps): # Gabble looks up our capabilities event = q.expect('stream-iq', to=contact, query_ns=ns.DISCO_INFO) assertEquals(client + '#' + caps['ver'], event.query['node']) return event.stanza
def setup_incoming_tests(q, bus, conn): handle, contact_name, listener = setup_tests(q, bus, conn) self_handle = conn.GetSelfHandle() self_handle_name = conn.InspectHandles(cs.HT_CONTACT, [self_handle])[0] AvahiListener(q).listen_for_service("_presence._tcp") e = q.expect('service-added', name=self_handle_name, protocol = avahi.PROTO_INET) service = e.service service.resolve() e = q.expect('service-resolved', service=service) outbound = connect_to_stream(q, contact_name, self_handle_name, str(e.pt), e.port) e = q.expect('connection-result') assert e.succeeded, e.reason e = q.expect('stream-opened', connection=outbound) iq = IQ(None, 'get') iq['id'] = 'le-loldongs' iq['from'] = contact_name iq['to'] = self_handle_name msg = iq.addElement((ycs.MESSAGE_NS, 'message')) msg['from-service'] = 'the.from.service' msg['to-service'] = 'the.to.service' msg['owl-companions'] = 'the pussy cat' msg['destination'] = 'sea' msg['seacraft'] = 'beautiful pea green boat' lol = msg.addElement((None, 'lol')) lol['some'] = 'stuff' lol['to'] = 'fill' lol['the'] = 'time' lol.addElement((None, 'look-into-my-eyes'), content='and tell me how boring writing these tests is') outbound.send(iq) e = q.expect('dbus-signal', signal='NewChannels', predicate=lambda e: e.args[0][0][1][cs.CHANNEL_TYPE] == ycs.CHANNEL_IFACE) path, props = e.args[0][0] assertEquals(handle, props[cs.INITIATOR_HANDLE]) assertEquals(contact_name, props[cs.INITIATOR_ID]) assertEquals(False, props[cs.REQUESTED]) assertEquals(handle, props[cs.TARGET_HANDLE]) assertEquals(cs.HT_CONTACT, props[cs.TARGET_HANDLE_TYPE]) assertEquals(contact_name, props[cs.TARGET_ID]) assertEquals('the.from.service', props[ycs.INITIATOR_SERVICE]) assertEquals('the.to.service', props[ycs.TARGET_SERVICE]) assertEquals(ycs.REQUEST_TYPE_GET, props[ycs.REQUEST_TYPE]) assertEquals({'destination': 'sea', 'owl-companions': 'the pussy cat', 'seacraft': 'beautiful pea green boat'}, props[ycs.REQUEST_ATTRIBUTES]) assertEquals('<?xml version="1.0" encoding="UTF-8"?>\n' \ '<message seacraft="beautiful pea green boat" ' \ 'from-service="the.from.service" destination="sea" ' \ 'owl-companions="the pussy cat" to-service="the.to.service" ' \ 'xmlns="urn:ytstenut:message">' \ '<lol to="fill" the="time" some="stuff">' \ '<look-into-my-eyes>and tell me how boring ' \ 'writing these tests is</look-into-my-eyes>' \ '</lol></message>\n', props[ycs.REQUEST_BODY]) # finally we have our channel chan = wrap_channel(bus, conn, path) # let's check we can't call Request() call_async(q, chan, 'Request') q.expect('dbus-error', method='Request') return chan, outbound, contact_name, self_handle_name
def check_message(message): assertEquals('message', message.name) assertEquals(ycs.MESSAGE_NS, message.uri) assertEquals('beautiful pea green boat', message['seacraft']) assertEquals('sea', message['destination']) assertEquals('the pussy cat', message['owl-companions']) assertEquals('the.from.service', message['from-service']) assertEquals('the.to.service', message['to-service']) assertEquals(1, len(message.children)) lol = message.children[0] assertEquals('lol', lol.name) assertEquals('fill', lol['to']) assertEquals('time', lol['the']) assertEquals('stuff', lol['some']) assertEquals(1, len(lol.children)) look = lol.children[0] assertEquals('look-into-my-eyes', look.name) assertEquals(1, len(look.children)) assertEquals('and tell me how boring writing these tests is', look.children[0])
def check_error(error): assertEquals('error', error.name) assertEquals('auth', error['type']) assertEquals(3, len(error.children)) for c in error.children: if c.name == 'auth': assertEquals(ns.STANZA, c.uri) elif c.name == 'omgwtfbbq': assertEquals(ycs.MESSAGE_NS, c.uri) elif c.name == 'text': assertEquals(ns.STANZA, c.uri) assertEquals(1, len(c.children)) assertEquals('I most certainly dont feel like dancing', c.children[0]) else: raise
def incoming_reply(q, bus, conn): chan, outbound, contact_name, self_handle_name = \ setup_incoming_tests(q, bus, conn) moar = Element((ycs.MESSAGE_NS, 'message')) moar['ninety-nine-problems'] = 'but a sauvignon blanc aint one' moar['also'] = 'my mum said hi' trollface = moar.addElement('trollface', content='problem?') call_async( q, chan, 'Reply', { 'ninety-nine-problems': 'but a sauvignon blanc aint one', 'also': 'my mum said hi' }, moar.toXml()) _, e = q.expect_many(EventPattern('dbus-return', method='Reply'), EventPattern('stream-iq', connection=outbound)) iq = e.stanza assertEquals('le-loldongs', iq['id']) assertEquals('result', iq['type']) assertEquals(self_handle_name, iq['from']) assertEquals(contact_name, iq['to']) assertEquals(1, len(iq.children)) message = iq.children[0] assertEquals('message', message.name) assertEquals(ycs.MESSAGE_NS, message.uri) assertEquals('my mum said hi', message['also']) assertEquals('but a sauvignon blanc aint one', message['ninety-nine-problems']) assertEquals('the.from.service', message['to-service']) assertEquals('the.to.service', message['from-service']) assertEquals(1, len(message.children)) trollface = message.children[0] assertEquals('trollface', trollface.name) assertEquals(1, len(trollface.children)) assertEquals('problem?', trollface.children[0]) # check we can't call anything any more call_async(q, chan, 'Fail', ycs.ERROR_TYPE_CANCEL, 'lol', 'whut', 'pear') q.expect('dbus-error', method='Fail') call_async(q, chan, 'Reply', {'lol': 'whut'}, '') q.expect('dbus-error', method='Reply') call_async(q, chan, 'Request') q.expect('dbus-error', method='Request')
# Gabble looks up our capabilities event = q.expect('stream-iq', to=contact, query_ns=ns.DISCO_INFO) assertEquals(client + '#' + caps['ver'], event.query['node']) return event.stanza def send_disco_reply(stream, stanza, identities, features, dataforms={}): stream.send( make_caps_disco_reply(stream, stanza, identities, features, dataforms)) if __name__ == '__main__': # example from XEP-0115 assertEquals('QgayPKawpkPSDYmwT/WM94uAlu0=', compute_caps_hash(['client/pc//Exodus 0.9.1'], ["http://jabber.org/protocol/disco#info", "http://jabber.org/protocol/disco#items", "http://jabber.org/protocol/muc", "http://jabber.org/protocol/caps"], {})) # another example from XEP-0115 identities = [u'client/pc/en/Psi 0.11', u'client/pc/el/Ψ 0.11'] features = [ u'http://jabber.org/protocol/caps', u'http://jabber.org/protocol/disco#info', u'http://jabber.org/protocol/disco#items', u'http://jabber.org/protocol/muc', ] dataforms = { u'urn:xmpp:dataforms:softwareinfo': { u'ip_version': [u'ipv4', u'ipv6'],
def test(q, bus, conn): contact_name = "test-hct@" + get_host_name() call_async(q, conn.Future, 'EnsureSidecar', ycs.STATUS_IFACE) conn.Connect() q.expect('dbus-signal', signal='StatusChanged', args=[0, 0]) e = q.expect('dbus-return', method='EnsureSidecar') path, props = e.value assertEquals({}, props) status = ProxyWrapper(bus.get_object(conn.bus_name, path), ycs.STATUS_IFACE, {}) discovered = status.Get(ycs.STATUS_IFACE, 'DiscoveredServices', dbus_interface=dbus.PROPERTIES_IFACE) assertEquals({}, discovered) self_handle = conn.GetSelfHandle() self_handle_name = conn.InspectHandles(cs.HT_CONTACT, [self_handle])[0] AvahiListener(q).listen_for_service("_presence._tcp") e = q.expect('service-added', name = self_handle_name, protocol = avahi.PROTO_INET) service = e.service service.resolve() q.expect('service-resolved', service=service) # now update the caps conn.ContactCapabilities.UpdateCapabilities([ ('well.gnome.name', [], ['org.freedesktop.ytstenut.xpmn.Channel/uid/org.gnome.Banshee', 'org.freedesktop.ytstenut.xpmn.Channel/type/application', 'org.freedesktop.ytstenut.xpmn.Channel/name/en_GB/Banshee Media Player', 'org.freedesktop.ytstenut.xpmn.Channel/name/fr/Banshee Lecteur de Musique', 'org.freedesktop.ytstenut.xpmn.Channel/caps/urn:ytstenut:capabilities:yts-caps-audio', 'org.freedesktop.ytstenut.xpmn.Channel/caps/urn:ytstenut:data:jingle:rtp'])]) e, _ = q.expect_many(EventPattern('service-resolved', service=service), EventPattern('dbus-signal', signal='ServiceAdded')) node = txt_get_key(e.txt, 'node') ver = txt_get_key(e.txt, 'ver') # don't check the hash here because it could change in salut and # it would break this test even though it wouldn't /actually/ be # broken outbound = connect_to_stream(q, contact_name, self_handle_name, str(e.pt), e.port) e = q.expect('connection-result') assert e.succeeded, e.reason e = q.expect('stream-opened', connection=outbound) # ask what this ver means request = \ elem_iq(outbound, 'get', from_=contact_name)( elem(ns.DISCO_INFO, 'query', node=(node + '#' + ver)) ) outbound.send(request) # receive caps e = q.expect('stream-iq', connection=outbound, query_ns=ns.DISCO_INFO, iq_id=request['id']) iq = e.stanza query = iq.children[0] x = None for child in query.children: if child.name == 'x' and child.uri == ns.X_DATA: # we should only have one child assert x is None x = child # don't break here as we can waste time to make sure x # isn't assigned twice assert x is not None for field in x.children: if field['var'] == 'FORM_TYPE': assertEquals('hidden', field['type']) assertEquals('urn:ytstenut:capabilities#org.gnome.Banshee', field.children[0].children[0]) elif field['var'] == 'type': assertEquals('application', field.children[0].children[0]) elif field['var'] == 'name': names = [a.children[0] for a in field.children] assertLength(2, names) assertContains('en_GB/Banshee Media Player', names) assertContains('fr/Banshee Lecteur de Musique', names) elif field['var'] == 'capabilities': caps = [a.children[0] for a in field.children] assertLength(2, caps) assertContains('urn:ytstenut:capabilities:yts-caps-audio', caps) assertContains('urn:ytstenut:data:jingle:rtp', caps) else: assert False # now add another service forbidden = [EventPattern('dbus-signal', signal='ServiceRemoved')] q.forbid_events(forbidden) conn.ContactCapabilities.UpdateCapabilities([ ('another.nice.gname', [], ['org.freedesktop.ytstenut.xpmn.Channel/uid/org.gnome.Eog', 'org.freedesktop.ytstenut.xpmn.Channel/type/application', 'org.freedesktop.ytstenut.xpmn.Channel/name/en_GB/Eye Of Gnome', 'org.freedesktop.ytstenut.xpmn.Channel/name/it/Occhio Di uno Gnomo', 'org.freedesktop.ytstenut.xpmn.Channel/caps/urn:ytstenut:capabilities:yts-picz'])]) e = q.expect('dbus-signal', signal='ServiceAdded') sync_dbus(bus, q, conn) discovered = status.Get(ycs.STATUS_IFACE, 'DiscoveredServices', dbus_interface=dbus.PROPERTIES_IFACE) assertEquals({'testsuite@testsuite': {'org.gnome.Banshee': ('application', {'en_GB': 'Banshee Media Player', 'fr': 'Banshee Lecteur de Musique'}, ['urn:ytstenut:capabilities:yts-caps-audio', 'urn:ytstenut:data:jingle:rtp']), 'org.gnome.Eog': ('application', {'en_GB': 'Eye Of Gnome', 'it': 'Occhio Di uno Gnomo'}, ['urn:ytstenut:capabilities:yts-picz'])}}, discovered) q.unforbid_events(forbidden) forbidden = [EventPattern('dbus-signal', signal='ServiceRemoved', args=[self_handle_name, 'org.gnome.Eog'])] q.forbid_events(forbidden) conn.ContactCapabilities.UpdateCapabilities([ ('well.gnome.name', [], [])]) e = q.expect('dbus-signal', signal='ServiceRemoved', args=[self_handle_name, 'org.gnome.Banshee']) sync_dbus(bus, q, conn) discovered = status.Get(ycs.STATUS_IFACE, 'DiscoveredServices', dbus_interface=dbus.PROPERTIES_IFACE) assertEquals({'testsuite@testsuite': {'org.gnome.Eog': ('application', {'en_GB': 'Eye Of Gnome', 'it': 'Occhio Di uno Gnomo'}, ['urn:ytstenut:capabilities:yts-picz'])}}, discovered) q.unforbid_events(forbidden) conn.ContactCapabilities.UpdateCapabilities([ ('another.nice.gname', [], [])]) e = q.expect('dbus-signal', signal='ServiceRemoved', args=[self_handle_name, 'org.gnome.Eog']) discovered = status.Get(ycs.STATUS_IFACE, 'DiscoveredServices', dbus_interface=dbus.PROPERTIES_IFACE) assertEquals({}, discovered)
def incoming_reply(q, bus, conn): chan, outbound, contact_name, self_handle_name = \ setup_incoming_tests(q, bus, conn) moar = Element((ycs.MESSAGE_NS, 'message')) moar['ninety-nine-problems'] = 'but a sauvignon blanc aint one' moar['also'] = 'my mum said hi' trollface = moar.addElement('trollface', content='problem?') call_async(q, chan, 'Reply', {'ninety-nine-problems': 'but a sauvignon blanc aint one', 'also': 'my mum said hi'}, moar.toXml()) _, e = q.expect_many(EventPattern('dbus-return', method='Reply'), EventPattern('stream-iq', connection=outbound)) iq = e.stanza assertEquals('le-loldongs', iq['id']) assertEquals('result', iq['type']) assertEquals(self_handle_name, iq['from']) assertEquals(contact_name, iq['to']) assertEquals(1, len(iq.children)) message = iq.children[0] assertEquals('message', message.name) assertEquals(ycs.MESSAGE_NS, message.uri) assertEquals('my mum said hi', message['also']) assertEquals('but a sauvignon blanc aint one', message['ninety-nine-problems']) assertEquals('the.from.service', message['to-service']) assertEquals('the.to.service', message['from-service']) assertEquals(1, len(message.children)) trollface = message.children[0] assertEquals('trollface', trollface.name) assertEquals(1, len(trollface.children)) assertEquals('problem?', trollface.children[0]) # check we can't call anything any more call_async(q, chan, 'Fail', ycs.ERROR_TYPE_CANCEL, 'lol', 'whut', 'pear') q.expect('dbus-error', method='Fail') call_async(q, chan, 'Reply', {'lol':'whut'}, '') q.expect('dbus-error', method='Reply') call_async(q, chan, 'Request') q.expect('dbus-error', method='Request')
def test(q, bus, conn): call_async(q, conn.Future, 'EnsureSidecar', ycs.STATUS_IFACE) conn.Connect() # Now we're connected, the call we made earlier should return. e = q.expect('dbus-return', method='EnsureSidecar') path, props = e.value assertEquals({}, props) status = ProxyWrapper(bus.get_object(conn.bus_name, path), ycs.STATUS_IFACE, {}) discovered = status.Get(ycs.STATUS_IFACE, 'DiscoveredServices', dbus_interface=dbus.PROPERTIES_IFACE) assertEquals({}, discovered) # announce a contact with the right caps ver = compute_caps_hash([], [], banshee) txt_record = { "txtvers": "1", "status": "avail", "node": CLIENT_NAME, "ver": ver, "hash": "sha-1" } contact_name = "test-service@" + get_host_name() listener, port = setup_stream_listener(q, contact_name) announcer = AvahiAnnouncer(contact_name, "_presence._tcp", port, txt_record) handle = wait_for_contact_in_publish(q, bus, conn, contact_name) # this is the first presence, Salut connects to the contact e = q.expect('incoming-connection', listener=listener) incoming = e.connection # Salut looks up its capabilities event = q.expect('stream-iq', connection=incoming, query_ns=ns.DISCO_INFO) query_node = xpath.queryForNodes('/iq/query', event.stanza)[0] assertEquals(CLIENT_NAME + '#' + ver, query_node.attributes['node']) contact_handle = conn.RequestHandles(cs.HT_CONTACT, [contact_name])[0] # send good reply result = make_result_iq(event.stanza) query = result.firstChildElement() query['node'] = CLIENT_NAME + '#' + ver x = query.addElement((ns.X_DATA, 'x')) x['type'] = 'result' # FORM_TYPE field = x.addElement((None, 'field')) field['var'] = 'FORM_TYPE' field['type'] = 'hidden' field.addElement((None, 'value'), content='urn:ytstenut:capabilities#org.gnome.Banshee') # type field = x.addElement((None, 'field')) field['var'] = 'type' field.addElement((None, 'value'), content='application') # name field = x.addElement((None, 'field')) field['var'] = 'name' field.addElement((None, 'value'), content='en_GB/Banshee Media Player') field.addElement((None, 'value'), content='fr/Banshee Lecteur de Musique') # capabilities field = x.addElement((None, 'field')) field['var'] = 'capabilities' field.addElement((None, 'value'), content='urn:ytstenut:capabilities:yts-caps-audio') field.addElement((None, 'value'), content='urn:ytstenut:data:jingle:rtp') incoming.send(result) # this will be fired as text channel caps will be fired _, e = q.expect_many( EventPattern('dbus-signal', signal='ContactCapabilitiesChanged', predicate=lambda e: contact_handle in e.args[0]), EventPattern('dbus-signal', signal='ServiceAdded')) contact_id, service_name, details = e.args assertEquals(contact_name, contact_id) assertEquals('org.gnome.Banshee', service_name) type, name_map, caps = details assertEquals('application', type) assertEquals( { 'en_GB': 'Banshee Media Player', 'fr': 'Banshee Lecteur de Musique' }, name_map) assertSameSets([ 'urn:ytstenut:capabilities:yts-caps-audio', 'urn:ytstenut:data:jingle:rtp' ], caps) discovered = status.Get(ycs.STATUS_IFACE, 'DiscoveredServices', dbus_interface=dbus.PROPERTIES_IFACE) assertEquals( { contact_name: { 'org.gnome.Banshee': ('application', { 'en_GB': 'Banshee Media Player', 'fr': 'Banshee Lecteur de Musique' }, [ 'urn:ytstenut:capabilities:yts-caps-audio', 'urn:ytstenut:data:jingle:rtp' ]) }, }, discovered) # add evince tmp = banshee.copy() tmp.update(evince) ver = compute_caps_hash([], [], tmp) txt_record['ver'] = ver announcer.update(txt_record) # Salut looks up our capabilities event = q.expect('stream-iq', connection=incoming, query_ns='http://jabber.org/protocol/disco#info') query_node = xpath.queryForNodes('/iq/query', event.stanza)[0] assert query_node.attributes['node'] == \ CLIENT_NAME + '#' + txt_record['ver'] # send good reply result['id'] = event.stanza['id'] query['node'] = CLIENT_NAME + '#' + ver x = query.addElement((ns.X_DATA, 'x')) x['type'] = 'result' # FORM_TYPE field = x.addElement((None, 'field')) field['var'] = 'FORM_TYPE' field['type'] = 'hidden' field.addElement((None, 'value'), content='urn:ytstenut:capabilities#org.gnome.Evince') # type field = x.addElement((None, 'field')) field['var'] = 'type' field.addElement((None, 'value'), content='application') # name field = x.addElement((None, 'field')) field['var'] = 'name' field.addElement((None, 'value'), content='en_GB/Evince Picture Viewer') field.addElement((None, 'value'), content='fr/Evince uh, ow do you say') # capabilities field = x.addElement((None, 'field')) field['var'] = 'capabilities' field.addElement((None, 'value'), content='urn:ytstenut:capabilities:pics') incoming.send(result) e = q.expect('dbus-signal', signal='ServiceAdded') contact_id, service_name, details = e.args assertEquals(contact_name, contact_id) assertEquals('org.gnome.Evince', service_name) type, name_map, caps = details assertEquals('application', type) assertEquals( { 'en_GB': 'Evince Picture Viewer', 'fr': 'Evince uh, ow do you say' }, name_map) assertSameSets(['urn:ytstenut:capabilities:pics'], caps) discovered = status.Get(ycs.STATUS_IFACE, 'DiscoveredServices', dbus_interface=dbus.PROPERTIES_IFACE) assertEquals( { contact_name: { 'org.gnome.Banshee': ('application', { 'en_GB': 'Banshee Media Player', 'fr': 'Banshee Lecteur de Musique' }, [ 'urn:ytstenut:capabilities:yts-caps-audio', 'urn:ytstenut:data:jingle:rtp' ]), 'org.gnome.Evince': ('application', { 'en_GB': 'Evince Picture Viewer', 'fr': 'Evince uh, ow do you say' }, ['urn:ytstenut:capabilities:pics']) } }, discovered) # remove evince ver = compute_caps_hash([], [], banshee) txt_record['ver'] = ver forbidden = [ EventPattern('dbus-signal', signal='stream-iq', connection=incoming) ] q.forbid_events(forbidden) announcer.update(txt_record) e = q.expect('dbus-signal', signal='ServiceRemoved') contact_id, service_name = e.args assertEquals(contact_name, contact_id) assertEquals('org.gnome.Evince', service_name) discovered = status.Get(ycs.STATUS_IFACE, 'DiscoveredServices', dbus_interface=dbus.PROPERTIES_IFACE) assertEquals( { contact_name: { 'org.gnome.Banshee': ('application', { 'en_GB': 'Banshee Media Player', 'fr': 'Banshee Lecteur de Musique' }, [ 'urn:ytstenut:capabilities:yts-caps-audio', 'urn:ytstenut:data:jingle:rtp' ]) }, }, discovered) sync_stream(q, incoming) q.unforbid_events(forbidden) # now just evince ver = compute_caps_hash([], [], evince) txt_record['ver'] = ver announcer.update(txt_record) # Salut looks up our capabilities event = q.expect('stream-iq', connection=incoming, query_ns='http://jabber.org/protocol/disco#info') query_node = xpath.queryForNodes('/iq/query', event.stanza)[0] assert query_node.attributes['node'] == \ CLIENT_NAME + '#' + txt_record['ver'] # send good reply result = make_result_iq(event.stanza) query = result.firstChildElement() query['node'] = CLIENT_NAME + '#' + ver x = query.addElement((ns.X_DATA, 'x')) x['type'] = 'result' # FORM_TYPE field = x.addElement((None, 'field')) field['var'] = 'FORM_TYPE' field['type'] = 'hidden' field.addElement((None, 'value'), content='urn:ytstenut:capabilities#org.gnome.Evince') # type field = x.addElement((None, 'field')) field['var'] = 'type' field.addElement((None, 'value'), content='application') # name field = x.addElement((None, 'field')) field['var'] = 'name' field.addElement((None, 'value'), content='en_GB/Evince Picture Viewer') field.addElement((None, 'value'), content='fr/Evince uh, ow do you say') # capabilities field = x.addElement((None, 'field')) field['var'] = 'capabilities' field.addElement((None, 'value'), content='urn:ytstenut:capabilities:pics') incoming.send(result) sa, sr = q.expect_many( EventPattern('dbus-signal', signal='ServiceAdded'), EventPattern('dbus-signal', signal='ServiceRemoved')) contact_id, service_name, details = sa.args assertEquals(contact_name, contact_id) assertEquals('org.gnome.Evince', service_name) type, name_map, caps = details assertEquals('application', type) assertEquals( { 'en_GB': 'Evince Picture Viewer', 'fr': 'Evince uh, ow do you say' }, name_map) assertSameSets(['urn:ytstenut:capabilities:pics'], caps) contact_id, service_name = sr.args assertEquals(contact_name, contact_id) assertEquals('org.gnome.Banshee', service_name) discovered = status.Get(ycs.STATUS_IFACE, 'DiscoveredServices', dbus_interface=dbus.PROPERTIES_IFACE) assertEquals( { contact_name: { 'org.gnome.Evince': ('application', { 'en_GB': 'Evince Picture Viewer', 'fr': 'Evince uh, ow do you say' }, ['urn:ytstenut:capabilities:pics']) } }, discovered) # just banshee again ver = compute_caps_hash([], [], banshee) txt_record['ver'] = ver forbidden = [ EventPattern('dbus-signal', signal='stream-iq', connection=incoming) ] q.forbid_events(forbidden) announcer.update(txt_record) sr, sa = q.expect_many( EventPattern('dbus-signal', signal='ServiceRemoved'), EventPattern('dbus-signal', signal='ServiceAdded')) contact_id, service_name = sr.args assertEquals(contact_name, contact_id) assertEquals('org.gnome.Evince', service_name) contact_id, service_name, details = sa.args assertEquals(contact_name, contact_id) assertEquals('org.gnome.Banshee', service_name) type, name_map, caps = details assertEquals('application', type) assertEquals( { 'en_GB': 'Banshee Media Player', 'fr': 'Banshee Lecteur de Musique' }, name_map) assertSameSets([ 'urn:ytstenut:capabilities:yts-caps-audio', 'urn:ytstenut:data:jingle:rtp' ], caps) discovered = status.Get(ycs.STATUS_IFACE, 'DiscoveredServices', dbus_interface=dbus.PROPERTIES_IFACE) assertEquals( { contact_name: { 'org.gnome.Banshee': ('application', { 'en_GB': 'Banshee Media Player', 'fr': 'Banshee Lecteur de Musique' }, [ 'urn:ytstenut:capabilities:yts-caps-audio', 'urn:ytstenut:data:jingle:rtp' ]) } }, discovered) sync_stream(q, incoming) q.unforbid_events(forbidden) # both again ver = compute_caps_hash([], [], tmp) txt_record['ver'] = ver announcer.update(txt_record) sa = q.expect('dbus-signal', signal='ServiceAdded') contact_id, service_name, details = sa.args assertEquals(contact_name, contact_id) assertEquals('org.gnome.Evince', service_name) type, name_map, caps = details assertEquals('application', type) assertEquals( { 'en_GB': 'Evince Picture Viewer', 'fr': 'Evince uh, ow do you say' }, name_map) assertSameSets(['urn:ytstenut:capabilities:pics'], caps) discovered = status.Get(ycs.STATUS_IFACE, 'DiscoveredServices', dbus_interface=dbus.PROPERTIES_IFACE) assertEquals( { contact_name: { 'org.gnome.Banshee': ('application', { 'en_GB': 'Banshee Media Player', 'fr': 'Banshee Lecteur de Musique' }, [ 'urn:ytstenut:capabilities:yts-caps-audio', 'urn:ytstenut:data:jingle:rtp' ]), 'org.gnome.Evince': ('application', { 'en_GB': 'Evince Picture Viewer', 'fr': 'Evince uh, ow do you say' }, ['urn:ytstenut:capabilities:pics']) } }, discovered) sync_stream(q, incoming) # and finally, nothing ver = compute_caps_hash([], [], {}) txt_record['ver'] = ver announcer.update(txt_record) # Salut looks up our capabilities event = q.expect('stream-iq', connection=incoming, query_ns='http://jabber.org/protocol/disco#info') query_node = xpath.queryForNodes('/iq/query', event.stanza)[0] assert query_node.attributes['node'] == \ CLIENT_NAME + '#' + txt_record['ver'] # send good reply result = make_result_iq(event.stanza) query = result.firstChildElement() query['node'] = CLIENT_NAME + '#' + ver incoming.send(result) q.expect_many( EventPattern('dbus-signal', signal='ServiceRemoved', args=[contact_name, 'org.gnome.Banshee']), EventPattern('dbus-signal', signal='ServiceRemoved', args=[contact_name, 'org.gnome.Evince'])) discovered = status.Get(ycs.STATUS_IFACE, 'DiscoveredServices', dbus_interface=dbus.PROPERTIES_IFACE) assertEquals({}, discovered)
def setup_incoming_tests(q, bus, conn): handle, contact_name, listener = setup_tests(q, bus, conn) self_handle = conn.GetSelfHandle() self_handle_name = conn.InspectHandles(cs.HT_CONTACT, [self_handle])[0] AvahiListener(q).listen_for_service("_presence._tcp") e = q.expect('service-added', name=self_handle_name, protocol=avahi.PROTO_INET) service = e.service service.resolve() e = q.expect('service-resolved', service=service) outbound = connect_to_stream(q, contact_name, self_handle_name, str(e.pt), e.port) e = q.expect('connection-result') assert e.succeeded, e.reason e = q.expect('stream-opened', connection=outbound) iq = IQ(None, 'get') iq['id'] = 'le-loldongs' iq['from'] = contact_name iq['to'] = self_handle_name msg = iq.addElement((ycs.MESSAGE_NS, 'message')) msg['from-service'] = 'the.from.service' msg['to-service'] = 'the.to.service' msg['owl-companions'] = 'the pussy cat' msg['destination'] = 'sea' msg['seacraft'] = 'beautiful pea green boat' lol = msg.addElement((None, 'lol')) lol['some'] = 'stuff' lol['to'] = 'fill' lol['the'] = 'time' lol.addElement((None, 'look-into-my-eyes'), content='and tell me how boring writing these tests is') outbound.send(iq) e = q.expect('dbus-signal', signal='NewChannels', predicate=lambda e: e.args[0][0][1][cs.CHANNEL_TYPE] == ycs. CHANNEL_IFACE) path, props = e.args[0][0] assertEquals(handle, props[cs.INITIATOR_HANDLE]) assertEquals(contact_name, props[cs.INITIATOR_ID]) assertEquals(False, props[cs.REQUESTED]) assertEquals(handle, props[cs.TARGET_HANDLE]) assertEquals(cs.HT_CONTACT, props[cs.TARGET_HANDLE_TYPE]) assertEquals(contact_name, props[cs.TARGET_ID]) assertEquals('the.from.service', props[ycs.INITIATOR_SERVICE]) assertEquals('the.to.service', props[ycs.TARGET_SERVICE]) assertEquals(ycs.REQUEST_TYPE_GET, props[ycs.REQUEST_TYPE]) assertEquals( { 'destination': 'sea', 'owl-companions': 'the pussy cat', 'seacraft': 'beautiful pea green boat' }, props[ycs.REQUEST_ATTRIBUTES]) assertEquals('<?xml version="1.0" encoding="UTF-8"?>\n' \ '<message seacraft="beautiful pea green boat" ' \ 'from-service="the.from.service" destination="sea" ' \ 'owl-companions="the pussy cat" to-service="the.to.service" ' \ 'xmlns="urn:ytstenut:message">' \ '<lol to="fill" the="time" some="stuff">' \ '<look-into-my-eyes>and tell me how boring ' \ 'writing these tests is</look-into-my-eyes>' \ '</lol></message>\n', props[ycs.REQUEST_BODY]) # finally we have our channel chan = wrap_channel(bus, conn, path) # let's check we can't call Request() call_async(q, chan, 'Request') q.expect('dbus-error', method='Request') return chan, outbound, contact_name, self_handle_name
def incoming_fail(q, bus, conn): chan, outbound, contact_name, self_handle_name = \ setup_incoming_tests(q, bus, conn) call_async(q, chan, 'Fail', ycs.ERROR_TYPE_AUTH, 'auth', 'omgwtfbbq', 'I most certainly dont feel like dancing') _, e = q.expect_many(EventPattern('dbus-return', method='Fail'), EventPattern('stream-iq', connection=outbound)) iq = e.stanza assertEquals('le-loldongs', iq['id']) assertEquals('error', iq['type']) assertEquals(self_handle_name, iq['from']) assertEquals(contact_name, iq['to']) assertEquals(2, len(iq.children)) def check_message(message): assertEquals('message', message.name) assertEquals(ycs.MESSAGE_NS, message.uri) assertEquals('beautiful pea green boat', message['seacraft']) assertEquals('sea', message['destination']) assertEquals('the pussy cat', message['owl-companions']) assertEquals('the.from.service', message['from-service']) assertEquals('the.to.service', message['to-service']) assertEquals(1, len(message.children)) lol = message.children[0] assertEquals('lol', lol.name) assertEquals('fill', lol['to']) assertEquals('time', lol['the']) assertEquals('stuff', lol['some']) assertEquals(1, len(lol.children)) look = lol.children[0] assertEquals('look-into-my-eyes', look.name) assertEquals(1, len(look.children)) assertEquals('and tell me how boring writing these tests is', look.children[0]) def check_error(error): assertEquals('error', error.name) assertEquals('auth', error['type']) assertEquals(3, len(error.children)) for c in error.children: if c.name == 'auth': assertEquals(ns.STANZA, c.uri) elif c.name == 'omgwtfbbq': assertEquals(ycs.MESSAGE_NS, c.uri) elif c.name == 'text': assertEquals(ns.STANZA, c.uri) assertEquals(1, len(c.children)) assertEquals('I most certainly dont feel like dancing', c.children[0]) else: raise for child in iq.children: if child.name == 'message': check_message(child) elif child.name == 'error': check_error(child) else: raise # check we can't call anything any more call_async(q, chan, 'Fail', ycs.ERROR_TYPE_CANCEL, 'lol', 'whut', 'pear') q.expect('dbus-error', method='Fail') call_async(q, chan, 'Reply', {'lol': 'whut'}, '') q.expect('dbus-error', method='Reply') call_async(q, chan, 'Request') q.expect('dbus-error', method='Request')
def test(q, bus, conn): # we won't be using any data forms, so these two shouldn't ever be # fired. q.forbid_events([ EventPattern('dbus-signal', signal='ServiceAdded'), EventPattern('dbus-signal', signal='ServiceRemoved') ]) call_async(q, conn.Future, 'EnsureSidecar', ycs.STATUS_IFACE) conn.Connect() # Now we're connected, the call we made earlier should return. e = q.expect('dbus-return', method='EnsureSidecar') path, props = e.value assertEquals({}, props) status = ProxyWrapper(bus.get_object(conn.bus_name, path), ycs.STATUS_IFACE, {}) # bad capability argument call_async(q, status, 'AdvertiseStatus', '', 'service.name', '') q.expect('dbus-error', method='AdvertiseStatus') # bad service name call_async(q, status, 'AdvertiseStatus', CAP_NAME, '', '') q.expect('dbus-error', method='AdvertiseStatus') # we can't test that the message type="headline" stanza is # actually received because it's thrown into the loopback stream # immediately. # announce a contact with the right caps ver = compute_caps_hash([], [CAP_NAME + '+notify'], {}) txt_record = { "txtvers": "1", "status": "avail", "node": CLIENT_NAME, "ver": ver, "hash": "sha-1" } contact_name = "test-status@" + get_host_name() listener, port = setup_stream_listener(q, contact_name) announcer = AvahiAnnouncer(contact_name, "_presence._tcp", port, txt_record) handle = wait_for_contact_in_publish(q, bus, conn, contact_name) # this is the first presence, Salut connects to the contact e = q.expect('incoming-connection', listener=listener) incoming = e.connection # Salut looks up its capabilities event = q.expect('stream-iq', connection=incoming, query_ns=ns.DISCO_INFO) query_node = xpath.queryForNodes('/iq/query', event.stanza)[0] assertEquals(CLIENT_NAME + '#' + ver, query_node.attributes['node']) contact_handle = conn.RequestHandles(cs.HT_CONTACT, [contact_name])[0] # send good reply result = make_result_iq(event.stanza) query = result.firstChildElement() query['node'] = CLIENT_NAME + '#' + ver feature = query.addElement('feature') feature['var'] = CAP_NAME + '+notify' incoming.send(result) # this will be fired as text channel caps will be fired q.expect('dbus-signal', signal='ContactCapabilitiesChanged', predicate=lambda e: contact_handle in e.args[0]) # okay now we know about the contact's caps, we can go ahead discovered = status.Get(ycs.STATUS_IFACE, 'DiscoveredStatuses', dbus_interface=dbus.PROPERTIES_IFACE) assertEquals({}, discovered) el = Element(('urn:ytstenut:status', 'status')) el['activity'] = 'messing-with-your-stuff' desc = el.addElement('ytstenut:description', content='Yeah sorry about that') desc['xml:lang'] = 'en-GB' call_async(q, status, 'AdvertiseStatus', CAP_NAME, 'ants.in.their.pants', el.toXml()) e, _, sig = q.expect_many( EventPattern('stream-message', connection=incoming), EventPattern('dbus-return', method='AdvertiseStatus'), EventPattern('dbus-signal', signal='StatusChanged', interface=ycs.STATUS_IFACE)) # check message message = e.stanza event = message.children[0] items = event.children[0] item = items.children[0] status_el = item.children[0] assertEquals('status', status_el.name) assertEquals('messing-with-your-stuff', status_el['activity']) assertEquals('ants.in.their.pants', status_el['from-service']) assertEquals(CAP_NAME, status_el['capability']) # check signal contact_id, capability, service_name, status_str = sig.args assertEquals(CAP_NAME, capability) assertEquals('ants.in.their.pants', service_name) assertNotEquals('', status_str) # check property discovered = status.Get(ycs.STATUS_IFACE, 'DiscoveredStatuses', dbus_interface=dbus.PROPERTIES_IFACE) assertEquals( { 'testsuite@testsuite': { CAP_NAME: { 'ants.in.their.pants': status_str } } }, discovered) # set another el = Element(('urn:ytstenut:status', 'status')) el['activity'] = 'rofling' desc = el.addElement('ytstenut:description', content='U MAD?') desc['xml:lang'] = 'en-GB' call_async(q, status, 'AdvertiseStatus', CAP_NAME, 'bananaman.on.holiday', el.toXml()) e, _, sig = q.expect_many( EventPattern('stream-message', connection=incoming), EventPattern('dbus-return', method='AdvertiseStatus'), EventPattern('dbus-signal', signal='StatusChanged', interface=ycs.STATUS_IFACE)) # check message message = e.stanza event = message.children[0] items = event.children[0] item = items.children[0] status_el = item.children[0] assertEquals('status', status_el.name) assertEquals('rofling', status_el['activity']) assertEquals('bananaman.on.holiday', status_el['from-service']) assertEquals(CAP_NAME, status_el['capability']) # check signal contact_id, capability, service_name, bananaman_status_str = sig.args assertEquals(CAP_NAME, capability) assertEquals('bananaman.on.holiday', service_name) assertNotEquals('', bananaman_status_str) # check property discovered = status.Get(ycs.STATUS_IFACE, 'DiscoveredStatuses', dbus_interface=dbus.PROPERTIES_IFACE) assertEquals( { 'testsuite@testsuite': { CAP_NAME: { 'ants.in.their.pants': status_str, 'bananaman.on.holiday': bananaman_status_str } } }, discovered) # unset the status from one service call_async(q, status, 'AdvertiseStatus', CAP_NAME, 'ants.in.their.pants', '') e, _, sig = q.expect_many( EventPattern('stream-message', connection=incoming), EventPattern('dbus-return', method='AdvertiseStatus'), EventPattern('dbus-signal', signal='StatusChanged', interface=ycs.STATUS_IFACE)) # check message message = e.stanza event = message.children[0] items = event.children[0] item = items.children[0] status_el = item.children[0] assertEquals('status', status_el.name) assertEquals('ants.in.their.pants', status_el['from-service']) assertEquals(CAP_NAME, status_el['capability']) assert 'activity' not in status_el.attributes assertEquals([], status_el.children) # check signal contact_id, capability, service_name, status_str = sig.args assertEquals(CAP_NAME, capability) assertEquals('ants.in.their.pants', service_name) assertEquals('', status_str) # check property discovered = status.Get(ycs.STATUS_IFACE, 'DiscoveredStatuses', dbus_interface=dbus.PROPERTIES_IFACE) assertEquals( { 'testsuite@testsuite': { CAP_NAME: { 'bananaman.on.holiday': bananaman_status_str } } }, discovered) # unset the status from the other service call_async(q, status, 'AdvertiseStatus', CAP_NAME, 'bananaman.on.holiday', '') e, _, sig = q.expect_many( EventPattern('stream-message', connection=incoming), EventPattern('dbus-return', method='AdvertiseStatus'), EventPattern('dbus-signal', signal='StatusChanged', interface=ycs.STATUS_IFACE)) # check message message = e.stanza event = message.children[0] items = event.children[0] item = items.children[0] status_el = item.children[0] assertEquals('status', status_el.name) assertEquals('bananaman.on.holiday', status_el['from-service']) assertEquals(CAP_NAME, status_el['capability']) assert 'activity' not in status_el.attributes assertEquals([], status_el.children) # check signal contact_id, capability, service_name, status_str = sig.args assertEquals(CAP_NAME, capability) assertEquals('bananaman.on.holiday', service_name) assertEquals('', status_str) # check property discovered = status.Get(ycs.STATUS_IFACE, 'DiscoveredStatuses', dbus_interface=dbus.PROPERTIES_IFACE) assertEquals({}, discovered)
assertEquals(client + '#' + caps['ver'], event.query['node']) return event.stanza def send_disco_reply(stream, stanza, identities, features, dataforms={}): stream.send( make_caps_disco_reply(stream, stanza, identities, features, dataforms)) if __name__ == '__main__': # example from XEP-0115 assertEquals( 'QgayPKawpkPSDYmwT/WM94uAlu0=', compute_caps_hash(['client/pc//Exodus 0.9.1'], [ "http://jabber.org/protocol/disco#info", "http://jabber.org/protocol/disco#items", "http://jabber.org/protocol/muc", "http://jabber.org/protocol/caps" ], {})) # another example from XEP-0115 identities = [u'client/pc/en/Psi 0.11', u'client/pc/el/Ψ 0.11'] features = [ u'http://jabber.org/protocol/caps', u'http://jabber.org/protocol/disco#info', u'http://jabber.org/protocol/disco#items', u'http://jabber.org/protocol/muc', ] dataforms = { u'urn:xmpp:dataforms:softwareinfo': { u'ip_version': [u'ipv4', u'ipv6'],
def test(q, bus, conn): forbidden = [EventPattern('dbus-signal', signal='ServiceAdded'), EventPattern('dbus-signal', signal='ServiceRemoved')] q.forbid_events(forbidden) conn.Connect() q.expect('dbus-signal', signal='StatusChanged', args=[0, 0]) # announce a contact with the right caps ver = compute_caps_hash([], [], banshee) txt_record = { "txtvers": "1", "status": "avail", "node": CLIENT_NAME, "ver": ver, "hash": "sha-1"} contact_name = "test-service@" + get_host_name() listener, port = setup_stream_listener(q, contact_name) announcer = AvahiAnnouncer(contact_name, "_presence._tcp", port, txt_record) handle = wait_for_contact_in_publish(q, bus, conn, contact_name) # this is the first presence, Salut connects to the contact e = q.expect('incoming-connection', listener=listener) incoming = e.connection # Salut looks up its capabilities event = q.expect('stream-iq', connection=incoming, query_ns=ns.DISCO_INFO) query_node = xpath.queryForNodes('/iq/query', event.stanza)[0] assertEquals(CLIENT_NAME + '#' + ver, query_node.attributes['node']) contact_handle = conn.RequestHandles(cs.HT_CONTACT, [contact_name])[0] # send good reply result = make_result_iq(event.stanza) query = result.firstChildElement() query['node'] = CLIENT_NAME + '#' + ver x = query.addElement((ns.X_DATA, 'x')) x['type'] = 'result' # FORM_TYPE field = x.addElement((None, 'field')) field['var'] = 'FORM_TYPE' field['type'] = 'hidden' field.addElement((None, 'value'), content='urn:ytstenut:capabilities#org.gnome.Banshee') # type field = x.addElement((None, 'field')) field['var'] = 'type' field.addElement((None, 'value'), content='application') # name field = x.addElement((None, 'field')) field['var'] = 'name' field.addElement((None, 'value'), content='en_GB/Banshee Media Player') field.addElement((None, 'value'), content='fr/Banshee Lecteur de Musique') # capabilities field = x.addElement((None, 'field')) field['var'] = 'capabilities' field.addElement((None, 'value'), content='urn:ytstenut:capabilities:yts-caps-audio') field.addElement((None, 'value'), content='urn:ytstenut:data:jingle:rtp') incoming.send(result) # this will be fired as text channel caps will be fired q.expect('dbus-signal', signal='ContactCapabilitiesChanged', predicate=lambda e: contact_handle in e.args[0]) # add evince tmp = banshee.copy() tmp.update(evince) ver = compute_caps_hash([], [], tmp) txt_record['ver'] = ver announcer.update(txt_record) # Salut looks up our capabilities event = q.expect('stream-iq', connection=incoming, query_ns='http://jabber.org/protocol/disco#info') query_node = xpath.queryForNodes('/iq/query', event.stanza)[0] assert query_node.attributes['node'] == \ CLIENT_NAME + '#' + txt_record['ver'] # send good reply result['id'] = event.stanza['id'] query['node'] = CLIENT_NAME + '#' + ver x = query.addElement((ns.X_DATA, 'x')) x['type'] = 'result' # FORM_TYPE field = x.addElement((None, 'field')) field['var'] = 'FORM_TYPE' field['type'] = 'hidden' field.addElement((None, 'value'), content='urn:ytstenut:capabilities#org.gnome.Evince') # type field = x.addElement((None, 'field')) field['var'] = 'type' field.addElement((None, 'value'), content='application') # name field = x.addElement((None, 'field')) field['var'] = 'name' field.addElement((None, 'value'), content='en_GB/Evince Picture Viewer') field.addElement((None, 'value'), content='fr/Evince uh, ow do you say') # capabilities field = x.addElement((None, 'field')) field['var'] = 'capabilities' field.addElement((None, 'value'), content='urn:ytstenut:capabilities:pics') incoming.send(result) # this will be fired as text channel caps will be fired q.expect('dbus-signal', signal='ContactCapabilitiesChanged', predicate=lambda e: contact_handle in e.args[0]) # now finally ensure the sidecar path, props = conn.Future.EnsureSidecar(ycs.STATUS_IFACE) assertEquals({}, props) status = ProxyWrapper(bus.get_object(conn.bus_name, path), ycs.STATUS_IFACE, {}) discovered = status.Get(ycs.STATUS_IFACE, 'DiscoveredServices', dbus_interface=dbus.PROPERTIES_IFACE) assertEquals({contact_name: { 'org.gnome.Banshee': ('application', {'en_GB': 'Banshee Media Player', 'fr': 'Banshee Lecteur de Musique'}, ['urn:ytstenut:capabilities:yts-caps-audio', 'urn:ytstenut:data:jingle:rtp']), 'org.gnome.Evince': ('application', {'en_GB': 'Evince Picture Viewer', 'fr': 'Evince uh, ow do you say'}, ['urn:ytstenut:capabilities:pics'])} }, discovered)
def incoming_fail(q, bus, conn): chan, outbound, contact_name, self_handle_name = \ setup_incoming_tests(q, bus, conn) call_async(q, chan, 'Fail', ycs.ERROR_TYPE_AUTH, 'auth', 'omgwtfbbq', 'I most certainly dont feel like dancing') _, e = q.expect_many(EventPattern('dbus-return', method='Fail'), EventPattern('stream-iq', connection=outbound)) iq = e.stanza assertEquals('le-loldongs', iq['id']) assertEquals('error', iq['type']) assertEquals(self_handle_name, iq['from']) assertEquals(contact_name, iq['to']) assertEquals(2, len(iq.children)) def check_message(message): assertEquals('message', message.name) assertEquals(ycs.MESSAGE_NS, message.uri) assertEquals('beautiful pea green boat', message['seacraft']) assertEquals('sea', message['destination']) assertEquals('the pussy cat', message['owl-companions']) assertEquals('the.from.service', message['from-service']) assertEquals('the.to.service', message['to-service']) assertEquals(1, len(message.children)) lol = message.children[0] assertEquals('lol', lol.name) assertEquals('fill', lol['to']) assertEquals('time', lol['the']) assertEquals('stuff', lol['some']) assertEquals(1, len(lol.children)) look = lol.children[0] assertEquals('look-into-my-eyes', look.name) assertEquals(1, len(look.children)) assertEquals('and tell me how boring writing these tests is', look.children[0]) def check_error(error): assertEquals('error', error.name) assertEquals('auth', error['type']) assertEquals(3, len(error.children)) for c in error.children: if c.name == 'auth': assertEquals(ns.STANZA, c.uri) elif c.name == 'omgwtfbbq': assertEquals(ycs.MESSAGE_NS, c.uri) elif c.name == 'text': assertEquals(ns.STANZA, c.uri) assertEquals(1, len(c.children)) assertEquals('I most certainly dont feel like dancing', c.children[0]) else: raise for child in iq.children: if child.name == 'message': check_message(child) elif child.name == 'error': check_error(child) else: raise # check we can't call anything any more call_async(q, chan, 'Fail', ycs.ERROR_TYPE_CANCEL, 'lol', 'whut', 'pear') q.expect('dbus-error', method='Fail') call_async(q, chan, 'Reply', {'lol':'whut'}, '') q.expect('dbus-error', method='Reply') call_async(q, chan, 'Request') q.expect('dbus-error', method='Request')