def test_cancel_job(self): # Setup Expected Response driver_output_resource_uri = 'driverOutputResourceUri-542229086' driver_control_files_uri = 'driverControlFilesUri207057643' expected_response = { 'driver_output_resource_uri': driver_output_resource_uri, 'driver_control_files_uri': driver_control_files_uri } expected_response = jobs_pb2.Job(**expected_response) # Mock the API response channel = ChannelStub(responses=[expected_response]) patch = mock.patch('google.api_core.grpc_helpers.create_channel') with patch as create_channel: create_channel.return_value = channel client = dataproc_v1.JobControllerClient() # Setup Request project_id = 'projectId-1969970175' region = 'region-934795532' job_id = 'jobId-1154752291' response = client.cancel_job(project_id, region, job_id) assert expected_response == response assert len(channel.requests) == 1 expected_request = jobs_pb2.CancelJobRequest(project_id=project_id, region=region, job_id=job_id) actual_request = channel.requests[0][1] assert expected_request == actual_request
def test_submit_job(self): # Setup Expected Response driver_output_resource_uri = 'driverOutputResourceUri-542229086' driver_control_files_uri = 'driverControlFilesUri207057643' expected_response = { 'driver_output_resource_uri': driver_output_resource_uri, 'driver_control_files_uri': driver_control_files_uri } expected_response = jobs_pb2.Job(**expected_response) # Mock the API response channel = ChannelStub(responses=[expected_response]) client = dataproc_v1.JobControllerClient(channel=channel) # Setup Request project_id = 'projectId-1969970175' region = 'region-934795532' job = {} response = client.submit_job(project_id, region, job) assert expected_response == response assert len(channel.requests) == 1 expected_request = jobs_pb2.SubmitJobRequest( project_id=project_id, region=region, job=job) actual_request = channel.requests[0][1] assert expected_request == actual_request
def test_submit_job(self): # Setup Expected Response driver_output_resource_uri = "driverOutputResourceUri-542229086" driver_control_files_uri = "driverControlFilesUri207057643" job_uuid = "jobUuid-1615012099" expected_response = { "driver_output_resource_uri": driver_output_resource_uri, "driver_control_files_uri": driver_control_files_uri, "job_uuid": job_uuid, } expected_response = jobs_pb2.Job(**expected_response) # Mock the API response channel = ChannelStub(responses=[expected_response]) patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = dataproc_v1.JobControllerClient() # Setup Request project_id = "projectId-1969970175" region = "region-934795532" job = {} response = client.submit_job(project_id, region, job) assert expected_response == response assert len(channel.requests) == 1 expected_request = jobs_pb2.SubmitJobRequest(project_id=project_id, region=region, job=job) actual_request = channel.requests[0][1] assert expected_request == actual_request