python-eventlet/tests/test__greenness.py

   1 """Test than modules in eventlet.green package are indeed green.
   2 To do that spawn a green server and then access it using a green socket.
   3 If either operation blocked the whole script would block and timeout.
   4 """
   5 import unittest
   6
   7 from eventlet.green import BaseHTTPServer
   8 from eventlet import spawn, kill
   9 from eventlet.support import six
  10
  11 if six.PY2:
  12     from eventlet.green.urllib2 import HTTPError, urlopen
  13 else:
  14     from eventlet.green.urllib.request import urlopen
  15     from eventlet.green.urllib.error import HTTPError
  16
  17
  18 class QuietHandler(BaseHTTPServer.BaseHTTPRequestHandler):
  19     protocol_version = "HTTP/1.0"
  20
  21     def log_message(self, *args, **kw):
  22         pass
  23
  24
  25 def start_http_server():
  26     server_address = ('localhost', 0)
  27     httpd = BaseHTTPServer.HTTPServer(server_address, QuietHandler)
  28     sa = httpd.socket.getsockname()
  29     # print("Serving HTTP on", sa[0], "port", sa[1], "...")
  30     httpd.request_count = 0
  31
  32     def serve():
  33         # increment the request_count before handling the request because
  34         # the send() for the response blocks (or at least appeared to be)
  35         httpd.request_count += 1
  36         httpd.handle_request()
  37     return spawn(serve), httpd, sa[1]
  38
  39
  40 class TestGreenness(unittest.TestCase):
  41
  42     def setUp(self):
  43         self.gthread, self.server, self.port = start_http_server()
  44         # print('Spawned the server')
  45
  46     def tearDown(self):
  47         self.server.server_close()
  48         kill(self.gthread)
  49
  50     def test_urllib(self):
  51         self.assertEqual(self.server.request_count, 0)
  52         try:
  53             urlopen('http://127.0.0.1:%s' % self.port)
  54             assert False, 'should not get there'
  55         except HTTPError as ex:
  56             assert ex.code == 501, repr(ex)
  57         self.assertEqual(self.server.request_count, 1)
  58
  59 if __name__ == '__main__':
  60     unittest.main()