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)