From 658d605273328e40f85201e67b6a1e43c0950a6a Mon Sep 17 00:00:00 2001
From: Li Qing <Qing.Li3@vce.com>
Date: Mon, 1 Jun 2020 13:33:00 +0800
Subject: [PATCH] implement context manager as per
 https://github.com/python/cpython/blob/master/Lib/socket.py#L236:L241

---
 httpretty/core.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/httpretty/core.py b/httpretty/core.py
index 83133545..e1e9ebab 100644
--- a/httpretty/core.py
+++ b/httpretty/core.py
@@ -395,6 +395,13 @@ def __init__(
             self.is_http = False
             self._bufsize = 32 * 1024
 
+        def __enter__(self):
+            return self
+
+        def __exit__(self, *args):
+            if not self._closed:
+                self.close()
+
         def create_socket(self):
             return old_socket(self.socket_family, self.socket_type, self.socket_proto)