Examples with Source org.talend.components.api.component.runtime.Source used on opensource projects

Example 1 with Source

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

the class SalesforceRuntimeTestUtil method initReader.

public Reader<?> initReader(TSalesforceBulkExecDefinition definition, String data_file, TSalesforceBulkExecProperties modelProperties, Schema schema, Schema output) {
    modelProperties.connection.userPassword.userId.setValue(username);
    modelProperties.connection.userPassword.password.setValue(password);
    modelProperties.connection.userPassword.securityKey.setValue(securityKey);
    modelProperties.connection.timeout.setValue(60000);
    modelProperties.connection.bulkConnection.setValue(true);
    modelProperties.bulkFilePath.setValue(data_file);
    modelProperties.bulkProperties.bytesToCommit.setValue(10 * 1024 * 1024);
    modelProperties.bulkProperties.rowsToCommit.setValue(10000);
    modelProperties.bulkProperties.concurrencyMode.setValue(Concurrency.Parallel);
    modelProperties.bulkProperties.waitTimeCheckBatchState.setValue(10000);
    modelProperties.module.moduleName.setValue(module);
    modelProperties.module.main.schema.setValue(schema);
    modelProperties.schemaFlow.schema.setValue(output);
    Source source = new SalesforceSource();
    source.initialize(null, modelProperties);
    ValidationResult vr = source.validate(null);
    if (vr.getStatus() == ValidationResult.Result.ERROR) {
        Assert.fail(vr.getMessage());
    }
    Reader<?> reader = source.createReader(null);
    return reader;
}

Example 2 with Source

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

the class SalesforceRuntimeTestUtil method initReader.

public Reader initReader(TSalesforceBulkExecDefinition definition, String data_file, TSalesforceBulkExecProperties modelProperties, Schema schema, Schema output) {
    modelProperties.connection.userPassword.userId.setValue(username);
    modelProperties.connection.userPassword.password.setValue(password);
    modelProperties.connection.userPassword.securityKey.setValue(securityKey);
    modelProperties.connection.timeout.setValue(60000);
    modelProperties.connection.bulkConnection.setValue(true);
    modelProperties.bulkFilePath.setValue(data_file);
    modelProperties.bulkProperties.bytesToCommit.setValue(10 * 1024 * 1024);
    modelProperties.bulkProperties.rowsToCommit.setValue(10000);
    modelProperties.bulkProperties.concurrencyMode.setValue(Concurrency.Parallel);
    modelProperties.bulkProperties.waitTimeCheckBatchState.setValue(10000);
    modelProperties.module.moduleName.setValue(module);
    modelProperties.module.main.schema.setValue(schema);
    modelProperties.schemaFlow.schema.setValue(output);
    Source source = new SalesforceSource();
    source.initialize(null, modelProperties);
    ValidationResult vr = source.validate(null);
    if (vr.getStatus() == ValidationResult.Result.ERROR) {
        Assert.fail(vr.getMessage());
    }
    Reader reader = source.createReader(null);
    return reader;
}

Example 3 with Source

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

the class AbstractMultiRuntimeComponentTests method testMultiRuntimeComponentRuntime.

@Test
public void testMultiRuntimeComponentRuntime() throws Exception {
    MultiRuntimeComponentDefinition def = (MultiRuntimeComponentDefinition) getComponentService().getComponentDefinition(MultiRuntimeComponentDefinition.COMPONENT_NAME);
    MultiRuntimeComponentProperties props = (MultiRuntimeComponentProperties) getComponentService().getComponentProperties(MultiRuntimeComponentDefinition.COMPONENT_NAME);
    props.version.setValue(Version.VERSION_0_1);
    try (SandboxedInstance sandboxedInstance = RuntimeUtil.createRuntimeClass(def.getRuntimeInfo(ExecutionEngine.DI, props, ConnectorTopology.OUTGOING), props.getClass().getClassLoader())) {
        Source source = (Source) sandboxedInstance.getInstance();
        source.initialize(null, props);
        assertEquals("Me", source.validate(null).getMessage());
    }
    props.version.setValue(Version.VERSION_0_2);
    try (SandboxedInstance sandboxedInstance = RuntimeUtil.createRuntimeClass(def.getRuntimeInfo(ExecutionEngine.DI, props, ConnectorTopology.OUTGOING), props.getClass().getClassLoader())) {
        Source source2 = (Source) sandboxedInstance.getInstance();
        source2.initialize(null, props);
        assertEquals("AnotherMe", source2.validate(null).getMessage());
    }
}

Example 4 with Source

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

the class TJiraInputDefinitionTest method testGetRuntime.

/**
 * Check {@link TJiraInputDefinition#getRuntime()} returns instance of {@link JiraSource}
 */
@Test
public void testGetRuntime() {
    TJiraInputDefinition definition = new TJiraInputDefinition();
    RuntimeInfo runtimeInfo = definition.getRuntimeInfo(ExecutionEngine.DI, null, ConnectorTopology.OUTGOING);
    SandboxedInstance sandboxedInstance = RuntimeUtil.createRuntimeClass(runtimeInfo, definition.getClass().getClassLoader());
    Source source = (Source) sandboxedInstance.getInstance();
    assertThat(source, is(instanceOf(JiraSource.class)));
}

Example 5 with Source

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

the class SalesforceProxyTestIT method testProxyWithBulkQuery.

@Test
public void testProxyWithBulkQuery() {
    TSalesforceInputDefinition definition = (TSalesforceInputDefinition) getComponentService().getComponentDefinition(TSalesforceInputDefinition.COMPONENT_NAME);
    TSalesforceInputProperties properties = (TSalesforceInputProperties) definition.createRuntimeProperties();
    properties.connection.bulkConnection.setValue(true);
    properties.queryMode.setValue(QueryMode.Bulk);
    SalesforceRuntimeTestUtil util = new SalesforceRuntimeTestUtil();
    properties.module.moduleName.setValue(util.getTestModuleName());
    properties.module.main.schema.setValue(util.getTestSchema1());
    properties.connection.userPassword.userId.setValue(userId);
    properties.connection.userPassword.password.setValue(password);
    properties.connection.userPassword.securityKey.setValue(securityKey);
    Source source = new SalesforceSource();
    source.initialize(null, properties);
    org.talend.daikon.properties.ValidationResult vr = source.validate(null);
    Assert.assertEquals(ValidationResult.Result.OK, vr.getStatus());
    Reader reader = source.createReader(null);
    try {
        reader.start();
        do {
            reader.getCurrent();
        } while (reader.advance());
        reader.close();
    } catch (Exception e) {
        Assert.fail(e.getMessage());
    }
}