def test__connect_inet46_ipv4(self):
connection = pyredis.connection.Connection(host='127.0.0.1')
sock = connection._connect_inet46()
self.socket_mock.socket.assert_called_with(
self.socket_mock.AF_INET, self.socket_mock.SOCK_STREAM)
sock.settimeout.assert_called_with(2)
sock.connect.assert_called_with(('127.0.0.1', 6379))
self.assertEqual(sock, self.socket_mock.socket())
def test__connect_inet46_ipv4(self):
connection = pyredis.connection.Connection(host='127.0.0.1')
sock = connection._connect_inet46()
self.socket_mock.socket.assert_called_with(
self.socket_mock.AF_INET,
self.socket_mock.SOCK_STREAM
)
sock.settimeout.assert_called_with(2)
sock.connect.assert_called_with(('127.0.0.1', 6379))
self.assertEqual(sock, self.socket_mock.socket())
def test__connect_inet46_ipv6(self):
sock_mock = Mock()
self.socket_mock.socket.side_effect = [socket.gaierror, sock_mock]
self.socket_mock.gaierror = socket.gaierror
connection = pyredis.connection.Connection(host='::1')
sock = connection._connect_inet46()
self.socket_mock.socket.assert_has_calls([
call(self.socket_mock.AF_INET, self.socket_mock.SOCK_STREAM),
call(self.socket_mock.AF_INET6, self.socket_mock.SOCK_STREAM)
])
sock.settimeout.assert_called_with(2)
sock.connect.assert_called_with(('::1', 6379))
self.assertEqual(sock, sock_mock)
def test__connect_inet46_ipv6(self):
sock_mock = Mock()
self.socket_mock.socket.side_effect = [socket.gaierror, sock_mock]
self.socket_mock.gaierror = socket.gaierror
connection = pyredis.connection.Connection(host='::1')
sock = connection._connect_inet46()
self.socket_mock.socket.assert_has_calls([
call(
self.socket_mock.AF_INET,
self.socket_mock.SOCK_STREAM
),
call(
self.socket_mock.AF_INET6,
self.socket_mock.SOCK_STREAM
)
])
sock.settimeout.assert_called_with(2)
sock.connect.assert_called_with(('::1', 6379))
self.assertEqual(sock, sock_mock)
