@Test
public void testRestartJob() throws Exception {
int containerId = 1;
List<String> expectedCommand =
asList(
"aurora job restart %s/%s %s %s %d",
JOB_SPEC, containerId, VERBOSE_CONFIG, BATCH_CONFIG, Integer.MAX_VALUE);
// Failed
Mockito.doReturn(false).when(controller).runProcess(Matchers.anyListOf(String.class));
Assert.assertFalse(controller.restartJob(containerId));
Mockito.verify(controller).runProcess(Mockito.eq(expectedCommand));
// Happy path
Mockito.doReturn(true).when(controller).runProcess(Matchers.anyListOf(String.class));
Assert.assertTrue(controller.restartJob(containerId));
Mockito.verify(controller, Mockito.times(2)).runProcess(expectedCommand);
}
