test: added wait for reader and writer

redis · Jul 11, 2023 · 7dec5c6 · 7dec5c6
1 parent d5ce9d3
commit 7dec5c6
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/plugins/riot/src/test/java/com/redis/riot/cli/AbstractTests.java b/plugins/riot/src/test/java/com/redis/riot/cli/AbstractTests.java
@@ -200,7 +200,14 @@ protected <T> JobExecution run(String name, int chunkSize, ItemReader<T> reader,
 			throws JobExecutionException {
 		SimpleStepBuilder<T, T> step = step(name, chunkSize, reader, writer);
 		Job job = jobBuilderFactory.get(name).start(step.build()).build();
-		return jobLauncher.run(job, new JobParameters());
+		JobExecution execution = jobLauncher.run(job, new JobParameters());
+		awaitClosed(reader);
+		awaitClosed(writer);
+		return execution;
+	}
+
+	private void awaitClosed(Object object) {
+
 	}
 
 	protected String id() {