@Test
public void testUnknownJob() {
long testJobId = 555l;
SqoopException exception =
new SqoopException(DriverError.DRIVER_0004, "Unknown job id: " + testJobId);
when(repositoryManagerMock.getRepository()).thenReturn(jdbcRepoMock);
when(jdbcRepoMock.findJob(testJobId)).thenReturn(null);
try {
jobManager.getJob(testJobId);
} catch (SqoopException ex) {
assertEquals(ex.getMessage(), exception.getMessage());
verify(repositoryManagerMock, times(1)).getRepository();
verify(jdbcRepoMock, times(1)).findJob(testJobId);
}
}
@Test
public void testDisabledJob() {
MJob testJob = job(123l, 456l);
testJob.setEnabled(false);
testJob.setPersistenceId(1111);
SqoopException exception =
new SqoopException(DriverError.DRIVER_0009, "Job id: " + testJob.getPersistenceId());
MJob mJobSpy = org.mockito.Mockito.spy(testJob);
when(repositoryManagerMock.getRepository()).thenReturn(jdbcRepoMock);
when(jdbcRepoMock.findJob(123l)).thenReturn(mJobSpy);
try {
jobManager.getJob(123l);
} catch (SqoopException ex) {
assertEquals(ex.getMessage(), exception.getMessage());
verify(repositoryManagerMock, times(1)).getRepository();
verify(jdbcRepoMock, times(1)).findJob(123l);
}
}
@Test
public void testDisabledLink() {
MLink testConnection = new MLink(123l, null);
testConnection.setPersistenceId(1234);
testConnection.setEnabled(false);
SqoopException exception =
new SqoopException(
DriverError.DRIVER_0010, "Connection id: " + testConnection.getPersistenceId());
MLink mConnectionSpy = org.mockito.Mockito.spy(testConnection);
when(repositoryManagerMock.getRepository()).thenReturn(jdbcRepoMock);
when(jdbcRepoMock.findLink(123l)).thenReturn(mConnectionSpy);
try {
jobManager.getLink(123l);
} catch (SqoopException ex) {
assertEquals(ex.getMessage(), exception.getMessage());
verify(repositoryManagerMock, times(1)).getRepository();
verify(jdbcRepoMock, times(1)).findLink(123l);
}
}
@Test
public void testUnsupportedDirectionForConnector() {
// invalid job id/ direction
SqoopException exception =
new SqoopException(
DriverError.DRIVER_0011,
"Connector: " + sqoopConnectorMock.getClass().getCanonicalName());
List<Direction> supportedDirections = getSupportedDirections();
when(sqoopConnectorMock.getSupportedDirections()).thenReturn(supportedDirections);
try {
// invalid direction
jobManager.validateSupportedDirection(sqoopConnectorMock, null);
} catch (SqoopException ex) {
assertEquals(ex.getMessage(), exception.getMessage());
verify(sqoopConnectorMock, times(1)).getSupportedDirections();
return;
}
fail("Should throw out an exception with message: " + exception.getMessage());
}
