testcontainers - 在初始化脚本之前在 postgres 测试容器中创建一个文件夹

Question

我有以下代码：

public static PostgreSQLContainer<?> postgreDBContainer = new PostgreSQLContainer<>("postgres:12")
        .withInitScript("init-database-test.sql")
        .withUsername("dba")
        .withPassword("dba");

在初始化脚本中，我正在创建一些表空间并关联文件夹：

CREATE TABLESPACE tsd01 OWNER dba LOCATION '/tsd01';
CREATE TABLESPACE tsi01 OWNER dba LOCATION '/tsi01';
CREATE TABLESPACE tsisecurity01 OWNER dba LOCATION '/tsisecurity01';

这些表空间文件夹应该在初始化脚本运行之前创建。我怎么能做到这一点？

标签： testcontainerstestcontainers-junit5

Answer 1

我能够通过扩展默认值PostgreSQLContainer并更改containerIsStarted方法来解决此问题：

public class CustomPostgreSQLContainer<SELF extends CustomPostgreSQLContainer<SELF>> extends PostgreSQLContainer<SELF> {

    private static final Logger log = LoggerFactory.getLogger(CustomPostgreSQLContainer.class);

    public CustomPostgreSQLContainer() {
        super("postgres:12");
    }

    @Override
    protected void containerIsStarted(InspectContainerResponse containerInfo) {
        try {
            log.debug("M=containerIsStarted, creating database namespace folders and setting permissions");
            execInContainer("mkdir", "/tsd01");
            execInContainer("chown", "-R", "postgres.postgres", "/tsd01/");
            execInContainer("mkdir", "/tsi01");
            execInContainer("chown", "-R", "postgres.postgres", "/tsi01/");
            execInContainer("mkdir", "/tsisecurity01");
            execInContainer("chown", "-R", "postgres.postgres", "/tsisecurity01/");
        } catch (IOException | InterruptedException e) {
            e.printStackTrace();
        }
        super.containerIsStarted(containerInfo);
    }
}