cassandra-container.ts

import { AbstractStartedContainer, GenericContainer, type StartedTestContainer } from "testcontainers";

const CASSANDRA_PORT = 9042;

export class CassandraContainer extends GenericContainer {
  private dc = "dc1";
  private rack = "rack1";
  private username = "cassandra";
  private password = "cassandra";

  constructor(image = "cassandra:5.0.2") {
    super(image);
    this.withExposedPorts(CASSANDRA_PORT).withStartupTimeout(120_000);
  }

  public withDatacenter(dc: string): this {
    this.dc = dc;
    return this;
  }

  public withRack(rack: string): this {
    this.rack = rack;
    return this;
  }

  public withUsername(username: string): this {
    this.username = username;
    return this;
  }

  public withPassword(password: string): this {
    this.password = password;
    return this;
  }

  public override async start(): Promise<StartedCassandraContainer> {
    this.withEnvironment({
      CASSANDRA_DC: this.dc,
      CASSANDRA_RACK: this.rack,
      CASSANDRA_LISTEN_ADDRESS: "auto",
      CASSANDRA_BROADCAST_ADDRESS: "auto",
      CASSANDRA_RPC_ADDRESS: "0.0.0.0",
      CASSANDRA_USERNAME: this.username,
      CASSANDRA_PASSWORD: this.password,
      CASSANDRA_SNITCH: "GossipingPropertyFileSnitch",
      CASSANDRA_ENDPOINT_SNITCH: "GossipingPropertyFileSnitch",
    });
    return new StartedCassandraContainer(await super.start(), this.dc, this.rack, this.username, this.password);
  }
}

export class StartedCassandraContainer extends AbstractStartedContainer {
  private readonly port: number;

  constructor(
    startedTestContainer: StartedTestContainer,
    private readonly dc: string,
    private readonly rack: string,
    private readonly username: string,
    private readonly password: string
  ) {
    super(startedTestContainer);
    this.port = startedTestContainer.getMappedPort(CASSANDRA_PORT);
  }

  public getPort(): number {
    return this.port;
  }

  public getDatacenter(): string {
    return this.dc;
  }

  public getRack(): string {
    return this.rack;
  }

  public getUsername(): string {
    return this.username;
  }

  public getPassword(): string {
    return this.password;
  }

  public getContactPoint(): string {
    return `${this.getHost()}:${this.getPort()}`;
  }
}