Class ProgramMock

tests/utils/mock.py:8–44

class ProgramMock:
    def __init__(self, directory: str):
        self.directory = Path(os.path.abspath(directory))
        os.makedirs(self.directory, exist_ok=True)

    def update_env(self, env: Dict[str, str]):
        path = str(self.directory)
        if "PATH" in env:
            path += f":{env['PATH']}"
        env["PATH"] = path

    def redirect_program_to_binary(self, mocked_program: str, target_binary: Path):
        """
        Mocks `mocked_program` so that when you try to execute it, `target_binary` will be executed
        instead.
        """
        link_path = self.directory / mocked_program
        link_path.unlink(missing_ok=True)
        os.symlink(dst=link_path, src=target_binary)
        # Make the link executable
        os.chmod(link_path, 0o700)

    @contextlib.contextmanager
    def mock_program_with_code(self, name: str, code: str):
        import textwrap

        content = f"#!{sys.executable}\n{textwrap.dedent(code)}"
        program_path = self.directory / name
        assert not program_path.is_file()

        with open(program_path, "w") as f:
            f.write(content)
        os.chmod(program_path, 0o700)

        yield program_path

        os.unlink(program_path)