import os

import psutil

from buildscripts.resmokelib import config


def test_analysis(logger, pids):
    """
    Write the pids out to a file and kill them instead of running analysis.
    This option will only be specified in resmoke selftests.
    """
    with open(os.path.join(config.DBPATH_PREFIX, "test_analysis.txt"), "w") as analysis_file:
        analysis_file.write("\n".join([str(pid) for pid in pids]))
        for pid in pids:
            try:
                proc = psutil.Process(pid)
                logger.info("Killing process pid %d", pid)
                proc.kill()
                proc.wait(
                    timeout=5
                )  # A zombie or defunct process won't end until it is reaped by its parent.
            except (psutil.NoSuchProcess, psutil.TimeoutExpired):
                # Process has already terminated or will need to be reaped by its parent.
                pass