1 package com.imcode.db.mock;
2
3 import com.imcode.db.DatabaseConnection;
4 import com.imcode.db.mock.MockConnection;
5 import com.imcode.db.mock.MockDatabase;
6 import org.apache.commons.dbutils.ResultSetHandler;
7
8 import java.sql.Connection;
9
10 public class MockDatabaseConnection implements DatabaseConnection {
11
12 private MockDatabase database;
13
14 public MockDatabaseConnection( MockDatabase mockDatabase ) {
15 this.database = mockDatabase ;
16 }
17
18 public Number executeUpdateAndGetGeneratedKey( String sql, Object[] parameters ) {
19 return (Number)database.getResultForSqlCall( sql, parameters ) ;
20 }
21
22 public Object executeQuery( String sqlQuery, Object[] parameters,
23 ResultSetHandler resultSetHandler ) {
24 return database.executeQuery(sqlQuery, parameters, resultSetHandler) ;
25 }
26
27 public Connection getConnection() {
28 return new MockConnection(database) ;
29 }
30
31 public int executeUpdate( String sql, Object[] parameters ) {
32 return database.executeUpdate( sql, parameters );
33 }
34 }