def test_ne(self):
cred_1 = credentials.ExternalCredentials()
cred_2 = credentials.ExternalCredentials()
cred_3 = ChildExternalCredentials()
cred_1.erase_on_connect = False
cred_2.erase_on_connect = True
self.assertNotEqual(cred_1, cred_2)
self.assertNotEqual(cred_2, cred_1)
cred_1.erase_on_connect = False
cred_3.erase_on_connect = True
self.assertNotEqual(cred_1, cred_3)
self.assertNotEqual(cred_3, cred_1)
self.assertNotEqual(cred_1, dict(erase_on_connect=False))
self.assertNotEqual(dict(erase_on_connect=False), cred_1)
class Foreign(object):
def __ne__(self, other):
return 'foobar'
self.assertEqual(credentials.ExternalCredentials() != Foreign(),
'foobar')
self.assertEqual(Foreign() != credentials.ExternalCredentials(),
'foobar')
def test_eq(self):
cred_1 = credentials.ExternalCredentials()
cred_2 = credentials.ExternalCredentials()
cred_3 = ChildExternalCredentials()
self.assertEqual(cred_1, cred_2)
self.assertEqual(cred_2, cred_1)
cred_1.erase_on_connect = True
cred_2.erase_on_connect = True
self.assertEqual(cred_1, cred_2)
self.assertEqual(cred_2, cred_1)
cred_1.erase_on_connect = False
cred_2.erase_on_connect = False
self.assertEqual(cred_1, cred_2)
self.assertEqual(cred_2, cred_1)
cred_1.erase_on_connect = False
cred_3.erase_on_connect = False
self.assertEqual(cred_1, cred_3)
self.assertEqual(cred_3, cred_1)
class Foreign(object):
def __eq__(self, other):
return 'foobar'
self.assertEqual(credentials.ExternalCredentials() == Foreign(),
'foobar')
self.assertEqual(Foreign() == credentials.ExternalCredentials(),
'foobar')
def test_ne(self):
cr1 = credentials.ExternalCredentials()
cr2 = credentials.ExternalCredentials()
cr2.erase_on_connect = not cr2.erase_on_connect
self.assertNotEqual(cr1, cr2)
self.assertNotEqual(cr2, cr1)
cred = credentials.ExternalCredentials()
self.assertNotEqual(cred, dict(erase_on_connect=cred.erase_on_connect))
self.assertNotEqual(dict(erase_on_connect=cred.erase_on_connect), cred)
self.assertNotEqual(credentials.ExternalCredentials(),
ImprovedExternalCredentials())
self.assertNotEqual(ImprovedExternalCredentials(),
credentials.ExternalCredentials())
def test_ne(self):
cr1 = credentials.ExternalCredentials()
cr2 = credentials.ExternalCredentials()
cr2.erase_on_connect = not cr2.erase_on_connect
self.assertNotEqual(cr1, cr2)
self.assertNotEqual(cr2, cr1)
cred = credentials.ExternalCredentials()
self.assertNotEqual(
cred,
dict(erase_on_connect=cred.erase_on_connect))
self.assertNotEqual(
dict(erase_on_connect=cred.erase_on_connect),
cred)
class ImprovedExternalCredentials(credentials.ExternalCredentials):
def __init__(self, *args, **kwargs):
super(ImprovedExternalCredentials, self).__init__(*args, **kwargs)
self.extra = 'e'
def __eq__(self, other):
return (isinstance(other, ImprovedExternalCredentials) and
self.extra == other.extra and
super(ImprovedExternalCredentials, self).__eq__(other))
def __ne__(self, other):
return not self == other
self.assertNotEqual(
credentials.ExternalCredentials(),
ImprovedExternalCredentials())
self.assertNotEqual(
ImprovedExternalCredentials(),
credentials.ExternalCredentials())
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_erase_credentials(self):
with mock.patch('pika.credentials.LOGGER', autospec=True) as logger:
cred = credentials.ExternalCredentials()
cred.erase_credentials()
logger.debug.assert_called_once_with('Not supported by this '
'Credentials type')
def test_erase_response_for_no_mechanism_match(self):
cred = credentials.ExternalCredentials()
start = spec.Connection.Start()
start.mechanisms = 'FOO BAR BAZ'
self.assertEqual(cred.response_for(start), (None, None))
def test_response_for(self):
cred = credentials.ExternalCredentials()
start = spec.Connection.Start()
start.mechanisms = 'PLAIN EXTERNAL'
self.assertEqual(cred.response_for(start), ('EXTERNAL', b''))
def test_eq(self):
self.assertEqual(
credentials.ExternalCredentials(),
credentials.ExternalCredentials())
def test_response_for(self):
obj = credentials.ExternalCredentials()
start = spec.Connection.Start()
start.mechanisms = b'PLAIN EXTERNAL'
self.assertEqual(obj.response_for(start), (b'EXTERNAL', ''))
