added test to check that socket.socket instances aren't leaking in hub internals

This commit is contained in:
Denis Bilenko
2008-10-29 20:32:14 +06:00
parent 5bf2cf7f8d
commit ef087d64be

View File

@@ -0,0 +1,71 @@
"""This test checks that socket instances (not GreenSockets but underlying sockets)
are not leaked by the hub.
"""
import sys
import unittest
from eventlet.green import socket
from eventlet.green.thread import start_new_thread
from eventlet.green.time import sleep
import weakref
import gc
address = ('0.0.0.0', 7878)
def init_server():
s = socket.socket()
s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
s.bind(address)
s.listen(5)
return s
def handle_request(s):
conn, address = s.accept()
#print 'handle_request - accepted'
res = conn.recv(100)
#print 'handle_request - recvd %r' % res
res = conn.send('bye')
#print 'handle_request - sent %r' % res
#print 'handle_request - conn refcount: %s' % sys.getrefcount(conn)
#conn.close()
def make_request():
#print 'make_request'
s = socket.socket()
s.connect(address)
#print 'make_request - connected'
res = s.send('hello')
#print 'make_request - sent %s' % res
res = s.recv(100)
#print 'make_request - recvd %r' % res
#s.close()
def run_interaction():
s = init_server()
start_new_thread(handle_request, (s, ))
start_new_thread(make_request, ())
sleep(0.2)
#print sys.getrefcount(s.fd)
#s.close()
return weakref.ref(s.fd)
def run_and_check():
w = run_interaction()
reactor = sys.modules.get('twisted.internet.reactor')
if w():
print gc.get_referrers(w())
for x in gc.get_referrers(w()):
print x
for y in gc.get_referrers(x):
print '-', y
raise AssertionError('server should be dead by now')
class test(unittest.TestCase):
def test(self):
run_and_check()
run_and_check()
if __name__=='__main__':
unittest.main()