01: /*
02: *
03: * The DbUnit Database Testing Framework
04: * Copyright (C)2002-2004, DbUnit.org
05: *
06: * This library is free software; you can redistribute it and/or
07: * modify it under the terms of the GNU Lesser General Public
08: * License as published by the Free Software Foundation; either
09: * version 2.1 of the License, or (at your option) any later version.
10: *
11: * This library is distributed in the hope that it will be useful,
12: * but WITHOUT ANY WARRANTY; without even the implied warranty of
13: * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14: * Lesser General Public License for more details.
15: *
16: * You should have received a copy of the GNU Lesser General Public
17: * License along with this library; if not, write to the Free Software
18: * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
19: *
20: */
21: package org.dbunit.operation;
22:
23: import org.slf4j.Logger;
24: import org.slf4j.LoggerFactory;
25:
26: import org.dbunit.DatabaseUnitException;
27: import org.dbunit.database.DatabaseConfig;
28: import org.dbunit.database.IDatabaseConnection;
29: import org.dbunit.dataset.*;
30:
31: import java.sql.SQLException;
32: import java.util.ArrayList;
33: import java.util.List;
34:
35: /**
36: * @author Manuel Laflamme
37: * @since Jan 17, 2004
38: * @version $Revision: 554 $
39: */
40: public abstract class AbstractOperation extends DatabaseOperation {
41:
42: /**
43: * Logger for this class
44: */
45: private static final Logger logger = LoggerFactory
46: .getLogger(AbstractOperation.class);
47:
48: protected String getQualifiedName(String prefix, String name,
49: IDatabaseConnection connection) {
50: logger.debug("getQualifiedName(prefix=" + prefix + ", name="
51: + name + ", connection=" + connection + ") - start");
52:
53: String escapePattern = (String) connection.getConfig()
54: .getProperty(DatabaseConfig.PROPERTY_ESCAPE_PATTERN);
55:
56: return DataSetUtils.getQualifiedName(prefix, name,
57: escapePattern);
58: }
59:
60: /**
61: * Returns the metadata to use in this operation.
62: *
63: * @param connection the database connection
64: * @param metaData the xml table metadata
65: */
66: static ITableMetaData getOperationMetaData(
67: IDatabaseConnection connection, ITableMetaData metaData)
68: throws DatabaseUnitException, SQLException {
69: logger.debug("getOperationMetaData(connection=" + connection
70: + ", metaData=" + metaData + ") - start");
71:
72: IDataSet databaseDataSet = connection.createDataSet();
73: String tableName = metaData.getTableName();
74:
75: ITableMetaData databaseMetaData = databaseDataSet
76: .getTableMetaData(tableName);
77: Column[] databaseColumns = databaseMetaData.getColumns();
78: Column[] columns = metaData.getColumns();
79:
80: List columnList = new ArrayList();
81: for (int j = 0; j < columns.length; j++) {
82: String columnName = columns[j].getColumnName();
83: Column column = DataSetUtils.getColumn(columnName,
84: databaseColumns);
85: if (column == null) {
86: throw new NoSuchColumnException(tableName + "."
87: + columnName);
88: }
89: columnList.add(column);
90: }
91:
92: return new DefaultTableMetaData(
93: databaseMetaData.getTableName(), (Column[]) columnList
94: .toArray(new Column[0]), databaseMetaData
95: .getPrimaryKeys());
96: }
97: }
|