[Telepathy-commits] [telepathy-salut/master] add file_transfer_helper.py

Guillaume Desmottes guillaume.desmottes at collabora.co.uk
Tue Dec 9 05:56:10 PST 2008


---
 tests/twisted/avahi/file_transfer_helper.py |  273 +++++++++++++++++++++++++++
 1 files changed, 273 insertions(+), 0 deletions(-)
 create mode 100644 tests/twisted/avahi/file_transfer_helper.py

diff --git a/tests/twisted/avahi/file_transfer_helper.py b/tests/twisted/avahi/file_transfer_helper.py
new file mode 100644
index 0000000..8082afb
--- /dev/null
+++ b/tests/twisted/avahi/file_transfer_helper.py
@@ -0,0 +1,273 @@
+import dbus
+import socket
+import md5
+import avahi
+import BaseHTTPServer
+import urllib
+
+from avahitest import AvahiAnnouncer, AvahiListener
+from avahitest import get_host_name
+
+from xmppstream import setup_stream_listener, connect_to_stream
+from servicetest import make_channel_proxy
+
+from twisted.words.xish import domish
+
+from dbus import PROPERTIES_IFACE
+
+CONNECTION_INTERFACE_REQUESTS = 'org.freedesktop.Telepathy.Connection.Interface.Requests'
+CHANNEL_INTERFACE ='org.freedesktop.Telepathy.Channel'
+CHANNEL_TYPE_FILE_TRANSFER = 'org.freedesktop.Telepathy.Channel.Type.FileTransfer.DRAFT'
+HT_CONTACT = 1
+HT_CONTACT_LIST = 3
+
+FT_STATE_NONE = 0
+FT_STATE_PENDING = 1
+FT_STATE_ACCEPTED = 2
+FT_STATE_OPEN = 3
+FT_STATE_COMPLETED = 4
+FT_STATE_CANCELLED = 5
+
+FT_STATE_CHANGE_REASON_NONE = 0
+FT_STATE_CHANGE_REASON_REQUESTED = 1
+FT_STATE_CHANGE_REASON_LOCAL_STOPPED = 2
+FT_STATE_CHANGE_REASON_REMOTE_STOPPED = 3
+FT_STATE_CHANGE_REASON_LOCAL_ERROR = 4
+FT_STATE_CHANGE_REASON_REMOTE_ERROR = 5
+
+FILE_HASH_TYPE_NONE = 0
+FILE_HASH_TYPE_MD5 = 1
+FILE_HASH_TYPE_SHA1 = 2
+FILE_HASH_TYPE_SHA256 = 3
+
+SOCKET_ADDRESS_TYPE_UNIX = 0
+SOCKET_ADDRESS_TYPE_ABSTRACT_UNIX = 1
+SOCKET_ADDRESS_TYPE_IPV4 = 2
+SOCKET_ADDRESS_TYPE_IPV6 = 3
+
+SOCKET_ACCESS_CONTROL_LOCALHOST = 0
+SOCKET_ACCESS_CONTROL_PORT = 1
+SOCKET_ACCESS_CONTROL_NETMASK = 2
+SOCKET_ACCESS_CONTROL_CREDENTIALS = 3
+
+class File(object):
+    DEFAULT_DATA = "What a nice file"
+    DEFAULT_NAME = "The foo.txt"
+    DEFAULT_CONTENT_TYPE = 'text/plain'
+    DEFAULT_DESCRIPTION = "A nice file to test"
+
+    def __init__(self, data=DEFAULT_DATA, name=DEFAULT_NAME,
+            content_type=DEFAULT_CONTENT_TYPE, description=DEFAULT_DESCRIPTION,
+            hash_type=FILE_HASH_TYPE_MD5):
+        self.data = data
+        self.size = len(self.data)
+        self.name = name
+
+        self.content_type = content_type
+        self.description = description
+
+        self.compute_hash(hash_type)
+
+    def compute_hash(self, hash_type):
+        assert hash_type == FILE_HASH_TYPE_MD5
+
+        self.hash_type = hash_type
+        m = md5.new()
+        m.update(self.data)
+        self.hash = m.hexdigest()
+
+class ReceiveFileTransferTest(object):
+    CONTACT_NAME = 'test-ft'
+
+    def __init__(self):
+        self.file = File()
+
+    def connect(self):
+        self.conn.Connect()
+        self.q.expect('dbus-signal', signal='StatusChanged', args=[0L, 0L])
+
+        self.self_handle = self.conn.GetSelfHandle()
+        self.self_handle_name =  self.conn.InspectHandles(HT_CONTACT, [self.self_handle])[0]
+
+    def announce_contact(self, name):
+        basic_txt = { "txtvers": "1", "status": "avail" }
+
+        self.contact_name = '%s@%s' % (name, get_host_name())
+        listener, port = setup_stream_listener(self.q, self.contact_name)
+
+        AvahiAnnouncer(self.contact_name, "_presence._tcp", port, basic_txt)
+
+    def wait_for_contact(self, name):
+        publish_handle = self.conn.RequestHandles(HT_CONTACT_LIST, ["publish"])[0]
+        publish = self.conn.RequestChannel(
+                "org.freedesktop.Telepathy.Channel.Type.ContactList",
+                HT_CONTACT_LIST, publish_handle, False)
+
+        self.handle = 0
+        # Wait until the record shows up in publish
+        while self.handle == 0:
+            e = self.q.expect('dbus-signal', signal='MembersChanged', path=publish)
+            for h in e.args[1]:
+                name = self.conn.InspectHandles(HT_CONTACT, [h])[0]
+                if name == self.contact_name:
+                    self.handle = h
+
+    def connect_to_salut(self):
+        AvahiListener(self.q).listen_for_service("_presence._tcp")
+        e = self.q.expect('service-added', name = self.self_handle_name,
+            protocol = avahi.PROTO_INET)
+        service = e.service
+        service.resolve()
+
+        e = self.q.expect('service-resolved', service = service)
+
+        self.outbound = connect_to_stream(self.q, self.contact_name,
+            self.self_handle_name, str(e.pt), e.port)
+
+        e = self.q.expect('connection-result')
+        assert e.succeeded, e.reason
+        self.q.expect('stream-opened', connection = self.outbound)
+
+    def setup_http_server(self):
+        class HTTPHandler(BaseHTTPServer.BaseHTTPRequestHandler):
+            def do_GET(self_):
+                # is that the right file ?
+                filename = self_.path.rsplit('/', 2)[-1]
+                assert filename == urllib.quote(self.file.name)
+
+                self_.send_response(200)
+                self_.send_header('Content-type', self.file.content_type)
+                self_.end_headers()
+                self_.wfile.write(self.file.data)
+
+        self.httpd = BaseHTTPServer.HTTPServer(('', 0), HTTPHandler)
+
+    def send_ft_offer_iq(self):
+        iq = domish.Element((None, 'iq'))
+        iq['to'] = self.self_handle_name
+        iq['from'] = self.contact_name
+        iq['type'] = 'set'
+        iq['id'] = 'gibber-file-transfer-0'
+        query = iq.addElement(('jabber:iq:oob', 'query'))
+        url = 'http://127.0.0.1:%u/gibber-file-transfer-0/%s' % \
+            (self.httpd.server_port, urllib.quote(self.file.name))
+        url_node = query.addElement('url', content=url)
+        url_node['type'] = 'file'
+        url_node['size'] = str(self.file.size)
+        url_node['mimeType'] = self.file.content_type
+        query.addElement('desc', content=self.file.description)
+        self.outbound.send(iq)
+
+    def check_new_channel(self):
+        e = self.q.expect('dbus-signal', signal='NewChannels')
+        channels = e.args[0]
+        assert len(channels) == 1
+        path, props = channels[0]
+
+        # check channel properties
+        # org.freedesktop.Telepathy.Channel D-Bus properties
+        assert props[CHANNEL_INTERFACE + '.ChannelType'] == CHANNEL_TYPE_FILE_TRANSFER
+        assert props[CHANNEL_INTERFACE + '.Interfaces'] == []
+        assert props[CHANNEL_INTERFACE + '.TargetHandle'] == self.handle
+        assert props[CHANNEL_INTERFACE + '.TargetID'] == self.contact_name
+        assert props[CHANNEL_INTERFACE + '.TargetHandleType'] == HT_CONTACT
+        assert props[CHANNEL_INTERFACE + '.Requested'] == False
+        assert props[CHANNEL_INTERFACE + '.InitiatorHandle'] == self.handle
+        assert props[CHANNEL_INTERFACE + '.InitiatorID'] == self.contact_name
+
+        # org.freedesktop.Telepathy.Channel.Type.FileTransfer D-Bus properties
+        assert props[CHANNEL_TYPE_FILE_TRANSFER + '.State'] == FT_STATE_PENDING
+        assert props[CHANNEL_TYPE_FILE_TRANSFER + '.ContentType'] == self.file.content_type
+        assert props[CHANNEL_TYPE_FILE_TRANSFER + '.Filename'] == self.file.name
+        assert props[CHANNEL_TYPE_FILE_TRANSFER + '.Size'] == self.file.size
+        # FT's protocol doesn't allow us the send the hash info
+        assert props[CHANNEL_TYPE_FILE_TRANSFER + '.ContentHashType'] == FILE_HASH_TYPE_NONE
+        assert props[CHANNEL_TYPE_FILE_TRANSFER + '.ContentHash'] == ''
+        assert props[CHANNEL_TYPE_FILE_TRANSFER + '.Description'] == self.file.description
+        # FT's protocol doesn't allow us the send the date info
+        assert props[CHANNEL_TYPE_FILE_TRANSFER + '.Date'] == 0
+        assert props[CHANNEL_TYPE_FILE_TRANSFER + '.AvailableSocketTypes'] == \
+            {SOCKET_ADDRESS_TYPE_UNIX: [SOCKET_ACCESS_CONTROL_LOCALHOST]}
+        assert props[CHANNEL_TYPE_FILE_TRANSFER + '.TransferredBytes'] == 0
+        assert props[CHANNEL_TYPE_FILE_TRANSFER + '.InitialOffset'] == 0
+
+        self.ft_path = path
+
+    def create_ft_channel(self):
+        self.channel = make_channel_proxy(self.conn, self.ft_path, 'Channel')
+        self.ft_channel = make_channel_proxy(self.conn, self.ft_path, 'Channel.Type.FileTransfer.DRAFT')
+        self.ft_props = dbus.Interface(self.bus.get_object(
+            self.conn.object.bus_name, self.ft_path), PROPERTIES_IFACE)
+
+    def accept_file(self):
+        self.address = self.ft_channel.AcceptFile(SOCKET_ADDRESS_TYPE_UNIX,
+                SOCKET_ACCESS_CONTROL_LOCALHOST, "", 5)
+
+        e = self.q.expect('dbus-signal', signal='FileTransferStateChanged')
+        state, reason = e.args
+        assert state == FT_STATE_ACCEPTED
+        assert reason == FT_STATE_CHANGE_REASON_REQUESTED
+
+        e = self.q.expect('dbus-signal', signal='InitialOffsetDefined')
+        offset = e.args[0]
+        # We don't support resume
+        assert offset == 0
+
+        e = self.q.expect('dbus-signal', signal='FileTransferStateChanged')
+        state, reason = e.args
+        assert state == FT_STATE_OPEN
+        assert reason == FT_STATE_CHANGE_REASON_NONE
+
+    def receive_file(self):
+        # Connect to Salut's socket
+        s = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
+        s.connect(self.address)
+
+        self.httpd.handle_request()
+
+        # Receiver inform us he finished to download the file
+        self.q.expect('stream-iq', iq_type='result')
+
+        # Read the file from Salut's socket
+        data = ''
+        read = 0
+        while read < self.file.size:
+            data += s.recv(self.file.size - read)
+            read = len(data)
+        assert data == self.file.data
+
+        e = self.q.expect('dbus-signal', signal='TransferredBytesChanged')
+        count = e.args[0]
+        while count < self.file.size:
+            # Catch TransferredBytesChanged until we transfered all the data
+            e = self.q.expect('dbus-signal', signal='TransferredBytesChanged')
+            count = e.args[0]
+
+        e = self.q.expect('dbus-signal', signal='FileTransferStateChanged')
+        state, reason = e.args
+        assert state == FT_STATE_COMPLETED
+        assert reason == FT_STATE_CHANGE_REASON_NONE
+
+    def close_channel(self):
+        self.channel.Close()
+        self.q.expect('dbus-signal', signal='Closed')
+
+    def run(self):
+        self.connect()
+        self.announce_contact(self.CONTACT_NAME)
+        self.wait_for_contact(self.CONTACT_NAME)
+        self.connect_to_salut()
+        self.setup_http_server()
+        self.send_ft_offer_iq()
+        self.check_new_channel()
+        self.create_ft_channel()
+        self.accept_file()
+        self.receive_file()
+        self.close_channel()
+
+    def test(self, q, bus, conn):
+        self.q = q
+        self.bus = bus
+        self.conn = conn
+
+        self.run()
-- 
1.5.6.5




More information about the Telepathy-commits mailing list