Add count and index options to MetafixListPaths

See https://gitlab.com/oersi/oersi-etl/-/issues/238

Author: fsteeg

metafix/src/main/java/org/metafacture/metafix/MetafixListPaths.java

@@ -35,12 +35,13 @@
 public class MetafixListPaths extends DefaultStreamPipe<ObjectReceiver<String>> {
 
     private Metafix fix;
+    private boolean count = true;
+    private boolean index;
 
     public MetafixListPaths() {
         try {
             fix = new Metafix("nothing()");
             fix.setRepeatedFieldsToEntities(true);
-            fix.setEntityMemberName("*");
         }
         catch (final FileNotFoundException e) {
             e.printStackTrace();
@@ -51,11 +52,14 @@ public MetafixListPaths() {
     protected void onSetReceiver() {
         final TripleCount tripleCount = new TripleCount();
         tripleCount.setCountBy(Compare.PREDICATE);
+        if (!index) {
+            fix.setEntityMemberName("*");
+        }
         fix
             .setReceiver(new StreamFlattener())
             .setReceiver(new StreamToTriples())
             .setReceiver(tripleCount)
-            .setReceiver(new ObjectTemplate<Triple>("${s}\t ${o}"))
+            .setReceiver(new ObjectTemplate<Triple>(count ? "${s}\t ${o}" : "${s}"))
             .setReceiver(getReceiver());
     }
 
@@ -93,4 +97,20 @@ protected void onCloseStream() {
     protected void onResetStream() {
         fix.resetStream();
     }
+
+    public void setCount(final boolean count) {
+        this.count = count;
+    }
+
+    public boolean getCount() {
+        return this.count;
+    }
+
+    public void setIndex(final boolean index) {
+        this.index = index;
+    }
+
+    public boolean getIndex() {
+        return this.index;
+    }
 }

metafix/src/test/java/org/metafacture/metafix/MetafixListPathsTest.java

@@ -40,19 +40,53 @@ public final class MetafixListPathsTest {
     public MetafixListPathsTest() {
         MockitoAnnotations.initMocks(this);
         lister = new MetafixListPaths();
-        lister.setReceiver(receiver);
     }
 
     @Test
     public void testShouldListPaths() {
+        processRecord();
+        verify("a.*\t 3");
+    }
+
+    @Test
+    public void testShouldListPathsNoCount() {
+        lister.setCount(false);
+        processRecord();
+        verify("a.*");
+    }
+
+    @Test
+    public void testShouldListPathsUseIndex() {
+        lister.setIndex(true);
+        processRecord();
+        verify("a.1\t 1");
+        verify("a.2\t 1");
+        verify("a.3\t 1");
+    }
+
+    @Test
+    public void testShouldListPathsNoCountUseIndex() {
+        lister.setCount(false);
+        lister.setIndex(true);
+        processRecord();
+        verify("a.1");
+        verify("a.2");
+        verify("a.3");
+    }
+
+    private void processRecord() {
+        lister.setReceiver(receiver);
         lister.startRecord("");
         lister.literal("a", "A");
         lister.literal("a", "B");
         lister.literal("a", "C");
         lister.endRecord();
         lister.closeStream();
+    }
+
+    private void verify(final String result) throws MockitoAssertionError {
         try {
-            Mockito.verify(receiver).process("a.*\t 3");
+            Mockito.verify(receiver).process(result);
         }
         catch (final MockitoAssertionError e) {
             System.out.println(Mockito.mockingDetails(receiver).printInvocations());