flink-connector-jdbc/src/test/java/org/apache/flink/connector/jdbc/dialect/oracle/OracleExactlyOnceSinkE2eTest.java

package org.apache.flink.connector.jdbc.dialect.oracle;

import org.apache.flink.connector.jdbc.databases.oracle.OracleXaDatabase;
import org.apache.flink.connector.jdbc.xa.JdbcExactlyOnceSinkE2eTest;
import org.apache.flink.util.function.SerializableSupplier;

import oracle.jdbc.xa.client.OracleXADataSource;

import javax.sql.XADataSource;

import java.sql.SQLException;

/** A simple end-to-end test for {@link JdbcExactlyOnceSinkE2eTest}. */
public class OracleExactlyOnceSinkE2eTest extends JdbcExactlyOnceSinkE2eTest
        implements OracleXaDatabase {

    @Override
    public SerializableSupplier<XADataSource> getDataSourceSupplier() {
        return () -> {
            try {
                OracleXADataSource xaDataSource = new OracleXADataSource();
                xaDataSource.setURL(CONTAINER.getJdbcUrl());
                xaDataSource.setUser(CONTAINER.getUsername());
                xaDataSource.setPassword(CONTAINER.getPassword());
                return xaDataSource;
            } catch (SQLException ex) {
                throw new RuntimeException(ex);
            }
        };
    }
}