def test_submit_job(self, mock_mgr, mock_check_dir):
        """Test submit_job"""

        mock_check_dir.return_value = "C:\\valid_dir"
        mock_mgr.return_value = mock.create_autospec(FileManager)

        config = mock.create_autospec(Configuration)
        auth = mock.create_autospec(Credentials)
        job_mgr = mock.create_autospec(JobManager)
        new_job = mock.create_autospec(JobSubmission)
        job_mgr.create_job.return_value = new_job
        new_job.required_files = mock.create_autospec(FileCollection)

        new_job.submit.side_effect = RestCallException(None, "RestCallExceptionTEST", None)

        with self.assertRaises(RuntimeError):
            client.submit_job(config, auth, job_mgr)

        self.assertTrue(new_job.submit.called)
        mock_mgr.assert_called_with(auth, cfg=config)
        mock_mgr.return_value.files_from_dir.assert_called_with("C:\\valid_dir")
Example #2
0
    def test_submit_job(self, mock_mgr, mock_check_dir):
        """Test submit_job"""

        mock_check_dir.return_value = "C:\\valid_dir"
        mock_mgr.return_value = mock.create_autospec(FileManager)

        config = mock.create_autospec(Configuration)
        auth = mock.create_autospec(Credentials)
        job_mgr = mock.create_autospec(JobManager)
        new_job = mock.create_autospec(JobSubmission)
        job_mgr.create_job.return_value = new_job
        new_job.required_files = mock.create_autospec(FileCollection)

        new_job.submit.side_effect = RestCallException(
            None, "RestCallExceptionTEST", None)

        with self.assertRaises(RuntimeError):
            client.submit_job(config, auth, job_mgr)

        self.assertTrue(new_job.submit.called)
        mock_mgr.assert_called_with(auth, cfg=config)
        mock_mgr.return_value.files_from_dir.assert_called_with(
            "C:\\valid_dir")