socket.create_connection cannot be used as a context manager

[miguelgrinberg]

Consider the following script:

import socket
with socket.create_connection(('google.com', 80)) as s:
    pass

With the regular standard library socket (PY3 only), this works fine. If the library is monkey patched, though, I get the following:

(py36) $ python test.py
Traceback (most recent call last):
  File "test.py", line 6, in <module>
    with socket.create_connection(('google.com', 80)) as s:
AttributeError: __enter__

The green version of socket seems to be missing the context manager __enter__ and __exit__ methods.