private void verifyHeaderSent() { boolean headerSent = channelProvider.isHeaderSent(HEADER_NAME, EXPECTED_HEADER_PATTERN); assertWithMessage("Header was sent").that(headerSent).isTrue(); } }
@Test @SuppressWarnings("all") public void getIamPolicyTest() { int version = 351608024; ByteString etag = ByteString.copyFromUtf8("21"); Policy expectedResponse = Policy.newBuilder().setVersion(version).setEtag(etag).build(); mockDeviceManager.addResponse(expectedResponse); RegistryName resource = RegistryName.of("[PROJECT]", "[LOCATION]", "[REGISTRY]"); Policy actualResponse = client.getIamPolicy(resource); Assert.assertEquals(expectedResponse, actualResponse); List<GeneratedMessageV3> actualRequests = mockDeviceManager.getRequests(); Assert.assertEquals(1, actualRequests.size()); GetIamPolicyRequest actualRequest = (GetIamPolicyRequest) actualRequests.get(0); Assert.assertEquals(Objects.toString(resource), Objects.toString(actualRequest.getResource())); Assert.assertTrue( channelProvider.isHeaderSent( ApiClientHeaderProvider.getDefaultApiClientHeaderKey(), GaxGrpcProperties.getDefaultApiClientHeaderPattern())); }
@Test @SuppressWarnings("all") public void batchAnnotateImagesTest() { BatchAnnotateImagesResponse expectedResponse = BatchAnnotateImagesResponse.newBuilder().build(); mockImageAnnotator.addResponse(expectedResponse); List<AnnotateImageRequest> requests = new ArrayList<>(); BatchAnnotateImagesResponse actualResponse = client.batchAnnotateImages(requests); Assert.assertEquals(expectedResponse, actualResponse); List<GeneratedMessageV3> actualRequests = mockImageAnnotator.getRequests(); Assert.assertEquals(1, actualRequests.size()); BatchAnnotateImagesRequest actualRequest = (BatchAnnotateImagesRequest) actualRequests.get(0); Assert.assertEquals(requests, actualRequest.getRequestsList()); Assert.assertTrue( channelProvider.isHeaderSent( ApiClientHeaderProvider.getDefaultApiClientHeaderKey(), GaxGrpcProperties.getDefaultApiClientHeaderPattern())); }
@Test @SuppressWarnings("all") public void batchAnnotateImagesTest() { BatchAnnotateImagesResponse expectedResponse = BatchAnnotateImagesResponse.newBuilder().build(); mockImageAnnotator.addResponse(expectedResponse); List<AnnotateImageRequest> requests = new ArrayList<>(); BatchAnnotateImagesResponse actualResponse = client.batchAnnotateImages(requests); Assert.assertEquals(expectedResponse, actualResponse); List<GeneratedMessageV3> actualRequests = mockImageAnnotator.getRequests(); Assert.assertEquals(1, actualRequests.size()); BatchAnnotateImagesRequest actualRequest = (BatchAnnotateImagesRequest) actualRequests.get(0); Assert.assertEquals(requests, actualRequest.getRequestsList()); Assert.assertTrue( channelProvider.isHeaderSent( ApiClientHeaderProvider.getDefaultApiClientHeaderKey(), GaxGrpcProperties.getDefaultApiClientHeaderPattern())); }
@Test @SuppressWarnings("all") public void batchAnnotateImagesTest() { BatchAnnotateImagesResponse expectedResponse = BatchAnnotateImagesResponse.newBuilder().build(); mockImageAnnotator.addResponse(expectedResponse); List<AnnotateImageRequest> requests = new ArrayList<>(); BatchAnnotateImagesResponse actualResponse = client.batchAnnotateImages(requests); Assert.assertEquals(expectedResponse, actualResponse); List<GeneratedMessageV3> actualRequests = mockImageAnnotator.getRequests(); Assert.assertEquals(1, actualRequests.size()); BatchAnnotateImagesRequest actualRequest = (BatchAnnotateImagesRequest) actualRequests.get(0); Assert.assertEquals(requests, actualRequest.getRequestsList()); Assert.assertTrue( channelProvider.isHeaderSent( ApiClientHeaderProvider.getDefaultApiClientHeaderKey(), GaxGrpcProperties.getDefaultApiClientHeaderPattern())); }
@Test @SuppressWarnings("all") public void batchAnnotateImagesTest() { BatchAnnotateImagesResponse expectedResponse = BatchAnnotateImagesResponse.newBuilder().build(); mockImageAnnotator.addResponse(expectedResponse); List<AnnotateImageRequest> requests = new ArrayList<>(); BatchAnnotateImagesResponse actualResponse = client.batchAnnotateImages(requests); Assert.assertEquals(expectedResponse, actualResponse); List<GeneratedMessageV3> actualRequests = mockImageAnnotator.getRequests(); Assert.assertEquals(1, actualRequests.size()); BatchAnnotateImagesRequest actualRequest = (BatchAnnotateImagesRequest) actualRequests.get(0); Assert.assertEquals(requests, actualRequest.getRequestsList()); Assert.assertTrue( channelProvider.isHeaderSent( ApiClientHeaderProvider.getDefaultApiClientHeaderKey(), GaxGrpcProperties.getDefaultApiClientHeaderPattern())); }
@Test @SuppressWarnings("all") public void rollbackTest() { Empty expectedResponse = Empty.newBuilder().build(); mockFirestore.addResponse(expectedResponse); String formattedDatabase = FirestoreClient.formatDatabaseRootName("[PROJECT]", "[DATABASE]"); ByteString transaction = ByteString.copyFromUtf8("-34"); client.rollback(formattedDatabase, transaction); List<GeneratedMessageV3> actualRequests = mockFirestore.getRequests(); Assert.assertEquals(1, actualRequests.size()); RollbackRequest actualRequest = (RollbackRequest) actualRequests.get(0); Assert.assertEquals(formattedDatabase, actualRequest.getDatabase()); Assert.assertEquals(transaction, actualRequest.getTransaction()); Assert.assertTrue( channelProvider.isHeaderSent( ApiClientHeaderProvider.getDefaultApiClientHeaderKey(), GaxGrpcProperties.getDefaultApiClientHeaderPattern())); }
@Test @SuppressWarnings("all") public void deleteUptimeCheckConfigTest() { Empty expectedResponse = Empty.newBuilder().build(); mockUptimeCheckService.addResponse(expectedResponse); String formattedName = UptimeCheckConfigName.format("[PROJECT]", "[UPTIME_CHECK_CONFIG]"); client.deleteUptimeCheckConfig(formattedName); List<GeneratedMessageV3> actualRequests = mockUptimeCheckService.getRequests(); Assert.assertEquals(1, actualRequests.size()); DeleteUptimeCheckConfigRequest actualRequest = (DeleteUptimeCheckConfigRequest) actualRequests.get(0); Assert.assertEquals(formattedName, actualRequest.getName()); Assert.assertTrue( channelProvider.isHeaderSent( ApiClientHeaderProvider.getDefaultApiClientHeaderKey(), GaxGrpcProperties.getDefaultApiClientHeaderPattern())); }
@Test @SuppressWarnings("all") public void deleteProductTest() { Empty expectedResponse = Empty.newBuilder().build(); mockProductSearch.addResponse(expectedResponse); String formattedName = ProductSearchClient.formatProductName("[PROJECT]", "[LOCATION]", "[PRODUCT]"); client.deleteProduct(formattedName); List<GeneratedMessageV3> actualRequests = mockProductSearch.getRequests(); Assert.assertEquals(1, actualRequests.size()); DeleteProductRequest actualRequest = (DeleteProductRequest) actualRequests.get(0); Assert.assertEquals(formattedName, actualRequest.getName()); Assert.assertTrue( channelProvider.isHeaderSent( ApiClientHeaderProvider.getDefaultApiClientHeaderKey(), GaxGrpcProperties.getDefaultApiClientHeaderPattern())); }
@Test @SuppressWarnings("all") public void deleteDocumentTest() { Empty expectedResponse = Empty.newBuilder().build(); mockFirestore.addResponse(expectedResponse); String formattedName = FirestoreClient.formatAnyPathName("[PROJECT]", "[DATABASE]", "[DOCUMENT]", "[ANY_PATH]"); client.deleteDocument(formattedName); List<GeneratedMessageV3> actualRequests = mockFirestore.getRequests(); Assert.assertEquals(1, actualRequests.size()); DeleteDocumentRequest actualRequest = (DeleteDocumentRequest) actualRequests.get(0); Assert.assertEquals(formattedName, actualRequest.getName()); Assert.assertTrue( channelProvider.isHeaderSent( ApiClientHeaderProvider.getDefaultApiClientHeaderKey(), GaxGrpcProperties.getDefaultApiClientHeaderPattern())); }
@Test @SuppressWarnings("all") public void deleteProductSetTest() { Empty expectedResponse = Empty.newBuilder().build(); mockProductSearch.addResponse(expectedResponse); String formattedName = ProductSearchClient.formatProductSetName("[PROJECT]", "[LOCATION]", "[PRODUCT_SET]"); client.deleteProductSet(formattedName); List<GeneratedMessageV3> actualRequests = mockProductSearch.getRequests(); Assert.assertEquals(1, actualRequests.size()); DeleteProductSetRequest actualRequest = (DeleteProductSetRequest) actualRequests.get(0); Assert.assertEquals(formattedName, actualRequest.getName()); Assert.assertTrue( channelProvider.isHeaderSent( ApiClientHeaderProvider.getDefaultApiClientHeaderKey(), GaxGrpcProperties.getDefaultApiClientHeaderPattern())); }
@Test @SuppressWarnings("all") public void listVoicesTest() { ListVoicesResponse expectedResponse = ListVoicesResponse.newBuilder().build(); mockTextToSpeech.addResponse(expectedResponse); String languageCode = "languageCode-412800396"; ListVoicesResponse actualResponse = client.listVoices(languageCode); Assert.assertEquals(expectedResponse, actualResponse); List<GeneratedMessageV3> actualRequests = mockTextToSpeech.getRequests(); Assert.assertEquals(1, actualRequests.size()); ListVoicesRequest actualRequest = (ListVoicesRequest) actualRequests.get(0); Assert.assertEquals(languageCode, actualRequest.getLanguageCode()); Assert.assertTrue( channelProvider.isHeaderSent( ApiClientHeaderProvider.getDefaultApiClientHeaderKey(), GaxGrpcProperties.getDefaultApiClientHeaderPattern())); }
@Test @SuppressWarnings("all") public void createTimeSeriesTest() { Empty expectedResponse = Empty.newBuilder().build(); mockMetricService.addResponse(expectedResponse); ProjectName name = ProjectName.of("[PROJECT]"); List<TimeSeries> timeSeries = new ArrayList<>(); client.createTimeSeries(name, timeSeries); List<GeneratedMessageV3> actualRequests = mockMetricService.getRequests(); Assert.assertEquals(1, actualRequests.size()); CreateTimeSeriesRequest actualRequest = (CreateTimeSeriesRequest) actualRequests.get(0); Assert.assertEquals(name, ProjectName.parse(actualRequest.getName())); Assert.assertEquals(timeSeries, actualRequest.getTimeSeriesList()); Assert.assertTrue( channelProvider.isHeaderSent( ApiClientHeaderProvider.getDefaultApiClientHeaderKey(), GaxGrpcProperties.getDefaultApiClientHeaderPattern())); }
@Test @SuppressWarnings("all") public void deleteIntentTest() { Empty expectedResponse = Empty.newBuilder().build(); mockIntents.addResponse(expectedResponse); IntentName name = IntentName.of("[PROJECT]", "[INTENT]"); client.deleteIntent(name); List<GeneratedMessageV3> actualRequests = mockIntents.getRequests(); Assert.assertEquals(1, actualRequests.size()); DeleteIntentRequest actualRequest = (DeleteIntentRequest) actualRequests.get(0); Assert.assertEquals(name, IntentName.parse(actualRequest.getName())); Assert.assertTrue( channelProvider.isHeaderSent( ApiClientHeaderProvider.getDefaultApiClientHeaderKey(), GaxGrpcProperties.getDefaultApiClientHeaderPattern())); }
@Test @SuppressWarnings("all") public void deleteSessionTest() { Empty expectedResponse = Empty.newBuilder().build(); mockSpanner.addResponse(expectedResponse); SessionName name = SessionName.of("[PROJECT]", "[INSTANCE]", "[DATABASE]", "[SESSION]"); client.deleteSession(name); List<GeneratedMessageV3> actualRequests = mockSpanner.getRequests(); Assert.assertEquals(1, actualRequests.size()); DeleteSessionRequest actualRequest = (DeleteSessionRequest) actualRequests.get(0); Assert.assertEquals(name, SessionName.parse(actualRequest.getName())); Assert.assertTrue( channelProvider.isHeaderSent( ApiClientHeaderProvider.getDefaultApiClientHeaderKey(), GaxGrpcProperties.getDefaultApiClientHeaderPattern())); }
@Test @SuppressWarnings("all") public void deleteTaskTest() { Empty expectedResponse = Empty.newBuilder().build(); mockCloudTasks.addResponse(expectedResponse); TaskName name = TaskName.of("[PROJECT]", "[LOCATION]", "[QUEUE]", "[TASK]"); client.deleteTask(name); List<GeneratedMessageV3> actualRequests = mockCloudTasks.getRequests(); Assert.assertEquals(1, actualRequests.size()); DeleteTaskRequest actualRequest = (DeleteTaskRequest) actualRequests.get(0); Assert.assertEquals(name, TaskName.parse(actualRequest.getName())); Assert.assertTrue( channelProvider.isHeaderSent( ApiClientHeaderProvider.getDefaultApiClientHeaderKey(), GaxGrpcProperties.getDefaultApiClientHeaderPattern())); }
@Test @SuppressWarnings("all") public void deleteGroupTest() { Empty expectedResponse = Empty.newBuilder().build(); mockGroupService.addResponse(expectedResponse); GroupName name = GroupName.of("[PROJECT]", "[GROUP]"); client.deleteGroup(name); List<GeneratedMessageV3> actualRequests = mockGroupService.getRequests(); Assert.assertEquals(1, actualRequests.size()); DeleteGroupRequest actualRequest = (DeleteGroupRequest) actualRequests.get(0); Assert.assertEquals(name, GroupName.parse(actualRequest.getName())); Assert.assertTrue( channelProvider.isHeaderSent( ApiClientHeaderProvider.getDefaultApiClientHeaderKey(), GaxGrpcProperties.getDefaultApiClientHeaderPattern())); }
@Test @SuppressWarnings("all") public void dropDatabaseTest() { Empty expectedResponse = Empty.newBuilder().build(); mockDatabaseAdmin.addResponse(expectedResponse); DatabaseName database = DatabaseName.of("[PROJECT]", "[INSTANCE]", "[DATABASE]"); client.dropDatabase(database); List<GeneratedMessageV3> actualRequests = mockDatabaseAdmin.getRequests(); Assert.assertEquals(1, actualRequests.size()); DropDatabaseRequest actualRequest = (DropDatabaseRequest) actualRequests.get(0); Assert.assertEquals(database, DatabaseName.parse(actualRequest.getDatabase())); Assert.assertTrue( channelProvider.isHeaderSent( ApiClientHeaderProvider.getDefaultApiClientHeaderKey(), GaxGrpcProperties.getDefaultApiClientHeaderPattern())); }
@Test @SuppressWarnings("all") public void finalizeStreamTest() { Empty expectedResponse = Empty.newBuilder().build(); mockBigQueryStorage.addResponse(expectedResponse); Stream stream = Stream.newBuilder().build(); client.finalizeStream(stream); List<GeneratedMessageV3> actualRequests = mockBigQueryStorage.getRequests(); Assert.assertEquals(1, actualRequests.size()); FinalizeStreamRequest actualRequest = (FinalizeStreamRequest) actualRequests.get(0); Assert.assertEquals(stream, actualRequest.getStream()); Assert.assertTrue( channelProvider.isHeaderSent( ApiClientHeaderProvider.getDefaultApiClientHeaderKey(), GaxGrpcProperties.getDefaultApiClientHeaderPattern())); }
@Test @SuppressWarnings("all") public void deleteOccurrenceTest() { Empty expectedResponse = Empty.newBuilder().build(); mockGrafeasV1Beta1.addResponse(expectedResponse); OccurrenceName name = OccurrenceName.of("[PROJECT]", "[OCCURRENCE]"); client.deleteOccurrence(name); List<GeneratedMessageV3> actualRequests = mockGrafeasV1Beta1.getRequests(); Assert.assertEquals(1, actualRequests.size()); DeleteOccurrenceRequest actualRequest = (DeleteOccurrenceRequest) actualRequests.get(0); Assert.assertEquals(name, OccurrenceName.parse(actualRequest.getName())); Assert.assertTrue( channelProvider.isHeaderSent( ApiClientHeaderProvider.getDefaultApiClientHeaderKey(), GaxGrpcProperties.getDefaultApiClientHeaderPattern())); }