1234567 |
- def test_job_configuration_and_create_job(self, mock_batch):
- hook = AzureBatchHook(azure_batch_conn_id=self.test_vm_conn_id)
- mock_instance = mock_batch.return_value.job.add
- job = hook.configure_job(job_id='myjob', pool_id='mypool')
- hook.create_job(job)
- assert isinstance(job, batch_models.JobAddParameter)
- mock_instance.assert_called_once_with(job)
|