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 _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 test_name_or_value_basic_consume_object(self): self.assertEqual(callback._name_or_value(spec.Basic.Consume()), 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_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 _on_rpc_complete(self, frame): key = callback._name_or_value(frame) self._replies.append(key) self._frames[key] = frame self._received_response = True