def test_list_traces_exception(self): channel = ChannelStub(responses=[CustomException()]) patch = mock.patch('google.api_core.grpc_helpers.create_channel') with patch as create_channel: create_channel.return_value = channel client = trace_v1.TraceServiceClient() # Setup request project_id = 'projectId-1969970175' paged_list_response = client.list_traces(project_id) with pytest.raises(CustomException): list(paged_list_response)
def test_patch_traces_exception(self): # Mock the API response channel = ChannelStub(responses=[CustomException()]) patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = trace_v1.TraceServiceClient() # Setup request project_id = "projectId-1969970175" traces = {} with pytest.raises(CustomException): client.patch_traces(project_id, traces)
def test_get_trace_exception(self): # Mock the API response channel = ChannelStub(responses=[CustomException()]) patch = mock.patch('google.api_core.grpc_helpers.create_channel') with patch as create_channel: create_channel.return_value = channel client = trace_v1.TraceServiceClient() # Setup request project_id = 'projectId-1969970175' trace_id = 'traceId1270300245' with pytest.raises(CustomException): client.get_trace(project_id, trace_id)
def test_patch_traces(self): channel = ChannelStub() client = trace_v1.TraceServiceClient(channel=channel) # Setup Request project_id = 'projectId-1969970175' traces = {} client.patch_traces(project_id, traces) assert len(channel.requests) == 1 expected_request = trace_pb2.PatchTracesRequest(project_id=project_id, traces=traces) actual_request = channel.requests[0][1] assert expected_request == actual_request
def test_patch_traces(self): channel = ChannelStub() patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = trace_v1.TraceServiceClient() # Setup Request project_id = "projectId-1969970175" traces = {} client.patch_traces(project_id, traces) assert len(channel.requests) == 1 expected_request = trace_pb2.PatchTracesRequest(project_id=project_id, traces=traces) actual_request = channel.requests[0][1] assert expected_request == actual_request
def client(): return trace_v1.TraceServiceClient()
def test_list_traces(self): project_id = os.environ['PROJECT_ID'] client = trace_v1.TraceServiceClient() project_id_2 = project_id response = client.list_traces(project_id_2)
def trace_id(): # list all traces in the project and return the first client = trace_v1.TraceServiceClient() traces = client.list_traces(project_id=PROJECT_ID) return list(traces)[0].trace_id