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)); + } }