external-software/github_OpenIdentityPlatform_OpenDJ.git

			@@ -11,7 +11,7 @@
			* Header, with the fields enclosed by brackets [] replaced by your own identifying
			* information: "Portions Copyright [year] [name of copyright owner]".
			*
			* Copyright 2025 3A Systems, LLC.
			* Copyright 2025-2026 3A Systems, LLC.
			*/
			package org.opends.server.backends.jdbc;

			@@ -19,9 +19,21 @@
			import org.testcontainers.containers.MySQLContainer;
			import org.testng.annotations.Test;

			import java.sql.Connection;
			import java.sql.DriverManager;
			import java.sql.SQLException;
			import java.sql.Statement;

			import static org.testng.Assert.assertEquals;
			import static org.testng.Assert.assertFalse;
			import static org.testng.Assert.assertTrue;
			import static org.testng.Assert.fail;

			//docker run --rm --name mysql -p 3306:3306 -e MYSQL_DATABASE=database_name -e MYSQL_ROOT_PASSWORD=password mysql:latest

			@Test
			// sequential, as every suite declaring a test of its own is: TestListener asks it of the class a
			// running test is declared by, and the inherited ones answer for the class that declares them
			@Test(sequential = true)
			public class MySqlTestCase extends TestCase {

			@Override
			@@ -48,4 +60,128 @@
			return "jdbc:mysql://root:password@localhost:" + ((container==null)?"3306":container.getMappedPort(3306)) + "/database_name";
			}

			/**
			* The vendor code a per-account connection limit is classified by is the code the server sends
			* for it, and it is the whole of what this verdict can be made of: an account whose grant caps
			* its simultaneous connections refuses the next connect with 1226 in the syntax error class -
			* 42000, where a statement the database rejected lands - rather than in a connection class of
			* its own. The unit tests pin what the pool does with the code; this pins that the code is the
			* one arriving from a real server through the driver this backend ships with (#1011).
			*/
			@Test(timeOut = 120000)
			public void testAPerAccountConnectionLimitIsWorthRetrying() throws Exception {
			final String url = getJdbcUrl();
			final String limited = url.replace("root:password", "limited1011:secret");
			// dropped first: a run this one was killed in the middle of leaves the account behind
			grant(url, "drop user if exists 'limited1011'@'%'",
			"create user 'limited1011'@'%' identified by 'secret' with max_user_connections 1",
			"grant all on database_name.* to 'limited1011'@'%'");
			try (final Connection held = DriverManager.getConnection(limited)) {
			assertTrue(held.isValid(CachedConnection.VALIDATION_TIMEOUT_SECONDS),
			"the account is refused its first connection already");
			try (final Connection second = DriverManager.getConnection(limited)) {
			fail("an account limited to one connection must be refused a second: " + second);
			} catch (SQLException refused) {
			assertEquals(refused.getErrorCode(), 1226,
			"the server reports a per-account connection limit as: " + refused);
			// the resource is what tells this code from the ones no wait clears, and the server
			// writes it as a literal of its own: read here from the server rather than assumed
			assertTrue(refused.getMessage().contains("'max_user_connections'"),
			"the server names the resource of the concurrent limit as: " + refused.getMessage());
			assertTrue(CachedConnection.isWorthRetrying(refused, CachedConnection.ConnectDialect.of(limited)),
			"a borrow must wait a per-account limit out rather than fail on it: " + refused);
			}
			} finally {
			grant(url, "drop user if exists 'limited1011'@'%'");
			}
			}

			/**
			* The connections an account is granted per hour carry the same 1226 as the ones it may hold at
			* once, and only the top of the hour clears them: a borrow waiting one out would park a worker
			* thread in every attempt for as long as the hour lasts. What tells the two apart is the
			* resource the server names, and this pins that name against the server itself (#1011).
			*/
			@Test(timeOut = 120000)
			public void testTheHourlyConnectionLimitIsNotWorthRetrying() throws Exception {
			final String url = getJdbcUrl();
			final String limited = url.replace("root:password", "perhour1011:secret");
			grant(url, "drop user if exists 'perhour1011'@'%'",
			"create user 'perhour1011'@'%' identified by 'secret' with max_connections_per_hour 1",
			"grant all on database_name.* to 'perhour1011'@'%'");
			try {
			try (final Connection spent = DriverManager.getConnection(limited)) {
			assertTrue(spent.isValid(CachedConnection.VALIDATION_TIMEOUT_SECONDS),
			"the account is refused the one connection of its hour already");
			}
			try (final Connection second = DriverManager.getConnection(limited)) {
			fail("an account granted one connection an hour must be refused a second: " + second);
			} catch (SQLException refused) {
			assertEquals(refused.getErrorCode(), 1226,
			"the server reports an hourly connection limit as: " + refused);
			assertTrue(refused.getMessage().contains("'max_connections_per_hour'"),
			"the server names the hourly resource as: " + refused.getMessage());
			assertFalse(CachedConnection.isWorthRetrying(refused, CachedConnection.ConnectDialect.of(limited)),
			"an hourly limit must be reported rather than waited out: " + refused);
			}
			} finally {
			grant(url, "drop user if exists 'perhour1011'@'%'");
			}
			}

			/**
			* The queries an account is granted per hour are named by the server without the _per_hour of
			* the GRANT keyword - max_questions - and they reach this gate on the road it guards: an
			* account whose quota is spent is refused the connect itself, Connector/J spending what is
			* left of the quota on the queries of its own login. Waited out, that would cost every borrow
			* and every catalog connect the whole deadline of the pool until the hour turned (#1011).
			*/
			@Test(timeOut = 120000)
			public void testTheHourlyQueryLimitIsNotWorthRetrying() throws Exception {
			final String url = getJdbcUrl();
			final String limited = url.replace("root:password", "hourly1011:secret");
			grant(url, "drop user if exists 'hourly1011'@'%'",
			"create user 'hourly1011'@'%' identified by 'secret' with max_queries_per_hour 1",
			"grant all on database_name.* to 'hourly1011'@'%'");
			try {
			final SQLException refused = spendTheHourlyQueries(limited);
			assertEquals(refused.getErrorCode(), 1226,
			"the server reports an hourly query limit as: " + refused);
			assertTrue(refused.getMessage().contains("'max_questions'"),
			"the server names the hourly query resource as: " + refused.getMessage());
			assertFalse(CachedConnection.isWorthRetrying(refused, CachedConnection.ConnectDialect.of(limited)),
			"an hourly query limit must be reported rather than waited out: " + refused);
			} finally {
			grant(url, "drop user if exists 'hourly1011'@'%'");
			}
			}

			/**
			* Spends the hourly query quota of an account and hands back what the server refused it with.
			* The connect is attempted rather than one statement over a connection held open, since the
			* login of the driver spends the quota as readily as a statement does: whichever of the two
			* meets the limit, the failure is the one a borrow of this pool would catch.
			*/
			private static SQLException spendTheHourlyQueries(String url) throws SQLException {
			for (int attempt = 0; attempt < 4; attempt++) {
			try (final Connection con = DriverManager.getConnection(url);
			final Statement st = con.createStatement()) {
			st.execute("select 1");
			} catch (SQLException refused) {
			return refused;
			}
			}
			throw new AssertionError("an account granted one query an hour must be refused within four connects");
			}

			/** The account of the test is made and unmade on the connection of the suite's own credentials. */
			private static void grant(String url, String... statements) throws SQLException {
			try (final Connection admin = DriverManager.getConnection(url);
			final Statement st = admin.createStatement()) {
			for (final String statement : statements) {
			st.execute(statement);
			}
			}
			}

			}