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

Tresdoce NestJS Toolkit - Utilities para testing

import { DockerComposeEnvironment, Wait } from 'testcontainers';
import _ from 'lodash';
let environment;
/**
 * Creates a Jest-compatible global setup function that boots a docker-compose environment.
 *
 * @param _services - Optional subset of docker-compose services to start.
 * @param _composeFilePath - Directory where the compose file is located.
 * @param _composeFile - Compose file name.
 * @param _startupTimeout - Maximum startup time in milliseconds.
 * @param _options - Additional wait strategy configuration.
 * @returns An async function that starts the compose environment.
 */
export const initDockerCompose = (_services, _composeFilePath = '.', _composeFile = 'docker-compose.yml', _startupTimeout = 60000, _options = {}) => {
    return async () => {
        console.info(`🐳 Initialize docker-compose...`);
        if (_.isEmpty(_services)) {
            console.log(`• All services from ${_composeFile}`);
        }
        else {
            console.log(`• Services from ${_composeFile}: ${_services.join(', ')}`);
        }
        try {
            let composeEnvironment = new DockerComposeEnvironment(_composeFilePath, _composeFile).withStartupTimeout(_startupTimeout);
            if (_options.useDefaultHealthCheckWaitStrategy) {
                composeEnvironment = composeEnvironment.withDefaultWaitStrategy(Wait.forHealthCheck());
            }
            for (const containerName of _options.healthCheckWaitStrategyNames || []) {
                composeEnvironment = composeEnvironment.withWaitStrategy(containerName, Wait.forHealthCheck());
            }
            environment = await composeEnvironment.up(_services);
            globalThis.__TESTCONTAINERS__ = environment;
            console.info(`✨ Container(s) initialized.`);
            return environment;
        }
        catch (_error) {
            /* istanbul ignore next */
            console.error(`😰 Error initializing container(s): ${_error}`);
            return undefined;
        }
    };
};
/**
 * Creates a Jest-compatible global teardown function that stops the docker-compose environment.
 *
 * @param _options - Optional docker compose down options.
 * @returns An async function that tears down the compose environment.
 */
export const closeDockerCompose = (_options) => {
    return async () => {
        console.info('🐳 Terminate docker-compose...');
        try {
            const startedEnvironment = environment ?? globalThis.__TESTCONTAINERS__;
            await startedEnvironment.down(_options);
            console.info(`👌 Container(s) stopped successfully.`);
        }
        catch (_error) {
            /* istanbul ignore next */
            console.error(`😒 Container(s) not initialized.`);
        }
    };
};