[telepathy-gabble/master] tubetestutil: add check_new_connection_access

[Guillaume Desmottes]

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

diff --git a/tests/twisted/tubes/tubetestutil.py b/tests/twisted/tubes/tubetestutil.py
index 4df419f..9450a98 100644
--- a/tests/twisted/tubes/tubetestutil.py
+++ b/tests/twisted/tubes/tubetestutil.py
@@ -7,7 +7,8 @@ import os
 
 import dbus
 
-from servicetest import unwrap, assertContains, EventProtocolClientFactory, EventProtocolFactory
+from servicetest import unwrap, assertContains, EventProtocolClientFactory,\
+    EventProtocolFactory, assertEquals
 from gabbletest import exec_test
 import constants as cs
 import bytestream
@@ -254,6 +255,18 @@ def create_server(q, address_type, factory=None):
     else:
         assert False
 
+def check_new_connection_access(access_control, access_control_param, protocol):
+    if access_control == cs.SOCKET_ACCESS_CONTROL_LOCALHOST:
+        # nothing to check
+        return
+    elif access_control == cs.SOCKET_ACCESS_CONTROL_PORT:
+        ip, port = access_control_param
+        address = protocol.transport.getPeer()
+        assertEquals(ip, address.host)
+        assertEquals(port, address.port)
+    else:
+        assert False
+
 def exec_tube_test(test, *args):
     for bytestream_cls in [
             bytestream.BytestreamIBBMsg,
-- 
1.5.6.5