diff --git a/launcher/src/test/java/org/apache/spark/launcher/LauncherServerSuite.java b/launcher/src/test/java/org/apache/spark/launcher/LauncherServerSuite.java index 13f72b757f30a074a208e73a61d23dad7d3f7233..5bf2babdd103636f175a52efa1e53f0f4ab9af97 100644 --- a/launcher/src/test/java/org/apache/spark/launcher/LauncherServerSuite.java +++ b/launcher/src/test/java/org/apache/spark/launcher/LauncherServerSuite.java @@ -83,17 +83,17 @@ public class LauncherServerSuite extends BaseSuite { client = new TestClient(s); client.send(new Hello(handle.getSecret(), "1.4.0")); - semaphore.tryAcquire(10, TimeUnit.MILLISECONDS); + assertTrue(semaphore.tryAcquire(1, TimeUnit.SECONDS)); // Make sure the server matched the client to the handle. assertNotNull(handle.getConnection()); client.send(new SetAppId("app-id")); - semaphore.tryAcquire(10, TimeUnit.MILLISECONDS); + assertTrue(semaphore.tryAcquire(1, TimeUnit.SECONDS)); assertEquals("app-id", handle.getAppId()); client.send(new SetState(SparkAppHandle.State.RUNNING)); - semaphore.tryAcquire(10, TimeUnit.MILLISECONDS); + assertTrue(semaphore.tryAcquire(1, TimeUnit.SECONDS)); assertEquals(SparkAppHandle.State.RUNNING, handle.getState()); handle.stop();