def test_estimate_write_capacity_client(self): testutil.DummyReadWorker((HOST, PORT)) client_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM) sleep(1) client_socket.connect((HOST, PORT)) buf_capacity = estimate_write_capacity(client_socket) i = 0 while buf_capacity > const.MTU: client_socket.sendall('\0' * const.MTU) buf_capacity = estimate_write_capacity(client_socket) print i, buf_capacity i += 1 self.assertTrue(buf_capacity <= const.MTU) client_socket.close()
def test_estimate_write_capacity_server(self): testutil.DummyWriteWorker((HOST, PORT)) listener = socket.socket(socket.AF_INET, socket.SOCK_STREAM) listener.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1) listener.bind((HOST, PORT)) listener.listen(1) (conn, _) = listener.accept() listener.close() try: i = 0 while True: conn.recv(4096) buf_capacity = estimate_write_capacity(conn) print i, buf_capacity i += 1 except Exception, e: print "[ReadWorker] Exception %s" % str(e)