def test_run(self, purge_): out = WhateverIO() a = purge(app=self.app, stdout=out) purge_.return_value = 0 a.run() self.assertIn('No messages purged', out.getvalue()) purge_.return_value = 100 a.run() self.assertIn('100 messages', out.getvalue())
def test_run(self, discard_all): out = WhateverIO() a = purge(app=self.app, stdout=out) discard_all.return_value = 0 a.run() self.assertIn("No messages purged", out.getvalue()) discard_all.return_value = 100 a.run() self.assertIn("100 messages", out.getvalue())
def test_run(self, purge_): out = WhateverIO() a = purge(app=self.app, stdout=out) purge_.return_value = 0 a.run(force=True) self.assertIn('No messages purged', out.getvalue()) purge_.return_value = 100 a.run(force=True) self.assertIn('100 messages', 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 purge(self, no_color, quiet, **arguments): command.purge(self.celery).run(no_color=no_color, quiet=quiet, **arguments)