[Telepathy-commits] [telepathy-salut/master] test-receive-file-not-found.py: use ReceiveFileTransferTest
Guillaume Desmottes
guillaume.desmottes at collabora.co.uk
Tue Dec 9 05:56:11 PST 2008
---
tests/twisted/avahi/test-receive-file-not-found.py | 256 ++++----------------
1 files changed, 47 insertions(+), 209 deletions(-)
diff --git a/tests/twisted/avahi/test-receive-file-not-found.py b/tests/twisted/avahi/test-receive-file-not-found.py
index 25c1429..140f331 100644
--- a/tests/twisted/avahi/test-receive-file-not-found.py
+++ b/tests/twisted/avahi/test-receive-file-not-found.py
@@ -1,215 +1,53 @@
-import httplib
-import urlparse
-import dbus
import socket
-import md5
-import avahi
import BaseHTTPServer
+import urllib
-from saluttest import exec_test
-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, EventPattern
-
-from twisted.words.xish import domish, xpath
-
-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
-TEXT_MESSAGE_TYPE_NORMAL = dbus.UInt32(0)
-
-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
-
-SOCKET_ADDRESS_TYPE_UNIX = 0
-SOCKET_ADDRESS_TYPE_IPV4 = 2
-
-SOCKET_ACCESS_CONTROL_LOCALHOST = 0
-
-# File to Offer
-FILE_DATA = "What a nice fileeeeeeeeeeeeeee"
-FILE_SIZE = len(FILE_DATA)
-FILE_NAME = 'foo.txt'
-FILE_CONTENT_TYPE = 'text/plain'
-FILE_DESCRIPTION = 'A nice file to test'
-FILE_HASH_TYPE = FILE_HASH_TYPE_MD5
-m = md5.new()
-m.update(FILE_DATA)
-FILE_HASH = m.hexdigest()
-
-def test(q, bus, conn):
- conn.Connect()
- q.expect('dbus-signal', signal='StatusChanged', args=[0L, 0L])
- basic_txt = { "txtvers": "1", "status": "avail" }
-
- self_handle = conn.GetSelfHandle()
- self_handle_name = conn.InspectHandles(HT_CONTACT, [self_handle])[0]
-
- contact_name = "test-file-sender@" + get_host_name()
- listener, port = setup_stream_listener(q, contact_name)
-
- AvahiAnnouncer(contact_name, "_presence._tcp", port, basic_txt)
-
- publish_handle = conn.RequestHandles(HT_CONTACT_LIST, ["publish"])[0]
- publish = conn.RequestChannel(
- "org.freedesktop.Telepathy.Channel.Type.ContactList",
- HT_CONTACT_LIST, publish_handle, False)
-
- handle = 0
- # Wait until the record shows up in publish
- while handle == 0:
- e = q.expect('dbus-signal', signal='MembersChanged', path=publish)
- for h in e.args[1]:
- name = conn.InspectHandles(HT_CONTACT, [h])[0]
- if name == contact_name:
- handle = h
-
- requests_iface = dbus.Interface(conn, CONNECTION_INTERFACE_REQUESTS)
-
- # Create a connection to send the FT stanza
- 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)
+from twisted.words.xish import xpath
- 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)
-
- # Setup the HTTP server
- httpd = BaseHTTPServer.HTTPServer(('', 0), HTTPHandler)
-
- # connected to Salut, now send the IQ
- iq = domish.Element((None, 'iq'))
- iq['to'] = self_handle_name
- iq['from'] = 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' % (httpd.server_port, FILE_NAME)
- url_node = query.addElement('url', content=url)
- url_node['type'] = 'file'
- url_node['size'] = str(FILE_SIZE)
- url_node['mimeType'] = FILE_CONTENT_TYPE
- query.addElement('desc', content=FILE_DESCRIPTION)
- outbound.send(iq)
-
- e =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'] == handle
- assert props[CHANNEL_INTERFACE + '.TargetID'] == contact_name
- assert props[CHANNEL_INTERFACE + '.TargetHandleType'] == HT_CONTACT
- assert props[CHANNEL_INTERFACE + '.Requested'] == False
- assert props[CHANNEL_INTERFACE + '.InitiatorHandle'] == handle
- assert props[CHANNEL_INTERFACE + '.InitiatorID'] == 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'] == FILE_CONTENT_TYPE
- assert props[CHANNEL_TYPE_FILE_TRANSFER + '.Filename'] == FILE_NAME
- assert props[CHANNEL_TYPE_FILE_TRANSFER + '.Size'] == 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'] == 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
-
- channel = make_channel_proxy(conn, path, 'Channel')
- ft_channel = make_channel_proxy(conn, path, 'Channel.Type.FileTransfer.DRAFT')
- ft_props = dbus.Interface(bus.get_object(conn.object.bus_name, path), PROPERTIES_IFACE)
-
- address = ft_channel.AcceptFile(SOCKET_ADDRESS_TYPE_UNIX, SOCKET_ACCESS_CONTROL_LOCALHOST, "", 5)
-
- e = q.expect('dbus-signal', signal='FileTransferStateChanged')
- state, reason = e.args
- assert state == FT_STATE_ACCEPTED
- assert reason == FT_STATE_CHANGE_REASON_REQUESTED
-
- e = q.expect('dbus-signal', signal='InitialOffsetDefined')
- offset = e.args[0]
- # We don't support resume
- assert offset == 0
-
- e = q.expect('dbus-signal', signal='FileTransferStateChanged')
- state, reason = e.args
- assert state == FT_STATE_OPEN
- assert reason == FT_STATE_CHANGE_REASON_NONE
-
- # Connect to Salut's socket
- s = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
- s.connect(address)
-
- httpd.handle_request()
-
- # Receiver inform us he finished to download the file
- e = q.expect('stream-iq', iq_type='error')
- iq = e.stanza
- error_node = xpath.queryForNodes("/iq/error", iq)[0]
- assert error_node['code'] == '404'
- assert error_node['type'] == 'cancel'
- not_found_node = error_node.firstChildElement()
- assert not_found_node.name == 'item-not-found'
-
- e = q.expect('dbus-signal', signal='FileTransferStateChanged')
- state, reason = e.args
- assert state == FT_STATE_CANCELLED
- assert reason == FT_STATE_CHANGE_REASON_LOCAL_ERROR
-
- transferred = ft_props.Get(CHANNEL_TYPE_FILE_TRANSFER, 'TransferredBytes')
- # no byte has been transferred as the transfer failed
- assert transferred == 0
-
- channel.Close()
- q.expect('dbus-signal', signal='Closed')
-
-class HTTPHandler(BaseHTTPServer.BaseHTTPRequestHandler):
- def do_GET(self):
- # is that the right file ?
- filename = self.path.rsplit('/', 2)[-1]
- assert filename == FILE_NAME
-
- self.send_response(404)
- self.end_headers()
- self.wfile.write(FILE_DATA)
+from saluttest import exec_test
+from file_transfer_helper import ReceiveFileTransferTest,FT_STATE_CANCELLED, \
+ FT_STATE_CHANGE_REASON_LOCAL_ERROR, CHANNEL_TYPE_FILE_TRANSFER
+
+class ReceiveFileNotFound(ReceiveFileTransferTest):
+ 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(404)
+ self_.end_headers()
+ self_.wfile.write(self.file.data)
+
+ self.httpd = BaseHTTPServer.HTTPServer(('', 0), HTTPHandler)
+
+ 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 can't download the file
+ e = self.q.expect('stream-iq', iq_type='error')
+ iq = e.stanza
+ print iq.toXml()
+ error_node = xpath.queryForNodes("/iq/error", iq)[0]
+ assert error_node['code'] == '404'
+ assert error_node['type'] == 'cancel'
+ not_found_node = error_node.firstChildElement()
+ assert not_found_node.name == 'item-not-found'
+
+ e = self.q.expect('dbus-signal', signal='FileTransferStateChanged')
+ state, reason = e.args
+ assert state == FT_STATE_CANCELLED
+ assert reason == FT_STATE_CHANGE_REASON_LOCAL_ERROR
+
+ transferred = self.ft_props.Get(CHANNEL_TYPE_FILE_TRANSFER, 'TransferredBytes')
+ # no byte has been transferred as the transfer failed
+ assert transferred == 0
if __name__ == '__main__':
- exec_test(test)
+ test = ReceiveFileNotFound()
+ exec_test(test.test)
--
1.5.6.5
More information about the Telepathy-commits
mailing list