def test_remove(self): class B(Exception): pass pyamf.ERROR_CLASS_MAP['abc'] = B self.assertRaises(TypeError, pyamf.remove_error_class, None) pyamf.remove_error_class('abc') self.assertFalse('abc' in pyamf.ERROR_CLASS_MAP.keys()) self.assertRaises(KeyError, pyamf.ERROR_CLASS_MAP.__getitem__, 'abc') pyamf.ERROR_CLASS_MAP['abc'] = B pyamf.remove_error_class(B) self.assertRaises(KeyError, pyamf.ERROR_CLASS_MAP.__getitem__, 'abc') self.assertRaises(ValueError, pyamf.remove_error_class, B) self.assertRaises(ValueError, pyamf.remove_error_class, 'abc')
def teardown(): pyamf.remove_error_class(RegisteredError) reactor.stop()