def test_stop(self, _stop_workers):
"""
Test with "stop" as an arg.
"""
manage_workers.main()
_stop_workers.assert_called_once_with()
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)
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)
