Search in sources :

Example 1 with ServiceSaleItem

use of com.netsuite.webservices.test.lists.accounting.ServiceSaleItem in project components by Talend.

the class SearchResultSetTest method testRecordFiltering.

@Test
public void testRecordFiltering() throws Exception {
    NetSuiteClientService<?> conn = mock(NetSuiteClientService.class);
    List<Record> page1 = new ArrayList<>();
    for (int i = 0; i < 1000; i++) {
        page1.add(new InventoryItem());
    }
    List<Record> page2 = new ArrayList<>();
    for (int i = 0; i < 750; i++) {
        page2.add(new ServiceSaleItem());
    }
    SearchResult result1 = new SearchResult();
    Status status = new Status();
    status.setIsSuccess(true);
    result1.setStatus(status);
    result1.setSearchId("abc123");
    result1.setPageIndex(1);
    result1.setTotalRecords(page1.size() + page2.size());
    result1.setTotalPages(2);
    result1.setRecordList(new RecordList());
    result1.getRecordList().getRecord().addAll(page1);
    SearchResult result2 = new SearchResult();
    result2.setStatus(status);
    result2.setSearchId(result1.getSearchId());
    result2.setPageIndex(2);
    result2.setTotalRecords(result1.getTotalRecords());
    result2.setTotalPages(result1.getTotalPages());
    result2.setRecordList(new RecordList());
    result2.getRecordList().getRecord().addAll(page2);
    SearchResponse response1 = new SearchResponse();
    response1.setSearchResult(result1);
    SearchMoreWithIdResponse response2 = new SearchMoreWithIdResponse();
    response2.setSearchResult(result2);
    ItemSearch nsSearchRecord1 = new ItemSearch();
    NsSearchResult nsSearchResult1 = TestNetSuiteClientService.toNsSearchResult(result1);
    NsSearchResult nsSearchResult2 = TestNetSuiteClientService.toNsSearchResult(result2);
    when(conn.search(eq(nsSearchRecord1))).thenReturn(nsSearchResult1);
    when(conn.searchMoreWithId(eq("abc123"), eq(2))).thenReturn(nsSearchResult2);
    NetSuiteClientService<?> clientService = new TestNetSuiteClientService();
    RecordTypeInfo recordTypeInfo = clientService.getMetaDataSource().getRecordType("InventoryItem");
    SearchRecordTypeDesc searchRecordTypeDesc = clientService.getMetaDataSource().getSearchRecordType(recordTypeInfo.getRecordType().getSearchRecordType());
    SearchResultSet<Record> resultSet = new SearchResultSet<>(conn, recordTypeInfo.getRecordType(), searchRecordTypeDesc, nsSearchResult1);
    List<Object> recordList = new ArrayList<>();
    while (resultSet.next()) {
        Object record = resultSet.get();
        assertNotNull(record);
        recordList.add(record);
    }
    assertEquals(page1.size(), recordList.size());
}
Also used : Status(com.netsuite.webservices.test.platform.core.Status) InventoryItem(com.netsuite.webservices.test.lists.accounting.InventoryItem) ServiceSaleItem(com.netsuite.webservices.test.lists.accounting.ServiceSaleItem) TestNetSuiteClientService(org.talend.components.netsuite.test.client.TestNetSuiteClientService) SearchMoreWithIdResponse(com.netsuite.webservices.test.platform.messages.SearchMoreWithIdResponse) ArrayList(java.util.ArrayList) SearchResult(com.netsuite.webservices.test.platform.core.SearchResult) ItemSearch(com.netsuite.webservices.test.lists.accounting.ItemSearch) SearchResponse(com.netsuite.webservices.test.platform.messages.SearchResponse) SearchRecordTypeDesc(org.talend.components.netsuite.client.model.SearchRecordTypeDesc) RecordList(com.netsuite.webservices.test.platform.core.RecordList) RecordTypeInfo(org.talend.components.netsuite.client.model.RecordTypeInfo) SearchResultSet(org.talend.components.netsuite.client.search.SearchResultSet) Record(com.netsuite.webservices.test.platform.core.Record) Test(org.junit.Test)

Aggregations

InventoryItem (com.netsuite.webservices.test.lists.accounting.InventoryItem)1 ItemSearch (com.netsuite.webservices.test.lists.accounting.ItemSearch)1 ServiceSaleItem (com.netsuite.webservices.test.lists.accounting.ServiceSaleItem)1 Record (com.netsuite.webservices.test.platform.core.Record)1 RecordList (com.netsuite.webservices.test.platform.core.RecordList)1 SearchResult (com.netsuite.webservices.test.platform.core.SearchResult)1 Status (com.netsuite.webservices.test.platform.core.Status)1 SearchMoreWithIdResponse (com.netsuite.webservices.test.platform.messages.SearchMoreWithIdResponse)1 SearchResponse (com.netsuite.webservices.test.platform.messages.SearchResponse)1 ArrayList (java.util.ArrayList)1 Test (org.junit.Test)1 RecordTypeInfo (org.talend.components.netsuite.client.model.RecordTypeInfo)1 SearchRecordTypeDesc (org.talend.components.netsuite.client.model.SearchRecordTypeDesc)1 SearchResultSet (org.talend.components.netsuite.client.search.SearchResultSet)1 TestNetSuiteClientService (org.talend.components.netsuite.test.client.TestNetSuiteClientService)1