예제 #1
0
    def test_start_job_no_token(self, mock_print, mock_bespin_api,
                                mock_config_file):
        commands = Commands(self.version_str, self.user_agent_str)
        commands.start_job(job_id=1)

        self.assertFalse(mock_bespin_api.return_value.authorize_job.called)
        mock_bespin_api.return_value.start_job.assert_called_with(1)
        mock_print.assert_has_calls([call("Started job 1")])
예제 #2
0
    def test_start_job_with_token(self, mock_print, mock_bespin_api,
                                  mock_config_file):
        commands = Commands(self.version_str, self.user_agent_str)
        commands.start_job(job_id=1, token='secret')

        mock_bespin_api.return_value.authorize_job.assert_called_with(
            1, 'secret')
        mock_bespin_api.return_value.start_job.assert_called_with(1)
        mock_print.assert_has_calls(
            [call('Set run token for job 1'),
             call('Started job 1')])