05ac779256
GreenSocket.send() wasn't blocking properly or raising a socket.timeout if the send buffer is full. added a call to trampoline() added tests for several GreenSocket methods to check for proper socket.timeout behavior fixed a couple of other tests with changed behavior due to the send() fix.
44 lines
1.5 KiB
Python
44 lines
1.5 KiB
Python
"""Test than modules in eventlet.green package are indeed green.
|
|
To do that spawn a green server and then access it using a green socket.
|
|
If either operation blocked the whole script would block and timeout.
|
|
"""
|
|
import unittest
|
|
from eventlet.green import urllib2, BaseHTTPServer
|
|
from eventlet.api import spawn, kill
|
|
|
|
def start_http_server():
|
|
server_address = ('localhost', 0)
|
|
BaseHTTPServer.BaseHTTPRequestHandler.protocol_version = "HTTP/1.0"
|
|
httpd = BaseHTTPServer.HTTPServer(server_address, BaseHTTPServer.BaseHTTPRequestHandler)
|
|
sa = httpd.socket.getsockname()
|
|
#print "Serving HTTP on", sa[0], "port", sa[1], "..."
|
|
httpd.request_count = 0
|
|
def serve():
|
|
# increment the request_count before handling the request because
|
|
# the send() for the response blocks (or at least appeared to be)
|
|
httpd.request_count += 1
|
|
httpd.handle_request()
|
|
return spawn(serve), httpd, sa[1]
|
|
|
|
class TestGreenness(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
self.gthread, self.server,self.port = start_http_server()
|
|
#print 'Spawned the server'
|
|
|
|
def tearDown(self):
|
|
self.server.server_close()
|
|
kill(self.gthread)
|
|
|
|
def test_urllib2(self):
|
|
self.assertEqual(self.server.request_count, 0)
|
|
try:
|
|
urllib2.urlopen('http://127.0.0.1:%s' % self.port)
|
|
assert False, 'should not get there'
|
|
except urllib2.HTTPError, ex:
|
|
assert ex.code == 501, `ex`
|
|
self.assertEqual(self.server.request_count, 1)
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|