|
@@ -109,6 +109,7 @@ import java.util.Map;
|
|
|
import static org.assertj.core.api.Assertions.assertThat;
|
|
|
import static junit.framework.TestCase.assertTrue;
|
|
|
import static org.junit.Assert.assertEquals;
|
|
|
+import static org.junit.Assert.assertFalse;
|
|
|
import static org.mockito.ArgumentMatchers.any;
|
|
|
import static org.mockito.ArgumentMatchers.anyLong;
|
|
|
import static org.mockito.ArgumentMatchers.eq;
|
|
@@ -486,8 +487,12 @@ public class TestRMAppTransitions {
|
|
|
private void assertAppStateLaunchTimeSaved(long expectedLaunchTime) {
|
|
|
ArgumentCaptor<ApplicationStateData> state =
|
|
|
ArgumentCaptor.forClass(ApplicationStateData.class);
|
|
|
- verify(store, times(1)).updateApplicationState(state.capture());
|
|
|
+ ArgumentCaptor<Boolean> notifyApp =
|
|
|
+ ArgumentCaptor.forClass(Boolean.class);
|
|
|
+ verify(store, times(1)).updateApplicationState(state.capture(),
|
|
|
+ notifyApp.capture());
|
|
|
assertEquals(expectedLaunchTime, state.getValue().getLaunchTime());
|
|
|
+ assertFalse(notifyApp.getValue());
|
|
|
}
|
|
|
|
|
|
private void assertKilled(RMApp application) {
|