Examples with MarkLogicConnectionProperties org.talend.components.marklogic.tmarklogicconnection.MarkLogicConnectionProperties used on opensource projects

Example 6 with MarkLogicConnectionProperties

use of org.talend.components.marklogic.tmarklogicconnection.MarkLogicConnectionProperties in project components by Talend.

the class MarkLogicDatasetPropertiesTest method setup.

@Before
public void setup() {
    properties = new MarkLogicDatasetProperties("dataset");
    datastore = new MarkLogicConnectionProperties("datastore");
    properties.setDatastoreProperties(datastore);
}

Example 7 with MarkLogicConnectionProperties

use of org.talend.components.marklogic.tmarklogicconnection.MarkLogicConnectionProperties in project components by Talend.

the class MarkLogicSourceOrSinkTest method testValidateWithReferenceConnection.

@Test
public void testValidateWithReferenceConnection() {
    RuntimeContainer container = mock(RuntimeContainer.class);
    DatabaseClient mockedClient = mock(DatabaseClient.class);
    when(container.getComponentData(anyString(), eq(MarkLogicConnection.CONNECTION))).thenReturn(mockedClient);
    MarkLogicConnectionProperties connectionProperties = new MarkLogicConnectionProperties("connection");
    connectionProperties.init();
    MarkLogicInputProperties inputProperties = new MarkLogicInputProperties("inputProperties");
    inputProperties.connection.referencedComponent.setReference(connectionProperties);
    inputProperties.connection.referencedComponent.componentInstanceId.setValue(MarkLogicConnectionDefinition.COMPONENT_NAME + "_1");
    inputProperties.init();
    sourceOrSink.initialize(container, inputProperties);
    ValidationResult vr = sourceOrSink.validate(container);
    assertEquals(ValidationResult.Result.OK, vr.getStatus());
}

Example 8 with MarkLogicConnectionProperties

use of org.talend.components.marklogic.tmarklogicconnection.MarkLogicConnectionProperties in project components by Talend.

the class TMarkLogicConnectionStandaloneTest method setup.

@Before
public void setup() {
    connectionStandalone = new TMarkLogicConnectionStandalone();
    PowerMockito.mockStatic(DatabaseClientFactory.class);
    client = Mockito.mock(DatabaseClient.class);
    Mockito.when(DatabaseClientFactory.newClient(Mockito.anyString(), Mockito.anyInt(), Mockito.anyString(), Mockito.any(SecurityContext.class))).thenReturn(client);
    container = Mockito.mock(RuntimeContainer.class);
    connectionProperties = new MarkLogicConnectionProperties("connection");
    connectionProperties.authentication.setValue("BASIC");
    Mockito.when(container.getComponentData(Mockito.anyString(), Mockito.eq(MarkLogicConnection.CONNECTION))).thenReturn(client);
}

Example 9 with MarkLogicConnectionProperties

use of org.talend.components.marklogic.tmarklogicconnection.MarkLogicConnectionProperties in project components by Talend.

the class MarkLogicExternalBulkLoadRunnerTest method setUp.

@Before
public void setUp() {
    bulkLoadRuntime = new MarkLogicBulkLoad();
    connectionProperties = new MarkLogicConnectionProperties("connectionProperties");
    bulkLoadProperties = new MarkLogicBulkLoadProperties("bulkLoadProperties");
}

Example 10 with MarkLogicConnectionProperties

use of org.talend.components.marklogic.tmarklogicconnection.MarkLogicConnectionProperties in project components by Talend.

the class MarkLogicDatasetRuntimeTest method testWorkflow.

@SuppressWarnings("unchecked")
@Test
public void testWorkflow() {
    RuntimeContainer container = Mockito.mock(RuntimeContainer.class);
    MarkLogicConnectionProperties connection = new MarkLogicConnectionProperties("datastore");
    connection.referencedComponent.componentInstanceId.setValue("reference");
    DatabaseClient connectionClient = Mockito.mock(DatabaseClient.class);
    GenericDocumentManager docManager = Mockito.mock(GenericDocumentManager.class);
    Mockito.when(connectionClient.newDocumentManager()).thenReturn(docManager);
    QueryManager queryManager = Mockito.mock(QueryManager.class);
    Mockito.when(connectionClient.newQueryManager()).thenReturn(queryManager);
    StringQueryDefinition stringQueryDefinition = Mockito.mock(StringQueryDefinition.class);
    Mockito.when(queryManager.newStringDefinition()).thenReturn(stringQueryDefinition);
    SearchHandle searchHandle = Mockito.mock(SearchHandle.class);
    Mockito.when(queryManager.search(Mockito.eq(stringQueryDefinition), Mockito.any(SearchHandle.class))).thenReturn(searchHandle);
    MatchDocumentSummary[] results = new MatchDocumentSummary[2];
    MatchDocumentSummary firstRecord = Mockito.mock(MatchDocumentSummary.class);
    MatchDocumentSummary secondRecord = Mockito.mock(MatchDocumentSummary.class);
    results[0] = firstRecord;
    results[1] = secondRecord;
    Mockito.when(container.getComponentData("reference", MarkLogicConnection.CONNECTION)).thenReturn(connectionClient);
    MarkLogicDatasetProperties properties = new MarkLogicDatasetProperties("dataset");
    connection.init();
    properties.setDatastoreProperties(connection);
    properties.init();
    datasetRuntime.initialize(container, properties);
    Assert.assertEquals(2, datasetRuntime.getSchema().getFields().size());
    Assert.assertEquals("docId", datasetRuntime.getSchema().getFields().get(0).name());
    Assert.assertEquals("docContent", datasetRuntime.getSchema().getFields().get(1).name());
    Mockito.when(searchHandle.getMatchResults()).thenReturn(results);
    Mockito.when(searchHandle.getTotalResults()).thenReturn(2l);
    Mockito.when(docManager.readAs(Mockito.anyString(), Mockito.any(Class.class))).thenReturn(new Object());
    Consumer<IndexedRecord> consumer = Mockito.mock(Consumer.class);
    datasetRuntime.getSample(2, consumer);
    Mockito.verify(consumer, Mockito.times(2)).accept(Mockito.any(IndexedRecord.class));
}