Fixed testFatalHandlerException in CrawlerTest

PreferredAI · Feb 26, 2019 · 9b19af7 · 9b19af7
1 parent 4d14599
commit 9b19af7
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/test/java/ai/preferred/venom/CrawlerTest.java b/src/test/java/ai/preferred/venom/CrawlerTest.java
@@ -276,12 +276,15 @@ public void testUrlRouterIntegration() throws Exception {
   @Test
   public void testFatalHandlerException() {
     Assertions.assertThrows(FatalHandlerException.class, () -> {
-      final List<FakeFetcher.Status> statuses = Arrays.asList(FakeFetcher.Status.COMPLETE, FakeFetcher.Status.COMPLETE,
+      final List<FakeFetcher.Status> statuses = Arrays.asList(
+          FakeFetcher.Status.COMPLETE,
+          FakeFetcher.Status.COMPLETE,
           FakeFetcher.Status.COMPLETE);
       final Fetcher fetcher = new FakeFetcher(new LinkedList<>(statuses));
       try (final Crawler crawler = Crawler.builder()
           .setFetcher(fetcher)
           .setMaxTries(1)
+          .setMaxConnections(1)
           .setScheduler(new FIFOScheduler())
           .setSleepScheduler(new SleepScheduler(0))
           .build()
@@ -292,11 +295,11 @@ public void testFatalHandlerException() {
         });
 
         crawler.getScheduler().add(vRequest, (request, response, scheduler, session, worker) -> {
-          throw new FatalHandlerException("FatalHandlerException #1");
+          throw new FatalHandlerException("FatalHandlerException");
         });
 
         crawler.getScheduler().add(vRequest, (request, response, scheduler, session, worker) -> {
-          throw new FatalHandlerException("FatalHandlerException #2");
+          // do nothing
         });
       }
     });