|
| 1 | +package com.bencodez.simpleapi.sql.mysql; |
| 2 | + |
| 3 | +import static org.junit.jupiter.api.Assertions.*; |
| 4 | + |
| 5 | +import java.lang.reflect.InvocationTargetException; |
| 6 | +import java.nio.file.Files; |
| 7 | +import java.nio.file.Path; |
| 8 | +import javax.tools.ToolProvider; |
| 9 | +import org.junit.jupiter.api.Test; |
| 10 | +import org.junit.jupiter.api.io.TempDir; |
| 11 | +import com.bencodez.simpleapi.tests.shared.SharedRuntimeClasspath; |
| 12 | + |
| 13 | +class MariaDbDriverFallbackTest { |
| 14 | + @TempDir Path directory; |
| 15 | + |
| 16 | + @Test void mysqlFallbackUsesMysqlSchemeAndKeepsConfiguredOptions() { |
| 17 | + ConnectionManager manager = manager(DbType.MARIADB); |
| 18 | + manager.setUseSSL(true); |
| 19 | + manager.setPublicKeyRetrieval(true); |
| 20 | + manager.setStr("&customOption=retained"); |
| 21 | + String url = manager.buildJdbcUrl("com.mysql.cj.jdbc.Driver"); |
| 22 | + assertTrue(url.startsWith("jdbc:mysql://localhost:3306/votes?")); |
| 23 | + assertTrue(url.contains("useSSL=true")); |
| 24 | + assertTrue(url.contains("allowPublicKeyRetrieval=true")); |
| 25 | + assertTrue(url.contains("&customOption=retained")); |
| 26 | + assertTrue(url.endsWith("&sslMode=REQUIRED&requireSSL=true")); |
| 27 | + assertEquals(DbType.MARIADB, manager.getDbType()); |
| 28 | + } |
| 29 | + |
| 30 | + @Test void nativeMariaDbDriverRetainsMariaDbScheme() { |
| 31 | + assertTrue(manager(DbType.MARIADB).buildJdbcUrl("org.mariadb.jdbc.Driver").startsWith("jdbc:mariadb:")); |
| 32 | + } |
| 33 | + |
| 34 | + @Test void explicitMariaDbDriverOverridesMysqlTypeAsBefore() { |
| 35 | + assertTrue(manager(DbType.MYSQL).buildJdbcUrl("org.mariadb.jdbc.Driver").startsWith("jdbc:mariadb:")); |
| 36 | + } |
| 37 | + |
| 38 | + @Test void customMariaDbDriverRetainsConfiguredScheme() { |
| 39 | + assertTrue(manager(DbType.MARIADB).buildJdbcUrl("example.CustomDriver").startsWith("jdbc:mariadb:")); |
| 40 | + } |
| 41 | + |
| 42 | + @Test void mysqlAndPostgresqlSelectionsRemainUnchanged() { |
| 43 | + assertTrue(manager(DbType.MYSQL).buildJdbcUrl("com.mysql.cj.jdbc.Driver").startsWith("jdbc:mysql:")); |
| 44 | + assertTrue(manager(DbType.POSTGRESQL).buildJdbcUrl("org.postgresql.Driver").startsWith("jdbc:postgresql:")); |
| 45 | + } |
| 46 | + |
| 47 | + @Test void absentMariaDbDriverResolvesToDriverThatAcceptsGeneratedUrl() throws Exception { |
| 48 | + try (var loader = isolatedMysqlDriver()) { |
| 49 | + assertThrows(ClassNotFoundException.class, () -> loader.loadClass("org.mariadb.jdbc.Driver")); |
| 50 | + Class<?> type = loader.loadClass(ConnectionManager.class.getName()); |
| 51 | + Object manager = type.getConstructor(String.class, String.class, String.class, String.class, String.class) |
| 52 | + .newInstance("localhost", "3306", "test", "", "votes"); |
| 53 | + Class<?> dbType = loader.loadClass(DbType.class.getName()); |
| 54 | + type.getMethod("setDbType", dbType).invoke(manager, dbType.getField("MARIADB").get(null)); |
| 55 | + var resolve = type.getDeclaredMethod("resolveDriver"); resolve.setAccessible(true); |
| 56 | + assertEquals("com.mysql.cj.jdbc.Driver", resolve.invoke(manager)); |
| 57 | + var build = type.getDeclaredMethod("buildJdbcUrl", String.class); build.setAccessible(true); |
| 58 | + String url = (String) build.invoke(manager, resolve.invoke(manager)); |
| 59 | + java.sql.Driver driver = (java.sql.Driver) loader.loadClass("com.mysql.cj.jdbc.Driver") |
| 60 | + .getConstructor().newInstance(); |
| 61 | + assertTrue(driver.acceptsURL(url)); |
| 62 | + type.getMethod("setMariadbFallbackToMysqlDriver", boolean.class).invoke(manager, false); |
| 63 | + InvocationTargetException rejected = assertThrows(InvocationTargetException.class, () -> resolve.invoke(manager)); |
| 64 | + assertInstanceOf(ClassNotFoundException.class, rejected.getCause()); |
| 65 | + } |
| 66 | + } |
| 67 | + |
| 68 | + private java.net.URLClassLoader isolatedMysqlDriver() throws Exception { |
| 69 | + Path source = directory.resolve("com/mysql/cj/jdbc/Driver.java"); |
| 70 | + Files.createDirectories(source.getParent()); |
| 71 | + Files.writeString(source, """ |
| 72 | + package com.mysql.cj.jdbc; |
| 73 | + public class Driver implements java.sql.Driver { |
| 74 | + public boolean acceptsURL(String url) { return url.startsWith("jdbc:mysql:"); } |
| 75 | + public java.sql.Connection connect(String url, java.util.Properties p) { return null; } |
| 76 | + public java.sql.DriverPropertyInfo[] getPropertyInfo(String u, java.util.Properties p) { return new java.sql.DriverPropertyInfo[0]; } |
| 77 | + public int getMajorVersion() { return 1; } |
| 78 | + public int getMinorVersion() { return 0; } |
| 79 | + public boolean jdbcCompliant() { return false; } |
| 80 | + public java.util.logging.Logger getParentLogger() { return java.util.logging.Logger.getGlobal(); } |
| 81 | + } |
| 82 | + """); |
| 83 | + assertEquals(0, ToolProvider.getSystemJavaCompiler().run(null, null, null, |
| 84 | + "-proc:none", "-d", directory.toString(), source.toString())); |
| 85 | + return SharedRuntimeClasspath.open(ConnectionManager.class.getProtectionDomain().getCodeSource().getLocation(), |
| 86 | + directory.toUri().toURL()); |
| 87 | + } |
| 88 | + |
| 89 | + private ConnectionManager manager(DbType type) { |
| 90 | + ConnectionManager result = new ConnectionManager("localhost", "3306", "test", "", "votes"); |
| 91 | + result.setDbType(type); |
| 92 | + return result; |
| 93 | + } |
| 94 | +} |
0 commit comments