import org.pentaho.di.core.exception.KettleDatabaseBatchException; //导入依赖的package包/类
public static KettleDatabaseBatchException createKettleDatabaseBatchException( String message, SQLException ex ) {
  KettleDatabaseBatchException kdbe = new KettleDatabaseBatchException( message, ex );
  if ( ex instanceof BatchUpdateException ) {
    kdbe.setUpdateCounts( ( (BatchUpdateException) ex ).getUpdateCounts() );
  } else {
    // Null update count forces rollback of batch
    kdbe.setUpdateCounts( null );
  }
  List<Exception> exceptions = new ArrayList<Exception>();
  SQLException nextException = ex.getNextException();
  SQLException oldException = null;

  // This construction is specifically done for some JDBC drivers, these
  // drivers
  // always return the same exception on getNextException() (and thus go
  // into an infinite loop).
  // So it's not "equals" but != (comments from Sven Boden).
  while ( ( nextException != null ) && ( oldException != nextException ) ) {
    exceptions.add( nextException );
    oldException = nextException;
    nextException = nextException.getNextException();
  }
  kdbe.setExceptionsList( exceptions );
  return kdbe;
}
 

import org.pentaho.di.core.exception.KettleDatabaseBatchException; //导入依赖的package包/类
@Test( expected = KettleDatabaseBatchException.class )
public void testInsertRowWithBatchAlwaysThrowsKettleBatchException() throws KettleDatabaseException, SQLException {
  DatabaseMeta mockDatabaseMeta = mock( DatabaseMeta.class );
  when( mockDatabaseMeta.supportsBatchUpdates() ).thenReturn( true );

  DatabaseMetaData mockDatabaseMetaData = mock( DatabaseMetaData.class );
  when( mockDatabaseMetaData.supportsBatchUpdates() ).thenReturn( true );
  Connection mockConnection = mockConnection( mockDatabaseMetaData );

  PreparedStatement ps = mock( PreparedStatement.class );
  when( ps.executeBatch() ).thenThrow( new SQLException() );

  Database database = new Database( mockLogger(), mockDatabaseMeta );
  database.setCommit( 1 );
  database.setConnection( mockConnection );
  database.insertRow( ps, true, true );
}
 

import org.pentaho.di.core.exception.KettleDatabaseBatchException; //导入依赖的package包/类
@Test( expected = KettleDatabaseException.class )
public void testInsertRowWithoutBatchDoesntThrowKettleBatchException() throws KettleDatabaseException, SQLException {
  DatabaseMeta mockDatabaseMeta = mock( DatabaseMeta.class );
  when( mockDatabaseMeta.supportsBatchUpdates() ).thenReturn( true );

  DatabaseMetaData mockDatabaseMetaData = mock( DatabaseMetaData.class );
  when( mockDatabaseMetaData.supportsBatchUpdates() ).thenReturn( true );
  Connection mockConnection = mockConnection( mockDatabaseMetaData );

  PreparedStatement ps = mock( PreparedStatement.class );
  when( ps.executeUpdate() ).thenThrow( new SQLException() );

  Database database = new Database( mockLogger(), mockDatabaseMeta );
  database.setConnection( mockConnection );
  try {
    database.insertRow( ps, true, true );
  } catch ( KettleDatabaseBatchException e ) {
    // noop
  }
}
 

import org.pentaho.di.core.exception.KettleDatabaseBatchException; //导入依赖的package包/类
@Test( expected = KettleDatabaseBatchException.class )
public void testEmptyAndCommitWithBatchAlwaysThrowsKettleBatchException() throws KettleDatabaseException,
  SQLException {
  DatabaseMeta mockDatabaseMeta = mock( DatabaseMeta.class );
  when( mockDatabaseMeta.supportsBatchUpdates() ).thenReturn( true );

  DatabaseMetaData mockDatabaseMetaData = mock( DatabaseMetaData.class );
  when( mockDatabaseMetaData.supportsBatchUpdates() ).thenReturn( true );
  Connection mockConnection = mockConnection( mockDatabaseMetaData );

  PreparedStatement ps = mock( PreparedStatement.class );
  when( ps.executeBatch() ).thenThrow( new SQLException() );

  Database database = new Database( mockLogger(), mockDatabaseMeta );
  database.setCommit( 1 );
  database.setConnection( mockConnection );
  database.emptyAndCommit( ps, true, 1 );
}
 

import org.pentaho.di.core.exception.KettleDatabaseBatchException; //导入依赖的package包/类
@Test( expected = KettleDatabaseException.class )
public void testEmptyAndCommitWithoutBatchDoesntThrowKettleBatchException()
  throws KettleDatabaseException, SQLException {
  DatabaseMeta mockDatabaseMeta = mock( DatabaseMeta.class );
  when( mockDatabaseMeta.supportsBatchUpdates() ).thenReturn( true );

  DatabaseMetaData mockDatabaseMetaData = mock( DatabaseMetaData.class );
  when( mockDatabaseMetaData.supportsBatchUpdates() ).thenReturn( true );
  Connection mockConnection = mockConnection( mockDatabaseMetaData );

  PreparedStatement ps = mock( PreparedStatement.class );
  doThrow( new SQLException() ).when( ps ).close();

  Database database = new Database( mockLogger(), mockDatabaseMeta );
  database.setConnection( mockConnection );
  try {
    database.emptyAndCommit( ps, true, 1 );
  } catch ( KettleDatabaseBatchException e ) {
    // noop
  }
}
 

import org.pentaho.di.core.exception.KettleDatabaseBatchException; //导入依赖的package包/类
@SuppressWarnings( "deprecation" )
@Test( expected = KettleDatabaseBatchException.class )
public void testInsertFinishedWithBatchAlwaysThrowsKettleBatchException() throws KettleDatabaseException,
  SQLException {
  DatabaseMeta mockDatabaseMeta = mock( DatabaseMeta.class );
  when( mockDatabaseMeta.supportsBatchUpdates() ).thenReturn( true );

  DatabaseMetaData mockDatabaseMetaData = mock( DatabaseMetaData.class );
  when( mockDatabaseMetaData.supportsBatchUpdates() ).thenReturn( true );
  Connection mockConnection = mockConnection( mockDatabaseMetaData );

  PreparedStatement ps = mock( PreparedStatement.class );
  when( ps.executeBatch() ).thenThrow( new SQLException() );

  Database database = new Database( mockLogger(), mockDatabaseMeta );
  database.setCommit( 1 );
  database.setConnection( mockConnection );
  database.insertFinished( ps, true );
}
 

import org.pentaho.di.core.exception.KettleDatabaseBatchException; //导入依赖的package包/类
@SuppressWarnings( "deprecation" )
@Test( expected = KettleDatabaseException.class )
public void testInsertFinishedWithoutBatchDoesntThrowKettleBatchException()
  throws KettleDatabaseException, SQLException {
  DatabaseMeta mockDatabaseMeta = mock( DatabaseMeta.class );
  when( mockDatabaseMeta.supportsBatchUpdates() ).thenReturn( true );

  DatabaseMetaData mockDatabaseMetaData = mock( DatabaseMetaData.class );
  when( mockDatabaseMetaData.supportsBatchUpdates() ).thenReturn( true );
  Connection mockConnection = mockConnection( mockDatabaseMetaData );

  PreparedStatement ps = mock( PreparedStatement.class );
  doThrow( new SQLException() ).when( ps ).close();

  Database database = new Database( mockLogger(), mockDatabaseMeta );
  database.setConnection( mockConnection );
  try {
    database.insertFinished( ps, true );
  } catch ( KettleDatabaseBatchException e ) {
    // noop
  }
}