def test_stop(self, _stop_workers): """ Test with "stop" as an arg. """ manage_workers.main() _stop_workers.assert_called_once_with()
def test_wrong_arg(self, stderr, exit): """ Test with an argument that isn't "stop" or "start". """ manage_workers.main() error_msg = 'This script may only be called with "start" or "stop" as an argument.\n' stderr.write.assert_called_once_with(error_msg) exit.assert_called_once_with(1)