Debug

Author: stscoundrel

packages/modules/mongodb/src/keyfile.ts

@@ -1,12 +1,10 @@
 import { AbstractStartedContainer, ExecResult, GenericContainer, StartedTestContainer, Wait } from "testcontainers";
-import createKeyfile from "./keyfile";
 
 const MONGODB_PORT = 27017;
 
 export class MongoDBContainer extends GenericContainer {
   private username: string | null = null;
   private password: string | null = null;
-  private keyfilePath: string | null = null;
 
   constructor(image = "mongo:4.0.1") {
     super(image);
@@ -28,34 +26,32 @@ export class MongoDBContainer extends GenericContainer {
 
   public override async start(): Promise<StartedMongoDBContainer> {
     if (this.username && this.password) {
-      this.keyfilePath = createKeyfile();
-      const containerKeyfilePath = "/etc/mongo-keyfile";
-      this.withBindMounts([
-        {
-          source: this.keyfilePath,
-          target: containerKeyfilePath,
-          mode: "ro",
-        },
+      const containerKeyfilePath = "/tmp/mongo-keyfile";
+      this.withCommand([
+        "/bin/sh",
+        "-c",
+        `
+        openssl rand -base64 756 > ${containerKeyfilePath} &&
+        chmod 600 ${containerKeyfilePath} &&
+        chown mongodb:mongodb ${containerKeyfilePath} &&
+        exec mongod --replSet rs0 --keyFile ${containerKeyfilePath} --bind_ip_all
+        `,
       ]);
-      this.withCommand(["--replSet", "rs0", "--keyFile", containerKeyfilePath, "--bind_ip_all"]);
       this.withEnvironment({ MONGO_INITDB_ROOT_USERNAME: this.username, MONGO_INITDB_ROOT_PASSWORD: this.password });
     }
 
     return new StartedMongoDBContainer(await super.start(), this.username, this.password);
   }
 
   protected override async containerStarted(startedTestContainer: StartedTestContainer): Promise<void> {
-    if (this.username && this.password) {
-      await this.waitForRootUser(startedTestContainer);
-    }
+    // if (this.username && this.password) {
+    //   await this.waitForRootUser(startedTestContainer);
+    // }
     await this.initReplicaSet(startedTestContainer);
   }
 
   private async initReplicaSet(startedTestContainer: StartedTestContainer) {
-    await this.executeMongoEvalCommand(
-      startedTestContainer,
-      "rs.initiate({ _id: 'rs0', members: [ { _id: 0, host: '127.0.0.1:27017' } ] });"
-    );
+    await this.executeMongoEvalCommand(startedTestContainer, `rs.initiate();`);
     await this.executeMongoEvalCommand(startedTestContainer, this.buildMongoWaitCommand());
   }
 
@@ -68,17 +64,7 @@ export class MongoDBContainer extends GenericContainer {
     const cmd = [this.getMongoCmdBasedOnImageTag()];
 
     if (this.username && this.password) {
-      cmd.push(
-        "admin",
-        "--port",
-        MONGODB_PORT.toString(),
-        "-u",
-        this.username,
-        "-p",
-        this.password,
-        "--authenticationDatabase",
-        "admin"
-      );
+      cmd.push("--username", this.username, "--password", this.password, "--authenticationDatabase", "admin");
     }
 
     cmd.push("--eval", command);
@@ -97,19 +83,23 @@ export class MongoDBContainer extends GenericContainer {
     }
   }
 
-  private async waitForRootUser(startedTestContainer: StartedTestContainer) {
-    const checkCommand = this.buildMongoEvalCommand("db.runCommand({ connectionStatus: 1 })");
-
-    for (let i = 0; i < 30; i++) {
-      const result = await startedTestContainer.exec(checkCommand);
-      if (result.exitCode === 0 && result.output.includes(`${this.username}`)) {
-        return;
-      }
-      await new Promise((res) => setTimeout(res, 1000));
-    }
-
-    throw new Error("Root user not ready after 30s");
-  }
+  // private async waitForRootUser(startedTestContainer: StartedTestContainer) {
+  //   const checkCommand = this.buildMongoEvalCommand("db.runCommand({ connectionStatus: 1 })");
+
+  //   for (let i = 0; i < 30; i++) {
+  //     const result = await startedTestContainer.exec(checkCommand);
+  //     if (
+  //       result.exitCode === 0 &&
+  //       result.output.includes(`${this.username}`) &&
+  //       !result.output.includes(`UserNotFound`)
+  //     ) {
+  //       return;
+  //     }
+  //     await new Promise((res) => setTimeout(res, 1000));
+  //   }
+
+  //   throw new Error("Root user not ready after 30s");
+  // }
 
   private buildMongoWaitCommand() {
     return `