@Test
public void testSuccess() {

    // Mock rest client
    Mockito.when(restClient.createJob(process, request)).thenReturn(SUCCESS_RESPONSE);
    // Test success response
    final SparkJobResponse response = new MockBatchJob().get();
    Assert.assertEquals(SUCCESS_RESPONSE, response);
    Assert.assertEquals(0, sleepMillis.get());
}