View Javadoc
1   /*
2    *
3    * The DbUnit Database Testing Framework
4    * Copyright (C)2002-2004, DbUnit.org
5    *
6    * This library is free software; you can redistribute it and/or
7    * modify it under the terms of the GNU Lesser General Public
8    * License as published by the Free Software Foundation; either
9    * 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.dataset;
22  
23  import org.dbunit.dataset.datatype.DataType;
24  
25  /**
26   * @author Manuel Laflamme
27   * @since Apr 12, 2003
28   * @version $Revision$
29   */
30  public class MockTableMetaData extends AbstractTableMetaData
31  {
32      private String _tableName;
33      private Column[] _columns = new Column[0];
34      private String[] _keyNames = new String[0];
35  
36      public MockTableMetaData()
37      {
38      }
39  
40      public MockTableMetaData(String tableName, String[] columnNames)
41      {
42          _tableName = tableName;
43          setupColumns(columnNames);
44      }
45  
46      public void setTableName(String tableName)
47      {
48          _tableName = tableName;
49      }
50  
51      public void setupColumns(Column[] columns)
52      {
53          _columns = columns;
54      }
55  
56      public void setupColumns(String[] columnNames)
57      {
58          Column[] columns = new Column[columnNames.length];
59          for (int i = 0; i < columnNames.length; i++)
60          {
61              String columnName = columnNames[i];
62              columns[i] = new Column(columnName, DataType.UNKNOWN);
63          }
64          _columns = columns;
65      }
66  
67      public void setupPrimaryKeys(String[] keyNames)
68      {
69          _keyNames = keyNames;
70      }
71  
72      ////////////////////////////////////////////////////////////////////////////
73      // ITableMetaData interface
74  
75      public String getTableName()
76      {
77          return _tableName;
78      }
79  
80      public Column[] getColumns() throws DataSetException
81      {
82          return _columns;
83      }
84  
85      public Column[] getPrimaryKeys() throws DataSetException
86      {
87          return Columns.getColumns(_keyNames, _columns);
88      }
89  }