def test_run(self):
out = WhateverIO()
a = purge(app=self.app, stdout=out)
a._purge = Mock(name='_purge')
a._purge.return_value = 0
a.run(force=True)
assert 'No messages purged' in out.getvalue()
a._purge.return_value = 100
a.run(force=True)
assert '100 messages' in out.getvalue()
a.out = Mock(name='out')
a.ask = Mock(name='ask')
a.run(force=False)
a.ask.assert_called_with(a.warn_prompt, ('yes', 'no'), 'no')
a.ask.return_value = 'yes'
a.run(force=False)
def test_run(self):
out = WhateverIO()
a = purge(app=self.app, stdout=out)
a._purge = Mock(name='_purge')
a._purge.return_value = 0
a.run(force=True)
assert 'No messages purged' in out.getvalue()
a._purge.return_value = 100
a.run(force=True)
assert '100 messages' in out.getvalue()
a.out = Mock(name='out')
a.ask = Mock(name='ask')
a.run(force=False)
a.ask.assert_called_with(a.warn_prompt, ('yes', 'no'), 'no')
a.ask.return_value = 'yes'
a.run(force=False)
from app import app
from celery.bin import purge
if __name__ == '__main__':
purge.purge() # Purge Celery
app.app.run(host="0.0.0.0", port=80) # Initialize app
def run(self):
"""Running the celery app"""
purge(CELERY)
CELERY.on_after_configure.connect(self.setup_periodic_tasks)
CELERY.worker_main(['--loglevel=info', '--beat'])