Search in sources :

Example 1 with SearchPagedResponse

use of com.google.cloud.retail.v2.SearchServiceClient.SearchPagedResponse in project java-retail by googleapis.

the class SearchServiceClientTest method searchTest.

@Test
public void searchTest() throws Exception {
    SearchResponse.SearchResult responsesElement = SearchResponse.SearchResult.newBuilder().build();
    SearchResponse expectedResponse = SearchResponse.newBuilder().setNextPageToken("").addAllResults(Arrays.asList(responsesElement)).build();
    mockSearchService.addResponse(expectedResponse);
    SearchRequest request = SearchRequest.newBuilder().setPlacement("placement1792938725").setBranch(BranchName.of("[PROJECT]", "[LOCATION]", "[CATALOG]", "[BRANCH]").toString()).setQuery("query107944136").setVisitorId("visitorId1880545833").setUserInfo(UserInfo.newBuilder().build()).setPageSize(883849137).setPageToken("pageToken873572522").setOffset(-1019779949).setFilter("filter-1274492040").setCanonicalFilter("canonicalFilter-722283124").setOrderBy("orderBy-1207110587").addAllFacetSpecs(new ArrayList<SearchRequest.FacetSpec>()).setDynamicFacetSpec(SearchRequest.DynamicFacetSpec.newBuilder().build()).setBoostSpec(SearchRequest.BoostSpec.newBuilder().build()).setQueryExpansionSpec(SearchRequest.QueryExpansionSpec.newBuilder().build()).addAllVariantRollupKeys(new ArrayList<String>()).addAllPageCategories(new ArrayList<String>()).build();
    SearchPagedResponse pagedListResponse = client.search(request);
    List<SearchResponse.SearchResult> resources = Lists.newArrayList(pagedListResponse.iterateAll());
    Assert.assertEquals(1, resources.size());
    Assert.assertEquals(expectedResponse.getResultsList().get(0), resources.get(0));
    List<AbstractMessage> actualRequests = mockSearchService.getRequests();
    Assert.assertEquals(1, actualRequests.size());
    SearchRequest actualRequest = ((SearchRequest) actualRequests.get(0));
    Assert.assertEquals(request.getPlacement(), actualRequest.getPlacement());
    Assert.assertEquals(request.getBranch(), actualRequest.getBranch());
    Assert.assertEquals(request.getQuery(), actualRequest.getQuery());
    Assert.assertEquals(request.getVisitorId(), actualRequest.getVisitorId());
    Assert.assertEquals(request.getUserInfo(), actualRequest.getUserInfo());
    Assert.assertEquals(request.getPageSize(), actualRequest.getPageSize());
    Assert.assertEquals(request.getPageToken(), actualRequest.getPageToken());
    Assert.assertEquals(request.getOffset(), actualRequest.getOffset());
    Assert.assertEquals(request.getFilter(), actualRequest.getFilter());
    Assert.assertEquals(request.getCanonicalFilter(), actualRequest.getCanonicalFilter());
    Assert.assertEquals(request.getOrderBy(), actualRequest.getOrderBy());
    Assert.assertEquals(request.getFacetSpecsList(), actualRequest.getFacetSpecsList());
    Assert.assertEquals(request.getDynamicFacetSpec(), actualRequest.getDynamicFacetSpec());
    Assert.assertEquals(request.getBoostSpec(), actualRequest.getBoostSpec());
    Assert.assertEquals(request.getQueryExpansionSpec(), actualRequest.getQueryExpansionSpec());
    Assert.assertEquals(request.getVariantRollupKeysList(), actualRequest.getVariantRollupKeysList());
    Assert.assertEquals(request.getPageCategoriesList(), actualRequest.getPageCategoriesList());
    Assert.assertEquals(request.getSearchMode(), actualRequest.getSearchMode());
    Assert.assertTrue(channelProvider.isHeaderSent(ApiClientHeaderProvider.getDefaultApiClientHeaderKey(), GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
}
Also used : AbstractMessage(com.google.protobuf.AbstractMessage) ArrayList(java.util.ArrayList) SearchPagedResponse(com.google.cloud.retail.v2.SearchServiceClient.SearchPagedResponse) Test(org.junit.Test)

Aggregations

SearchPagedResponse (com.google.cloud.retail.v2.SearchServiceClient.SearchPagedResponse)1 AbstractMessage (com.google.protobuf.AbstractMessage)1 ArrayList (java.util.ArrayList)1 Test (org.junit.Test)1