Polish "Fix handling of env vars in Bitnami's Postgres image"

See gh-43783

Author: wilkinsona

spring-boot-project/spring-boot-docker-compose/build.gradle

@@ -20,6 +20,8 @@ dependencies {
 	dockerTestRuntimeOnly("com.microsoft.sqlserver:mssql-jdbc")
 	dockerTestRuntimeOnly("com.oracle.database.r2dbc:oracle-r2dbc")
 	dockerTestRuntimeOnly("io.r2dbc:r2dbc-mssql")
+	dockerTestRuntimeOnly("org.postgresql:postgresql")
+	dockerTestRuntimeOnly("org.postgresql:r2dbc-postgresql")
 
 	implementation("com.fasterxml.jackson.core:jackson-databind")
 	implementation("com.fasterxml.jackson.module:jackson-module-parameter-names")

spring-boot-project/spring-boot-docker-compose/src/dockerTest/java/org/springframework/boot/docker/compose/service/connection/postgres/PostgresJdbcDockerComposeConnectionDetailsFactoryIntegrationTests.java

@@ -1,5 +1,5 @@
 /*
- * Copyright 2012-2024 the original author or authors.
+ * Copyright 2012-2025 the original author or authors.
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -16,9 +16,15 @@
 
 package org.springframework.boot.docker.compose.service.connection.postgres;
 
+import java.sql.Driver;
+
 import org.springframework.boot.autoconfigure.jdbc.JdbcConnectionDetails;
 import org.springframework.boot.docker.compose.service.connection.test.DockerComposeTest;
+import org.springframework.boot.jdbc.DatabaseDriver;
 import org.springframework.boot.testsupport.container.TestImage;
+import org.springframework.jdbc.core.JdbcTemplate;
+import org.springframework.jdbc.datasource.SimpleDriverDataSource;
+import org.springframework.util.ClassUtils;
 
 import static org.assertj.core.api.Assertions.assertThat;
 
@@ -51,4 +57,21 @@ private void assertConnectionDetails(JdbcConnectionDetails connectionDetails) {
 		assertThat(connectionDetails.getJdbcUrl()).startsWith("jdbc:postgresql://").endsWith("/mydatabase");
 	}
 
+	private void checkDatabaseAccess(JdbcConnectionDetails connectionDetails) throws ClassNotFoundException {
+		assertThat(executeQuery(connectionDetails, DatabaseDriver.POSTGRESQL.getValidationQuery(), Integer.class))
+			.isEqualTo(1);
+	}
+
+	@SuppressWarnings("unchecked")
+	private <T> T executeQuery(JdbcConnectionDetails connectionDetails, String sql, Class<T> result)
+			throws ClassNotFoundException {
+		SimpleDriverDataSource dataSource = new SimpleDriverDataSource();
+		dataSource.setUrl(connectionDetails.getJdbcUrl());
+		dataSource.setUsername(connectionDetails.getUsername());
+		dataSource.setPassword(connectionDetails.getPassword());
+		dataSource.setDriverClass((Class<? extends Driver>) ClassUtils.forName(connectionDetails.getDriverClassName(),
+				getClass().getClassLoader()));
+		return new JdbcTemplate(dataSource).queryForObject(sql, result);
+	}
+
 }

spring-boot-project/spring-boot-docker-compose/src/dockerTest/java/org/springframework/boot/docker/compose/service/connection/postgres/PostgresR2dbcDockerComposeConnectionDetailsFactoryIntegrationTests.java

@@ -1,5 +1,5 @@
 /*
- * Copyright 2012-2024 the original author or authors.
+ * Copyright 2012-2025 the original author or authors.
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -16,11 +16,16 @@
 
 package org.springframework.boot.docker.compose.service.connection.postgres;
 
+import java.time.Duration;
+
+import io.r2dbc.spi.ConnectionFactories;
 import io.r2dbc.spi.ConnectionFactoryOptions;
 
 import org.springframework.boot.autoconfigure.r2dbc.R2dbcConnectionDetails;
 import org.springframework.boot.docker.compose.service.connection.test.DockerComposeTest;
+import org.springframework.boot.jdbc.DatabaseDriver;
 import org.springframework.boot.testsupport.container.TestImage;
+import org.springframework.r2dbc.core.DatabaseClient;
 
 import static org.assertj.core.api.Assertions.assertThat;
 
@@ -53,4 +58,18 @@ private void assertConnectionDetails(R2dbcConnectionDetails connectionDetails) {
 		assertThat(connectionFactoryOptions.getRequiredValue(ConnectionFactoryOptions.PASSWORD)).isEqualTo("secret");
 	}
 
+	private void checkDatabaseAccess(R2dbcConnectionDetails connectionDetails) {
+		assertThat(executeQuery(connectionDetails, DatabaseDriver.POSTGRESQL.getValidationQuery(), Integer.class))
+			.isEqualTo(1);
+	}
+
+	private <T> T executeQuery(R2dbcConnectionDetails connectionDetails, String sql, Class<T> result) {
+		ConnectionFactoryOptions connectionFactoryOptions = connectionDetails.getConnectionFactoryOptions();
+		return DatabaseClient.create(ConnectionFactories.get(connectionFactoryOptions))
+			.sql(sql)
+			.mapValue(result)
+			.first()
+			.block(Duration.ofSeconds(30));
+	}
+
 }

spring-boot-project/spring-boot-docker-compose/src/main/java/org/springframework/boot/docker/compose/service/connection/postgres/PostgresEnvironment.java

@@ -32,12 +32,12 @@
  */
 class PostgresEnvironment {
 
-	private static final String[] USERNAME_KEYS = new String[] { "POSTGRES_USER", "POSTGRES_USERNAME",
-			"POSTGRESQL_USER", "POSTGRESQL_USERNAME" };
+	private static final String[] USERNAME_KEYS = new String[] { "POSTGRES_USER", "POSTGRESQL_USER",
+			"POSTGRESQL_USERNAME" };
 
 	private static final String DEFAULT_USERNAME = "postgres";
 
-	private static final String[] DATABASE_KEYS = new String[] { "POSTGRES_DB", "POSTGRES_DATABASE",
+	private static final String[] DATABASE_KEYS = new String[] { "POSTGRES_DB", "POSTGRESQL_DB",
 			"POSTGRESQL_DATABASE" };
 
 	private final String username;
@@ -47,18 +47,18 @@ class PostgresEnvironment {
 	private final String database;
 
 	PostgresEnvironment(Map<String, String> env) {
-		this.username = extractUsername(env);
+		this.username = extract(env, USERNAME_KEYS, DEFAULT_USERNAME);
 		this.password = extractPassword(env);
-		this.database = extractDatabase(env);
+		this.database = extract(env, DATABASE_KEYS, this.username);
 	}
 
-	private String extractUsername(Map<String, String> env) {
-		for (String key : USERNAME_KEYS) {
+	private String extract(Map<String, String> env, String[] keys, String defaultValue) {
+		for (String key : keys) {
 			if (env.containsKey(key)) {
 				return env.get(key);
 			}
 		}
-		return DEFAULT_USERNAME;
+		return defaultValue;
 	}
 
 	private String extractPassword(Map<String, String> env) {
@@ -67,15 +67,6 @@ private String extractPassword(Map<String, String> env) {
 		return password;
 	}
 
-	private String extractDatabase(Map<String, String> env) {
-		for (String key : DATABASE_KEYS) {
-			if (env.containsKey(key)) {
-				return env.get(key);
-			}
-		}
-		return this.username;
-	}
-
 	String getUsername() {
 		return this.username;
 	}

spring-boot-project/spring-boot-docker-compose/src/test/java/org/springframework/boot/docker/compose/service/connection/postgres/PostgresEnvironmentTests.java

@@ -67,13 +67,6 @@ void getUsernameWhenHasPostgresqlUser() {
 		assertThat(environment.getUsername()).isEqualTo("me");
 	}
 
-	@Test
-	void getUsernameWhenHasPostgresUsername() {
-		PostgresEnvironment environment = new PostgresEnvironment(
-				Map.of("POSTGRES_USERNAME", "me", "POSTGRESQL_PASSWORD", "secret"));
-		assertThat(environment.getUsername()).isEqualTo("me");
-	}
-
 	@Test
 	void getUsernameWhenHasPostgresqlUsername() {
 		PostgresEnvironment environment = new PostgresEnvironment(
@@ -134,9 +127,9 @@ void getDatabaseWhenHasPostgresDb() {
 	}
 
 	@Test
-	void getDatabaseWhenHasPostgresDatabase() {
+	void getDatabaseWhenHasPostgresqlDb() {
 		PostgresEnvironment environment = new PostgresEnvironment(
-				Map.of("POSTGRES_DATABASE", "db", "POSTGRESQL_PASSWORD", "secret"));
+				Map.of("POSTGRESQL_DB", "db", "POSTGRESQL_PASSWORD", "secret"));
 		assertThat(environment.getDatabase()).isEqualTo("db");
 	}