|
| 1 | +package org.testcontainers.containers; |
| 2 | + |
| 3 | +import com.google.auto.service.AutoService; |
| 4 | +import io.r2dbc.spi.ConnectionFactoryMetadata; |
| 5 | +import io.r2dbc.spi.ConnectionFactoryOptions; |
| 6 | +import org.jetbrains.annotations.Nullable; |
| 7 | +import org.testcontainers.r2dbc.R2DBCDatabaseContainer; |
| 8 | +import org.testcontainers.r2dbc.R2DBCDatabaseContainerProvider; |
| 9 | + |
| 10 | +@AutoService(R2DBCDatabaseContainerProvider.class) |
| 11 | +public class OracleR2DBCDatabaseContainerProvider implements R2DBCDatabaseContainerProvider { |
| 12 | + |
| 13 | + static final String DRIVER = "oracle"; |
| 14 | + |
| 15 | + @Override |
| 16 | + public boolean supports(ConnectionFactoryOptions options) { |
| 17 | + return DRIVER.equals(options.getRequiredValue(ConnectionFactoryOptions.DRIVER)); |
| 18 | + } |
| 19 | + |
| 20 | + @Override |
| 21 | + public R2DBCDatabaseContainer createContainer(ConnectionFactoryOptions options) { |
| 22 | + String image = OracleContainer.IMAGE + ":" + options.getRequiredValue(IMAGE_TAG_OPTION); |
| 23 | + OracleContainer container = new OracleContainer(image) |
| 24 | + .withDatabaseName((String) options.getRequiredValue(ConnectionFactoryOptions.DATABASE)); |
| 25 | + if (Boolean.TRUE.equals(options.getValue(REUSABLE_OPTION))) { |
| 26 | + container.withReuse(true); |
| 27 | + } |
| 28 | + return new OracleR2DBCDatabaseContainer(container); |
| 29 | + } |
| 30 | + |
| 31 | + @Nullable |
| 32 | + @Override |
| 33 | + public ConnectionFactoryMetadata getMetadata(ConnectionFactoryOptions options) { |
| 34 | + ConnectionFactoryOptions.Builder builder = options.mutate(); |
| 35 | + if (!options.hasOption(ConnectionFactoryOptions.USER)) { |
| 36 | + builder.option(ConnectionFactoryOptions.USER, OracleContainer.APP_USER); |
| 37 | + } |
| 38 | + if (!options.hasOption(ConnectionFactoryOptions.PASSWORD)) { |
| 39 | + builder.option(ConnectionFactoryOptions.PASSWORD, OracleContainer.APP_USER_PASSWORD); |
| 40 | + } |
| 41 | + return R2DBCDatabaseContainerProvider.super.getMetadata(builder.build()); |
| 42 | + } |
| 43 | +} |
0 commit comments