Merge pull request #222 from ento/true-socket-host-and-port

When creating true socket connection, use host and port stored in instance variables, not in the Mocket class's variable

mocket/mocket.py

@@ -379,7 +379,7 @@ def true_sendall(self, data, *args, **kwargs):
             encoded_response = hexload(response_dict["response"])
         # if not available, call the real sendall
         except KeyError:
-            host, port = Mocket._address
+            host, port = self._host, self._port
             host = true_gethostbyname(host)
 
             if isinstance(self.true_socket, true_socket) and self._secure_socket:

tests/main/test_mocket.py

@@ -146,6 +146,36 @@ def test_makefile(self):
             self.assertEqual(fp.read().strip(), encode_to_bytes("Show me."))
             self.assertEqual(len(Mocket.request_list()), 1)
 
+    @pytest.mark.skipif(
+        'os.getenv("SKIP_TRUE_HTTP", False) or os.getenv("SKIP_TRUE_REDIS", False)'
+    )
+    def test_multiple_socket_connections(self):
+        redis_addr = ("localhost", 6379)
+        httpbin_addr = ("localhost", 80)
+
+        redis_buffer = io.BytesIO()
+        httpbin_buffer = io.BytesIO()
+
+        with Mocketizer():
+            redis_so = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+            redis_so.connect(redis_addr)
+
+            httpbin_so = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+            httpbin_so.connect(httpbin_addr)
+
+            # Creating another socket that connects to a different address
+            # should not cause the first connection to go awry.
+            redis_so.sendall(b"ping\r\n")
+            redis_so.recv_into(redis_buffer)
+
+            httpbin_so.sendall(b"GET / HTTP/1.1\r\nHost: localhost\r\n\r\n")
+            httpbin_so.recv_into(httpbin_buffer)
+
+        redis_buffer.seek(0)
+        assert redis_buffer.read() == b"+PONG\r\n"
+        httpbin_buffer.seek(0)
+        assert httpbin_buffer.read().startswith(b"HTTP/1.1 200 OK\r\n")
+
     def test_socket_as_context_manager(self):
         addr = ("localhost", 80)
         Mocket.register(MocketEntry(addr, ["Show me.\r\n"]))