Made fixes and added to integration test

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

Author: svats0001

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

@@ -19,20 +19,50 @@
 import static org.assertj.core.api.Assertions.assertThat;
 import static org.junit.jupiter.api.Assertions.assertThrows;
 
+import java.time.Duration;
+
 import org.junit.Assume;
+import org.junit.jupiter.api.BeforeAll;
 import org.junit.jupiter.api.Test;
 import org.junit.jupiter.api.extension.ExtendWith;
 
 import io.asyncer.r2dbc.mysql.constant.SslMode;
 import io.asyncer.r2dbc.mysql.internal.util.TestContainerExtension;
 import io.asyncer.r2dbc.mysql.internal.util.TestServerUtil;
 import io.r2dbc.spi.R2dbcPermissionDeniedException;
+import io.r2dbc.spi.ValidationDepth;
+import reactor.core.publisher.Mono;
+import reactor.test.StepVerifier;
 
 @ExtendWith(TestContainerExtension.class)
 public class RsaEncryptionIntegrationTest {
+
+    @BeforeAll
+    static void initCachingSha2Password() {
+        // If the server uses caching_sha2_password, the first time a client connects to the server, the
+        // server will require a native SSL connection. So all the SSL tunnel tests should be run after
+        // the caching_sha2_password initialization.
+
+        MySqlConnectionConfiguration configuration = MySqlConnectionConfiguration.builder()
+            .host(TestServerUtil.getHost())
+            .port(TestServerUtil.getPort())
+            .connectTimeout(Duration.ofSeconds(3))
+            .user(TestServerUtil.getUsername())
+            .password(TestServerUtil.getPassword())
+            .database(TestServerUtil.getDatabase())
+            .createDatabaseIfNotExist(true)
+            .build();
+
+        MySqlConnectionFactory.from(configuration).create()
+            .flatMap(connection -> connection.validate(ValidationDepth.REMOTE)
+                .flatMap(it -> connection.close().then(Mono.just(it))))
+            .as(StepVerifier::create)
+            .expectNext(true)
+            .verifyComplete();
+    }
 
     @Test
-    void rsaEnabledSslDisabledTest() {
+    void sslDisabledTest() {
         Assume.assumeTrue(TestServerUtil.getServerVersion().getMajor() >= 8);
 
         MySqlConnectionConfiguration configuration = MySqlConnectionConfiguration.builder()
@@ -45,6 +75,11 @@ void rsaEnabledSslDisabledTest() {
             .build();
 
         assertThrows(R2dbcPermissionDeniedException.class, () -> MySqlConnectionFactory.from(configuration));
+    }
+
+    @Test
+    void sslDisabledInvalidRsaConfig() {
+        Assume.assumeTrue(TestServerUtil.getServerVersion().getMajor() >= 8);
 
         MySqlConnectionConfiguration configuration2 = MySqlConnectionConfiguration.builder()
             .host("localhost")
@@ -57,6 +92,11 @@ void rsaEnabledSslDisabledTest() {
             .build();
 
         assertThrows(IllegalArgumentException.class, () -> MySqlConnectionFactory.from(configuration2));
+    }
+
+    @Test
+    void sslDisabledValidRsaConfig() {
+        Assume.assumeTrue(TestServerUtil.getServerVersion().getMajor() >= 8);
 
         MySqlConnectionConfiguration configuration3 = MySqlConnectionConfiguration.builder()
             .host("localhost")
@@ -68,6 +108,6 @@ void rsaEnabledSslDisabledTest() {
             .serverRSAPublicKeyFile("/usr/local/mysql/data/public_key.pem")
             .build();
 
-        assertThat(MySqlConnectionFactory.from(configuration2)).isExactlyInstanceOf(MySqlConnectionFactory.class);
+        assertThat(MySqlConnectionFactory.from(configuration3)).isExactlyInstanceOf(MySqlConnectionFactory.class);
     }
 }