Fixed unit test so that multiple runs won't invalidate the results

JumpMind · Apr 22, 2008 · 53fbce4 · 53fbce4
1 parent 8cdb5b3
commit 53fbce4
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/symmetric/src/test/java/org/jumpmind/symmetric/db/DbTriggerTest.java b/symmetric/src/test/java/org/jumpmind/symmetric/db/DbTriggerTest.java
@@ -25,6 +25,7 @@
 import java.sql.SQLException;
 import java.sql.Statement;
 import java.sql.Types;
+import java.util.List;
 
 import org.apache.commons.logging.Log;
 import org.apache.commons.logging.LogFactory;
@@ -138,17 +139,18 @@ public void validateTestTableTriggers() throws Exception {
         assert match : "Received " + csvString + ", Expected the string to end with " + EXPECTED_INSERT1_CSV_ENDSWITH;
     }
 
+    @SuppressWarnings("unchecked")
     @Test(groups = "continuous", dependsOnMethods = "validateTestTableTriggers")
     public void testInitialLoadSql() throws Exception {
         IConfigurationService service = (IConfigurationService) getSymmetricEngine().getApplicationContext().getBean(
                 "configurationService");
         service.getTriggerFor(TEST_TRIGGERS_TABLE, TestConstants.TEST_ROOT_NODE_GROUP);
         String sql = getDbDialect(getSymmetricEngine()).createInitalLoadSqlFor(new Node("1", null, "1.0"),
                 service.getTriggerFor(TEST_TRIGGERS_TABLE, TestConstants.TEST_ROOT_NODE_GROUP));
-        String csvString = (String) getJdbcTemplate(getSymmetricEngine()).queryForObject(sql, String.class);
-        boolean match = false;
-        match = csvString.endsWith(EXPECTED_INSERT1_CSV_ENDSWITH);
-        assert match : "Received " + csvString + ", Expected the string to end with " + EXPECTED_INSERT1_CSV_ENDSWITH;
+        List<String> csvStrings = getJdbcTemplate(getSymmetricEngine()).queryForList(sql, String.class);
+        Assert.assertTrue(csvStrings.size() > 0);
+        String csvString = csvStrings.get(0);
+        Assert.assertTrue(csvString.endsWith(EXPECTED_INSERT1_CSV_ENDSWITH), "Received " + csvString + ", Expected the string to end with " + EXPECTED_INSERT1_CSV_ENDSWITH);
     }
 
     @Test(groups = "continuous", dependsOnMethods = "testInitialLoadSql")