commit fec838bc8bc0632d7681698303bbf731c9b46b0d
author Lorenzo Colitti <lorenzo@google.com> Tue Nov 10 17:13:43 2015 +0900
committer Lorenzo Colitti <lorenzo@google.com> Wed Dec 09 22:51:07 2015 +0900
tree 8cde105e812a8c290cbfdf85bd4461f92bc87f6b
parent 925068328f2be292486629f692e32573dfd2ee25

Move CreateSocketPair to net_test.py so we can use it elsewhere.

Change-Id: Ic46e1b9ffb9d1d6082c0453baaad71a3d43b432f

net/test/net_test.py

diff --git a/net/test/net_test.py b/net/test/net_test.py
index ada7eba..5c5a4c5 100755
--- a/net/test/net_test.py
+++ b/net/test/net_test.py
@@ -151,6 +151,17 @@
   return s
 
 
+def CreateSocketPair(family, socktype, addr):
+  clientsock = socket(family, socktype, 0)
+  listensock = socket(family, socktype, 0)
+  listensock.bind((addr, 0))
+  addr = listensock.getsockname()
+  listensock.listen(1)
+  clientsock.connect(addr)
+  acceptedsock, _ = listensock.accept()
+  return clientsock, acceptedsock
+
+
 def GetInterfaceIndex(ifname):
   s = IPv4PingSocket()
   ifr = struct.pack("16si", ifname, 0)

net/test/tcp_nuke_addr_test.py

diff --git a/net/test/tcp_nuke_addr_test.py b/net/test/tcp_nuke_addr_test.py
index fd3cded..b0ba27d 100755
--- a/net/test/tcp_nuke_addr_test.py
+++ b/net/test/tcp_nuke_addr_test.py
@@ -135,16 +135,12 @@
     except Exception, e:
       self.exception = e
 
+# For convenience.
+def CreateIPv4SocketPair():
+  return net_test.CreateSocketPair(AF_INET, SOCK_STREAM, IPV4_LOOPBACK_ADDR)
 
-def CreateSocketPair(family, addr):
-  clientsock = socket(family, SOCK_STREAM, 0)
-  listensock = socket(family, SOCK_STREAM, 0)
-  listensock.bind((addr, 0))
-  addr = listensock.getsockname()
-  listensock.listen(1)
-  clientsock.connect(addr)
-  acceptedsock, _ = listensock.accept()
-  return clientsock, acceptedsock
+def CreateIPv6SocketPair():
+  return net_test.CreateSocketPair(AF_INET6, SOCK_STREAM, IPV6_LOOPBACK_ADDR)
 
 
 class TcpNukeAddrTest(net_test.NetworkTest):
@@ -168,7 +164,7 @@
     threadpairs = []
 
     for i in xrange(DEFAULT_TEST_RUNS):
-      clientsock, acceptedsock = CreateSocketPair(AF_INET6, IPV6_LOOPBACK_ADDR)
+      clientsock, acceptedsock = CreateIPv6SocketPair()
       clientthread = ExceptionalReadThread(clientsock)
       clientthread.start()
       serverthread = ExceptionalReadThread(acceptedsock)
@@ -201,7 +197,7 @@
       self.assertTrue(sock.getpeername())
 
   def testAddresses(self):
-    socketpair = CreateSocketPair(AF_INET, IPV4_LOOPBACK_ADDR)
+    socketpair = CreateIPv4SocketPair()
     KillAddrIoctl("::")
     self.assertSocketsNotClosed(socketpair)
     KillAddrIoctl("::1")
@@ -213,7 +209,7 @@
     KillAddrIoctl("127.0.0.1")
     self.assertSocketsClosed(socketpair)
 
-    socketpair = CreateSocketPair(AF_INET6, IPV6_LOOPBACK_ADDR)
+    socketpair = CreateIPv6SocketPair()
     KillAddrIoctl("0.0.0.0")
     self.assertSocketsNotClosed(socketpair)
     KillAddrIoctl("127.0.0.1")
@@ -239,8 +235,8 @@
   def testClosesAllSockets(self):
     socketpairs = []
     for i in xrange(HASH_TEST_RUNS):
-      socketpairs.append(CreateSocketPair(AF_INET, IPV4_LOOPBACK_ADDR))
-      socketpairs.append(CreateSocketPair(AF_INET6, IPV6_LOOPBACK_ADDR))
+      socketpairs.append(CreateIPv4SocketPair())
+      socketpairs.append(CreateIPv6SocketPair())
 
     KillAddrIoctl(IPV4_LOOPBACK_ADDR)
     KillAddrIoctl(IPV6_LOOPBACK_ADDR)