def test_sanitize_name_in_self(self):
"""
Verify NAME is gotten from test_cls.NAME
"""
result = callback._name_or_value(self.test_cls)
self.assertEqual(result, self.test_cls.NAME)
def test_name_or_value_unicode(self):
value = u"Это тест значения"
expectation = (
"\xd0\xad\xd1\x82\xd0\xbe \xd1\x82\xd0\xb5\xd1\x81\xd1"
"\x82 \xd0\xb7\xd0\xbd\xd0\xb0\xd1\x87\xd0\xb5\xd0\xbd"
"\xd0\xb8\xd1\x8f"
)
self.assertEqual(callback._name_or_value(value), expectation)
def test_sanitize_name_c(self):
"""
Verify NAME is gotten from test_cls.__dict__['NAME']
"""
delattr(self.test_cls, 'NAME')
self.test_cls.__dict__['NAME'] = 'in __dict__'
result = callback._name_or_value(self.test_cls)
self.assertEqual(result, self.test_cls.__dict__['NAME'])
def test_sanitize_name_in_method(self):
"""
Verify NAME is gotten from test_cls.method.NAME
"""
# put a 'method' class in our test_cls
# for the sanitize method to find.
self.test_cls.method = Mock(spec=['NAME'])
self.test_cls.method.NAME = 'attr of attr'
result = callback._name_or_value(self.test_cls)
self.assertEqual(result, self.test_cls.method.NAME)
def test_name_or_value_amqpobject_class(self):
self.assertEqual(callback._name_or_value(self.PREFIX_CLASS),
self.PREFIX)
def _remove_reply(self, frame):
key = callback._name_or_value(frame)
if key in self._replies:
self._replies.remove(key)
def _remove_reply(self, frame):
key = callback._name_or_value(frame)
if key in self._replies:
self._replies.remove(key)
def _add_reply(self, reply):
reply = callback._name_or_value(reply)
self._replies.append(reply)
def add_reply(self, reply):
reply = _name_or_value(reply)
self._replies.append(reply)
def test_name_or_value_str(self):
value = 'Test String Value'
expectation = value
self.assertEqual(callback._name_or_value(value), expectation)
def test_name_or_value_method_object(self):
value = spec.Basic.Consume()
self.assertEqual(callback._name_or_value(value), self.PREFIX)
def remove_reply(self, frame):
key = _name_or_value(frame)
if key in self._replies:
self._replies.remove(key)
def add_reply(self, reply):
reply = _name_or_value(reply)
self._replies.append(reply)
def test_name_or_value_basic_consume_object(self):
self.assertEqual(callback._name_or_value(spec.Basic.Consume()),
self.PREFIX)
def test_name_or_value_method_object(self):
value = spec.Basic.Consume()
self.assertEqual(callback._name_or_value(value), self.PREFIX)
def test_name_or_value_protocol_header(self):
self.assertEqual(callback._name_or_value(frame.ProtocolHeader()),
'ProtocolHeader')
def test_name_or_value_method_frame(self):
value = frame.Method(1, self.PREFIX_CLASS())
self.assertEqual(callback._name_or_value(value), self.PREFIX)
def test_name_or_value_basic_consume_object(self):
self.assertEqual(callback._name_or_value(spec.Basic.Consume()),
self.PREFIX)
def test_name_or_value_unicode(self):
value = u'Это тест значения'
expectation = ('\xd0\xad\xd1\x82\xd0\xbe \xd1\x82\xd0\xb5\xd1\x81\xd1'
'\x82 \xd0\xb7\xd0\xbd\xd0\xb0\xd1\x87\xd0\xb5\xd0\xbd'
'\xd0\xb8\xd1\x8f')
self.assertEqual(callback._name_or_value(value), expectation)
def test_name_or_value_amqpobject_class(self):
self.assertEqual(callback._name_or_value(self.PREFIX_CLASS),
self.PREFIX)
def remove_reply(self, frame):
key = _name_or_value(frame)
if key in self._replies:
self._replies.remove(key)
def test_name_or_value_protocol_header(self):
self.assertEqual(callback._name_or_value(frame.ProtocolHeader()),
'ProtocolHeader')
def _on_rpc_complete(self, frame):
key = callback._name_or_value(frame)
self._replies.append(key)
self._frames[key] = frame
self._received_response = True
def test_name_or_value_method_frame(self):
value = frame.Method(1, self.PREFIX_CLASS())
self.assertEqual(callback._name_or_value(value), self.PREFIX)
def _on_rpc_complete(self, frame):
key = callback._name_or_value(frame)
self._replies.append(key)
self._frames[key] = frame
self._received_response = True
def test_name_or_value_str(self):
value = 'Test String Value'
expectation = value
self.assertEqual(callback._name_or_value(value), expectation)
def _add_reply(self, reply):
reply = callback._name_or_value(reply)
self._replies.append(reply)
def test_name_or_value_unicode(self):
value = u'Это тест значения'
expectation = ('\xd0\xad\xd1\x82\xd0\xbe \xd1\x82\xd0\xb5\xd1\x81\xd1'
'\x82 \xd0\xb7\xd0\xbd\xd0\xb0\xd1\x87\xd0\xb5\xd0\xbd'
'\xd0\xb8\xd1\x8f')
self.assertEqual(callback._name_or_value(value), expectation)
