YARN-5657. Fix TestDefaultContainerExecutor. (asuresh)

This commit is contained in:
Arun Suresh 2016-09-17 09:29:03 -07:00
parent 4174b9756c
commit f67237cbe7
1 changed files with 5 additions and 3 deletions

View File

@ -22,6 +22,7 @@ import static org.apache.hadoop.fs.CreateFlag.CREATE;
import static org.apache.hadoop.fs.CreateFlag.OVERWRITE; import static org.apache.hadoop.fs.CreateFlag.OVERWRITE;
import static org.junit.Assert.assertTrue; import static org.junit.Assert.assertTrue;
import static org.mockito.Matchers.any; import static org.mockito.Matchers.any;
import static org.mockito.Matchers.anyBoolean;
import static org.mockito.Matchers.isA; import static org.mockito.Matchers.isA;
import static org.mockito.Mockito.doAnswer; import static org.mockito.Mockito.doAnswer;
import static org.mockito.Mockito.mock; import static org.mockito.Mockito.mock;
@ -386,8 +387,8 @@ public class TestDefaultContainerExecutor {
} }
} }
return null; return null;
} }}).when(mockUtil).copy(any(Path.class), any(Path.class),
}).when(mockUtil).copy(any(Path.class), any(Path.class)); anyBoolean(), anyBoolean());
doAnswer(new Answer() { doAnswer(new Answer() {
@Override @Override
@ -478,7 +479,8 @@ public class TestDefaultContainerExecutor {
} }
// Verify that the calls happen the expected number of times // Verify that the calls happen the expected number of times
verify(mockUtil, times(1)).copy(any(Path.class), any(Path.class)); verify(mockUtil, times(1)).copy(any(Path.class), any(Path.class),
anyBoolean(), anyBoolean());
verify(mockLfs, times(2)).getFsStatus(any(Path.class)); verify(mockLfs, times(2)).getFsStatus(any(Path.class));
} }