def test_prompting(self, mock_prompt):
self.bs1.client.use('some-tube')
for i in range(100):
self.bs1.client.put_job(str(i))
parser = flush.setup_parser()
args = parser.parse_args([
'-H', self.bs1.host, '-p', str(self.bs1.port),
])
flush.run(args)
mock_prompt.assert_called_once_with('Are you sure you want to flush tubes default, some-tube (y/N)? ')
def test_flush_runnable(self):
self.bs1.client.use('some-tube')
for i in range(100):
self.bs1.client.put_job(str(i))
assert self.bs1.status()['current-jobs-ready'] == 100
parser = flush.setup_parser()
args = parser.parse_args([
'-H', self.bs1.host, '-p', str(self.bs1.port),
'-y',
'some-tube'
])
flush.run(args)
assert self.bs1.status()['current-jobs-ready'] == 0
assert self.bs1.status()['current-jobs-delayed'] == 0
assert self.bs1.status()['current-jobs-buried'] == 0
