def setUp(self):
_setUp(self)
self.client = BeanstalkClient(consumeDisconnects=False, noisy=True)
class BeanstalkClientTestCase(unittest.TestCase):
def setUp(self):
_setUp(self)
self.client = BeanstalkClient(consumeDisconnects=False, noisy=True)
def tearDown(self):
if self.client.protocol:
d = self.client.disconnect()
else:
d = defer.fail("Not connected")
return d.addCallbacks(self.fail, lambda _: spawner.terminate_all())
def test_store_host_and_port(self):
def check(client):
self.failUnlessEqual(self.host, client.host)
self.failUnlessEqual(self.port, client.port)
return self.client.connectTCP(self.host, self.port).addCallback(check)
def test_connect_and_disconnect(self):
self.connected_count = 0
self.disconnected_count = 0
def check_connected(client):
self.connected_count += 1
self.failUnlessEqual(self.client, client)
self.failUnless(self.client.protocol)
return self.client.put("tube", 1).addCallbacks(
lambda res: self.failUnlessEqual('ok', res['state']),
self.fail)
def check_disconnected(failure):
self.disconnected_count += 1
client, reason = failure.value
self.failUnlessIsInstance(reason, ConnectionDone)
self.failIf(self.client.protocol)
def check_count(_):
self.failUnlessEqual(1, self.connected_count)
self.failUnlessEqual(1, self.disconnected_count)
return self.client.connectTCP(self.host, self.port).addCallbacks(check_connected, self.fail) \
.addCallback(lambda _: self.client.disconnect()).addCallbacks(self.fail, check_disconnected) \
.addCallback(check_count)
def test_retry_connect(self):
self.connected_count = 0
self.disconnected_count = 0
def check_connected(client):
self.connected_count += 1
self.failUnlessEqual(self.client, client)
self.failUnless(self.client.protocol)
return self.client.put("tube", 1).addCallbacks(
lambda res: self.failUnlessEqual('ok', res['state']),
self.fail)
def check_disconnected(failure):
self.disconnected_count += 1
client, reason = failure.value
self.failUnlessIsInstance(reason,
(ConnectionDone, ConnectionRefusedError))
self.failIf(self.client.protocol)
def check_count(_):
self.failUnlessEqual(1, self.connected_count)
self.failUnlessEqual(1, self.disconnected_count)
spawner.terminate_all()
return self.client.connectTCP(self.host, self.port).addCallbacks(self.fail, check_disconnected) \
.addCallback(lambda _: _setUp(self)).addCallback(lambda _: self.client.deferred).addCallbacks(check_connected, self.fail) \
.addCallback(check_count)
def test_reconnect(self):
self.connected_count = 0
self.disconnected_count = 0
def check_connected(client):
self.connected_count += 1
self.failUnlessEqual(self.client, client)
self.failUnless(self.client.protocol)
return self.client.put("tube", 1).addCallbacks(
lambda res: self.failUnlessEqual('ok', res['state']),
self.fail)
def check_disconnected(failure):
self.disconnected_count += 1
client, reason = failure.value
self.failUnlessIsInstance(reason, ConnectionDone)
self.failIf(self.client.protocol)
def check_count(_):
self.failUnlessEqual(2, self.connected_count)
self.failUnlessEqual(1, self.disconnected_count)
return self.client.connectTCP(self.host, self.port).addCallbacks(check_connected, self.fail) \
.addCallback(lambda _: spawner.terminate_all()).addCallback(lambda _: self.client.deferred).addCallbacks(self.fail, check_disconnected) \
.addCallback(lambda _: _setUp(self)).addCallback(lambda _: self.client.deferred).addCallbacks(check_connected, self.fail) \
.addCallback(check_count)
def test_restore_state1(self):
self.callbacked = False
def change_state(_):
return self.client.use("used") \
.addCallback(lambda _: self.client.watch("watched")) \
.addCallback(lambda _: self.client.ignore("default"))
def restart_beanstalkd(_):
def put(_):
def mark(_):
self.callbacked = True
self.client.put("JOB").addCallback(mark)
def start(_):
_setUp(self)
return self.client.deferred
spawner.terminate_all()
return self.client.deferred.addCallbacks(
self.fail, put).addCallback(
start) # put job after we knew about disconnect
def check(_):
def check_job(job):
self.failUnless(self.callbacked)
self.failUnlessEqual("JOB", job['data'])
return self.client.stats_job(
job['jid']).addCallback(lambda stats: self.failUnlessEqual(
"used", stats['data']['tube']))
return self.client.list_tube_used().addCallback(lambda res: self.failUnlessEqual("used", res['tube'])) \
.addCallback(lambda _: self.client.list_tubes_watched()).addCallback(lambda res: self.failUnlessEqual(["watched"], res['data'])) \
.addCallback(lambda _: self.client.peek_ready()).addCallback(check_job)
return self.client.connectTCP(self.host, self.port).addCallback(
change_state).addCallback(restart_beanstalkd).addCallback(check)
def test_restore_state2(self):
self.callbacked = False
def change_state(_):
return self.client.use("used") \
.addCallback(lambda _: self.client.watch("watched")) \
.addCallback(lambda _: self.client.ignore("default"))
def restart_beanstalkd(_):
def put():
def mark(_):
self.callbacked = True
self.client.put("JOB").addCallback(mark)
return self.client.deferred
def start(_):
_setUp(self)
return self.client.deferred
spawner.terminate_all()
return put().addCallbacks(
self.fail, start) # put job before we knew about disconnect
def check(_):
def check_job(job):
self.failUnless(self.callbacked)
self.failUnlessEqual("JOB", job['data'])
return self.client.stats_job(
job['jid']).addCallback(lambda stats: self.failUnlessEqual(
"used", stats['data']['tube']))
return self.client.list_tube_used().addCallback(lambda res: self.failUnlessEqual("used", res['tube'])) \
.addCallback(lambda _: self.client.list_tubes_watched()).addCallback(lambda res: self.failUnlessEqual(["watched"], res['data'])) \
.addCallback(lambda _: self.client.peek_ready()).addCallback(check_job)
return self.client.connectTCP(self.host, self.port).addCallback(
change_state).addCallback(restart_beanstalkd).addCallback(check)
def test_reset_state(self):
self.callbacked = False
def change_state(_):
return self.client.use("used") \
.addCallback(lambda _: self.client.watch("watched")) \
.addCallback(lambda _: self.client.ignore("default"))
def reconnect(_):
def put(_):
def mark(_):
self.callbacked = True
self.client.put("JOB").addCallbacks(
mark, lambda fail: fail.trap(
'twisted.internet.error.NotConnectingError'))
def connect(_):
return self.client.connectTCP(self.host, self.port)
return self.client.disconnect().addCallbacks(
self.fail, put).addCallback(connect)
def check(_):
def check_error(fail):
fail.trap('errors.NotFound')
self.failIf(self.callbacked)
return self.client.list_tube_used().addCallback(lambda res: self.failUnlessEqual("default", res['tube'])) \
.addCallback(lambda _: self.client.list_tubes_watched()).addCallback(lambda res: self.failUnlessEqual(["default"], res['data'])) \
.addCallback(lambda _: self.client.peek_ready()).addCallbacks(self.fail, check_error)
return self.client.connectTCP(self.host, self.port).addCallback(
change_state).addCallback(reconnect).addCallback(check)
