select queries now work, and print tabular output.

[squelch.git] / src / test / java / net / jaekl / squelch / stmt / TabularMock.java

diff --git a/src/test/java/net/jaekl/squelch/stmt/TabularMock.java b/src/test/java/net/jaekl/squelch/stmt/TabularMock.java
new file mode 100644 (file)
index 0000000..b19a946
--- /dev/null
+++ b/src/test/java/net/jaekl/squelch/stmt/TabularMock.java
@@ -0,0 +1,41 @@
+package net.jaekl.squelch.stmt;
+
+import java.sql.SQLException;
+import java.util.ArrayList;
+
+import net.jaekl.squelch.sql.Column;
+import net.jaekl.squelch.sql.Row;
+
+public class TabularMock extends Tabular {
+       private Column[] m_cols;
+       private ArrayList<Row> m_rows;
+       private int m_rowIdx;
+       
+       public TabularMock() {
+               m_cols = new Column[0];
+               m_rows = new ArrayList<Row>();
+               m_rowIdx = (-1);
+       }
+       
+       @Override
+       Column[] getCols() throws SQLException {
+               return m_cols.clone();
+       }
+
+       @Override
+       Row getNext() {
+               if (m_rowIdx >= (m_rows.size() - 1)) {
+                       return null;
+               }
+               m_rowIdx++;
+               return m_rows.get(m_rowIdx);
+       }
+       
+       public void mock_setCols(Column[] cols) {
+               m_cols = cols.clone();
+       }
+       
+       public void mock_addRow(Row row) {
+               m_rows.add(row);
+       }
+}