From f5a4dabd6cca72462ee5942a291bb2f91dd3850b Mon Sep 17 00:00:00 2001
From: Alberto Bertogli <albertito@gmail.com>
Date: Mon, 7 Jan 2008 07:18:11 -0200
Subject: [PATCH] Add unit tests for TIPC socket support.

---
 Lib/test/test_socket.py |   63 +++++++++++++++++++++++++++++++++++++++++++++++
 1 files changed, 63 insertions(+), 0 deletions(-)

diff --git a/Lib/test/test_socket.py b/Lib/test/test_socket.py
index 8f69a40..8ccdd8e 100644
--- a/Lib/test/test_socket.py
+++ b/Lib/test/test_socket.py
@@ -1084,6 +1084,66 @@ class BufferIOTest(SocketConnectedTest):
         buf = buffer(MSG)
         self.serv_conn.send(buf)
 
+
+TIPC_STYPE = 2000
+TIPC_LOWER = 200
+TIPC_UPPER = 210
+
+class TIPCTest (unittest.TestCase):
+	def testRDM(self):
+		srv = socket.socket(socket.AF_TIPC, socket.SOCK_RDM)
+		cli = socket.socket(socket.AF_TIPC, socket.SOCK_RDM)
+
+		srv.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
+		srvaddr = (socket.TIPC_ADDR_NAMESEQ, TIPC_STYPE,
+				TIPC_LOWER, TIPC_UPPER)
+		srv.bind(srvaddr)
+
+		sendaddr = (socket.TIPC_ADDR_NAME, TIPC_STYPE,
+				TIPC_LOWER + (TIPC_UPPER - TIPC_LOWER) / 2, 0)
+		cli.sendto(MSG, sendaddr)
+
+		msg, recvaddr = srv.recvfrom(1024)
+
+	        self.assertEqual(cli.getsockname(), recvaddr)
+	        self.assertEqual(msg, MSG)
+
+class TIPCThreadableTest (unittest.TestCase, ThreadableTest):
+	def __init__(self, methodName = 'runTest'):
+		unittest.TestCase.__init__(self, methodName = methodName)
+		ThreadableTest.__init__(self)
+
+	def setUp(self):
+		self.srv = socket.socket(socket.AF_TIPC, socket.SOCK_STREAM)
+		self.srv.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
+		srvaddr = (socket.TIPC_ADDR_NAMESEQ, TIPC_STYPE,
+				TIPC_LOWER, TIPC_UPPER)
+		self.srv.bind(srvaddr)
+		self.srv.listen(5)
+		self.serverExplicitReady()
+		self.conn, self.connaddr = self.srv.accept()
+
+	def clientSetUp(self):
+		# The is a hittable race between serverExplicitReady() and the
+		# accept() call; sleep a little while to avoid it, otherwise
+		# we could get an exception
+		time.sleep(0.1)
+		self.cli = socket.socket(socket.AF_TIPC, socket.SOCK_STREAM)
+		addr = (socket.TIPC_ADDR_NAME, TIPC_STYPE,
+				TIPC_LOWER + (TIPC_UPPER - TIPC_LOWER) / 2, 0)
+		self.cli.connect(addr)
+		self.cliaddr = self.cli.getsockname()
+
+	def testStream(self):
+		msg = self.conn.recv(1024)
+		self.assertEqual(msg, MSG)
+		self.assertEqual(self.cliaddr, self.connaddr)
+
+	def _testStream(self):
+		self.cli.send(MSG)
+		self.cli.close()
+
+
 def test_main():
     tests = [GeneralModuleTests, BasicTCPTest, TCPCloserTest, TCPTimeoutTest,
              TestExceptions, BufferIOTest, BasicTCPTest2]
@@ -1105,6 +1165,9 @@ def test_main():
         tests.append(BasicSocketPairTest)
     if sys.platform == 'linux2':
         tests.append(TestLinuxAbstractNamespace)
+    if hasattr(socket, "AF_TIPC"):
+        tests.append(TIPCTest)
+	tests.append(TIPCThreadableTest)
 
     thread_info = test_support.threading_setup()
     test_support.run_unittest(*tests)
-- 
1.5.4.rc2.1164.g6451