tests/plugins/common/index.ts

/*
 * Licensed to the SkyAPM under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You under the Apache License, Version 2.0
 * (the "License"); you may not use this file except in compliance with
 * the License.  You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

import { spawn, exec } from 'child_process';

// @ts-ignore
import wait from '../../utils/wait';

const ifRebuild = process.env.SW_BUILD !== 'false';

export function setUp(composeFile: string, ready: () => boolean) {
  return new Promise((resolve, reject) => {
    const options = ['--force-recreate', '-d'];
    if (ifRebuild) {
      options.push('--build');
    }

    const composeUp = spawn('docker-compose', ['-f', composeFile, 'up', ...options]);

    composeUp.stdout.on('data', (data) => process.stdout.write(data.toString()));
    composeUp.stderr.on('data', (data) => process.stderr.write(data.toString()));

    wait({
      condition: ready,
      done: (success) => (success ? resolve() : reject()),
    });
  });
}

export function tearDown(composeFile: string) {
  return new Promise((resolve) => {
    exec(`docker-compose -f ${composeFile} down`, (err, stdout, stderr) => {
      process.stdout.write(stdout);
      process.stderr.write(stderr);
      resolve();
    });
  });
}