@BeforeClass public static void startStaticServer() { mockTraceService = new MockTraceService(); serviceHelper = new MockServiceHelper("in-process-1", Arrays.<MockGrpcService>asList(mockTraceService)); serviceHelper.start(); }
@Test @SuppressWarnings("all") public void getTraceExceptionTest() throws Exception { StatusRuntimeException exception = new StatusRuntimeException(Status.INVALID_ARGUMENT); mockTraceService.addException(exception); try { String projectId = "projectId-1969970175"; String traceId = "traceId1270300245"; client.getTrace(projectId, traceId); Assert.fail("No exception raised"); } catch (InvalidArgumentException e) { // Expected exception } }
@Test @SuppressWarnings("all") public void listTracesTest() { String nextPageToken = ""; Trace tracesElement = Trace.newBuilder().build(); List<Trace> traces = Arrays.asList(tracesElement); ListTracesResponse expectedResponse = ListTracesResponse.newBuilder() .setNextPageToken(nextPageToken) .addAllTraces(traces) .build(); mockTraceService.addResponse(expectedResponse); String projectId = "projectId-1969970175"; ListTracesPagedResponse pagedListResponse = client.listTraces(projectId); List<Trace> resources = Lists.newArrayList(pagedListResponse.iterateAll()); Assert.assertEquals(1, resources.size()); Assert.assertEquals(expectedResponse.getTracesList().get(0), resources.get(0)); List<GeneratedMessageV3> actualRequests = mockTraceService.getRequests(); Assert.assertEquals(1, actualRequests.size()); ListTracesRequest actualRequest = (ListTracesRequest) actualRequests.get(0); Assert.assertEquals(projectId, actualRequest.getProjectId()); Assert.assertTrue( channelProvider.isHeaderSent( ApiClientHeaderProvider.getDefaultApiClientHeaderKey(), GaxGrpcProperties.getDefaultApiClientHeaderPattern())); }
@Test @SuppressWarnings("all") public void patchTracesTest() { Empty expectedResponse = Empty.newBuilder().build(); mockTraceService.addResponse(expectedResponse); String projectId = "projectId-1969970175"; Traces traces = Traces.newBuilder().build(); client.patchTraces(projectId, traces); List<GeneratedMessageV3> actualRequests = mockTraceService.getRequests(); Assert.assertEquals(1, actualRequests.size()); PatchTracesRequest actualRequest = (PatchTracesRequest) actualRequests.get(0); Assert.assertEquals(projectId, actualRequest.getProjectId()); Assert.assertEquals(traces, actualRequest.getTraces()); Assert.assertTrue( channelProvider.isHeaderSent( ApiClientHeaderProvider.getDefaultApiClientHeaderKey(), GaxGrpcProperties.getDefaultApiClientHeaderPattern())); }
@Test @SuppressWarnings("all") public void getTraceTest() { String projectId2 = "projectId2939242356"; String traceId2 = "traceId2987826376"; Trace expectedResponse = Trace.newBuilder().setProjectId(projectId2).setTraceId(traceId2).build(); mockTraceService.addResponse(expectedResponse); String projectId = "projectId-1969970175"; String traceId = "traceId1270300245"; Trace actualResponse = client.getTrace(projectId, traceId); Assert.assertEquals(expectedResponse, actualResponse); List<GeneratedMessageV3> actualRequests = mockTraceService.getRequests(); Assert.assertEquals(1, actualRequests.size()); GetTraceRequest actualRequest = (GetTraceRequest) actualRequests.get(0); Assert.assertEquals(projectId, actualRequest.getProjectId()); Assert.assertEquals(traceId, actualRequest.getTraceId()); Assert.assertTrue( channelProvider.isHeaderSent( ApiClientHeaderProvider.getDefaultApiClientHeaderKey(), GaxGrpcProperties.getDefaultApiClientHeaderPattern())); }
@Test @SuppressWarnings("all") public void listTracesExceptionTest() throws Exception { StatusRuntimeException exception = new StatusRuntimeException(Status.INVALID_ARGUMENT); mockTraceService.addException(exception); try { String projectId = "projectId-1969970175"; client.listTraces(projectId); Assert.fail("No exception raised"); } catch (InvalidArgumentException e) { // Expected exception } } }
@Test @SuppressWarnings("all") public void patchTracesExceptionTest() throws Exception { StatusRuntimeException exception = new StatusRuntimeException(Status.INVALID_ARGUMENT); mockTraceService.addException(exception); try { String projectId = "projectId-1969970175"; Traces traces = Traces.newBuilder().build(); client.patchTraces(projectId, traces); Assert.fail("No exception raised"); } catch (InvalidArgumentException e) { // Expected exception } }