def test_generic_callback_ok(self): from kazoo.client import _generic_callback from kazoo.handlers.threading import SequentialThreadingHandler handler = SequentialThreadingHandler() # No args asy = handler.async_result() _generic_callback(asy, 0, zookeeper.OK) eq_(asy.get(), None) # One arg thats not a dict asy = handler.async_result() _generic_callback(asy, 0, zookeeper.OK, 12) eq_(asy.get(), 12) # One arg thats a node struct asy = handler.async_result() _generic_callback(asy, 0, zookeeper.OK, dummy_dict) eq_(asy.get().acl_version, 1) # two args, second is struct asy = handler.async_result() _generic_callback(asy, 0, zookeeper.OK, 11, dummy_dict) val = asy.get() eq_(val[1].acl_version, 1) eq_(val[0], 11)
def test_exists_callback(self): from kazoo.client import _exists_callback from kazoo.handlers.threading import SequentialThreadingHandler handler = SequentialThreadingHandler() asy = handler.async_result() _exists_callback(asy, 0, zookeeper.OK, True) eq_(asy.get(), True) asy = handler.async_result() _exists_callback(asy, 0, zookeeper.CONNECTIONLOSS, False) self.assertRaises(zookeeper.ConnectionLossException, asy.get)
def test_generic_callback_error(self): from kazoo.client import _generic_callback from kazoo.handlers.threading import SequentialThreadingHandler handler = SequentialThreadingHandler() asy = handler.async_result() _generic_callback(asy, 0, zookeeper.CONNECTIONLOSS) self.assertRaises(zookeeper.ConnectionLossException, asy.get)