Merge pull request #61 from profeg/OLogTransformerTest

O log transformer test

src/main/java/com/orientechnologies/orient/etl/transformer/OLogTransformer.java

@@ -66,7 +66,7 @@ public Object executeTransform(final Object input) {
     if (postfix != null && !postfix.isEmpty())
       buffer.append(resolve(postfix));
 
-    out.println(buffer.toString()); //TODO    log(OETLProcessor.LOG_LEVELS.INFO, buffer.toString());
+      log(OETLProcessor.LOG_LEVELS.INFO, buffer.toString());
 
     return input;
   }

src/test/java/com/orientechnologies/orient/etl/transformer/OLogTransformerTest.java

@@ -0,0 +1,49 @@
+package com.orientechnologies.orient.etl.transformer;
+
+import com.orientechnologies.orient.core.record.impl.ODocument;
+import com.orientechnologies.orient.etl.ETLBaseTest;
+import org.junit.After;
+import org.junit.Before;
+import org.junit.Test;
+
+import java.io.ByteArrayOutputStream;
+import java.io.PrintStream;
+import java.util.List;
+
+import static org.junit.Assert.*;
+import static org.junit.Assert.assertEquals;
+
+public class OLogTransformerTest extends ETLBaseTest {
+
+    private final PrintStream OUT = System.out;
+
+    @Test
+    public void testPrefix() throws Exception {
+        ByteArrayOutputStream output = getByteArrayOutputStream();
+        String cfgJson = "{source: { content: { value: 'id,text\n1,Hello\n2,Bye'} }, extractor : { row : {} }, transformers : [{ csv : {} },{ log : {prefix:'-> '}}], loader : { test: {} } }";
+        process(cfgJson);
+        List<ODocument> res = getResult();
+        ODocument doc = res.get(0);
+        String[] stringList = output.toString().split("\n");
+        assertEquals("[1:log] INFO -> {id:1,text:Hello}", stringList[1]);
+        assertEquals("[2:log] INFO -> {id:2,text:Bye}", stringList[2]);
+    }
+    @Test
+    public void testPostfix() throws Exception {
+        ByteArrayOutputStream output = getByteArrayOutputStream();
+        String cfgJson = "{source: { content: { value: 'id,text\n1,Hello\n2,Bye'} }, extractor : { row : {} }, transformers : [{ csv : {} },{ log : {postfix:'-> '}}], loader : { test: {} } }";
+        process(cfgJson);
+        List<ODocument> res = getResult();
+        ODocument doc = res.get(0);
+        String[] stringList = output.toString().split("\n");
+        assertEquals("[1:log] INFO {id:1,text:Hello}-> ", stringList[1]);
+        assertEquals("[2:log] INFO {id:2,text:Bye}-> ", stringList[2]);
+    }
+
+    private ByteArrayOutputStream getByteArrayOutputStream() {
+        ByteArrayOutputStream output = new ByteArrayOutputStream();
+        System.setOut(new PrintStream(output, true));
+        return output;
+    }
+
+}