Fix tests for FileWorkUnitCalculator

atlanhq · Sep 11, 2018 · c5f6a43 · c5f6a43
1 parent d6906fa
commit c5f6a43
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/marmaray/src/test/java/com/uber/marmaray/common/sources/file/TestFileWorkUnitCalculator.java b/marmaray/src/test/java/com/uber/marmaray/common/sources/file/TestFileWorkUnitCalculator.java
@@ -38,6 +38,7 @@ public class TestFileWorkUnitCalculator {
     public void computeWorkUnitsNoJson() throws Exception{
         final Configuration conf = new Configuration();
         conf.setProperty(FileSourceConfiguration.TYPE, "json");
+        conf.setProperty(FileSourceConfiguration.SCHEMA, "{}");
         final Path testDir = Files.createTempDirectory(null);
         try {
             conf.setProperty(FileSourceConfiguration.DIRECTORY, testDir.toString());
@@ -54,6 +55,7 @@ public void computeWorkUnitsNoJson() throws Exception{
     public void computeWorkUnitsNoSuchDirectory() {
         final Configuration conf = new Configuration();
         conf.setProperty(FileSourceConfiguration.TYPE, "json");
+        conf.setProperty(FileSourceConfiguration.SCHEMA, "{}");
         conf.setProperty(FileSourceConfiguration.DIRECTORY, "path/not/exist");
         final FileWorkUnitCalculator workUnitCalculator = new FileWorkUnitCalculator(new FileSourceConfiguration(conf));
         final FileWorkUnitCalculator.FileWorkUnitCalculatorResult result = workUnitCalculator.computeWorkUnits();
@@ -68,6 +70,7 @@ public void computeWorkUnitsSuccess() throws Exception {
             createFile(testDir, "file3.csv");
             final Configuration conf = new Configuration();
             conf.setProperty(FileSourceConfiguration.TYPE, "json");
+            conf.setProperty(FileSourceConfiguration.SCHEMA, "{}");
             conf.setProperty(FileSourceConfiguration.DIRECTORY, testDir.toString());
             final FileWorkUnitCalculator workUnitCalculator = new FileWorkUnitCalculator(new FileSourceConfiguration(conf));
             final FileWorkUnitCalculator.FileWorkUnitCalculatorResult result = workUnitCalculator.computeWorkUnits();