diff options
| -rw-r--r-- | tests/twisted/Makefile.am | 1 | ||||
| -rw-r--r-- | tests/twisted/simple-caps.py | 74 |
2 files changed, 75 insertions, 0 deletions
diff --git a/tests/twisted/Makefile.am b/tests/twisted/Makefile.am index f8e1fe2..3f89664 100644 --- a/tests/twisted/Makefile.am +++ b/tests/twisted/Makefile.am @@ -1,5 +1,6 @@ TWISTED_TESTS = \ avatar-requirements.py \ + simple-caps.py \ cm/protocols.py \ connect/fail.py \ connect/success.py \ diff --git a/tests/twisted/simple-caps.py b/tests/twisted/simple-caps.py new file mode 100644 index 0000000..519d0bd --- /dev/null +++ b/tests/twisted/simple-caps.py @@ -0,0 +1,74 @@ + +""" +Make sure ContactCaps works well enough. +""" + +from twisted.words.xish import domish + +from servicetest import assertEquals, assertContains, EventPattern +from hazetest import exec_test +import constants as cs + +import ns + +# assert this list of RCCs is only text +def check_text_only(rccs): + assertEquals([({ + cs.CHANNEL_TYPE: cs.CHANNEL_TYPE_TEXT, + cs.TARGET_HANDLE_TYPE: cs.HT_CONTACT + }, [cs.TARGET_HANDLE])], rccs) + +# assert GetContactCaps and GetContactAttributes returns just text caps +def check_rccs(conn, handle): + rccs = conn.ContactCapabilities.GetContactCapabilities([handle]) + assertEquals(1, len(rccs)) + check_text_only(rccs[handle]) + + attrs = conn.Contacts.GetContactAttributes([handle], + [cs.CONN_IFACE_CONTACT_CAPS], + False) + rccs = attrs[handle][cs.CONN_IFACE_CONTACT_CAPS + '/capabilities'] + check_text_only(rccs) + +# do the self handle which will just be text +def test_self_handle(q, bus, conn, stream): + conn.Connect() + q.expect('dbus-signal', signal='StatusChanged', + args=[cs.CONN_STATUS_CONNECTED, cs.CSR_REQUESTED]) + + self_handle = conn.Properties.Get(cs.CONN, 'SelfHandle') + + check_rccs(conn, self_handle) + +# do someone else which will also just be text +def test_someone_else(q, bus, conn, stream): + conn.Connect() + q.expect_many( + EventPattern('dbus-signal', signal='StatusChanged', + args=[cs.CONN_STATUS_CONNECTING, cs.CSR_REQUESTED]), + EventPattern('stream-authenticated'), + ) + + event = q.expect('stream-iq', query_ns=ns.ROSTER) + event.stanza['type'] = 'result' + + item = event.query.addElement('item') + item['jid'] = 'bob@foo.com' + item['subscription'] = 'both' + + stream.send(event.stanza) + + q.expect('dbus-signal', signal='StatusChanged', + args=[cs.CONN_STATUS_CONNECTED, cs.CSR_REQUESTED]) + + bob_handle = conn.RequestHandles(cs.HT_CONTACT, ['bob@foo.com'])[0] + check_rccs(conn, bob_handle) + + # now a randomer who isn't even in our contact list + amy_handle = conn.RequestHandles(cs.HT_CONTACT, ['amy@foo.com'])[0] + check_rccs(conn, amy_handle) + +if __name__ == '__main__': + exec_test(test_self_handle) + exec_test(test_someone_else) + |