Add unit tests for FileMap. (#471)

Author: blackwinter

.editorconfig

@@ -34,7 +34,7 @@ indent_size = 2
 [metafacture-io/src/test/resources/org/metafacture/io/compressed.txt]
 insert_final_newline = false
 
-[metamorph/src/test/resources/org/metafacture/metamorph/maps/file-map-test.txt]
+[metamorph/src/test/resources/org/metafacture/metamorph/maps/file-map-test*.txt]
 trim_trailing_whitespace = false
 
 [metafacture-runner/src/main/dist/config/java-options.conf]

metamorph/src/test/java/org/metafacture/metamorph/maps/FileMapTest.java

@@ -18,13 +18,16 @@
 
 import static org.metafacture.metamorph.TestHelpers.assertMorph;
 
+import org.junit.Assert;
 import org.junit.Rule;
 import org.junit.Test;
 import org.metafacture.framework.StreamReceiver;
 import org.mockito.Mock;
 import org.mockito.junit.MockitoJUnit;
 import org.mockito.junit.MockitoRule;
 
+import java.util.function.Consumer;
+
 /**
  * Tests for class {@link FileMap}.
  *
@@ -39,14 +42,16 @@ public final class FileMapTest {
     @Mock
     private StreamReceiver receiver;
 
+    private static String MAPS = "org/metafacture/metamorph/maps/";
+
     private static String MORPH =
         "<rules>" +
         "  <data source='1'>" +
         "    <%s='map1' />" +
         "  </data>" +
         "</rules>" +
         "<maps>" +
-        "  <filemap name='map1' files='org/metafacture/metamorph/maps/%s' %s/>" +
+        "  <filemap name='map1' files='" + MAPS + "%s' %s/>" +
         "</maps>";
 
     @Test
@@ -206,6 +211,97 @@ public void shouldLookupValuesInBlockedGzipFileMap() {
         );
     }
 
+    @Test
+    public void shouldLoadFile() {
+        assertMap(379, i -> {
+            Assert.assertEquals("Puerto Rico", i.get("pr"));
+            Assert.assertNull(i.get("zz"));
+        });
+    }
+
+    @Test
+    public void shouldLoadFileWithEmptyValues() {
+        assertMap(380, i -> {
+            i.setAllowEmptyValues(true);
+
+            Assert.assertEquals("Puerto Rico", i.get("pr"));
+            Assert.assertEquals("", i.get("zz"));
+        });
+    }
+
+    @Test
+    public void shouldLoadFileWithSeparator() {
+        assertMap(99, i -> {
+            i.setSeparator(" ");
+
+            Assert.assertNull(i.get("pp\tPapua"));
+            Assert.assertEquals("Rico", i.get("pr\tPuerto"));
+        });
+    }
+
+    @Test
+    public void shouldLoadFileWithKeyColumn() {
+        assertMap(21, i -> {
+            i.setSeparator(" ");
+            i.setKeyColumn(2);
+
+            Assert.assertEquals("New", i.get("Guinea"));
+        });
+    }
+
+    @Test
+    public void shouldLoadFileWithValueColumn() {
+        assertMap(24, i -> {
+            i.setSeparator(" ");
+            i.setValueColumn(2);
+
+            Assert.assertEquals("Guinea", i.get("pp\tPapua"));
+        });
+    }
+
+    @Test
+    public void shouldLoadFileWithKeyAndValueColumn() {
+        assertMap(66, i -> {
+            i.setSeparator(" ");
+            i.setKeyColumn(1);
+            i.setValueColumn(0);
+
+            Assert.assertEquals("pr\tPuerto", i.get("Rico"));
+        });
+    }
+
+    @Test
+    public void shouldLoadFileWithTrailingColumns() {
+        assertMap(149, i -> {
+            i.setSeparator(" ");
+            i.setAllowTrailingColumns(true);
+
+            Assert.assertEquals("New", i.get("pp\tPapua"));
+        });
+    }
+
+    @Test
+    public void shouldNotLoadFileWithOutOfRangeKeyColumn() {
+        assertMap(0, i -> {
+            i.setKeyColumn(2);
+        });
+    }
+
+    @Test
+    public void shouldNotLoadFileWithOutOfRangeValueColumn() {
+        assertMap(0, i -> {
+            i.setValueColumn(2);
+        });
+    }
+
+    private void assertMap(final int size, final Consumer<FileMap> consumer) {
+        final FileMap fileMap = new FileMap();
+        fileMap.setFile(MAPS + "file-map-test-columns.txt");
+
+        consumer.accept(fileMap);
+        Assert.assertEquals(size, fileMap.keySet().size());
+    }
+
     private String buildMorph(final String data, final String options) {
         return buildMorph(data, "file-map-test.txt", options);
     }