def test_jobmgr_submit(self, mock_api, mock_creds, mock_cfg):
"""Test submit"""
_callback = mock.Mock()
job = mock.create_autospec(JobSubmission)
job.name = "test"
job.source = "test"
job.required_files = mock.create_autospec(FileCollection)
mgr = JobManager(mock_creds, cfg=mock_cfg)
mgr.submit(job)
self.assertTrue(job.submit.called)
job.required_files.upload.assert_called_with(threads=None,
callback=None,
block=4096)
mgr.submit(job, upload_threads=10, callback=_callback, block=11111)
self.assertTrue(job.submit.called)
job.required_files.upload.assert_called_with(threads=10,
callback=_callback,
block=11111)
with self.assertRaises(TypeError):
mgr.submit("test")
job.required_files.upload.return_value = ["oops"]
with self.assertRaises(Exception):
mgr.submit(job)
def test_jobmgr_submit(self, mock_api, mock_creds, mock_cfg):
"""Test submit"""
_callback = mock.Mock()
job = mock.create_autospec(JobSubmission)
job.name = "test"
job.source = "test"
job.required_files = mock.create_autospec(FileCollection)
mgr = JobManager(mock_creds, cfg=mock_cfg)
mgr.submit(job)
self.assertTrue(job.submit.called)
job.required_files.upload.assert_called_with(threads=None, callback=None, block=4096)
mgr.submit(job, upload_threads=10, callback=_callback, block=11111)
self.assertTrue(job.submit.called)
job.required_files.upload.assert_called_with(threads=10, callback=_callback, block=11111)
with self.assertRaises(TypeError):
mgr.submit("test")
job.required_files.upload.return_value = ["oops"]
with self.assertRaises(Exception):
mgr.submit(job)
