Debug

Author: stscoundrel

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

@@ -1,5 +1,10 @@
 import { AbstractStartedContainer, ExecResult, GenericContainer, StartedTestContainer, Wait } from "testcontainers";
 
+import crypto from "crypto";
+import { promises as fs } from "fs";
+import os from "os";
+import path from "path";
+
 const MONGODB_PORT = 27017;
 
 export class MongoDBContainer extends GenericContainer {
@@ -26,32 +31,36 @@ export class MongoDBContainer extends GenericContainer {
 
   public override async start(): Promise<StartedMongoDBContainer> {
     if (this.username && this.password) {
-      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
-        `,
+      const keyfileContainerPath = "/tmp/mongodb-keyfile";
+
+      const keyfileContent = crypto.randomBytes(32).toString("base64");
+
+      const tempDir = await fs.mkdtemp(path.join(os.tmpdir(), "mongo-keyfile-"));
+      const keyfileHostPath = path.join(tempDir, "mongodb-keyfile");
+      await fs.writeFile(keyfileHostPath, keyfileContent, { mode: 0o600 });
+
+      this.withCopyFilesToContainer([
+        {
+          source: keyfileHostPath,
+          target: keyfileContainerPath,
+          mode: 0o600,
+        },
       ]);
+
+      this.withCommand(["--replSet", "rs0", "--keyFile", keyfileContainerPath, "--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);
-    // }
     await this.initReplicaSet(startedTestContainer);
   }
 
   private async initReplicaSet(startedTestContainer: StartedTestContainer) {
-    await this.executeMongoEvalCommand(startedTestContainer, `rs.initiate();`);
+    await this.executeMongoEvalCommand(startedTestContainer, "rs.initiate();");
     await this.executeMongoEvalCommand(startedTestContainer, this.buildMongoWaitCommand());
   }
 
@@ -64,7 +73,7 @@ export class MongoDBContainer extends GenericContainer {
     const cmd = [this.getMongoCmdBasedOnImageTag()];
 
     if (this.username && this.password) {
-      cmd.push("--username", this.username, "--password", this.password, "--authenticationDatabase", "admin");
+      cmd.push("--username", this.username, "--password", this.password);
     }
 
     cmd.push("--eval", command);
@@ -83,24 +92,6 @@ 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}`) &&
-  //       !result.output.includes(`UserNotFound`)
-  //     ) {
-  //       return;
-  //     }
-  //     await new Promise((res) => setTimeout(res, 1000));
-  //   }
-
-  //   throw new Error("Root user not ready after 30s");
-  // }
-
   private buildMongoWaitCommand() {
     return `
     var attempt = 0;