def testSendsPidToPortServer(self):
server = mock.Mock()
server.recv.return_value = b'42768\n'
with mock.patch.object(socket, 'socket', return_value=server):
port = portpicker.get_port_from_port_server('portserver', pid=1234)
server.sendall.assert_called_once_with(b'1234\n')
self.assertEqual(port, 42768)
def testGetPortFromPortServer(self):
"""Exercise the get_port_from_port_server() helper function."""
for _ in range(10):
port = portpicker.get_port_from_port_server(
os.environ['PORTSERVER_ADDRESS'])
self.assertTrue(self.IsUnusedTCPPort(port))
self.assertTrue(self.IsUnusedUDPPort(port))
def testPidDefaultsToOwnPid(self):
server = mock.Mock()
server.recv.return_value = b'52768\n'
with mock.patch.object(socket, 'socket', return_value=server):
with mock.patch.object(os, 'getpid', return_value=9876):
port = portpicker.get_port_from_port_server('portserver')
server.sendall.assert_called_once_with(b'9876\n')
self.assertEqual(port, 52768)
