mirror of
https://github.com/Tygs/0bin.git
synced 2023-08-10 21:13:00 +03: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)
|