@tresdoce-nestjs-toolkit/test-utils/dist-src/testcontainers/testContainersTD.js

Tresdoce NestJS Toolkit - Utilities para testing

import { GenericContainer, RandomUuid, Wait, } from 'testcontainers';
import _ from 'lodash';
export default class TestContainersTD {
    /* istanbul ignore next */
    constructor(_image = 'postgres:13', _options, _isSingleton = false) {
        this._image = _image;
        this._options = _options;
        this._isSingleton = _isSingleton;
        if (_isSingleton && TestContainersTD._instance)
            throw new Error('Use testContainers.getInstance() instead of new.');
        TestContainersTD._instance = this;
    }
    /**
     * Get instance
     */
    /* istanbul ignore next */
    static getInstance(_image, _options) {
        return (TestContainersTD._instance ??
            (TestContainersTD._instance = new TestContainersTD(_image, _options, true)));
    }
    prepareContainer(options) {
        const genericContainer = new GenericContainer(`${this._image}`);
        /* Add container name*/
        /* istanbul ignore next */
        if (_.has(options, 'containerName') && !_.isEmpty(options.containerName)) {
            genericContainer.withName(options.containerName);
        }
        else {
            /* istanbul ignore next */
            genericContainer.withName(`test-container-${new RandomUuid().nextUuid()}`);
        }
        /* Add container ports */
        /* istanbul ignore next */
        if (_.has(options, 'networkName') && !_.isEmpty(options.networkName)) {
            genericContainer.withNetworkMode(options.networkName);
        }
        /* Add container ports */
        /* istanbul ignore next */
        if (_.has(options, 'ports') && !_.isEmpty(options.ports)) {
            genericContainer.withExposedPorts(...options.ports);
        }
        /* Add container envs */
        /* istanbul ignore next */
        if (_.has(options, 'envs') && !_.isEmpty(options.envs)) {
            genericContainer.withEnvironment(options.envs);
        }
        /* Add command to container */
        /* istanbul ignore next */
        if (_.has(options, 'command') && !_.isEmpty(options.command)) {
            genericContainer.withCommand(options.command);
        }
        /* Add startup timeout container */
        /* istanbul ignore next */
        if (_.has(options, 'startupTimeout')) {
            genericContainer.withStartupTimeout(options.startupTimeout);
        }
        /* Add strategy to start container */
        /* istanbul ignore next */
        if (_.has(options, 'strategyHealthCheck') && options.strategyHealthCheck) {
            genericContainer.withWaitStrategy(Wait.forHealthCheck());
        }
        /* Add container reuse*/
        /* istanbul ignore next */
        if (_.has(options, 'reuse') && options.reuse) {
            genericContainer.withReuse();
        }
        return genericContainer;
    }
    /**
     * Started container
     */
    async start() {
        try {
            this._container = await this.prepareContainer(this._options).start();
            global.hostContainer = this._container.getHost();
            console.info(`✨ Container initialized: ${this.getName()}`);
        }
        catch (e) {
            /* istanbul ignore next */
            console.error(`😰 [${this._options.containerName}] Error initializing container: ${e}`);
        }
    }
    /**
     * Stop container
     * @param options optional stop options of test containers.
     */
    async stop(options) {
        try {
            const containerName = this.getName();
            await this._container.stop(options);
            console.info(`👌 Container stopped successfully: ${containerName}`);
        }
        catch (e) {
            /* istanbul ignore next */
            console.error(`😒 [${this._options.containerName}] Container not initialized`);
        }
    }
    /*=============================*/
    /**
     * Get envs
     */
    /* istanbul ignore next */
    getEnvs() {
        return _.has(this._options, 'envs') ? this._options.envs : null;
    }
    /**
     * Get container
     */
    getContainer() {
        return this._container;
    }
    /**
     * Get host of container
     */
    getHost() {
        return this._container.getHost();
    }
    /**
     * Get container name
     */
    getName() {
        return this._container.getName();
    }
    /**
     * Get mapped ports
     */
    getMappedPort(port) {
        return this._container.getMappedPort(port);
    }
}