Fixed additional broken tests

Author: vladvildanov

tests/test_connection.py

@@ -428,7 +428,8 @@ def test_clears_cache_on_disconnect(self, mock_connection, cache_conf):
 
         cache.set(
             CacheEntry(
-                cache_key=cache_key, cache_value=b"bar", status=CacheEntryStatus.VALID
+                cache_key=cache_key, cache_value=b"bar", status=CacheEntryStatus.VALID,
+                connection_ref=mock_connection
             )
         )
         assert cache.get(cache_key).cache_value == b"bar"
@@ -438,7 +439,7 @@ def test_clears_cache_on_disconnect(self, mock_connection, cache_conf):
         mock_connection.host = "mock"
         mock_connection.port = "mock"
 
-        proxy_connection = CacheProxyConnection(mock_connection, cache)
+        proxy_connection = CacheProxyConnection(mock_connection, cache, threading.Lock())
         proxy_connection.disconnect()
 
         assert len(cache.get_collection()) == 0
@@ -456,28 +457,32 @@ def test_read_response_returns_cached_reply(self, mock_cache, mock_connection):
                 cache_key=CacheKey(command="GET", redis_keys=("foo",)),
                 cache_value=CacheProxyConnection.DUMMY_CACHE_VALUE,
                 status=CacheEntryStatus.IN_PROGRESS,
+                connection_ref=mock_connection
             ),
             CacheEntry(
                 cache_key=CacheKey(command="GET", redis_keys=("foo",)),
                 cache_value=b"bar",
                 status=CacheEntryStatus.VALID,
+                connection_ref=mock_connection
             ),
             CacheEntry(
                 cache_key=CacheKey(command="GET", redis_keys=("foo",)),
                 cache_value=b"bar",
                 status=CacheEntryStatus.VALID,
+                connection_ref=mock_connection
             ),
             CacheEntry(
                 cache_key=CacheKey(command="GET", redis_keys=("foo",)),
                 cache_value=b"bar",
                 status=CacheEntryStatus.VALID,
+                connection_ref=mock_connection
             ),
         ]
         mock_connection.send_command.return_value = Any
         mock_connection.read_response.return_value = b"bar"
         mock_connection.can_read.return_value = False
 
-        proxy_connection = CacheProxyConnection(mock_connection, mock_cache)
+        proxy_connection = CacheProxyConnection(mock_connection, mock_cache, threading.Lock())
         proxy_connection.send_command(*["GET", "foo"], **{"keys": ["foo"]})
         assert proxy_connection.read_response() == b"bar"
         assert proxy_connection.read_response() == b"bar"
@@ -490,13 +495,15 @@ def test_read_response_returns_cached_reply(self, mock_cache, mock_connection):
                         cache_key=CacheKey(command="GET", redis_keys=("foo",)),
                         cache_value=CacheProxyConnection.DUMMY_CACHE_VALUE,
                         status=CacheEntryStatus.IN_PROGRESS,
+                        connection_ref=mock_connection
                     )
                 ),
                 call(
                     CacheEntry(
                         cache_key=CacheKey(command="GET", redis_keys=("foo",)),
                         cache_value=b"bar",
                         status=CacheEntryStatus.VALID,
+                        connection_ref=mock_connection
                     )
                 ),
             ]