Neo4j Testcontainers 集成测试中加载数据库快照指南

.withCopyFileToContainer(MountableFile.forClasspathResource("neo4j.dump"), "/var/lib/neo4j/data/dumps/neo4j.dump")

#!/bin/bash -eu
/var/lib/neo4j/bin/neo4j-admin database load neo4j --from=/var/lib/neo4j/data/dumps/neo4j.dump --force
/startup/docker-entrypoint.sh neo4j

.withCopyToContainer(Transferable.of("""
    #!/bin/bash -eu
    /var/lib/neo4j/bin/neo4j-admin database load neo4j --from=/var/lib/neo4j/data/dumps/neo4j.dump --force
    /startup/docker-entrypoint.sh neo4j
    """, 0100555), "/startup/load-dump-and-start.sh")

.withCommand("/startup/load-dump-and-start.sh")

import org.junit.jupiter.api.AfterAll;
import org.junit.jupiter.api.Assertions;
import org.junit.jupiter.api.BeforeAll;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.TestInstance;
import org.neo4j.driver.AuthTokens;
import org.neo4j.driver.Driver;
import org.neo4j.driver.GraphDatabase;
import org.testcontainers.containers.Neo4jContainer;
import org.testcontainers.images.builder.Transferable;
import org.testcontainers.utility.MountableFile;

@TestInstance(TestInstance.Lifecycle.PER_CLASS)
public class LoadDumpTest {

    Neo4jContainer<?> neo4j;
    Driver driver;

    @BeforeAll
    void initNeo4j() {
        // 初始化Neo4j容器
        neo4j = new Neo4jContainer<>("neo4j:5.3.0") // 指定Neo4j版本
            // 将本地的neo4j.dump文件复制到容器内的指定路径
            .withCopyFileToContainer(MountableFile.forClasspathResource("neo4j.dump"),
                "/var/lib/neo4j/data/dumps/neo4j.dump")
            // 创建并复制自定义启动脚本到容器内，并赋予执行权限
            .withCopyToContainer(Transferable.of("""
                #!/bin/bash -eu
                # 加载数据库快照，--force 强制覆盖
                /var/lib/neo4j/bin/neo4j-admin database load neo4j --from=/var/lib/neo4j/data/dumps/neo4j.dump --force
                # 委托给原始的Neo4j启动脚本
                /startup/docker-entrypoint.sh neo4j
                """, 0100555), "/startup/load-dump-and-start.sh")
            // 将容器的启动命令设置为我们自定义的脚本
            .withCommand("/startup/load-dump-and-start.sh")
            // 捕获容器日志，方便调试
            .withLogConsumer(f -> System.out.print(f.getUtf8String()));

        neo4j.start(); // 启动容器
        // 获取Neo4j驱动，用于连接数据库
        driver = GraphDatabase.driver(neo4j.getBoltUrl(), AuthTokens.basic("neo4j", neo4j.getAdminPassword()));
    }

    @Test
    void dataShouldHaveBeenLoaded() {
        // 验证数据是否已成功加载
        try (var session = driver.session()) {
            var numNodes = session.run("MATCH (n) RETURN count(n)").single().get(0).asLong();
            Assertions.assertTrue(numNodes > 0, "数据库中应包含节点数据");
        }
    }

    @AfterAll
    void stopNeo4j() {
        // 测试结束后停止容器
        if (neo4j != null) {
            neo4j.stop();
        }
        if (driver != null) {
            driver.close();
        }
    }
}