[telepathy-gabble/master] add ReceiveFileTestS5B

[Guillaume Desmottes]

---
 .../twisted/file-transfer/file_transfer_helper.py  |   27 +++++++++++++++++++-
 1 files changed, 26 insertions(+), 1 deletions(-)

diff --git a/tests/twisted/file-transfer/file_transfer_helper.py b/tests/twisted/file-transfer/file_transfer_helper.py
index 9420b79..f2c0b1f 100644
--- a/tests/twisted/file-transfer/file_transfer_helper.py
+++ b/tests/twisted/file-transfer/file_transfer_helper.py
@@ -7,7 +7,8 @@ from servicetest import EventPattern
 from gabbletest import acknowledge_iq, sync_stream
 import ns
 from bytestream import parse_si_offer, create_si_reply, parse_ibb_open, parse_ibb_msg_data,\
-    create_si_offer, parse_si_reply, send_ibb_open, send_ibb_msg_data
+    create_si_offer, parse_si_reply, send_ibb_open, send_ibb_msg_data, listen_socks5, \
+    send_socks5_init, socks5_expect_connection
 
 from twisted.words.xish import domish, xpath
 
@@ -309,6 +310,30 @@ class ReceiveFileTestIBB(ReceiveFileTest):
 
         self.seq += 1
 
+class ReceiveFileTestS5B(ReceiveFileTest):
+    def __init__(self):
+        ReceiveFileTest.__init__(self)
+
+        self.bytestream = ns.BYTESTREAMS
+
+    def open_bytestream(self):
+        port = listen_socks5(self.q)
+
+        send_socks5_init(self.stream, self.contact_name, 'test at localhost/Resource',
+            'alpha', 'tcp', [(self.contact_name, '127.0.0.1', port)])
+
+        self.transport = socks5_expect_connection(self.q, 'alpha',
+            self.contact_name, 'test at localhost/Resource')
+
+        offset_event, state_event = self.q.expect_many(
+            EventPattern('dbus-signal', signal='InitialOffsetDefined'),
+            EventPattern('dbus-signal', signal='FileTransferStateChanged'))
+
+        return offset_event, state_event
+
+    def send_data(self, data):
+        self.transport.write(data)
+
 class SendFileTest(FileTransferTest):
     def __init__(self):
         FileTransferTest.__init__(self)
-- 
1.5.6.5