1 package net.jaekl.cfb.db;
3 import java.util.ArrayList;
4 import static org.junit.Assert.*;
6 public class TableMock extends Table {
8 private ArrayList<Row> m_rows;
10 public TableMock(String name, Column[] columns) {
12 m_rows = new ArrayList<Row>();
15 public TableMock(Table table) {
16 this(table.m_name, table.m_columns.toArray(new Column[table.m_columns.size()]));
19 public ArrayList<Row> mock_getRows() { return m_rows; }
21 public boolean mock_hasColumn(Column expectedCol) {
22 for (Column col : m_columns) {
23 if (col.equals(expectedCol)) {
30 public Column[] mock_getColumns() {
31 return m_columns.toArray(new Column[m_columns.size()]);
34 public void mock_insert(Row row) {
35 assertEquals(getNumColumns(), row.getNumColumns());
36 for (int i = 0; i < getNumColumns(); ++i) {
37 assertEquals(getColumn(i).getType(), row.getColumn(i).getType());
43 public String toString()
45 StringBuilder sb = new StringBuilder("" + getName() + "={\n");
46 for (Row row : m_rows) {
47 sb.append("" + row + "\n");