eventlet/tests/stdlib/test_socket_ssl.py

   1 #!/usr/bin/env python
   2
   3 from eventlet import patcher
   4 from eventlet.green import socket
   5
   6 # enable network resource
   7 import test.test_support
   8 i_r_e = test.test_support.is_resource_enabled
   9
  10
  11 def is_resource_enabled(resource):
  12     if resource == 'network':
  13         return True
  14     else:
  15         return i_r_e(resource)
  16 test.test_support.is_resource_enabled = is_resource_enabled
  17
  18 try:
  19     socket.ssl
  20     socket.sslerror
  21 except AttributeError:
  22     raise ImportError("Socket module doesn't support ssl")
  23
  24 patcher.inject('test.test_socket_ssl', globals())
  25
  26 test_basic = patcher.patch_function(test_basic)
  27 test_rude_shutdown = patcher.patch_function(test_rude_shutdown)
  28
  29
  30 def test_main():
  31     if not hasattr(socket, "ssl"):
  32         raise test_support.TestSkipped("socket module has no ssl support")
  33     test_rude_shutdown()
  34     test_basic()
  35     test_timeout()
  36
  37
  38 if __name__ == "__main__":
  39     test_main()