self.app.conf.BROKER_POOL_LIMIT = None
try:
self.app._pool = None
self.assertFreed(its, task1.delay)
finally:
self.app.conf.BROKER_POOL_LIMIT = pool_limit
def test_task_apply_leak_with_pool(self, its=1000):
self.assertNotEqual(self.app.conf.BROKER_TRANSPORT, 'memory')
@self.app.task
def task2():
pass
try:
pool_limit = self.app.conf.BROKER_POOL_LIMIT
except AttributeError:
raise SkipTest('This version does not support autopool')
self.app.conf.BROKER_POOL_LIMIT = 10
try:
self.app._pool = None
self.assertFreed(its, task2.delay)
finally:
self.app.conf.BROKER_POOL_LIMIT = pool_limit
if __name__ == '__main__':
unittest.main()
self.app.conf.BROKER_POOL_LIMIT = None
try:
self.app._pool = None
self.assertFreed(its, task1.delay)
finally:
self.app.conf.BROKER_POOL_LIMIT = pool_limit
def test_task_apply_leak_with_pool(self, its=1000):
self.assertNotEqual(self.app.conf.BROKER_TRANSPORT, 'memory')
@self.app.task
def task2():
pass
try:
pool_limit = self.app.conf.BROKER_POOL_LIMIT
except AttributeError:
raise SkipTest('This version does not support autopool')
self.app.conf.BROKER_POOL_LIMIT = 10
try:
self.app._pool = None
self.assertFreed(its, task2.delay)
finally:
self.app.conf.BROKER_POOL_LIMIT = pool_limit
if __name__ == '__main__':
unittest.main()