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;
}
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;
}
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();
}
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;
}
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));
}
Aggregations