def function_result(client, **kwargs):
# pylint: disable=function-redefined,unused-argument
error = models.BatchError()
error.code = 'InvalidHeaderValue'
error.message = models.ErrorMessage('en-US', 'The value for one of the HTTP '
'headers is not in the correct format')
error.values = [
models.BatchErrorDetail('HeaderName', 'Content-Type'),
models.BatchErrorDetail('HeaderValue', 'application/json')
]
exp = models.BatchErrorException(lambda x, y: error, None)
raise exp
def test_submit_job_and_add_task(self):
tasks = [MagicMock]
ab = AzureBatch(
batch_client=MagicMock(),
config_file=CONFIG_FILE,
logger=logging.getLogger(__name__),
)
ab.batch_client.configure_mock(
**{
'pool.exists.return_value':
False,
'job.get.side_effect':
batchmodels.BatchErrorException(MagicMock(), "response"),
})
ab.submit_job_and_add_task(POOL_ID, JOB_ID, tasks, VM_SIZE, VM_COUNT,
MAX_TASKS)
self.assertTrue(ab.batch_client.pool.add.called)
self.assertTrue(ab.batch_client.job.add.called)
self.assertTrue(ab.batch_client.task.add_collection.called)
def test_create_job(self):
mocked_batch_client = MagicMock()
mocked_batch_client.configure_mock(
**{
'credentials.return_value': MagicMock(),
'batch_url.return_value': BATCH_SERVICE_URL,
})
ab = AzureBatch(
batch_client=mocked_batch_client,
config_file=CONFIG_FILE,
logger=logging.getLogger(__name__),
)
ab.batch_client.configure_mock(
**{
'job.get.side_effect':
batchmodels.BatchErrorException(MagicMock(), "response"),
})
ab.create_job(POOL_ID, JOB_ID)
self.assertTrue(ab.batch_client.job.add.called)
self.assertEqual(ab.config_file, CONFIG_FILE)