Add python-eventlet 0.16.1

[packages/trusty/python-eventlet.git] / eventlet / tests / test__refcount.py

diff --git a/eventlet/tests/test__refcount.py b/eventlet/tests/test__refcount.py
new file mode 100644 (file)
index 0000000..d38cf86
--- /dev/null
+++ b/eventlet/tests/test__refcount.py
@@ -0,0 +1,92 @@
+"""This test checks that socket instances (not GreenSockets but underlying sockets)
+are not leaked by the hub.
+"""
+import gc
+from pprint import pformat
+import unittest
+import weakref
+
+from eventlet.support import clear_sys_exc_info
+from eventlet.green import socket
+from eventlet.green.thread import start_new_thread
+from eventlet.green.time import sleep
+
+SOCKET_TIMEOUT = 0.1
+
+
+def init_server():
+    s = socket.socket()
+    s.settimeout(SOCKET_TIMEOUT)
+    s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
+    s.bind(('localhost', 0))
+    s.listen(5)
+    return s, s.getsockname()[1]
+
+
+def handle_request(s, raise_on_timeout):
+    try:
+        conn, address = s.accept()
+    except socket.timeout:
+        if raise_on_timeout:
+            raise
+        else:
+            return
+    # print('handle_request - accepted')
+    res = conn.recv(100)
+    assert res == b'hello', repr(res)
+    # print('handle_request - recvd %r' % res)
+    res = conn.send(b'bye')
+    # print('handle_request - sent %r' % res)
+    # print('handle_request - conn refcount: %s' % sys.getrefcount(conn))
+    # conn.close()
+
+
+def make_request(port):
+    # print('make_request')
+    s = socket.socket()
+    s.connect(('localhost', port))
+    # print('make_request - connected')
+    res = s.send(b'hello')
+    # print('make_request - sent %s' % res)
+    res = s.recv(100)
+    assert res == b'bye', repr(res)
+    # print('make_request - recvd %r' % res)
+    # s.close()
+
+
+def run_interaction(run_client):
+    s, port = init_server()
+    start_new_thread(handle_request, (s, run_client))
+    if run_client:
+        start_new_thread(make_request, (port,))
+    sleep(0.1 + SOCKET_TIMEOUT)
+    # print(sys.getrefcount(s.fd))
+    # s.close()
+    return weakref.ref(s.fd)
+
+
+def run_and_check(run_client):
+    w = run_interaction(run_client=run_client)
+    clear_sys_exc_info()
+    gc.collect()
+    if w():
+        print(pformat(gc.get_referrers(w())))
+        for x in gc.get_referrers(w()):
+            print(pformat(x))
+            for y in gc.get_referrers(x):
+                print('- {0}'.format(pformat(y)))
+        raise AssertionError('server should be dead by now')
+
+
+def test_clean_exit():
+    run_and_check(True)
+    run_and_check(True)
+
+
+def test_timeout_exit():
+    run_and_check(False)
+    run_and_check(False)
+
+
+if __name__ == '__main__':
+    unittest.main()