testcontainers 8.11.0 → 8.13.0

package/README.md

@@ -162,6 +162,17 @@ const container2 = await new GenericContainer("alpine")
 assert(container1.getId() === container2.getId());
 ```
 
+Restarting a container:
+
+```javascript
+const { GenericContainer } = require("testcontainers");
+
+const container = await new GenericContainer("alpine")
+  .start();
+  
+await container.restart();
+```
+
 Creating a container with a specified name:
 
 ```javascript

package/dist/docker/docker-client.d.ts

@@ -5,5 +5,5 @@ declare type DockerClient = {
     host: Host;
     dockerode: Dockerode;
 };
-export declare const dockerClient: Promise<DockerClient>;
+export declare const dockerClient: () => Promise<DockerClient>;
 export {};

package/dist/docker/docker-client.js

@@ -153,4 +153,11 @@ const findGateway = (dockerode) => __awaiter(void 0, void 0, void 0, function* (
 });
 const findDefaultGateway = (dockerode) => __awaiter(void 0, void 0, void 0, function* () { return (0, run_in_container_1.runInContainer)(dockerode, "alpine:3.14", ["sh", "-c", "ip route|awk '/default/ { print $3 }'"]); });
 const isInContainer = () => (0, fs_1.existsSync)("/.dockerenv");
-exports.dockerClient = getDockerClient();
+let _dockerClient;
+const dockerClient = () => {
+    if (!_dockerClient) {
+        _dockerClient = getDockerClient();
+    }
+    return _dockerClient;
+};
+exports.dockerClient = dockerClient;

package/dist/docker/functions/container/create-container.js

@@ -17,7 +17,7 @@ const create_labels_1 = require("../create-labels");
 const createContainer = (options) => __awaiter(void 0, void 0, void 0, function* () {
     try {
         logger_1.log.info(`Creating container for image: ${options.imageName}`);
-        const { dockerode } = yield docker_client_1.dockerClient;
+        const { dockerode } = yield (0, docker_client_1.dockerClient)();
         return yield dockerode.createContainer({
             name: options.name,
             User: options.user,

package/dist/docker/functions/container/get-container.js

@@ -15,7 +15,7 @@ const logger_1 = require("../../../logger");
 const labels_1 = require("../../../labels");
 const getContainerById = (id) => __awaiter(void 0, void 0, void 0, function* () {
     try {
-        const { dockerode } = yield docker_client_1.dockerClient;
+        const { dockerode } = yield (0, docker_client_1.dockerClient)();
         return dockerode.getContainer(id);
     }
     catch (err) {
@@ -26,7 +26,7 @@ const getContainerById = (id) => __awaiter(void 0, void 0, void 0, function* ()
 exports.getContainerById = getContainerById;
 const getContainerByHash = (hash) => __awaiter(void 0, void 0, void 0, function* () {
     try {
-        const { dockerode } = yield docker_client_1.dockerClient;
+        const { dockerode } = yield (0, docker_client_1.dockerClient)();
         const containers = yield dockerode.listContainers({
             limit: 1,
             filters: {

package/dist/docker/functions/container/list-containers.js

@@ -14,7 +14,7 @@ const docker_client_1 = require("../../docker-client");
 const logger_1 = require("../../../logger");
 const listContainers = () => __awaiter(void 0, void 0, void 0, function* () {
     try {
-        const { dockerode } = yield docker_client_1.dockerClient;
+        const { dockerode } = yield (0, docker_client_1.dockerClient)();
         return yield dockerode.listContainers();
     }
     catch (err) {

package/dist/docker/functions/container/restart-container.d.ts

@@ -0,0 +1,5 @@
+import Dockerode from "dockerode";
+export declare type RestartContainerOptions = {
+    timeout: number;
+};
+export declare const restartContainer: (container: Dockerode.Container, options: RestartContainerOptions) => Promise<void>;

package/dist/docker/functions/container/restart-container.js

@@ -0,0 +1,23 @@
+"use strict";
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.restartContainer = void 0;
+const logger_1 = require("../../../logger");
+const restartContainer = (container, options) => __awaiter(void 0, void 0, void 0, function* () {
+    try {
+        yield container.restart({ t: options.timeout / 1000 });
+    }
+    catch (err) {
+        logger_1.log.error(`Failed to restart container ${container.id}: ${err}`);
+        throw err;
+    }
+});
+exports.restartContainer = restartContainer;

package/dist/docker/functions/demux-stream.js

@@ -16,7 +16,7 @@ const docker_client_1 = require("../docker-client");
 const demuxStream = (stream) => __awaiter(void 0, void 0, void 0, function* () {
     try {
         const demuxedStream = new stream_1.PassThrough({ autoDestroy: true, encoding: "utf-8" });
-        const { dockerode } = yield docker_client_1.dockerClient;
+        const { dockerode } = yield (0, docker_client_1.dockerClient)();
         dockerode.modem.demuxStream(stream, demuxedStream, demuxedStream);
         stream.on("end", () => demuxedStream.end());
         demuxedStream.on("close", () => {

package/dist/docker/functions/get-info.js

@@ -12,7 +12,7 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.getDockerInfo = void 0;
 const docker_client_1 = require("../docker-client");
 const getDockerInfo = () => __awaiter(void 0, void 0, void 0, function* () {
-    const { dockerode } = yield docker_client_1.dockerClient;
+    const { dockerode } = yield (0, docker_client_1.dockerClient)();
     const info = yield dockerode.info();
     return {
         serverVersion: info.ServerVersion,

package/dist/docker/functions/image/build-image.js

@@ -36,7 +36,7 @@ const buildImage = (options) => __awaiter(void 0, void 0, void 0, function* () {
                 }
             },
         });
-        const { dockerode } = yield docker_client_1.dockerClient;
+        const { dockerode } = yield (0, docker_client_1.dockerClient)();
         return new Promise((resolve) => dockerode
             .buildImage(tarStream, {
             dockerfile: options.dockerfileName,

package/dist/docker/functions/image/list-images.js

@@ -15,7 +15,7 @@ const docker_client_1 = require("../../docker-client");
 const logger_1 = require("../../../logger");
 const listImages = () => __awaiter(void 0, void 0, void 0, function* () {
     try {
-        const { dockerode } = yield docker_client_1.dockerClient;
+        const { dockerode } = yield (0, docker_client_1.dockerClient)();
         const images = yield dockerode.listImages();
         return images.reduce((dockerImageNames, image) => {
             var _a;

package/dist/docker/functions/image/pull-image.js

@@ -21,7 +21,7 @@ const pullImage = (options) => __awaiter(void 0, void 0, void 0, function* () {
             return;
         }
         logger_1.log.info(`Pulling image: ${options.imageName}`);
-        const { dockerode } = yield docker_client_1.dockerClient;
+        const { dockerode } = yield (0, docker_client_1.dockerClient)();
         const stream = yield dockerode.pull(options.imageName.toString(), { authconfig: options.authConfig });
         yield new pull_stream_parser_1.PullStreamParser(options.imageName, logger_1.log).consume(stream);
     }

package/dist/docker/functions/network/connect-network.js

@@ -15,7 +15,7 @@ const docker_client_1 = require("../../docker-client");
 const connectNetwork = (options) => __awaiter(void 0, void 0, void 0, function* () {
     try {
         logger_1.log.info(`Connecting container ${options.containerId} to network ${options.networkId}`);
-        const { dockerode } = yield docker_client_1.dockerClient;
+        const { dockerode } = yield (0, docker_client_1.dockerClient)();
         const network = dockerode.getNetwork(options.networkId);
         yield network.connect({ Container: options.containerId, EndpointConfig: { Aliases: options.networkAliases } });
     }

package/dist/docker/functions/network/create-network.js

@@ -16,7 +16,7 @@ const create_labels_1 = require("../create-labels");
 const createNetwork = (options) => __awaiter(void 0, void 0, void 0, function* () {
     try {
         logger_1.log.info(`Creating network ${options.name}`);
-        const { dockerode } = yield docker_client_1.dockerClient;
+        const { dockerode } = yield (0, docker_client_1.dockerClient)();
         const network = yield dockerode.createNetwork({
             Name: options.name,
             CheckDuplicate: options.checkDuplicate,

package/dist/docker/functions/network/remove-network.js

@@ -15,7 +15,7 @@ const docker_client_1 = require("../../docker-client");
 const removeNetwork = (id) => __awaiter(void 0, void 0, void 0, function* () {
     try {
         logger_1.log.info(`Removing network ${id}`);
-        const { dockerode } = yield docker_client_1.dockerClient;
+        const { dockerode } = yield (0, docker_client_1.dockerClient)();
         const network = dockerode.getNetwork(id);
         const { message } = yield network.remove();
         if (message) {

package/dist/docker-compose-environment/docker-compose-environment.js

@@ -119,7 +119,7 @@ class DockerComposeEnvironment {
                     }
                     throw err;
                 }
-                return new started_generic_container_1.StartedGenericContainer(container, (yield docker_client_1.dockerClient).host, inspectResult, boundPorts, containerName);
+                return new started_generic_container_1.StartedGenericContainer(container, (yield (0, docker_client_1.dockerClient)()).host, inspectResult, boundPorts, containerName, this.getWaitStrategy((yield (0, docker_client_1.dockerClient)()).host, container, containerName).withStartupTimeout(this.startupTimeout));
             })))).reduce((map, startedGenericContainer) => {
                 const containerName = startedGenericContainer.getName();
                 return Object.assign(Object.assign({}, map), { [containerName]: startedGenericContainer });
@@ -130,7 +130,7 @@ class DockerComposeEnvironment {
     }
     waitForContainer(container, containerName, boundPorts) {
         return __awaiter(this, void 0, void 0, function* () {
-            const waitStrategy = this.getWaitStrategy((yield docker_client_1.dockerClient).host, container, containerName);
+            const waitStrategy = this.getWaitStrategy((yield (0, docker_client_1.dockerClient)()).host, container, containerName);
             yield waitStrategy.withStartupTimeout(this.startupTimeout).waitUntilReady(container, boundPorts);
         });
     }

package/dist/generic-container/generic-container.js

@@ -119,7 +119,7 @@ class GenericContainer {
             const inspectResult = yield (0, inspect_container_1.inspectContainer)(startedContainer);
             const boundPorts = bound_ports_1.BoundPorts.fromInspectResult(inspectResult).filter(this.ports);
             yield this.waitForContainer(startedContainer, boundPorts);
-            return new started_generic_container_1.StartedGenericContainer(startedContainer, (yield docker_client_1.dockerClient).host, inspectResult, boundPorts, inspectResult.name);
+            return new started_generic_container_1.StartedGenericContainer(startedContainer, (yield (0, docker_client_1.dockerClient)()).host, inspectResult, boundPorts, inspectResult.name, this.getWaitStrategy((yield (0, docker_client_1.dockerClient)()).host, startedContainer).withStartupTimeout(this.startupTimeout));
         });
     }
     startContainer(createContainerOptions) {
@@ -145,7 +145,7 @@ class GenericContainer {
                 });
             }
             if (this.tarToCopy) {
-                yield this.tarToCopy.finalize();
+                this.tarToCopy.finalize();
                 yield (0, put_container_archive_1.putContainerArchive)({ container, stream: this.tarToCopy, containerPath: "/" });
             }
             logger_1.log.info(`Starting container ${this.imageName} with ID: ${container.id}`);
@@ -156,7 +156,7 @@ class GenericContainer {
             const inspectResult = yield (0, inspect_container_1.inspectContainer)(container);
             const boundPorts = bound_ports_1.BoundPorts.fromInspectResult(inspectResult).filter(this.ports);
             yield this.waitForContainer(container, boundPorts);
-            const startedContainer = new started_generic_container_1.StartedGenericContainer(container, (yield docker_client_1.dockerClient).host, inspectResult, boundPorts, inspectResult.name);
+            const startedContainer = new started_generic_container_1.StartedGenericContainer(container, (yield (0, docker_client_1.dockerClient)()).host, inspectResult, boundPorts, inspectResult.name, this.getWaitStrategy((yield (0, docker_client_1.dockerClient)()).host, container).withStartupTimeout(this.startupTimeout));
             if (this.postStart) {
                 yield this.postStart(startedContainer, inspectResult, boundPorts);
             }
@@ -263,7 +263,7 @@ class GenericContainer {
     waitForContainer(container, boundPorts) {
         return __awaiter(this, void 0, void 0, function* () {
             logger_1.log.debug(`Waiting for container to be ready: ${container.id}`);
-            const waitStrategy = this.getWaitStrategy((yield docker_client_1.dockerClient).host, container);
+            const waitStrategy = this.getWaitStrategy((yield (0, docker_client_1.dockerClient)()).host, container);
             try {
                 yield waitStrategy.withStartupTimeout(this.startupTimeout).waitUntilReady(container, boundPorts);
                 logger_1.log.info("Container is ready");

package/dist/generic-container/started-generic-container.d.ts

@@ -1,20 +1,24 @@
 /// <reference types="node" />
-import { StartedTestContainer, StopOptions, StoppedTestContainer } from "../test-container";
+import { RestartOptions, StartedTestContainer, StopOptions, StoppedTestContainer } from "../test-container";
 import Dockerode from "dockerode";
 import { Command, ContainerName, ExecResult, Host, Id as ContainerId, Labels } from "../docker/types";
 import { InspectResult } from "../docker/functions/container/inspect-container";
 import { BoundPorts } from "../bound-ports";
 import { Port } from "../port";
 import { Readable } from "stream";
+import { WaitStrategy } from "../wait-strategy";
 export declare class StartedGenericContainer implements StartedTestContainer {
     private readonly container;
     private readonly host;
-    private readonly inspectResult;
-    private readonly boundPorts;
+    private inspectResult;
+    private boundPorts;
     private readonly name;
-    constructor(container: Dockerode.Container, host: Host, inspectResult: InspectResult, boundPorts: BoundPorts, name: ContainerName);
+    private readonly waitStrategy;
+    constructor(container: Dockerode.Container, host: Host, inspectResult: InspectResult, boundPorts: BoundPorts, name: ContainerName, waitStrategy: WaitStrategy);
     stop(options?: Partial<StopOptions>): Promise<StoppedTestContainer>;
+    restart(options?: Partial<RestartOptions>): Promise<void>;
     private stopContainer;
+    private waitForContainer;
     getHost(): Host;
     getMappedPort(port: Port): Port;
     getId(): ContainerId;

package/dist/generic-container/started-generic-container.js

@@ -10,25 +10,39 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.StartedGenericContainer = void 0;
+const inspect_container_1 = require("../docker/functions/container/inspect-container");
+const bound_ports_1 = require("../bound-ports");
 const logger_1 = require("../logger");
 const remove_container_1 = require("../docker/functions/container/remove-container");
 const exec_container_1 = require("../docker/functions/container/exec-container");
 const container_logs_1 = require("../docker/functions/container/container-logs");
 const stopped_generic_container_1 = require("./stopped-generic-container");
 const stop_container_1 = require("../docker/functions/container/stop-container");
+const restart_container_1 = require("../docker/functions/container/restart-container");
 class StartedGenericContainer {
-    constructor(container, host, inspectResult, boundPorts, name) {
+    constructor(container, host, inspectResult, boundPorts, name, waitStrategy) {
         this.container = container;
         this.host = host;
         this.inspectResult = inspectResult;
         this.boundPorts = boundPorts;
         this.name = name;
+        this.waitStrategy = waitStrategy;
     }
     stop(options = {}) {
         return __awaiter(this, void 0, void 0, function* () {
             return this.stopContainer(options);
         });
     }
+    restart(options = {}) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const resolvedOptions = Object.assign({ timeout: 0 }, options);
+            yield (0, restart_container_1.restartContainer)(this.container, resolvedOptions);
+            // Inspect the restarted container and update the bound ports
+            this.inspectResult = yield (0, inspect_container_1.inspectContainer)(this.container);
+            this.boundPorts = yield bound_ports_1.BoundPorts.fromInspectResult(this.inspectResult).filter(Array.from(this.boundPorts.iterator()).map((port) => port[0]));
+            yield this.waitForContainer(this.container, this.boundPorts);
+        });
+    }
     stopContainer(options = {}) {
         return __awaiter(this, void 0, void 0, function* () {
             logger_1.log.info(`Stopping container with ID: ${this.container.id}`);
@@ -38,6 +52,26 @@ class StartedGenericContainer {
             return new stopped_generic_container_1.StoppedGenericContainer();
         });
     }
+    waitForContainer(container, boundPorts) {
+        return __awaiter(this, void 0, void 0, function* () {
+            logger_1.log.debug(`Waiting for container to be ready: ${container.id}`);
+            try {
+                yield this.waitStrategy.waitUntilReady(container, boundPorts);
+                logger_1.log.info("Container is ready");
+            }
+            catch (err) {
+                logger_1.log.error(`Container failed to be ready: ${err}`);
+                try {
+                    yield (0, stop_container_1.stopContainer)(container, { timeout: 0 });
+                    yield (0, remove_container_1.removeContainer)(container, { removeVolumes: true });
+                }
+                catch (stopErr) {
+                    logger_1.log.error(`Failed to stop container after it failed to be ready: ${stopErr}`);
+                }
+                throw err;
+            }
+        });
+    }
     getHost() {
         return this.host;
     }

package/dist/index.d.ts

@@ -16,3 +16,4 @@ export { ArangoDBContainer, StartedArangoContainer } from "./modules/arangodb/ar
 export { ElasticsearchContainer, StartedElasticsearchContainer } from "./modules/elasticsearch/elasticsearch-container";
 export { MySqlContainer, StartedMySqlContainer } from "./modules/mysql/mysql-container";
 export { NatsContainer, StartedNatsContainer, NatsConnectionOptions, NatsTlsOptions, } from "./modules/nats/nats-container";
+export { MongoDBContainer, StartedMongoDBContainer } from "./modules/mongodb/mongodb-container";

package/dist/index.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.StartedNatsContainer = exports.NatsContainer = exports.StartedMySqlContainer = exports.MySqlContainer = exports.StartedElasticsearchContainer = exports.ElasticsearchContainer = exports.StartedArangoContainer = exports.ArangoDBContainer = exports.StartedNeo4jContainer = exports.Neo4jContainer = exports.StartedKafkaContainer = exports.KafkaContainer = exports.StartedPostgreSqlContainer = exports.PostgreSqlContainer = exports.AlwaysPullPolicy = exports.DefaultPullPolicy = exports.Wait = exports.StoppedNetwork = exports.StartedNetwork = exports.Network = exports.DownedDockerComposeEnvironment = exports.StoppedDockerComposeEnvironment = exports.StartedDockerComposeEnvironment = exports.DockerComposeEnvironment = exports.TestContainers = exports.GenericContainerBuilder = exports.GenericContainer = void 0;
+exports.StartedMongoDBContainer = exports.MongoDBContainer = exports.StartedNatsContainer = exports.NatsContainer = exports.StartedMySqlContainer = exports.MySqlContainer = exports.StartedElasticsearchContainer = exports.ElasticsearchContainer = exports.StartedArangoContainer = exports.ArangoDBContainer = exports.StartedNeo4jContainer = exports.Neo4jContainer = exports.StartedKafkaContainer = exports.KafkaContainer = exports.StartedPostgreSqlContainer = exports.PostgreSqlContainer = exports.AlwaysPullPolicy = exports.DefaultPullPolicy = exports.Wait = exports.StoppedNetwork = exports.StartedNetwork = exports.Network = exports.DownedDockerComposeEnvironment = exports.StoppedDockerComposeEnvironment = exports.StartedDockerComposeEnvironment = exports.DockerComposeEnvironment = exports.TestContainers = exports.GenericContainerBuilder = exports.GenericContainer = void 0;
 var generic_container_1 = require("./generic-container/generic-container");
 Object.defineProperty(exports, "GenericContainer", { enumerable: true, get: function () { return generic_container_1.GenericContainer; } });
 var generic_container_builder_1 = require("./generic-container/generic-container-builder");
@@ -45,3 +45,6 @@ Object.defineProperty(exports, "StartedMySqlContainer", { enumerable: true, get:
 var nats_container_1 = require("./modules/nats/nats-container");
 Object.defineProperty(exports, "NatsContainer", { enumerable: true, get: function () { return nats_container_1.NatsContainer; } });
 Object.defineProperty(exports, "StartedNatsContainer", { enumerable: true, get: function () { return nats_container_1.StartedNatsContainer; } });
+var mongodb_container_1 = require("./modules/mongodb/mongodb-container");
+Object.defineProperty(exports, "MongoDBContainer", { enumerable: true, get: function () { return mongodb_container_1.MongoDBContainer; } });
+Object.defineProperty(exports, "StartedMongoDBContainer", { enumerable: true, get: function () { return mongodb_container_1.StartedMongoDBContainer; } });

package/dist/modules/abstract-started-container.d.ts

@@ -1,5 +1,5 @@
 /// <reference types="node" />
-import { StartedTestContainer, StopOptions, StoppedTestContainer } from "../test-container";
+import { RestartOptions, StartedTestContainer, StopOptions, StoppedTestContainer } from "../test-container";
 import { Port } from "../port";
 import { ContainerName, Host, Id, Command, ExecResult, Labels } from "../docker/types";
 import { Readable } from "stream";
@@ -7,6 +7,7 @@ export declare class AbstractStartedContainer {
     protected readonly startedTestContainer: StartedTestContainer;
     constructor(startedTestContainer: StartedTestContainer);
     stop(options?: Partial<StopOptions>): Promise<StoppedTestContainer>;
+    restart(options?: Partial<RestartOptions>): Promise<void>;
     getHost(): Host;
     getMappedPort(port: Port): Port;
     getName(): ContainerName;

package/dist/modules/abstract-started-container.js

@@ -1,4 +1,13 @@
 "use strict";
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.AbstractStartedContainer = void 0;
 class AbstractStartedContainer {
@@ -8,6 +17,11 @@ class AbstractStartedContainer {
     stop(options) {
         return this.startedTestContainer.stop(options);
     }
+    restart(options) {
+        return __awaiter(this, void 0, void 0, function* () {
+            return this.startedTestContainer.restart(options);
+        });
+    }
     getHost() {
         return this.startedTestContainer.getHost();
     }

package/dist/modules/mongodb/mongodb-container.d.ts

@@ -0,0 +1,19 @@
+import { GenericContainer } from "../../generic-container/generic-container";
+import { AbstractStartedContainer } from "../abstract-started-container";
+import { StartedTestContainer } from "../../test-container";
+export declare class MongoDBContainer extends GenericContainer {
+    private readonly dockerImageName;
+    constructor(image?: string);
+    start(): Promise<StartedMongoDBContainer>;
+    protected postStart(startedTestContainer: StartedTestContainer): Promise<void>;
+    private initReplicaSet;
+    private executeMongoEvalCommand;
+    private buildMongoEvalCommand;
+    private getMongoCmdBasedOnImageTag;
+    private checkMongoNodeExitCode;
+    private buildMongoWaitCommand;
+}
+export declare class StartedMongoDBContainer extends AbstractStartedContainer {
+    constructor(startedTestContainer: StartedTestContainer);
+    getConnectionString(): string;
+}

package/dist/modules/mongodb/mongodb-container.js

@@ -0,0 +1,85 @@
+"use strict";
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.StartedMongoDBContainer = exports.MongoDBContainer = void 0;
+const generic_container_1 = require("../../generic-container/generic-container");
+const abstract_started_container_1 = require("../abstract-started-container");
+const wait_1 = require("../../wait");
+const docker_image_name_1 = require("../../docker-image-name");
+const MONGODB_PORT = 27017;
+class MongoDBContainer extends generic_container_1.GenericContainer {
+    constructor(image = "mongo:4.0.1") {
+        super(image);
+        this.dockerImageName = docker_image_name_1.DockerImageName.fromString(this.image);
+    }
+    start() {
+        const _super = Object.create(null, {
+            start: { get: () => super.start }
+        });
+        return __awaiter(this, void 0, void 0, function* () {
+            this.withExposedPorts(MONGODB_PORT)
+                .withCmd(["--replSet", "rs0"])
+                .withWaitStrategy(wait_1.Wait.forLogMessage(new RegExp(".*waiting for connections.*", "i")))
+                .withStartupTimeout(120000);
+            return new StartedMongoDBContainer(yield _super.start.call(this));
+        });
+    }
+    postStart(startedTestContainer) {
+        return __awaiter(this, void 0, void 0, function* () {
+            yield this.initReplicaSet(startedTestContainer);
+        });
+    }
+    initReplicaSet(startedTestContainer) {
+        return __awaiter(this, void 0, void 0, function* () {
+            yield this.executeMongoEvalCommand(startedTestContainer, "rs.initiate();");
+            yield this.executeMongoEvalCommand(startedTestContainer, this.buildMongoWaitCommand());
+        });
+    }
+    executeMongoEvalCommand(startedTestContainer, command) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const execResult = yield startedTestContainer.exec(this.buildMongoEvalCommand(command));
+            this.checkMongoNodeExitCode(execResult);
+        });
+    }
+    buildMongoEvalCommand(command) {
+        return [this.getMongoCmdBasedOnImageTag(), "--eval", command];
+    }
+    getMongoCmdBasedOnImageTag() {
+        return parseInt(this.dockerImageName.tag[0]) >= 5 ? "mongosh" : "mongo";
+    }
+    checkMongoNodeExitCode(execResult) {
+        const { exitCode, output } = execResult;
+        if (execResult.exitCode !== 0) {
+            throw new Error(`Error running mongo command. Exit code ${exitCode}: ${output}`);
+        }
+    }
+    buildMongoWaitCommand() {
+        return `
+    var attempt = 0;
+    while(db.runCommand({isMaster: 1}).ismaster==false) {
+      if (attempt > 60) {
+        quit(1);
+      }
+      print(attempt); sleep(100); attempt++; 
+    }
+    `;
+    }
+}
+exports.MongoDBContainer = MongoDBContainer;
+class StartedMongoDBContainer extends abstract_started_container_1.AbstractStartedContainer {
+    constructor(startedTestContainer) {
+        super(startedTestContainer);
+    }
+    getConnectionString() {
+        return `mongodb://${this.getHost()}:${this.getMappedPort(MONGODB_PORT)}`;
+    }
+}
+exports.StartedMongoDBContainer = StartedMongoDBContainer;

package/dist/port-forwarder.js

@@ -69,7 +69,7 @@ class PortForwarderInstance {
                 `echo "${username}:$PASSWORD" | chpasswd && /usr/sbin/sshd -D -o PermitRootLogin=yes -o AddressFamily=inet -o GatewayPorts=yes`,
             ])
                 .start();
-            const host = (yield docker_client_1.dockerClient).host;
+            const host = (yield (0, docker_client_1.dockerClient)()).host;
             const port = container.getMappedPort(22);
             logger_1.log.debug(`Connecting to Port Forwarder on ${host}:${port}`);
             const connection = yield (0, ssh_remote_port_forward_1.createSshConnection)({ host, port, username, password });

package/dist/test-container.d.ts

@@ -23,12 +23,16 @@ export interface TestContainer {
     withCopyFileToContainer(sourcePath: string, containerPath: string): this;
     withCopyContentToContainer(content: string | Buffer | Readable, containerPath: string): this;
 }
+export interface RestartOptions {
+    timeout: number;
+}
 export interface StopOptions {
     timeout: number;
     removeVolumes: boolean;
 }
 export interface StartedTestContainer {
     stop(options?: Partial<StopOptions>): Promise<StoppedTestContainer>;
+    restart(options?: Partial<RestartOptions>): Promise<void>;
     getHost(): Host;
     getMappedPort(port: Port): Port;
     getName(): ContainerName;

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "testcontainers",
   "author": "Cristian Greco",
-  "version": "8.11.0",
+  "version": "8.13.0",
   "main": "dist/index",
   "types": "dist/index",
   "keywords": [
@@ -41,6 +41,7 @@
     "docker-compose": "^0.23.17",
     "dockerode": "^3.3.1",
     "get-port": "^5.1.1",
+    "mongoose": "^5.13.15",
     "properties-reader": "^2.2.0",
     "ssh-remote-port-forward": "^1.0.4",
     "tar-fs": "^2.1.1"
@@ -66,6 +67,7 @@
     "kafkajs": "^1.16.0",
     "lint-staged": "^12.3.8",
     "mysql2": "^2.3.3",
+    "nats": "^2.7.1",
     "neo4j-driver": "^4.4.5",
     "node-fetch": "^2.6.7",
     "pg": "^8.7.3",
@@ -74,8 +76,7 @@
     "ts-jest": "^27.1.4",
     "ts-node": "^10.7.0",
     "typescript": "^4.6.3",
-    "wait-for-expect": "^3.0.2",
-    "nats": "^2.7.1"
+    "wait-for-expect": "^3.0.2"
   },
   "lint-staged": {
     "src/**/*.ts": [