Search in sources :

Example 1 with StringCustomFieldRef

use of com.netsuite.webservices.v2016_2.platform.core.StringCustomFieldRef in project components by Talend.

the class NetSuiteOutputWriterIT method testAddCustomRecord.

@Test
public void testAddCustomRecord() throws Exception {
    final NetSuiteClientService<NetSuitePortType> clientService = webServiceTestFixture.getClientService();
    clientService.getMetaDataSource().setCustomizationEnabled(true);
    RuntimeContainer container = mock(RuntimeContainer.class);
    NetSuiteOutputProperties properties = new NetSuiteOutputProperties("test");
    properties.init();
    properties.connection.endpoint.setValue(webServiceTestFixture.getEndpointUrl());
    properties.connection.email.setValue(webServiceTestFixture.getCredentials().getEmail());
    properties.connection.password.setValue(webServiceTestFixture.getCredentials().getPassword());
    properties.connection.account.setValue(webServiceTestFixture.getCredentials().getAccount());
    properties.connection.role.setValue(Integer.valueOf(webServiceTestFixture.getCredentials().getRoleId()));
    properties.connection.applicationId.setValue(webServiceTestFixture.getCredentials().getApplicationId());
    properties.module.moduleName.setValue("customrecordqacomp_custom_recordtype");
    properties.module.action.setValue(OutputAction.ADD);
    NetSuiteRuntimeImpl runtime = new NetSuiteRuntimeImpl();
    runtime.setClientFactory(clientFactory);
    NetSuiteDatasetRuntime dataSetRuntime = runtime.getDatasetRuntime(properties.getConnectionProperties());
    Schema schema = dataSetRuntime.getSchema(properties.module.moduleName.getValue());
    Schema targetSchema = TestUtils.makeRecordSchema(schema, Arrays.asList("name", "custrecord79", "custrecord80"));
    properties.module.main.schema.setValue(targetSchema);
    Schema targetFlowSchema = dataSetRuntime.getSchemaForUpdateFlow(properties.module.moduleName.getValue(), targetSchema);
    properties.module.flowSchema.schema.setValue(targetFlowSchema);
    Schema targetRejectSchema = dataSetRuntime.getSchemaForUpdateReject(properties.module.moduleName.getValue(), targetSchema);
    properties.module.rejectSchema.schema.setValue(targetRejectSchema);
    GenericRecord indexedRecordToAdd = new GenericData.Record(targetSchema);
    String testId = Long.toString(System.currentTimeMillis());
    indexedRecordToAdd.put("Name", "Test Project " + testId);
    indexedRecordToAdd.put("Custrecord79", "Test Project " + testId);
    indexedRecordToAdd.put("Custrecord80", "0.1.0");
    List<IndexedRecord> indexedRecordList = new ArrayList<>();
    indexedRecordList.add(indexedRecordToAdd);
    // Add records
    NetSuiteSink sink = new NetSuiteSinkImpl();
    sink.setClientFactory(clientFactory);
    sink.initialize(container, properties);
    NetSuiteWriteOperation writeOperation = (NetSuiteWriteOperation) sink.createWriteOperation();
    NetSuiteOutputWriter<?, CustomRecordRef> writer = (NetSuiteOutputWriter) writeOperation.createWriter(container);
    writer.open(UUID.randomUUID().toString());
    for (IndexedRecord indexedRecord : indexedRecordList) {
        writer.write(indexedRecord);
    }
    Result writerResult = writer.close();
    assertNotNull(writerResult);
    assertEquals(indexedRecordList.size(), writerResult.totalCount);
    assertEquals(indexedRecordList.size(), writerResult.successCount);
    final List<CustomRecordRef> refList = new ArrayList<>(indexedRecordList.size());
    for (NsWriteResponse<CustomRecordRef> response : writer.getWriteResponses()) {
        CustomRecordRef recordRef = response.getRef();
        refList.add(recordRef);
    }
    // Re-read updated records
    List<NsReadResponse<CustomRecord>> readResponseList = clientService.execute(new NetSuiteClientService.PortOperation<List<NsReadResponse<CustomRecord>>, NetSuitePortType>() {

        @Override
        public List<NsReadResponse<CustomRecord>> execute(NetSuitePortType port) throws Exception {
            GetListRequest request = new GetListRequest();
            request.getBaseRef().addAll(refList);
            return NetSuiteClientServiceImpl.toNsReadResponseList(port.getList(request).getReadResponseList());
        }
    });
    int index = 0;
    for (NsReadResponse<CustomRecord> readResponse : readResponseList) {
        assertTrue(readResponse.getStatus().isSuccess());
        GenericRecord inputRecord = (GenericRecord) indexedRecordList.get(index);
        CustomRecord record = readResponse.getRecord();
        CustomFieldList customFieldList = record.getCustomFieldList();
        assertNotNull(customFieldList);
        Map<String, CustomFieldRef> customFieldRefMap = new HashMap<>();
        for (CustomFieldRef fieldRef : customFieldList.getCustomField()) {
            customFieldRefMap.put(fieldRef.getScriptId(), fieldRef);
        }
        StringCustomFieldRef customFieldRef1 = (StringCustomFieldRef) customFieldRefMap.get("custrecord79");
        assertNotNull(customFieldRef1);
        assertEquals(inputRecord.get("Custrecord79"), customFieldRef1.getValue());
        StringCustomFieldRef customFieldRef2 = (StringCustomFieldRef) customFieldRefMap.get("custrecord80");
        assertNotNull(customFieldRef2);
        assertEquals(inputRecord.get("Custrecord80"), customFieldRef2.getValue());
    }
    clientService.deleteList(refList);
}
Also used : NetSuiteSinkImpl(org.talend.components.netsuite.v2016_2.NetSuiteSinkImpl) IndexedRecord(org.apache.avro.generic.IndexedRecord) HashMap(java.util.HashMap) Schema(org.apache.avro.Schema) ArrayList(java.util.ArrayList) NetSuiteDatasetRuntime(org.talend.components.netsuite.NetSuiteDatasetRuntime) NetSuiteWriteOperation(org.talend.components.netsuite.output.NetSuiteWriteOperation) CustomRecordRef(com.netsuite.webservices.v2016_2.platform.core.CustomRecordRef) CustomFieldList(com.netsuite.webservices.v2016_2.platform.core.CustomFieldList) NsReadResponse(org.talend.components.netsuite.client.NsReadResponse) StringCustomFieldRef(com.netsuite.webservices.v2016_2.platform.core.StringCustomFieldRef) NetSuiteRuntimeImpl(org.talend.components.netsuite.v2016_2.NetSuiteRuntimeImpl) Result(org.talend.components.api.component.runtime.Result) NsSearchResult(org.talend.components.netsuite.client.NsSearchResult) CustomRecord(com.netsuite.webservices.v2016_2.setup.customization.CustomRecord) IndexedRecord(org.apache.avro.generic.IndexedRecord) GenericRecord(org.apache.avro.generic.GenericRecord) CustomFieldList(com.netsuite.webservices.v2016_2.platform.core.CustomFieldList) List(java.util.List) ArrayList(java.util.ArrayList) RecordRefList(com.netsuite.webservices.v2016_2.platform.core.RecordRefList) CustomRecord(com.netsuite.webservices.v2016_2.setup.customization.CustomRecord) GenericRecord(org.apache.avro.generic.GenericRecord) NetSuiteOutputWriter(org.talend.components.netsuite.output.NetSuiteOutputWriter) NetSuiteClientService(org.talend.components.netsuite.client.NetSuiteClientService) NetSuiteOutputProperties(org.talend.components.netsuite.output.NetSuiteOutputProperties) NetSuiteException(org.talend.components.netsuite.client.NetSuiteException) NetSuitePortType(com.netsuite.webservices.v2016_2.platform.NetSuitePortType) CustomFieldRef(com.netsuite.webservices.v2016_2.platform.core.CustomFieldRef) StringCustomFieldRef(com.netsuite.webservices.v2016_2.platform.core.StringCustomFieldRef) BooleanCustomFieldRef(com.netsuite.webservices.v2016_2.platform.core.BooleanCustomFieldRef) NetSuiteSink(org.talend.components.netsuite.NetSuiteSink) GetListRequest(com.netsuite.webservices.v2016_2.platform.messages.GetListRequest) RuntimeContainer(org.talend.components.api.container.RuntimeContainer) Test(org.junit.Test)

Aggregations

NetSuitePortType (com.netsuite.webservices.v2016_2.platform.NetSuitePortType)1 BooleanCustomFieldRef (com.netsuite.webservices.v2016_2.platform.core.BooleanCustomFieldRef)1 CustomFieldList (com.netsuite.webservices.v2016_2.platform.core.CustomFieldList)1 CustomFieldRef (com.netsuite.webservices.v2016_2.platform.core.CustomFieldRef)1 CustomRecordRef (com.netsuite.webservices.v2016_2.platform.core.CustomRecordRef)1 RecordRefList (com.netsuite.webservices.v2016_2.platform.core.RecordRefList)1 StringCustomFieldRef (com.netsuite.webservices.v2016_2.platform.core.StringCustomFieldRef)1 GetListRequest (com.netsuite.webservices.v2016_2.platform.messages.GetListRequest)1 CustomRecord (com.netsuite.webservices.v2016_2.setup.customization.CustomRecord)1 ArrayList (java.util.ArrayList)1 HashMap (java.util.HashMap)1 List (java.util.List)1 Schema (org.apache.avro.Schema)1 GenericRecord (org.apache.avro.generic.GenericRecord)1 IndexedRecord (org.apache.avro.generic.IndexedRecord)1 Test (org.junit.Test)1 Result (org.talend.components.api.component.runtime.Result)1 RuntimeContainer (org.talend.components.api.container.RuntimeContainer)1 NetSuiteDatasetRuntime (org.talend.components.netsuite.NetSuiteDatasetRuntime)1 NetSuiteSink (org.talend.components.netsuite.NetSuiteSink)1