Add MSSQLR2DBCDatabaseContainer compatible with org.testcontainers.mssqlserver.MSSQLServerContainer

Author: eddumelendez

modules/mssqlserver/src/main/java/org/testcontainers/mssqlserver/MSSQLR2DBCDatabaseContainer.java

@@ -0,0 +1,46 @@
+package org.testcontainers.mssqlserver;
+
+import io.r2dbc.mssql.MssqlConnectionFactoryProvider;
+import io.r2dbc.spi.ConnectionFactoryOptions;
+import org.testcontainers.r2dbc.R2DBCDatabaseContainer;
+
+public class MSSQLR2DBCDatabaseContainer implements R2DBCDatabaseContainer {
+
+    private final MSSQLServerContainer container;
+
+    public MSSQLR2DBCDatabaseContainer(MSSQLServerContainer container) {
+        this.container = container;
+    }
+
+    public static ConnectionFactoryOptions getOptions(MSSQLServerContainer container) {
+        ConnectionFactoryOptions options = ConnectionFactoryOptions
+            .builder()
+            .option(ConnectionFactoryOptions.DRIVER, MssqlConnectionFactoryProvider.MSSQL_DRIVER)
+            .build();
+
+        return new MSSQLR2DBCDatabaseContainer(container).configure(options);
+    }
+
+    @Override
+    public ConnectionFactoryOptions configure(ConnectionFactoryOptions options) {
+        return options
+            .mutate()
+            .option(ConnectionFactoryOptions.HOST, container.getHost())
+            .option(ConnectionFactoryOptions.PORT, container.getMappedPort(MSSQLServerContainer.MS_SQL_SERVER_PORT))
+            // TODO enable if/when MSSQLServerContainer adds support for customizing the DB name
+            // .option(ConnectionFactoryOptions.DATABASE, container.getDatabasseName())
+            .option(ConnectionFactoryOptions.USER, container.getUsername())
+            .option(ConnectionFactoryOptions.PASSWORD, container.getPassword())
+            .build();
+    }
+
+    @Override
+    public void start() {
+        this.container.start();
+    }
+
+    @Override
+    public void stop() {
+        this.container.stop();
+    }
+}

modules/mssqlserver/src/test/java/org/testcontainers/mssqlserver/MSSQLR2DBCDatabaseContainerTest.java

@@ -0,0 +1,23 @@
+package org.testcontainers.mssqlserver;
+
+import io.r2dbc.spi.ConnectionFactoryOptions;
+import org.testcontainers.MSSQLServerTestImages;
+import org.testcontainers.r2dbc.AbstractR2DBCDatabaseContainerTest;
+
+class MSSQLR2DBCDatabaseContainerTest extends AbstractR2DBCDatabaseContainerTest<MSSQLServerContainer> {
+
+    @Override
+    protected ConnectionFactoryOptions getOptions(MSSQLServerContainer container) {
+        return MSSQLR2DBCDatabaseContainer.getOptions(container);
+    }
+
+    @Override
+    protected String createR2DBCUrl() {
+        return "r2dbc:tc:sqlserver:///?TC_IMAGE_TAG=2022-CU14-ubuntu-22.04";
+    }
+
+    @Override
+    protected MSSQLServerContainer createContainer() {
+        return new MSSQLServerContainer(MSSQLServerTestImages.MSSQL_SERVER_IMAGE);
+    }
+}