Skip to content

Commit a0df99d

Browse files
authored
Add MySQLR2DBCDatabaseContainer compatible with org.testcontainers.mysql.MySQLContainer (#11119)
1 parent ca450d0 commit a0df99d

File tree

2 files changed

+81
-0
lines changed

2 files changed

+81
-0
lines changed
Lines changed: 58 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,58 @@
1+
package org.testcontainers.mysql;
2+
3+
import io.asyncer.r2dbc.mysql.MySqlConnectionFactoryProvider;
4+
import io.r2dbc.spi.ConnectionFactoryOptions;
5+
import org.testcontainers.lifecycle.Startable;
6+
import org.testcontainers.r2dbc.R2DBCDatabaseContainer;
7+
8+
import java.util.Set;
9+
10+
public class MySQLR2DBCDatabaseContainer implements R2DBCDatabaseContainer {
11+
12+
private final MySQLContainer container;
13+
14+
public MySQLR2DBCDatabaseContainer(MySQLContainer container) {
15+
this.container = container;
16+
}
17+
18+
public static ConnectionFactoryOptions getOptions(MySQLContainer container) {
19+
ConnectionFactoryOptions options = ConnectionFactoryOptions
20+
.builder()
21+
.option(ConnectionFactoryOptions.DRIVER, MySqlConnectionFactoryProvider.MYSQL_DRIVER)
22+
.build();
23+
24+
return new MySQLR2DBCDatabaseContainer(container).configure(options);
25+
}
26+
27+
@Override
28+
public ConnectionFactoryOptions configure(ConnectionFactoryOptions options) {
29+
return options
30+
.mutate()
31+
.option(ConnectionFactoryOptions.HOST, container.getHost())
32+
.option(ConnectionFactoryOptions.PORT, container.getMappedPort(MySQLContainer.MYSQL_PORT))
33+
.option(ConnectionFactoryOptions.DATABASE, container.getDatabaseName())
34+
.option(ConnectionFactoryOptions.USER, container.getUsername())
35+
.option(ConnectionFactoryOptions.PASSWORD, container.getPassword())
36+
.build();
37+
}
38+
39+
@Override
40+
public Set<Startable> getDependencies() {
41+
return this.container.getDependencies();
42+
}
43+
44+
@Override
45+
public void start() {
46+
this.container.start();
47+
}
48+
49+
@Override
50+
public void stop() {
51+
this.container.stop();
52+
}
53+
54+
@Override
55+
public void close() {
56+
this.container.close();
57+
}
58+
}
Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,23 @@
1+
package org.testcontainers.mysql;
2+
3+
import io.r2dbc.spi.ConnectionFactoryOptions;
4+
import org.testcontainers.MySQLTestImages;
5+
import org.testcontainers.r2dbc.AbstractR2DBCDatabaseContainerTest;
6+
7+
public class MySQLR2DBCDatabaseContainerTest extends AbstractR2DBCDatabaseContainerTest<MySQLContainer> {
8+
9+
@Override
10+
protected ConnectionFactoryOptions getOptions(MySQLContainer container) {
11+
return MySQLR2DBCDatabaseContainer.getOptions(container);
12+
}
13+
14+
@Override
15+
protected String createR2DBCUrl() {
16+
return "r2dbc:tc:mysql:///db?TC_IMAGE_TAG=" + MySQLTestImages.MYSQL_80_IMAGE.getVersionPart();
17+
}
18+
19+
@Override
20+
protected MySQLContainer createContainer() {
21+
return new MySQLContainer(MySQLTestImages.MYSQL_80_IMAGE);
22+
}
23+
}

0 commit comments

Comments
 (0)