From a5b2d8de339400e4c4084fbde189f3fadfc1b042 Mon Sep 17 00:00:00 2001 From: chenson42 Date: Fri, 17 Feb 2012 00:02:38 +0000 Subject: [PATCH] new method in the abstract writer test --- .../symmetric/io/data/writer/AbstractWriterTest.java | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/symmetric/symmetric-io/src/test/java/org/jumpmind/symmetric/io/data/writer/AbstractWriterTest.java b/symmetric/symmetric-io/src/test/java/org/jumpmind/symmetric/io/data/writer/AbstractWriterTest.java index 0ff1efb582..badaaa944c 100644 --- a/symmetric/symmetric-io/src/test/java/org/jumpmind/symmetric/io/data/writer/AbstractWriterTest.java +++ b/symmetric/symmetric-io/src/test/java/org/jumpmind/symmetric/io/data/writer/AbstractWriterTest.java @@ -64,9 +64,13 @@ protected Table buildSourceTable(String tableName, String[] keyNames, String[] c protected void writeData(CsvData data, String[] expectedValues) { writeData(data, expectedValues, TEST_COLUMNS); } + + protected String getTestTable() { + return TEST_TABLE; + } protected void writeData(CsvData data, String[] expectedValues, String[] columnNames) { - writeData(data, expectedValues, TEST_TABLE, TEST_KEYS, columnNames); + writeData(data, expectedValues, getTestTable(), TEST_KEYS, columnNames); } protected void writeData(CsvData data, String[] expectedValues, String tableName, @@ -125,7 +129,7 @@ protected long writeData(IDataWriter writer, TableCsvData... datas) { } protected void assertTestTableEquals(String testTableId, String[] expectedValues) { - String sql = "select " + getSelect(TEST_COLUMNS) + " from " + TEST_TABLE + " where " + String sql = "select " + getSelect(TEST_COLUMNS) + " from " + getTestTable() + " where " + getWhere(TEST_KEYS); Map results = platform.getSqlTemplate().queryForMap(sql, new Object[] { new Long(testTableId) }); @@ -245,5 +249,9 @@ public TableCsvData(Table table, List data) { } } + + protected long countRows(String tableName) { + return platform.getSqlTemplate().queryForInt(String.format("select count(*) from %s", tableName)); + } }