Search in sources :

Example 46 with Result

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

the class AzureStorageQueueListReader method getReturnValues.

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

Example 47 with Result

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

the class AzureStorageReader method getReturnValues.

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

Example 48 with Result

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

the class CouchbaseWriterTest method testWriter.

@Test
public void testWriter() throws IOException {
    // Prepare objects.
    CouchbaseConnection connection = Mockito.mock(CouchbaseConnection.class);
    Mockito.when(sink.getConnection()).thenReturn(connection);
    Schema schema = SchemaBuilder.builder().record("record").fields().requiredString("idFieldName").endRecord();
    IndexedRecord record = new GenericRecordBuilder(schema).set("idFieldName", 1).build();
    // Calling real methods.
    writer.open("random");
    writer.write(record);
    Result result = writer.close();
    Assert.assertEquals(1, result.totalCount);
    // TODO: We need to ask about increasing success count and reject count.
    // Verifying mock calls.
    Mockito.verify(connection, Mockito.times(1)).upsert(Mockito.anyString(), Mockito.anyString());
    Mockito.verify(connection, Mockito.times(1)).increment();
    Mockito.verify(connection, Mockito.times(1)).decrement();
}
Also used : IndexedRecord(org.apache.avro.generic.IndexedRecord) Schema(org.apache.avro.Schema) GenericRecordBuilder(org.apache.avro.generic.GenericRecordBuilder) Result(org.talend.components.api.component.runtime.Result) Test(org.junit.Test)

Example 49 with Result

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

the class CouchbaseWriter method open.

@Override
public void open(String uId) throws IOException {
    if (opened) {
        LOGGER.debug("Writer is already opened");
        return;
    }
    connection = sink.getConnection();
    connection.increment();
    result = new Result(uId);
    opened = true;
}
Also used : Result(org.talend.components.api.component.runtime.Result)

Example 50 with Result

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

the class CouchbaseWriteOperationTest method testFinalize.

@Test
public void testFinalize() {
    Set<Result> resultList = new HashSet<>(2);
    Result result1 = new Result("result1", 2, 1, 1);
    resultList.add(result1);
    Result result2 = new Result("result2", 5, 3, 2);
    resultList.add(result2);
    Map<String, Object> resultMap = writeOperation.finalize(resultList, null);
    Assert.assertEquals(7, resultMap.get(ComponentDefinition.RETURN_TOTAL_RECORD_COUNT));
    Assert.assertEquals(4, resultMap.get(ComponentDefinition.RETURN_SUCCESS_RECORD_COUNT));
    Assert.assertEquals(3, resultMap.get(ComponentDefinition.RETURN_REJECT_RECORD_COUNT));
}
Also used : Result(org.talend.components.api.component.runtime.Result) HashSet(java.util.HashSet) 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