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 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 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 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 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 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 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 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 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)
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
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')
