use of com.netsuite.webservices.v2016_2.platform.core.Record in project components by Talend.
the class NetSuiteMockTestBase method makeRecordPages.
public static <T extends Record> List<SearchResult> makeRecordPages(List<T> recordList, int pageSize) throws Exception {
int count = recordList.size();
int totalPages = count / pageSize;
if (count % pageSize != 0) {
totalPages += 1;
}
String searchId = UUID.randomUUID().toString();
List<SearchResult> pageResults = new ArrayList<>();
SearchResult result = null;
Iterator<T> recordIterator = recordList.iterator();
while (recordIterator.hasNext() && count > 0) {
T record = recordIterator.next();
if (result == null) {
result = new SearchResult();
result.setSearchId(searchId);
result.setTotalPages(totalPages);
result.setTotalRecords(count);
result.setPageIndex(pageResults.size() + 1);
result.setPageSize(pageSize);
result.setStatus(createSuccessStatus());
}
if (result.getRecordList() == null) {
result.setRecordList(new RecordList());
}
result.getRecordList().getRecord().add(record);
if (result.getRecordList().getRecord().size() == pageSize) {
pageResults.add(result);
result = null;
}
count--;
}
if (result != null) {
pageResults.add(result);
}
return pageResults;
}
use of com.netsuite.webservices.v2016_2.platform.core.Record in project components by Talend.
the class NetSuiteMockTestBase method makeIndexedRecords.
public static <T> List<IndexedRecord> makeIndexedRecords(NetSuiteClientService<?> clientService, Schema schema, ObjectComposer<T> objectComposer, int count) throws Exception {
NsObjectInputTransducer transducer = new NsObjectInputTransducer(clientService, schema, schema.getName());
List<IndexedRecord> recordList = new ArrayList<>();
while (count > 0) {
T nsRecord = objectComposer.composeObject();
IndexedRecord convertedRecord = transducer.read(nsRecord);
Schema recordSchema = convertedRecord.getSchema();
GenericRecord record = new GenericData.Record(recordSchema);
for (Schema.Field field : schema.getFields()) {
Object value = convertedRecord.get(field.pos());
record.put(field.pos(), value);
}
recordList.add(record);
count--;
}
return recordList;
}
use of com.netsuite.webservices.v2016_2.platform.core.Record in project components by Talend.
the class NetSuiteMockTestBase method mockGetListRequestResults.
protected <T extends Record> void mockGetListRequestResults(final List<T> records) throws Exception {
final NetSuitePortType port = webServiceMockTestFixture.getPortMock();
when(port.getList(any(GetListRequest.class))).then(new Answer<GetListResponse>() {
@Override
public GetListResponse answer(InvocationOnMock invocationOnMock) throws Throwable {
GetListRequest request = (GetListRequest) invocationOnMock.getArguments()[0];
GetListResponse response = new GetListResponse();
ReadResponseList readResponseList = new ReadResponseList();
int count = request.getBaseRef().size();
for (int i = 0; i < count; i++) {
ReadResponse readResponse = new ReadResponse();
T record = records != null ? records.get(i) : null;
if (record != null) {
readResponse.setStatus(createSuccessStatus());
} else {
readResponse.setStatus(createNotFoundStatus());
}
readResponse.setRecord(record);
readResponseList.getReadResponse().add(readResponse);
}
response.setReadResponseList(readResponseList);
return response;
}
});
}
use of com.netsuite.webservices.v2016_2.platform.core.Record in project components by Talend.
the class NetSuiteMockTestBase method mockSearchRequestResults.
protected <T extends Record> void mockSearchRequestResults(List<T> recordList, int pageSize) throws Exception {
final NetSuitePortType port = webServiceMockTestFixture.getPortMock();
final List<SearchResult> pageResults = makeRecordPages(recordList, pageSize);
when(port.search(any(SearchRequest.class))).then(new Answer<SearchResponse>() {
@Override
public SearchResponse answer(InvocationOnMock invocationOnMock) throws Throwable {
SearchResponse response = new SearchResponse();
response.setSearchResult(pageResults.get(0));
return response;
}
});
when(port.searchMoreWithId(any(SearchMoreWithIdRequest.class))).then(new Answer<SearchMoreWithIdResponse>() {
@Override
public SearchMoreWithIdResponse answer(InvocationOnMock invocationOnMock) throws Throwable {
SearchMoreWithIdRequest request = (SearchMoreWithIdRequest) invocationOnMock.getArguments()[0];
SearchMoreWithIdResponse response = new SearchMoreWithIdResponse();
response.setSearchResult(pageResults.get(request.getPageIndex() - 1));
return response;
}
});
}
use of com.netsuite.webservices.v2016_2.platform.core.Record in project components by Talend.
the class NetSuiteOutputWriterTest method testUpdate.
@Test
public void testUpdate() throws Exception {
final NetSuitePortType port = webServiceMockTestFixture.getPortMock();
final TypeDesc typeDesc = webServiceMockTestFixture.getClientService().getMetaDataSource().getTypeInfo(RecordTypeEnum.OPPORTUNITY.getTypeName());
mockGetListRequestResults(null);
final List<Opportunity> updatedRecordList = new ArrayList<>();
when(port.updateList(any(UpdateListRequest.class))).then(new Answer<UpdateListResponse>() {
@Override
public UpdateListResponse answer(InvocationOnMock invocationOnMock) throws Throwable {
UpdateListRequest request = (UpdateListRequest) invocationOnMock.getArguments()[0];
assertFalse(request.getRecord().isEmpty());
UpdateListResponse response = new UpdateListResponse();
WriteResponseList writeResponseList = new WriteResponseList();
writeResponseList.setStatus(createSuccessStatus());
for (int i = 0; i < request.getRecord().size(); i++) {
Opportunity record = (Opportunity) request.getRecord().get(i);
assertNotNull(record);
assertNotNull(record.getInternalId());
RecordRef recordRef = new RecordRef();
recordRef.setInternalId(record.getInternalId());
recordRef.setType(RecordType.OPPORTUNITY);
updatedRecordList.add(record);
WriteResponse writeResponse = new WriteResponse();
writeResponse.setStatus(createSuccessStatus());
writeResponse.setBaseRef(recordRef);
writeResponseList.getWriteResponse().add(writeResponse);
}
response.setWriteResponseList(writeResponseList);
return response;
}
});
properties.module.moduleName.setValue(typeDesc.getTypeName());
properties.module.action.setValue(OutputAction.UPDATE);
NetSuiteRuntime netSuiteRuntime = new NetSuiteRuntimeImpl();
NetSuiteDatasetRuntime dataSetRuntime = netSuiteRuntime.getDatasetRuntime(properties.getConnectionProperties());
Schema schema = dataSetRuntime.getSchema(properties.module.moduleName.getValue());
properties.module.main.schema.setValue(schema);
NetSuiteSink sink = new NetSuiteSinkImpl();
sink.initialize(mockTestFixture.getRuntimeContainer(), properties);
NetSuiteClientService<?> clientService = sink.getClientService();
NetSuiteWriteOperation writeOperation = (NetSuiteWriteOperation) sink.createWriteOperation();
NetSuiteOutputWriter writer = (NetSuiteOutputWriter) writeOperation.createWriter(mockTestFixture.getRuntimeContainer());
writer.open(UUID.randomUUID().toString());
List<IndexedRecord> indexedRecordList = makeIndexedRecords(clientService, schema, new SimpleObjectComposer<>(typeDesc.getTypeClass()), 150);
for (IndexedRecord record : indexedRecordList) {
writer.write(record);
}
Result writerResult = writer.close();
assertNotNull(writerResult);
assertEquals(indexedRecordList.size(), writerResult.totalCount);
verify(port, times(2)).updateList(any(UpdateListRequest.class));
assertEquals(indexedRecordList.size(), updatedRecordList.size());
}
Aggregations