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)
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)
