Fixed integration test

Signed-off-by: S V <vats02581@gmail.com>

Author: svats0001

r2dbc-mysql/src/test/java/io/asyncer/r2dbc/mysql/RsaEncryptionIntegrationTest.java

@@ -36,8 +36,10 @@
 import io.netty.handler.ssl.SslContext;
 import io.netty.handler.ssl.SslContextBuilder;
 import io.netty.handler.ssl.util.SelfSignedCertificate;
+import io.r2dbc.spi.R2dbcPermissionDeniedException;
 import io.r2dbc.spi.ValidationDepth;
 
+import org.junit.Assume;
 import org.junit.jupiter.api.AfterEach;
 import org.junit.jupiter.api.BeforeAll;
 import org.junit.jupiter.api.BeforeEach;
@@ -52,6 +54,7 @@
 import java.time.Duration;
 
 import static org.assertj.core.api.Assertions.assertThat;
+import static org.junit.Assert.assertThrows;
 
 @ExtendWith(TestContainerExtension.class)
 public class SslTunnelIntegrationTest {
@@ -131,6 +134,59 @@ void sslTunnelConnectionTest() {
         connection.close().block();
     }
 
+    @Test
+    void sslDisabledTest() {
+        Assume.assumeTrue(TestServerUtil.getServerVersion().getMajor() >= 8);
+
+        MySqlConnectionConfiguration configuration = MySqlConnectionConfiguration.builder()
+            .host("localhost")
+            .port(sslTunnelServer.getLocalPort())
+            .user(TestServerUtil.getUsername())
+            .password(TestServerUtil.getPassword())
+            .database(TestServerUtil.getDatabase())
+            .createDatabaseIfNotExist(true)
+            .sslMode(SslMode.DISABLED)
+            .build();
+            
+        assertThrows(R2dbcPermissionDeniedException.class, () -> MySqlConnectionFactory.from(configuration).create().block());
+    }
+
+    @Test
+    void sslDisabledInvalidRsaConfig() {
+        Assume.assumeTrue(TestServerUtil.getServerVersion().getMajor() >= 8);
+
+        MySqlConnectionConfiguration configuration2 = MySqlConnectionConfiguration.builder()
+            .host("localhost")
+            .port(sslTunnelServer.getLocalPort())
+            .user(TestServerUtil.getUsername())
+            .password(TestServerUtil.getPassword())
+            .database(TestServerUtil.getDatabase())
+            .createDatabaseIfNotExist(true)
+            .sslMode(SslMode.DISABLED)
+            .serverRSAPublicKeyFile("/path/to/mysql/serverRSAPublicKey.pem")
+            .build();
+
+        assertThrows(IllegalArgumentException.class, () -> MySqlConnectionFactory.from(configuration2).create().block());
+    }
+
+    @Test
+    void sslDisabledValidRsaConfig() {
+        Assume.assumeTrue(TestServerUtil.getServerVersion().getMajor() >= 8);
+
+        MySqlConnectionConfiguration configuration3 = MySqlConnectionConfiguration.builder()
+            .host("localhost")
+            .port(sslTunnelServer.getLocalPort())
+            .user(TestServerUtil.getUsername())
+            .password(TestServerUtil.getPassword())
+            .database(TestServerUtil.getDatabase())
+            .createDatabaseIfNotExist(true)
+            .sslMode(SslMode.DISABLED)
+            .serverRSAPublicKeyFile("/usr/local/mysql/data/public_key.pem")
+            .build();
+
+        assertThat(MySqlConnectionFactory.from(configuration3).create().block()).isExactlyInstanceOf(MySqlSimpleConnection.class);
+    }
+
     private static class SslTunnelServer {
 
         private final String remoteHost;