Search in sources :

Example 36 with Result

use of org.talend.components.api.component.runtime.Result in project components by Talend.

the class MarketoBulkExecReader method getReturnValues.

@Override
public Map<String, Object> getReturnValues() {
    Result result = new Result();
    result.totalCount = apiCalls;
    Map<String, Object> res = result.toMap();
    res.put(RETURN_NB_CALL, apiCalls);
    res.put(RETURN_ERROR_MESSAGE, errorMessage);
    return res;
}
Also used : MarketoRecordResult(org.talend.components.marketo.runtime.client.type.MarketoRecordResult) Result(org.talend.components.api.component.runtime.Result)

Example 37 with Result

use of org.talend.components.api.component.runtime.Result in project components by Talend.

the class MarketoListOperationWriterTestIT method testRemoveFromREST.

@Test
public void testRemoveFromREST() throws Exception {
    props = getRESTProperties();
    props.listOperation.setValue(ListOperation.removeFrom);
    // 
    Schema s = props.schemaInput.schema.getValue();
    IndexedRecord record = new GenericData.Record(s);
    record.put(0, UNDX_TEST_LIST_SMALL_ID);
    record.put(1, createdLeads.get(4));
    // 
    writer = getWriter(props);
    writer.open("test");
    writer.write(record);
    writer.close();
    assertEquals(1, writer.result.getTotalCount());
    assertEquals(1, writer.result.getSuccessCount());
    assertEquals(0, writer.result.getRejectCount());
    // 
    props.listOperation.setValue(ListOperation.isMemberOf);
    writer = getWriter(props);
    writer.open("test");
    writer.write(record);
    for (IndexedRecord success : writer.getSuccessfulWrites()) {
        assertEquals("notmemberof", success.get(props.schemaFlow.schema.getValue().getField(FIELD_STATUS).pos()));
    }
    Result result = writer.close();
    LOG.debug("resultMap = {}.", result);
    assertEquals(1, result.getTotalCount());
    assertEquals(1, result.getSuccessCount());
    assertEquals(0, result.getRejectCount());
    assertEquals(1, result.toMap().get(MarketoComponentDefinition.RETURN_NB_CALL));
}
Also used : IndexedRecord(org.apache.avro.generic.IndexedRecord) Schema(org.apache.avro.Schema) IndexedRecord(org.apache.avro.generic.IndexedRecord) Result(org.talend.components.api.component.runtime.Result) Test(org.junit.Test)

Example 38 with Result

use of org.talend.components.api.component.runtime.Result in project components by Talend.

the class TFilterRowTest method testTFilterRowRuntime.

@Test
public void testTFilterRowRuntime() throws Exception {
    TFilterRowProperties props = (TFilterRowProperties) getComponentService().getComponentProperties("tFilterRow_POC");
    props.schemaListener.afterSchema();
    assertThat(props.conditionsTable.isEditable(), is(false));
    Schema schema = SchemaBuilder.builder().record("testRecord").fields().name("field1").type().stringType().noDefault().endRecord();
    props.schemaMain.schema.setValue(schema);
    final List<Schema.Field> additionalRejectFields = new ArrayList<Schema.Field>();
    Schema.Field field = null;
    field = new Schema.Field(TFilterRowProperties.FIELD_ERROR_MESSAGE, Schema.create(Schema.Type.STRING), null, (Object) null);
    field.addProp(SchemaConstants.TALEND_IS_LOCKED, "false");
    field.addProp(SchemaConstants.TALEND_FIELD_GENERATED, "true");
    field.addProp(SchemaConstants.TALEND_COLUMN_DB_LENGTH, "255");
    additionalRejectFields.add(field);
    Schema rejectSchema = newSchema(schema, "rejectOutput", additionalRejectFields);
    props.schemaListener.afterSchema();
    assertThat(props.schemaReject.schema.getValue(), is((Schema) rejectSchema));
    assertThat(props.schemaFlow.schema.getValue(), is((Schema) schema));
    assertThat(props.conditionsTable.isEditable(), is(true));
    props.conditionsTable.columnName.setValue(Arrays.asList("field1"));
    props.conditionsTable.function.setValue(Arrays.asList(FunctionType.EMPTY));
    props.conditionsTable.operator.setValue(Arrays.asList(OperatorType.EQUALS));
    props.conditionsTable.value.setValue(Arrays.asList((Object) "test"));
    props.logicalOperator.setValue(LogicalOperator.Or);
    Sink sink = new TFilterRowSink();
    sink.initialize(null, props);
    WriteOperation<?> writeOperation = sink.createWriteOperation();
    writeOperation.initialize(null);
    WriterWithFeedback<?, ?, ?> writer = (WriterWithFeedback<?, ?, ?>) writeOperation.createWriter(null);
    writer.open("uId");
    IndexedRecord record = new GenericData.Record(schema);
    record.put(schema.getField("field1").pos(), "test");
    writer.write(record);
    Iterable<?> success = writer.getSuccessfulWrites();
    Iterator<?> successIterator = success.iterator();
    assertThat(successIterator.hasNext(), is(true));
    IndexedRecord successWrite = (IndexedRecord) successIterator.next();
    assertThat(successWrite.getSchema(), is(schema));
    assertThat(successWrite, is(record));
    writer.cleanWrites();
    record = new GenericData.Record(schema);
    record.put(schema.getField("field1").pos(), "test123");
    writer.write(record);
    success = writer.getSuccessfulWrites();
    successIterator = success.iterator();
    assertThat(successIterator.hasNext(), is(false));
    Iterable<?> reject = writer.getRejectedWrites();
    Iterator<?> rejectIterator = reject.iterator();
    assertThat(rejectIterator.hasNext(), is(true));
    IndexedRecord rejectWrite = (IndexedRecord) rejectIterator.next();
    assertThat(rejectWrite.getSchema(), is(rejectSchema));
    Result result = (Result) writer.close();
    assertThat(result.totalCount, is(2));
    assertThat(result.successCount, is(1));
    assertThat(result.rejectCount, is(1));
}
Also used : IndexedRecord(org.apache.avro.generic.IndexedRecord) Schema(org.apache.avro.Schema) ArrayList(java.util.ArrayList) GenericData(org.apache.avro.generic.GenericData) Result(org.talend.components.api.component.runtime.Result) TFilterRowSink(org.talend.components.filterrow.runtime.TFilterRowSink) Sink(org.talend.components.api.component.runtime.Sink) IndexedRecord(org.apache.avro.generic.IndexedRecord) TFilterRowSink(org.talend.components.filterrow.runtime.TFilterRowSink) WriterWithFeedback(org.talend.components.api.component.runtime.WriterWithFeedback) Test(org.junit.Test)

Example 39 with Result

use of org.talend.components.api.component.runtime.Result in project components by Talend.

the class NetSuiteSearchInputReader method getReturnValues.

@Override
public Map<String, Object> getReturnValues() {
    Result result = new Result();
    result.totalCount = dataCount;
    return result.toMap();
}
Also used : Result(org.talend.components.api.component.runtime.Result)

Example 40 with Result

use of org.talend.components.api.component.runtime.Result in project components by Talend.

the class MarkLogicWriterTest method testClose.

@Test
public void testClose() throws IOException {
    prepareDocManagerText(MarkLogicOutputProperties.Action.DELETE, MarkLogicOutputProperties.DocType.MIXED);
    Result result = writer.close();
    assertEquals(1, result.totalCount);
    verify(writer.container).setComponentData(anyString(), eq("NB_LINE_DELETED"), eq(1));
}
Also used : Result(org.talend.components.api.component.runtime.Result) Test(org.junit.Test)

Aggregations

Result (org.talend.components.api.component.runtime.Result)76 Test (org.junit.Test)42 IndexedRecord (org.apache.avro.generic.IndexedRecord)31 ArrayList (java.util.ArrayList)22 Schema (org.apache.avro.Schema)17 TSalesforceOutputProperties (org.talend.components.salesforce.tsalesforceoutput.TSalesforceOutputProperties)13 ComponentDefinition (org.talend.components.api.component.ComponentDefinition)11 TSalesforceOutputDefinition (org.talend.components.salesforce.tsalesforceoutput.TSalesforceOutputDefinition)11 IOException (java.io.IOException)10 DefaultComponentRuntimeContainerImpl (org.talend.components.api.container.DefaultComponentRuntimeContainerImpl)8 NetSuiteDatasetRuntime (org.talend.components.netsuite.NetSuiteDatasetRuntime)8 NetSuiteSink (org.talend.components.netsuite.NetSuiteSink)8 NetSuiteOutputWriter (org.talend.components.netsuite.output.NetSuiteOutputWriter)8 NetSuiteWriteOperation (org.talend.components.netsuite.output.NetSuiteWriteOperation)8 ValidationResult (org.talend.daikon.properties.ValidationResult)7 NetSuitePortType (com.netsuite.webservices.v2016_2.platform.NetSuitePortType)6 SQLException (java.sql.SQLException)6 List (java.util.List)6 NetSuiteRuntimeImpl (org.talend.components.netsuite.v2016_2.NetSuiteRuntimeImpl)6 NetSuiteSinkImpl (org.talend.components.netsuite.v2016_2.NetSuiteSinkImpl)6