86 lines
2.7 KiB
Python
86 lines
2.7 KiB
Python
|
"""
|
||
|
|
Tests for pika.base_connection.BaseConnection
|
||
|
|
|
||
|
|
"""
|
||
|
|
|
||
|
|
import socket
|
||
|
|
import unittest
|
||
|
|
|
||
|
|
import mock
|
||
|
|
|
||
|
|
import pika
|
||
|
|
import sys
|
||
|
|
import ssl
|
||
|
|
|
||
|
|
import pika.tcp_socket_opts
|
||
|
|
from pika.adapters import base_connection
|
||
|
|
|
||
|
|
|
||
|
|
# If this is missing, set it manually. We need it to test tcp opt setting.
|
||
|
|
try:
|
||
|
|
socket.TCP_KEEPIDLE
|
||
|
|
except AttributeError:
|
||
|
|
socket.TCP_KEEPIDLE = 4
|
||
|
|
|
||
|
|
|
||
|
|
class BaseConnectionTests(unittest.TestCase):
|
||
|
|
def setUp(self):
|
||
|
|
with mock.patch('pika.connection.Connection.connect'):
|
||
|
|
self.connection = base_connection.BaseConnection()
|
||
|
|
self.connection._set_connection_state(
|
||
|
|
base_connection.BaseConnection.CONNECTION_OPEN)
|
||
|
|
|
||
|
|
def test_repr(self):
|
||
|
|
text = repr(self.connection)
|
||
|
|
self.assertTrue(text.startswith('<BaseConnection'), text)
|
||
|
|
|
||
|
|
def test_should_raise_value_exception_with_no_params_func_instead(self):
|
||
|
|
def foo():
|
||
|
|
return True
|
||
|
|
|
||
|
|
self.assertRaises(ValueError, base_connection.BaseConnection, foo)
|
||
|
|
|
||
|
|
def test_tcp_options_with_dict_tcp_options(self):
|
||
|
|
|
||
|
|
tcp_options = dict(TCP_KEEPIDLE=60)
|
||
|
|
params = pika.ConnectionParameters(tcp_options=tcp_options)
|
||
|
|
self.assertEqual(params.tcp_options, tcp_options)
|
||
|
|
|
||
|
|
with mock.patch.dict('pika.tcp_socket_opts._SUPPORTED_TCP_OPTIONS', {
|
||
|
|
'TCP_KEEPIDLE': socket.TCP_KEEPIDLE
|
||
|
|
}):
|
||
|
|
sock_mock = mock.Mock()
|
||
|
|
pika.tcp_socket_opts.set_sock_opts(params.tcp_options, sock_mock)
|
||
|
|
|
||
|
|
expected = [
|
||
|
|
mock.call.setsockopt(socket.SOL_SOCKET, socket.SO_KEEPALIVE,
|
||
|
|
1),
|
||
|
|
mock.call.setsockopt(socket.SOL_TCP, socket.TCP_KEEPIDLE, 60)
|
||
|
|
]
|
||
|
|
self.assertEquals(sock_mock.method_calls, expected)
|
||
|
|
|
||
|
|
def test_tcp_options_with_invalid_tcp_options(self):
|
||
|
|
|
||
|
|
tcp_options = dict(TCP_EVIL_OPTION=1234)
|
||
|
|
params = pika.ConnectionParameters(tcp_options=tcp_options)
|
||
|
|
self.assertEqual(params.tcp_options, tcp_options)
|
||
|
|
|
||
|
|
sock_mock = mock.Mock()
|
||
|
|
pika.tcp_socket_opts.set_sock_opts(params.tcp_options, sock_mock)
|
||
|
|
|
||
|
|
keepalive_call = mock.call.setsockopt(socket.SOL_SOCKET,
|
||
|
|
socket.SO_KEEPALIVE, 1)
|
||
|
|
self.assertNotIn(keepalive_call, sock_mock.method_calls)
|
||
|
|
|
||
|
|
def test_tcp_options_with_none_tcp_options(self):
|
||
|
|
|
||
|
|
params = pika.ConnectionParameters(tcp_options=None)
|
||
|
|
self.assertIsNone(params.tcp_options)
|
||
|
|
|
||
|
|
sock_mock = mock.Mock()
|
||
|
|
pika.tcp_socket_opts.set_sock_opts(params.tcp_options, sock_mock)
|
||
|
|
|
||
|
|
keepalive_call = mock.call.setsockopt(socket.SOL_SOCKET,
|
||
|
|
socket.SO_KEEPALIVE, 1)
|
||
|
|
self.assertNotIn(keepalive_call, sock_mock.method_calls)
|