[telepathy-gabble/master] add self.create_socket()

Guillaume Desmottes guillaume.desmottes at collabora.co.uk
Fri Apr 3 09:26:26 PDT 2009


---
 .../twisted/file-transfer/file_transfer_helper.py  |   10 ++++++++--
 1 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/tests/twisted/file-transfer/file_transfer_helper.py b/tests/twisted/file-transfer/file_transfer_helper.py
index fd395d3..5ebb00b 100644
--- a/tests/twisted/file-transfer/file_transfer_helper.py
+++ b/tests/twisted/file-transfer/file_transfer_helper.py
@@ -128,6 +128,12 @@ class FileTransferTest(object):
             if fct():
                 break
 
+    def create_socket(self):
+        if self.address_type == cs.SOCKET_ADDRESS_TYPE_UNIX:
+            return socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
+        else:
+            assert False
+
 class ReceiveFileTest(FileTransferTest):
     def __init__(self, bytestream_cls, address_type, access_control, access_control_param):
         FileTransferTest.__init__(self, bytestream_cls, address_type, access_control, access_control_param)
@@ -222,7 +228,7 @@ class ReceiveFileTest(FileTransferTest):
 
     def receive_file(self):
         # Connect to Gabble's socket
-        s = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
+        s = self.create_socket()
         s.connect(self.address)
 
         # send the rest of the file
@@ -362,7 +368,7 @@ class SendFileTest(FileTransferTest):
         self.bytestream.wait_bytestream_open()
 
     def send_file(self):
-        s = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
+        s = self.create_socket()
         s.connect(self.address)
         s.send(self.file.data)
 
-- 
1.5.6.5




More information about the telepathy-commits mailing list