Fixing credential tests and linters errors

Author: petyaslavova

tests/test_credentials.py

@@ -11,7 +11,10 @@
 from redis import AuthenticationError, DataError, Redis, ResponseError
 from redis.auth.err import RequestTokenErr
 from redis.backoff import NoBackoff
-from redis.connection import ConnectionInterface, ConnectionPool
+from redis.connection import (
+    ConnectionInterface,
+    ConnectionPool,
+)
 from redis.credentials import CredentialProvider, UsernamePasswordCredentialProvider
 from redis.exceptions import ConnectionError, RedisError
 from redis.retry import Retry
@@ -428,6 +431,8 @@ def re_auth_callback(token):
     def test_re_auth_pub_sub_in_resp3(self, credential_provider):
         mock_pubsub_connection = Mock(spec=ConnectionInterface)
         mock_pubsub_connection.get_protocol.return_value = 3
+
+        mock_pubsub_connection.should_reconnect = Mock(return_value=False)
         mock_pubsub_connection.credential_provider = credential_provider
         mock_pubsub_connection.retry = Retry(NoBackoff(), 3)
         mock_another_connection = Mock(spec=ConnectionInterface)
@@ -488,6 +493,7 @@ def re_auth_callback(token):
     def test_do_not_re_auth_pub_sub_in_resp2(self, credential_provider):
         mock_pubsub_connection = Mock(spec=ConnectionInterface)
         mock_pubsub_connection.get_protocol.return_value = 2
+        mock_pubsub_connection.should_reconnect = Mock(return_value=False)
         mock_pubsub_connection.credential_provider = credential_provider
         mock_pubsub_connection.retry = Retry(NoBackoff(), 3)
         mock_another_connection = Mock(spec=ConnectionInterface)

tests/test_maint_notifications.py

@@ -1,4 +1,3 @@
-from socket import socket
 import threading
 from unittest.mock import Mock, call, patch, MagicMock
 import pytest

tests/test_maint_notifications_handling.py

@@ -10,7 +10,6 @@
 from redis.cache import CacheConfig
 from redis.connection import (
     AbstractConnection,
-    CacheProxyConnection,
     Connection,
     ConnectionPool,
     BlockingConnectionPool,