Fix invalid connection URLs returned by PostgreSQLContainer (#1976)

modules/postgresql/src/main/java/org/testcontainers/containers/PostgreSQLContainer.java

@@ -7,6 +7,7 @@
 import java.util.HashSet;
 import java.util.Set;
 
+import static java.lang.String.format;
 import static java.time.temporal.ChronoUnit.SECONDS;
 
 /**
@@ -24,6 +25,8 @@ public class PostgreSQLContainer<SELF extends PostgreSQLContainer<SELF>> extends
 
     private static final String FSYNC_OFF_OPTION = "fsync=off";
 
+    private static final String QUERY_PARAM_SEPARATOR = "&";
+
     public PostgreSQLContainer() {
         this(IMAGE + ":" + DEFAULT_TAG);
     }
@@ -59,7 +62,14 @@ public String getDriverClassName() {
     @Override
     public String getJdbcUrl() {
         // Disable Postgres driver use of java.util.logging to reduce noise at startup time
-        return "jdbc:postgresql://" + getContainerIpAddress() + ":" + getMappedPort(POSTGRESQL_PORT) + "/" + databaseName + "?loggerLevel=OFF";
+        return format("jdbc:postgresql://%s:%d/%s?loggerLevel=OFF", getContainerIpAddress(), getMappedPort(POSTGRESQL_PORT), databaseName);
+    }
+
+    @Override
+    protected String constructUrlForConnection(String queryString) {
+        return "".equals(queryString)
+            ? getJdbcUrl()
+            : getJdbcUrl() + QUERY_PARAM_SEPARATOR + queryString.substring(1);
     }
 
     @Override

modules/postgresql/src/test/java/org/testcontainers/containers/PostgreSQLConnectionURLTest.java

@@ -0,0 +1,34 @@
+package org.testcontainers.containers;
+
+import org.junit.Test;
+
+import static org.rnorth.visibleassertions.VisibleAssertions.assertEquals;
+import static org.rnorth.visibleassertions.VisibleAssertions.assertFalse;
+import static org.rnorth.visibleassertions.VisibleAssertions.assertTrue;
+
+public class PostgreSQLConnectionURLTest {
+
+    @Test
+    public void shouldCorrectlyAppendQueryString() {
+        PostgreSQLContainer postgres = new FixedJdbcUrlPostgreSQLContainer();
+        String connectionUrl = postgres.constructUrlForConnection("?stringtype=unspecified&stringtype=unspecified");
+        String queryString = connectionUrl.substring(connectionUrl.indexOf('?'));
+
+        assertTrue("Query String contains expected params", queryString.contains("&stringtype=unspecified&stringtype=unspecified"));
+        assertEquals("Query String starts with '?'", 0, queryString.indexOf('?'));
+        assertFalse("Query String does not contain extra '?'", queryString.substring(1).contains("?"));
+    }
+
+    public static class FixedJdbcUrlPostgreSQLContainer extends PostgreSQLContainer {
+
+        @Override
+        public String getContainerIpAddress() {
+            return "localhost";
+        }
+
+        @Override
+        public Integer getMappedPort(int originalPort) {
+            return 34532;
+        }
+    }
+}