Test ControllableParallelExecutionHandler rethrows an exception occurred at createReader

Author: Taichi Yamakawa

src/test/java/lerna/nablarch/batch/parallelizable/handler/ControllableParallelExecutionHandlerTest.java

@@ -324,6 +324,31 @@ public Result handle(String s, ExecutionContext executionContext) {
         assertThat(handledCount.get(), is(0));
     }
 
+    @Test
+    public void createReaderで例外が発生したら親ハンドラにその例外がリスローされる() {
+        final int parallelism = 5;
+        final ControllableParallelExecutionHandler handler = new ControllableParallelExecutionHandler();
+        handler.setParallelism(parallelism);
+        handler.setConnectionFactory(connectionName -> new TestTransactionManagerConnectionBase());
+        handler.setTransactionFactory(connectionName -> new TestTransactionBase());
+
+        final IllegalStateException expectedException =
+                new IllegalStateException("Could not create a DataReader.");
+
+        final ExecutionContext context = new ExecutionContext();
+        context.addHandler(handler);
+        context.addHandler(new TestControllableParallelExecutorBase() {
+            @Override
+            public DataReader<String> createReader(ExecutionContext executionContext) {
+                throw expectedException;
+            }
+        });
+
+        final IllegalStateException thrownException =
+                assertThrows(IllegalStateException.class, () -> context.handleNext("input"));
+        assertThat(thrownException, is(expectedException));
+    }
+
     @Test
     public void executionId毎にcommitIntervalを迎えたときと終了時に1度コミットされる() {
         int numberOfData = 100;