mirror of
https://github.com/moparisthebest/SickRage
synced 2024-12-15 12:32:18 -05:00
60 lines
1.4 KiB
Python
60 lines
1.4 KiB
Python
|
"""Tests for refleaks."""
|
||
|
|
|
||
|
|
from cherrypy._cpcompat import HTTPConnection, HTTPSConnection, ntob
|
||
|
|
import threading
|
||
|
|
|
||
|
|
import cherrypy
|
||
|
|
|
||
|
|
|
||
|
|
data = object()
|
||
|
|
|
||
|
|
|
||
|
|
from cherrypy.test import helper
|
||
|
|
|
||
|
|
|
||
|
|
class ReferenceTests(helper.CPWebCase):
|
||
|
|
|
||
|
|
def setup_server():
|
||
|
|
|
||
|
|
class Root:
|
||
|
|
|
||
|
|
def index(self, *args, **kwargs):
|
||
|
|
cherrypy.request.thing = data
|
||
|
|
return "Hello world!"
|
||
|
|
index.exposed = True
|
||
|
|
|
||
|
|
cherrypy.tree.mount(Root())
|
||
|
|
setup_server = staticmethod(setup_server)
|
||
|
|
|
||
|
|
def test_threadlocal_garbage(self):
|
||
|
|
success = []
|
||
|
|
|
||
|
|
def getpage():
|
||
|
|
host = '%s:%s' % (self.interface(), self.PORT)
|
||
|
|
if self.scheme == 'https':
|
||
|
|
c = HTTPSConnection(host)
|
||
|
|
else:
|
||
|
|
c = HTTPConnection(host)
|
||
|
|
try:
|
||
|
|
c.putrequest('GET', '/')
|
||
|
|
c.endheaders()
|
||
|
|
response = c.getresponse()
|
||
|
|
body = response.read()
|
||
|
|
self.assertEqual(response.status, 200)
|
||
|
|
self.assertEqual(body, ntob("Hello world!"))
|
||
|
|
finally:
|
||
|
|
c.close()
|
||
|
|
success.append(True)
|
||
|
|
|
||
|
|
ITERATIONS = 25
|
||
|
|
ts = []
|
||
|
|
for _ in range(ITERATIONS):
|
||
|
|
t = threading.Thread(target=getpage)
|
||
|
|
ts.append(t)
|
||
|
|
t.start()
|
||
|
|
|
||
|
|
for t in ts:
|
||
|
|
t.join()
|
||
|
|
|
||
|
|
self.assertEqual(len(success), ITERATIONS)
|