def test_credentials(self):
params = connection.Parameters()
plain_cred = credentials.PlainCredentials('very', 'reliable')
params.credentials = plain_cred
self.assertEqual(params.credentials, plain_cred)
ext_cred = credentials.ExternalCredentials()
params.credentials = ext_cred
self.assertEqual(params.credentials, ext_cred)
with self.assertRaises(TypeError):
params.credentials = connection.Parameters()
with self.assertRaises(TypeError):
params.credentials = repr(plain_cred)
def test_heartbeat(self):
params = connection.Parameters()
params.heartbeat = None
self.assertIsNone(params.heartbeat)
params.heartbeat = 0
self.assertEqual(params.heartbeat, 0)
params.heartbeat = 600
self.assertEqual(params.heartbeat, 600)
with self.assertRaises(TypeError):
params.heartbeat = 1.5
with self.assertRaises(TypeError):
params.heartbeat = '99'
with self.assertRaises(ValueError):
params.heartbeat = -1
def heartbeat_callback(_conn, _val):
return 1
params.heartbeat = heartbeat_callback
self.assertTrue(callable(params.heartbeat))
self.assertIs(params.heartbeat, heartbeat_callback)
def test_frame_max(self):
params = connection.Parameters()
params.frame_max = spec.FRAME_MIN_SIZE
self.assertEqual(params.frame_max, spec.FRAME_MIN_SIZE)
params.frame_max = spec.FRAME_MIN_SIZE + 1
self.assertEqual(params.frame_max, spec.FRAME_MIN_SIZE + 1)
params.frame_max = spec.FRAME_MAX_SIZE
self.assertEqual(params.frame_max, spec.FRAME_MAX_SIZE)
params.frame_max = spec.FRAME_MAX_SIZE - 1
self.assertEqual(params.frame_max, spec.FRAME_MAX_SIZE - 1)
with self.assertRaises(TypeError):
params.frame_max = 10000.9
with self.assertRaises(TypeError):
params.frame_max = '10000'
with self.assertRaises(ValueError):
params.frame_max = spec.FRAME_MIN_SIZE - 1
with self.assertRaises(ValueError):
params.frame_max = spec.FRAME_MAX_SIZE + 1
def test_ne(self):
params_1 = connection.Parameters()
params_2 = connection.Parameters()
params_3 = ChildParameters()
params_1.host = 'localhost'
params_1.port = 5672
params_2.host = 'myserver.com'
params_2.port = 5672
self.assertNotEqual(params_1, params_2)
self.assertNotEqual(params_2, params_1)
params_1.host = 'localhost'
params_1.port = 5672
params_2.host = 'localhost'
params_2.port = 5671
self.assertNotEqual(params_1, params_2)
self.assertNotEqual(params_2, params_1)
params_1.host = 'localhost'
params_1.port = 5672
params_3.host = 'myserver.com'
params_3.port = 5672
self.assertNotEqual(params_1, params_3)
self.assertNotEqual(params_3, params_1)
params_1.host = 'localhost'
params_1.port = 5672
params_3.host = 'localhost'
params_3.port = 5671
self.assertNotEqual(params_1, params_3)
self.assertNotEqual(params_3, params_1)
self.assertNotEqual(params_1, dict(host='localhost', port=5672))
self.assertNotEqual(dict(host='localhost', port=5672), params_1)
class Foreign(object):
def __ne__(self, other):
return 'foobar'
self.assertEqual(params_1 != Foreign(), 'foobar')
self.assertEqual(Foreign() != params_1, 'foobar')
def test_locale(self):
params = connection.Parameters()
params.locale = 'en_UK'
self.assertEqual(params.locale, 'en_UK')
params.locale = u'en_UK'
self.assertEqual(params.locale, u'en_UK')
with self.assertRaises(TypeError):
params.locale = 127
def test_retry_delay(self):
params = connection.Parameters()
params.retry_delay = 0
self.assertEqual(params.retry_delay, 0)
params.retry_delay = 0.1
self.assertEqual(params.retry_delay, 0.1)
with self.assertRaises(TypeError):
params.retry_delay = '0.1'
def test_ssl_options(self):
params = connection.Parameters()
opt = dict(key='value', key2=2, key3=dict(a=1))
params.ssl_options = copy.deepcopy(opt)
self.assertEqual(params.ssl_options, opt)
params.ssl_options = None
self.assertIsNone(params.ssl_options)
with self.assertRaises(TypeError):
params.ssl_options = str(opt)
def test_ssl_options(self):
params = connection.Parameters()
ssl_options = connection.SSLOptions(ssl.create_default_context())
params.ssl_options = ssl_options
self.assertIs(params.ssl_options, ssl_options)
params.ssl_options = None
self.assertIsNone(params.ssl_options)
with self.assertRaises(TypeError):
params.ssl_options = dict()
def test_tcp_options(self):
params = connection.Parameters()
opt = dict(TCP_KEEPIDLE=60, TCP_KEEPINTVL=2, TCP_KEEPCNT=1, TCP_USER_TIMEOUT=1000)
params.tcp_options = copy.deepcopy(opt)
self.assertEqual(params.tcp_options, opt)
params.tcp_options = None
self.assertIsNone(params.tcp_options)
with self.assertRaises(TypeError):
params.tcp_options = str(opt)
def test_port(self):
params = connection.Parameters()
params.port = 0
self.assertEqual(params.port, 0)
params.port = 5672
self.assertEqual(params.port, 5672)
with self.assertRaises(TypeError):
params.port = 1.5
with self.assertRaises(TypeError):
params.port = '5672'
def test_virtual_host(self):
params = connection.Parameters()
params.virtual_host = '/'
self.assertEqual(params.virtual_host, '/')
params.virtual_host = u'/'
self.assertEqual(params.virtual_host, u'/')
params.virtual_host = 'test-vhost'
self.assertEqual(params.virtual_host, 'test-vhost')
with self.assertRaises(TypeError):
params.virtual_host = 99
def test_eq(self):
params_1 = connection.Parameters()
params_2 = connection.Parameters()
params_3 = ChildParameters()
self.assertEqual(params_1, params_2)
self.assertEqual(params_2, params_1)
params_1.host = 'localhost'
params_1.port = 5672
params_1.virtual_host = '/'
params_1.credentials = credentials.PlainCredentials('u', 'p')
params_2.host = 'localhost'
params_2.port = 5672
params_2.virtual_host = '//'
params_2.credentials = credentials.PlainCredentials('uu', 'pp')
self.assertEqual(params_1, params_2)
self.assertEqual(params_2, params_1)
params_1.host = 'localhost'
params_1.port = 5672
params_1.virtual_host = '/'
params_1.credentials = credentials.PlainCredentials('u', 'p')
params_3.host = 'localhost'
params_3.port = 5672
params_3.virtual_host = '//'
params_3.credentials = credentials.PlainCredentials('uu', 'pp')
self.assertEqual(params_1, params_3)
self.assertEqual(params_3, params_1)
class Foreign(object):
def __eq__(self, other):
return 'foobar'
self.assertEqual(params_1 == Foreign(), 'foobar')
self.assertEqual(Foreign() == params_1, 'foobar')
def test_heartbeat(self):
params = connection.Parameters()
params.heartbeat = 0
self.assertEqual(params.heartbeat, 0)
params.heartbeat = 600
self.assertEqual(params.heartbeat, 600)
with self.assertRaises(TypeError):
params.heartbeat = 1.5
with self.assertRaises(TypeError):
params.heartbeat = '99'
with self.assertRaises(ValueError):
params.heartbeat = -1
def test_host(self):
params = connection.Parameters()
params.host = '127.0.0.1'
self.assertEqual(params.host, '127.0.0.1')
params.host = 'myserver.com'
self.assertEqual(params.host, 'myserver.com')
params.host = u'myserver.com'
self.assertEqual(params.host, u'myserver.com')
with self.assertRaises(TypeError):
params.host = 127
with self.assertRaises(TypeError):
params.host = ('127.0.0.1', 5672)
def test_backpressure_detection(self):
params = connection.Parameters()
params.backpressure_detection = False
self.assertEqual(params.backpressure_detection, False)
params.backpressure_detection = True
self.assertEqual(params.backpressure_detection, True)
with self.assertRaises(TypeError):
params.backpressure_detection = 1
with self.assertRaises(TypeError):
params.backpressure_detection = 'true'
with self.assertRaises(TypeError):
params.backpressure_detection = 'f'
def test_blocked_connection_timeout(self):
params = connection.Parameters()
params.blocked_connection_timeout = 60
self.assertEqual(params.blocked_connection_timeout, 60)
params.blocked_connection_timeout = 93.5
self.assertEqual(params.blocked_connection_timeout, 93.5)
params.blocked_connection_timeout = None
self.assertIsNone(params.blocked_connection_timeout)
with self.assertRaises(TypeError):
params.blocked_connection_timeout = '1.5'
with self.assertRaises(ValueError):
params.blocked_connection_timeout = -40
def test_socket_timeout(self):
params = connection.Parameters()
params.socket_timeout = 1
self.assertEqual(params.socket_timeout, 1)
params.socket_timeout = 0.5
self.assertEqual(params.socket_timeout, 0.5)
params.socket_timeout = 60.5
self.assertEqual(params.socket_timeout, 60.5)
with self.assertRaises(TypeError):
params.socket_timeout = '60.5'
with self.assertRaises(ValueError):
params.socket_timeout = -1
with self.assertRaises(ValueError):
params.socket_timeout = 0
def test_connection_attempts(self):
params = connection.Parameters()
params.connection_attempts = 1
self.assertEqual(params.connection_attempts, 1)
params.connection_attempts = 10
self.assertEqual(params.connection_attempts, 10)
with self.assertRaises(TypeError):
params.connection_attempts = 1.5
with self.assertRaises(TypeError):
params.connection_attempts = '99'
with self.assertRaises(ValueError):
params.connection_attempts = 0
with self.assertRaises(ValueError):
params.connection_attempts = -40
def test_channel_max(self):
params = connection.Parameters()
params.channel_max = 1
self.assertEqual(params.channel_max, 1)
params.channel_max = channel.MAX_CHANNELS
self.assertEqual(params.channel_max, channel.MAX_CHANNELS)
with self.assertRaises(TypeError):
params.channel_max = 1.5
with self.assertRaises(TypeError):
params.channel_max = '99'
with self.assertRaises(ValueError):
params.channel_max = 0
with self.assertRaises(ValueError):
params.channel_max = -40
with self.assertRaises(ValueError):
params.channel_max = channel.MAX_CHANNELS + 1
def test_default_property_values(self):
self.assert_default_parameter_values(connection.Parameters())