[telepathy-gabble/master] tubetestutil: add connect_socket

[Guillaume Desmottes]

---
 tests/twisted/tubes/tubetestutil.py |   13 ++++++++++++-
 1 files changed, 12 insertions(+), 1 deletions(-)

diff --git a/tests/twisted/tubes/tubetestutil.py b/tests/twisted/tubes/tubetestutil.py
index e8998c1..326375d 100644
--- a/tests/twisted/tubes/tubetestutil.py
+++ b/tests/twisted/tubes/tubetestutil.py
@@ -7,7 +7,7 @@ import os
 
 import dbus
 
-from servicetest import unwrap, assertContains
+from servicetest import unwrap, assertContains, EventProtocolClientFactory
 from gabbletest import exec_test
 import constants as cs
 import bytestream
@@ -223,6 +223,17 @@ def set_up_echo(name):
     reactor.listenUNIX(full_path, factory)
     return full_path
 
+def connect_socket(q, address_type, address):
+    factory = EventProtocolClientFactory(q)
+    if address_type == cs.SOCKET_ADDRESS_TYPE_UNIX:
+        reactor.connectUNIX(address, factory)
+    elif address_type == cs.SOCKET_ADDRESS_TYPE_IPV4:
+        ip, port = address
+        assert port > 0
+        reactor.connectTCP(ip, port, factory)
+    else:
+        assert False
+
 def exec_tube_test(test, *args):
     for bytestream_cls in [
             bytestream.BytestreamIBBMsg,
-- 
1.5.6.5