fix #988

Author: digital88

packages/modules/mongodb/src/mongodb-container.test.ts

@@ -64,4 +64,23 @@ describe("MongodbContainer", { timeout: 240_000 }, () => {
     await mongodbContainer.stop();
   });
   // }
+  test.for([["mongo:4.0.1"], ["mongo:6.0.1"], ["mongo:8.0"]])(
+    "should connect to %s with credentials and rs disabled",
+    async ([image]) => {
+      const mongodbContainer = await new MongoDBContainer(image)
+        .withUsername("mongo_user")
+        .withPassword("mongo_password")
+        .withReplicaSetEnabled(false)
+        .start();
+      const db = await mongoose.connect(mongodbContainer.getConnectionString(), { directConnection: true });
+      expect(db.connection.readyState).toBe(1);
+      const result = await db.connection.collection("testcontainers").insertOne({ title: "testcontainers" });
+      const id = result.insertedId.toString();
+      expect(id).not.toBeNull();
+      expect(id).not.toBe("");
+
+      await mongoose.disconnect();
+      await mongodbContainer.stop();
+    }
+  );
 });

packages/modules/mongodb/src/mongodb-container.ts

@@ -3,20 +3,49 @@ import { AbstractStartedContainer, ExecResult, GenericContainer, StartedTestCont
 const MONGODB_PORT = 27017;
 
 export class MongoDBContainer extends GenericContainer {
+  private username = "";
+  private password = "";
+  private initRs = true;
+
   constructor(image: string) {
     super(image);
     this.withExposedPorts(MONGODB_PORT)
-      .withCommand(["--replSet", "rs0"])
       .withWaitStrategy(Wait.forLogMessage(/.*waiting for connections.*/i))
       .withStartupTimeout(120_000);
   }
 
+  public withReplicaSetEnabled(enabled: boolean): this {
+    this.initRs = enabled;
+    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<StartedMongoDBContainer> {
-    return new StartedMongoDBContainer(await super.start());
+    if (this.username && this.password) {
+      this.withEnvironment({
+        MONGO_INITDB_ROOT_USERNAME: this.username,
+        MONGO_INITDB_ROOT_PASSWORD: this.password,
+      });
+    }
+    if (this.initRs) {
+      this.withCommand(["--replSet", "rs0"]);
+    }
+    return new StartedMongoDBContainer(await super.start(), this.username, this.password);
   }
 
   protected override async containerStarted(startedTestContainer: StartedTestContainer): Promise<void> {
-    await this.initReplicaSet(startedTestContainer);
+    if (this.initRs) {
+      await this.initReplicaSet(startedTestContainer);
+    }
   }
 
   private async initReplicaSet(startedTestContainer: StartedTestContainer) {
@@ -58,11 +87,18 @@ export class MongoDBContainer extends GenericContainer {
 }
 
 export class StartedMongoDBContainer extends AbstractStartedContainer {
-  constructor(startedTestContainer: StartedTestContainer) {
+  private readonly username: string = "";
+  private readonly password: string = "";
+
+  constructor(startedTestContainer: StartedTestContainer, username: string, password: string) {
     super(startedTestContainer);
+    this.username = username;
+    this.password = password;
   }
 
   public getConnectionString(): string {
+    if (this.username && this.password)
+      return `mongodb://${this.username}:${this.password}@${this.getHost()}:${this.getMappedPort(MONGODB_PORT)}`;
     return `mongodb://${this.getHost()}:${this.getMappedPort(MONGODB_PORT)}`;
   }
 }