Search in sources :

Example 1 with MarkLogicOutputProperties

use of org.talend.components.marklogic.tmarklogicoutput.MarkLogicOutputProperties in project components by Talend.

the class MarkLogicSourceOrSinkTest method testInitializeOKOutputProperties.

@Test
public void testInitializeOKOutputProperties() {
    MarkLogicOutputProperties outputProperties = new MarkLogicOutputProperties("outputProperties");
    outputProperties.init();
    ValidationResult vr = sourceOrSink.initialize(null, outputProperties);
    assertEquals(ValidationResult.Result.OK, vr.getStatus());
}
Also used : MarkLogicOutputProperties(org.talend.components.marklogic.tmarklogicoutput.MarkLogicOutputProperties) ValidationResult(org.talend.daikon.properties.ValidationResult) PrepareForTest(org.powermock.core.classloader.annotations.PrepareForTest) Test(org.junit.Test)

Example 2 with MarkLogicOutputProperties

use of org.talend.components.marklogic.tmarklogicoutput.MarkLogicOutputProperties in project components by Talend.

the class MarkLogicWriterTest method testInitDocManagerBinary.

@Test
public void testInitDocManagerBinary() throws IOException {
    MarkLogicSink sink = new MarkLogicSink();
    MarkLogicOutputProperties properties = new MarkLogicOutputProperties("outputProperties");
    properties.init();
    properties.connection.referencedComponent.componentInstanceId.setValue("Not null value");
    properties.docType.setValue(MarkLogicOutputProperties.DocType.BINARY);
    sink.ioProperties = properties;
    RuntimeContainer mockedContainer = mock(RuntimeContainer.class);
    DatabaseClient mockedClient = mock(DatabaseClient.class);
    Mockito.when(mockedContainer.getComponentData(anyString(), anyString())).thenReturn(mockedClient);
    writer = sink.createWriteOperation().createWriter(mockedContainer);
    writer.open("123");
    verify(mockedClient).newBinaryDocumentManager();
}
Also used : DatabaseClient(com.marklogic.client.DatabaseClient) MarkLogicOutputProperties(org.talend.components.marklogic.tmarklogicoutput.MarkLogicOutputProperties) RuntimeContainer(org.talend.components.api.container.RuntimeContainer) Test(org.junit.Test)

Example 3 with MarkLogicOutputProperties

use of org.talend.components.marklogic.tmarklogicoutput.MarkLogicOutputProperties in project components by Talend.

the class MarkLogicWriterTest method testWriteNotIndexedRecord.

@Test
public void testWriteNotIndexedRecord() throws IOException {
    DocumentManager markLogicDocMngrMock = mock(DocumentManager.class);
    MarkLogicSink someSink = new MarkLogicSink();
    someSink.ioProperties = new MarkLogicOutputProperties("outputProps");
    writer = someSink.createWriteOperation().createWriter(null);
    writer.docMgr = markLogicDocMngrMock;
    writer.write(new Object());
    verifyZeroInteractions(markLogicDocMngrMock);
}
Also used : MarkLogicOutputProperties(org.talend.components.marklogic.tmarklogicoutput.MarkLogicOutputProperties) DocumentManager(com.marklogic.client.document.DocumentManager) Matchers.anyObject(org.mockito.Matchers.anyObject) Test(org.junit.Test)

Example 4 with MarkLogicOutputProperties

use of org.talend.components.marklogic.tmarklogicoutput.MarkLogicOutputProperties in project components by Talend.

the class MarkLogicWriterTest method testInitDocManagerText.

@Test
public void testInitDocManagerText() throws IOException {
    MarkLogicSink sink = new MarkLogicSink();
    MarkLogicOutputProperties properties = new MarkLogicOutputProperties("outputProperties");
    properties.init();
    properties.connection.referencedComponent.componentInstanceId.setValue("Not null value");
    properties.docType.setValue(MarkLogicOutputProperties.DocType.PLAIN_TEXT);
    sink.ioProperties = properties;
    RuntimeContainer mockedContainer = mock(RuntimeContainer.class);
    DatabaseClient mockedClient = mock(DatabaseClient.class);
    Mockito.when(mockedContainer.getComponentData(anyString(), anyString())).thenReturn(mockedClient);
    writer = sink.createWriteOperation().createWriter(mockedContainer);
    writer.open("123");
    verify(mockedClient).newTextDocumentManager();
}
Also used : DatabaseClient(com.marklogic.client.DatabaseClient) MarkLogicOutputProperties(org.talend.components.marklogic.tmarklogicoutput.MarkLogicOutputProperties) RuntimeContainer(org.talend.components.api.container.RuntimeContainer) Test(org.junit.Test)

Example 5 with MarkLogicOutputProperties

use of org.talend.components.marklogic.tmarklogicoutput.MarkLogicOutputProperties in project components by Talend.

the class MarkLogicWriterTest method testGetWriteOperation.

@Test
public void testGetWriteOperation() {
    MarkLogicSink sink = mock(MarkLogicSink.class);
    MarkLogicOutputProperties properties = new MarkLogicOutputProperties("outputProperties");
    properties.init();
    MarkLogicWriteOperation writeOperation = new MarkLogicWriteOperation(sink, properties);
    Mockito.when(sink.connect((RuntimeContainer) anyObject())).thenReturn(null);
    writer = writeOperation.createWriter(null);
    assertEquals(writeOperation, writer.getWriteOperation());
}
Also used : MarkLogicOutputProperties(org.talend.components.marklogic.tmarklogicoutput.MarkLogicOutputProperties) Test(org.junit.Test)

Aggregations

MarkLogicOutputProperties (org.talend.components.marklogic.tmarklogicoutput.MarkLogicOutputProperties)16 Test (org.junit.Test)14 DatabaseClient (com.marklogic.client.DatabaseClient)7 RuntimeContainer (org.talend.components.api.container.RuntimeContainer)7 DocumentManager (com.marklogic.client.document.DocumentManager)4 ValidationResult (org.talend.daikon.properties.ValidationResult)3 GenericData (org.apache.avro.generic.GenericData)2 IndexedRecord (org.apache.avro.generic.IndexedRecord)2 DocumentDescriptor (com.marklogic.client.document.DocumentDescriptor)1 DocumentUriTemplate (com.marklogic.client.document.DocumentUriTemplate)1 FileHandle (com.marklogic.client.io.FileHandle)1 AbstractWriteHandle (com.marklogic.client.io.marker.AbstractWriteHandle)1 File (java.io.File)1 Before (org.junit.Before)1 Matchers.anyObject (org.mockito.Matchers.anyObject)1 PrepareForTest (org.powermock.core.classloader.annotations.PrepareForTest)1