#534 shutdown

cqfn · Aug 31, 2023 · 1a82b30 · 1a82b30
1 parent 19c2214
commit 1a82b30
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 6 deletions.
diff --git a/src/main/java/org/jpeek/web/Futures.java b/src/main/java/org/jpeek/web/Futures.java
@@ -33,6 +33,7 @@
 import java.util.concurrent.ExecutorService;
 import java.util.concurrent.Executors;
 import java.util.concurrent.Future;
+import java.util.concurrent.TimeUnit;
 import org.cactoos.BiFunc;
 import org.cactoos.Func;
 import org.cactoos.Text;
@@ -170,4 +171,14 @@ public String asString() throws Exception {
             ).asString()
         );
     }
+
+    /**
+     * Shut it down.
+     * @return TRUE if terminated OK
+     * @throws InterruptedException If interrupted while waiting
+     */
+    public boolean shutdown() throws InterruptedException {
+        this.service.shutdown();
+        return this.service.awaitTermination(1L, TimeUnit.SECONDS);
+    }
 }
diff --git a/src/test/java/org/jpeek/web/TkReportTest.java b/src/test/java/org/jpeek/web/TkReportTest.java
@@ -45,17 +45,16 @@ public final class TkReportTest {
     @Test
     @DisabledOnOs(OS.WINDOWS)
     public void rendersEmptySvgBadge(@TempDir final File folder)
-        throws IOException {
+        throws IOException, InterruptedException {
+        final Futures futures = new Futures(
+            new Reports(folder.toPath())
+        );
         new Assertion<>(
             "Must render the badge",
             XhtmlMatchers.xhtml(
                 new RsPrint(
                     new TkReport(
-                        new AsyncReports(
-                            new Futures(
-                                new Reports(folder.toPath())
-                            )
-                        ),
+                        new AsyncReports(futures),
                         new Results()
                     ).act(
                         new RqRegex.Fake(
@@ -68,6 +67,7 @@ public void rendersEmptySvgBadge(@TempDir final File folder)
             ),
             XhtmlMatchers.hasXPath("//svg:svg")
         ).affirm();
+        futures.shutdown();
     }
 
 }