mssql test fixes

hapifhir · Jul 6, 2023 · 8d55781 · 8d55781
1 parent 806a628
commit 8d55781
Showing 1 changed file with 15 additions and 2 deletions.
diff --git a/...paserver-test-utilities/src/main/java/ca/uhn/fhir/jpa/embedded/MsSqlEmbeddedDatabase.java b/...paserver-test-utilities/src/main/java/ca/uhn/fhir/jpa/embedded/MsSqlEmbeddedDatabase.java
@@ -20,6 +20,8 @@
 package ca.uhn.fhir.jpa.embedded;
 
 import ca.uhn.fhir.jpa.migrate.DriverTypeEnum;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
 import org.testcontainers.containers.MSSQLServerContainer;
 import org.testcontainers.utility.DockerImageName;
 
@@ -34,6 +36,7 @@
  * @see <a href="https://www.testcontainers.org/modules/databases/mssqlserver/">MS SQL Server TestContainer</a>
  */
 public class MsSqlEmbeddedDatabase extends JpaEmbeddedDatabase {
+	private static final Logger ourLog = LoggerFactory.getLogger(MsSqlEmbeddedDatabase.class);
 
 	private final MSSQLServerContainer myContainer;
 
@@ -70,8 +73,18 @@ private void dropForeignKeys() {
 	private void dropRemainingConstraints() {
 		List<Map<String, Object>> queryResults = getJdbcTemplate().queryForList("SELECT * FROM INFORMATION_SCHEMA.TABLE_CONSTRAINTS");
 		for(Map<String, Object> row : queryResults){
-			String tableName = row.get("TABLE_NAME").toString();
-			String constraintName = row.get("CONSTRAINT_NAME").toString();
+			Object tableNameEntry = row.get("TABLE_NAME");
+			if (tableNameEntry == null) {
+				ourLog.warn("Found a constraint with no table name: {}", row);
+				continue;
+			}
+			String tableName = tableNameEntry.toString();
+			Object constraintNameEntry = row.get("CONSTRAINT_NAME");
+			if (constraintNameEntry == null) {
+				ourLog.warn("Found a constraint with no constraint name: {}", row);
+				continue;
+			}
+			String constraintName = constraintNameEntry.toString();
 			getJdbcTemplate().execute(String.format("ALTER TABLE \"%s\" DROP CONSTRAINT \"%s\"", tableName, constraintName));
 		}
 	}