Increase timeouts in TestStateMachine

prestodb · Nov 7, 2018 · 9011cfb · 9011cfb
1 parent b09ea18
commit 9011cfb
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/presto-main/src/test/java/com/facebook/presto/execution/TestStateMachine.java b/presto-main/src/test/java/com/facebook/presto/execution/TestStateMachine.java
@@ -240,8 +240,8 @@ private static void assertStateChange(StateMachine<State> stateMachine, StateCha
 
         assertEquals(stateMachine.get(), expectedState);
 
-        assertEquals(futureChange.get(1, SECONDS), expectedState);
-        assertEquals(listenerChange.get(1, SECONDS), expectedState);
+        assertEquals(futureChange.get(10, SECONDS), expectedState);
+        assertEquals(listenerChange.get(10, SECONDS), expectedState);
 
         // listeners should not be retained if we are in a terminal state
         boolean isTerminalState = stateMachine.isTerminalState(expectedState);
@@ -272,7 +272,7 @@ private static void assertNoStateChange(StateMachine<State> stateMachine, StateC
         boolean isTerminalState = stateMachine.isTerminalState(initialState);
         if (isTerminalState) {
             assertEquals(stateMachine.getStateChangeListeners(), ImmutableSet.of());
-            assertEquals(listenerChange.get(1, SECONDS), initialState);
+            assertEquals(listenerChange.get(10, SECONDS), initialState);
         }
 
         stateChange.run();