def setup(): global gateway gateway = TwistedGateway(logger=logging) gateway.isLeaf = True root = Resource() root.putChild('gw', gateway) root.putChild('bad', BadResource()) root.putChild('badtype', BadContentTypeResource()) site = Site(root) reactor.listenTCP(11111, site) pyamf.add_error_class(RegisteredError, u'RegisteredError')
def test_add(self): class A: pass class B(Exception): pass self.assertRaises(TypeError, pyamf.add_error_class, None, 'a') # class A does not sub-class Exception self.assertRaises(TypeError, pyamf.add_error_class, A, 'a') pyamf.add_error_class(B, 'b') self.assertEqual(pyamf.ERROR_CLASS_MAP['b'], B) pyamf.add_error_class(B, 'a') self.assertEqual(pyamf.ERROR_CLASS_MAP['a'], B) class C(Exception): pass self.assertRaises(ValueError, pyamf.add_error_class, C, 'b')
def test_add(self): class A: pass class B(Exception): pass self.assertRaises(TypeError, pyamf.add_error_class, None, "a") # class A does not sub-class Exception self.assertRaises(TypeError, pyamf.add_error_class, A, "a") pyamf.add_error_class(B, "b") self.assertEquals(pyamf.ERROR_CLASS_MAP["b"], B) pyamf.add_error_class(B, "a") self.assertEquals(pyamf.ERROR_CLASS_MAP["a"], B) class C(Exception): pass self.assertRaises(ValueError, pyamf.add_error_class, C, "b")
REPLACE_GATEWAY_URL = 'ReplaceGatewayUrl' REQUEST_PERSISTENT_HEADER = 'RequestPersistentHeader' class RemotingError(pyamf.BaseError): """ Generic remoting error class. """ class RemotingCallFailed(RemotingError): """ Raised if B{Server.Call.Failed} received. """ pyamf.add_error_class(RemotingCallFailed, ERROR_CODES[ERROR_CALL_FAILED]) class HeaderCollection(dict): """ Collection of AMF message headers. """ def __init__(self, raw_headers={}): self.required = [] for (k, ig, v) in raw_headers: self[k] = v if ig: self.required.append(k)