use of com.google.cloud.vision.v1.Product in project java-vision by googleapis.
the class ITSystemTest method listProductsInProductSetTest.
@Test
public void listProductsInProductSetTest() {
ListProductsInProductSetRequest request = ListProductsInProductSetRequest.newBuilder().setName(formatProductSetName).build();
for (Product actualProduct : productSearchClient.listProductsInProductSet(request).iterateAll()) {
assertEquals(product.getName(), actualProduct.getName());
assertEquals(PRODUCT_DISPLAY_NAME, actualProduct.getDisplayName());
assertEquals(PRODUCT_CATEGORY, actualProduct.getProductCategory());
}
}
use of com.google.cloud.vision.v1.Product in project java-vision by googleapis.
the class PurgeProductsInProductSet method purgeProductsInProductSet.
// Delete all products in a product set.
public static void purgeProductsInProductSet(String projectId, String location, String productSetId) throws Exception {
try (ProductSearchClient client = ProductSearchClient.create()) {
String parent = LocationName.format(projectId, location);
ProductSetPurgeConfig productSetPurgeConfig = ProductSetPurgeConfig.newBuilder().setProductSetId(productSetId).build();
PurgeProductsRequest request = PurgeProductsRequest.newBuilder().setParent(parent).setProductSetPurgeConfig(productSetPurgeConfig).setForce(true).build();
OperationFuture<Empty, BatchOperationMetadata> response = client.purgeProductsAsync(request);
response.getPollingFuture().get(180, TimeUnit.SECONDS);
System.out.println("Products removed from product set.");
}
}
use of com.google.cloud.vision.v1.Product in project java-vision by googleapis.
the class ProductSearchClientTest method listProductsInProductSetTest.
@Test
public void listProductsInProductSetTest() throws Exception {
Product responsesElement = Product.newBuilder().build();
ListProductsInProductSetResponse expectedResponse = ListProductsInProductSetResponse.newBuilder().setNextPageToken("").addAllProducts(Arrays.asList(responsesElement)).build();
mockProductSearch.addResponse(expectedResponse);
ProductSetName name = ProductSetName.of("[PROJECT]", "[LOCATION]", "[PRODUCT_SET]");
ListProductsInProductSetPagedResponse pagedListResponse = client.listProductsInProductSet(name);
List<Product> resources = Lists.newArrayList(pagedListResponse.iterateAll());
Assert.assertEquals(1, resources.size());
Assert.assertEquals(expectedResponse.getProductsList().get(0), resources.get(0));
List<AbstractMessage> actualRequests = mockProductSearch.getRequests();
Assert.assertEquals(1, actualRequests.size());
ListProductsInProductSetRequest actualRequest = ((ListProductsInProductSetRequest) actualRequests.get(0));
Assert.assertEquals(name.toString(), actualRequest.getName());
Assert.assertTrue(channelProvider.isHeaderSent(ApiClientHeaderProvider.getDefaultApiClientHeaderKey(), GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
}
use of com.google.cloud.vision.v1.Product in project java-vision by googleapis.
the class ProductSearchClientTest method listReferenceImagesTest.
@Test
public void listReferenceImagesTest() throws Exception {
ReferenceImage responsesElement = ReferenceImage.newBuilder().build();
ListReferenceImagesResponse expectedResponse = ListReferenceImagesResponse.newBuilder().setNextPageToken("").addAllReferenceImages(Arrays.asList(responsesElement)).build();
mockProductSearch.addResponse(expectedResponse);
ProductName parent = ProductName.of("[PROJECT]", "[LOCATION]", "[PRODUCT]");
ListReferenceImagesPagedResponse pagedListResponse = client.listReferenceImages(parent);
List<ReferenceImage> resources = Lists.newArrayList(pagedListResponse.iterateAll());
Assert.assertEquals(1, resources.size());
Assert.assertEquals(expectedResponse.getReferenceImagesList().get(0), resources.get(0));
List<AbstractMessage> actualRequests = mockProductSearch.getRequests();
Assert.assertEquals(1, actualRequests.size());
ListReferenceImagesRequest actualRequest = ((ListReferenceImagesRequest) actualRequests.get(0));
Assert.assertEquals(parent.toString(), actualRequest.getParent());
Assert.assertTrue(channelProvider.isHeaderSent(ApiClientHeaderProvider.getDefaultApiClientHeaderKey(), GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
}
use of com.google.cloud.vision.v1.Product in project java-vision by googleapis.
the class ProductSearchClientTest method listProductsTest2.
@Test
public void listProductsTest2() throws Exception {
Product responsesElement = Product.newBuilder().build();
ListProductsResponse expectedResponse = ListProductsResponse.newBuilder().setNextPageToken("").addAllProducts(Arrays.asList(responsesElement)).build();
mockProductSearch.addResponse(expectedResponse);
String parent = "parent-995424086";
ListProductsPagedResponse pagedListResponse = client.listProducts(parent);
List<Product> resources = Lists.newArrayList(pagedListResponse.iterateAll());
Assert.assertEquals(1, resources.size());
Assert.assertEquals(expectedResponse.getProductsList().get(0), resources.get(0));
List<AbstractMessage> actualRequests = mockProductSearch.getRequests();
Assert.assertEquals(1, actualRequests.size());
ListProductsRequest actualRequest = ((ListProductsRequest) actualRequests.get(0));
Assert.assertEquals(parent, actualRequest.getParent());
Assert.assertTrue(channelProvider.isHeaderSent(ApiClientHeaderProvider.getDefaultApiClientHeaderKey(), GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
}
Aggregations