Java 17环境下Cassandra集成测试的替代方案咨询

1. 使用Cassandra官方嵌入式模式 #

Cassandra原生支持嵌入式启动，通过官方cassandra-all依赖可在JVM内直接启动实例，无需外部进程，完全兼容Java 17。

实现步骤： #

• 添加Maven依赖（选用4.1+版本，确保Java 17兼容性）：

<dependency>
    <groupId>org.apache.cassandra</groupId>
    <artifactId>cassandra-all</artifactId>
    <version>4.1.3</version>
    <scope>test</scope>
</dependency>

• 编写测试基类管理实例生命周期：

import org.apache.cassandra.config.DatabaseDescriptor;
import org.apache.cassandra.service.CassandraDaemon;
import org.junit.jupiter.api.AfterAll;
import org.junit.jupiter.api.BeforeAll;
import java.io.File;

public abstract class EmbeddedCassandraTest {
    private static CassandraDaemon cassandraDaemon;

    @BeforeAll
    static void startCassandra() throws Exception {
        // 配置临时目录，避免污染本地环境
        System.setProperty("cassandra.config", "classpath:cassandra-test.yaml");
        System.setProperty("cassandra.storagedir", new File("target/cassandra-test-data").getAbsolutePath());
        
        DatabaseDescriptor.daemonInitialization();
        cassandraDaemon = new CassandraDaemon();
        cassandraDaemon.init(null);
        cassandraDaemon.start();
    }

    @AfterAll
    static void stopCassandra() {
        if (cassandraDaemon != null) {
            cassandraDaemon.stop();
        }
    }
}

• 在test/resources下创建简化配置文件cassandra-test.yaml：

cluster_name: 'Test Cluster'
listen_address: 127.0.0.1
native_transport_port: 9042
seed_provider:
    - class_name: org.apache.cassandra.locator.SimpleSeedProvider
      parameters:
          - seeds: "127.0.0.1"
data_file_directories:
    - ${cassandra.storagedir}/data
commitlog_directory: ${cassandra.storagedir}/commitlog
saved_caches_directory: ${cassandra.storagedir}/saved_caches
start_native_transport: true
authenticator: AllowAllAuthenticator
authorizer: AllowAllAuthorizer

2. 本地Cassandra进程管理（适配CI预安装环境） #

若CI环境已预安装Cassandra，可通过代码控制本地进程启停，用临时配置隔离测试环境。

示例代码： #

import org.junit.jupiter.api.AfterAll;
import org.junit.jupiter.api.BeforeAll;
import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Path;

public class LocalCassandraTest {
    private static Process cassandraProcess;

    @BeforeAll
    static void startCassandra() throws IOException, InterruptedException {
        // 生成临时配置文件
        Path tempConfig = Files.createTempFile("cassandra-test", ".yaml");
        String configContent = """
            cluster_name: 'Test Cluster'
            listen_address: 127.0.0.1
            native_transport_port: 9042
            seed_provider:
                - class_name: org.apache.cassandra.locator.SimpleSeedProvider
                  parameters:
                      - seeds: "127.0.0.1"
            start_native_transport: true
            authenticator: AllowAllAuthenticator
            """;
        Files.write(tempConfig, configContent.getBytes());

        // 启动Cassandra进程
        ProcessBuilder pb = new ProcessBuilder("cassandra", "-f", "-Dcassandra.config=" + tempConfig.toAbsolutePath());
        cassandraProcess = pb.start();
        // 等待服务启动（可通过检查9042端口优化等待逻辑）
        Thread.sleep(10000);
    }

    @AfterAll
    static void stopCassandra() throws InterruptedException {
        if (cassandraProcess != null) {
            cassandraProcess.destroy();
            cassandraProcess.waitFor();
        }
    }
}

3. 驱动层Mock（轻量替代方案） #

若无需验证真实存储逻辑，仅需校验代码与Cassandra驱动的交互，可使用Mockito模拟驱动核心类。

示例： #

import com.datastax.oss.driver.api.core.CqlSession;
import com.datastax.oss.driver.api.core.cql.ResultSet;
import org.junit.jupiter.api.Test;
import org.mockito.Mock;
import org.mockito.junit.jupiter.MockitoExtension;
import org.junit.jupiter.api.extension.ExtendWith;
import static org.mockito.Mockito.when;

@ExtendWith(MockitoExtension.class)
public class CassandraMockTest {
    @Mock
    private CqlSession mockSession;
    @Mock
    private ResultSet mockResultSet;

    @Test
    void testDataFetch() {
        // 模拟查询返回结果
        when(mockSession.execute("SELECT * FROM test.user")).thenReturn(mockResultSet);
        
        // 注入mock对象到待测试服务
        UserService userService = new UserService(mockSession);
        userService.getUsers();
        
        // 验证交互逻辑（此处可补充Mockito断言）
    }
}

内容的提问来源于stack exchange，提问作者alexe