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)