travis build: fix infinite loop

jknack · jknack · commit 25991d52e678 · 2018-06-25T19:31:10.000-03:00
diff --git a/jooby/src/test/java/org/jooby/AssetForwardingTest.java b/jooby/src/test/java/org/jooby/AssetForwardingTest.java
@@ -1,16 +1,14 @@
 package org.jooby;
 
 import static org.easymock.EasyMock.expect;
+import org.jooby.test.MockUnit;
 import static org.junit.Assert.assertEquals;
+import org.junit.Test;
 
+import java.io.File;
 import java.io.InputStream;
 import java.net.URL;
 
-import org.jooby.Asset;
-import org.jooby.MediaType;
-import org.jooby.test.MockUnit;
-import org.junit.Test;
-
 public class AssetForwardingTest {
 
   @Test
@@ -75,13 +73,14 @@ public void path() throws Exception {
 
   @Test
   public void url() throws Exception {
-    new MockUnit(Asset.class, URL.class)
+    URL url = new File("pom.xml").toURI().toURL();
+    new MockUnit(Asset.class)
         .expect(unit -> {
           Asset asset = unit.get(Asset.class);
-          expect(asset.resource()).andReturn(unit.get(URL.class));
+          expect(asset.resource()).andReturn(url);
         })
         .run(unit -> {
-          assertEquals(unit.get(URL.class), new Asset.Forwarding(unit.get(Asset.class)).resource());
+          assertEquals(url, new Asset.Forwarding(unit.get(Asset.class)).resource());
         });
   }
 
diff --git a/modules/coverage-report/src/test/java/org/jooby/ws/WebSocketPauseResumeFeature.java b/modules/coverage-report/src/test/java/org/jooby/ws/WebSocketPauseResumeFeature.java
@@ -1,6 +1,16 @@
 package org.jooby.ws;
 
+import com.google.common.collect.Sets;
+import com.ning.http.client.AsyncHttpClient;
+import com.ning.http.client.AsyncHttpClientConfig;
+import com.ning.http.client.ws.WebSocket;
+import com.ning.http.client.ws.WebSocketTextListener;
+import com.ning.http.client.ws.WebSocketUpgradeHandler;
+import org.jooby.test.ServerFeature;
+import org.junit.After;
 import static org.junit.Assert.assertEquals;
+import org.junit.Before;
+import org.junit.Test;
 
 import java.util.HashSet;
 import java.util.Set;
@@ -9,40 +19,23 @@
 import java.util.concurrent.ScheduledExecutorService;
 import java.util.concurrent.TimeUnit;
 
-import org.jooby.test.ServerFeature;
-import org.junit.After;
-import org.junit.Before;
-import org.junit.Test;
-
-import com.google.common.collect.Sets;
-import com.ning.http.client.AsyncHttpClient;
-import com.ning.http.client.AsyncHttpClientConfig;
-import com.ning.http.client.ws.WebSocket;
-import com.ning.http.client.ws.WebSocketTextListener;
-import com.ning.http.client.ws.WebSocketUpgradeHandler;
-
 public class WebSocketPauseResumeFeature extends ServerFeature {
 
   static final ScheduledExecutorService executor = Executors.newSingleThreadScheduledExecutor();
+
   {
     ws("/ws", ws -> {
-      CountDownLatch latch = new CountDownLatch(1);
-
       ws.pause();
       // 2nd ignored
       ws.pause();
 
-      executor.schedule(() -> {
-        ws.resume();
-        // 2nd call ignored
-        ws.resume();
-        latch.countDown();
-      }, 1, TimeUnit.SECONDS);
+      ws.resume();
+      // 2nd call ignored
+      ws.resume();
 
       ws.onMessage(message -> {
 
         ws.send("=" + message.value(), () -> {
-          latch.await();
           ws.close();
         });
 
