############################################################################
# Copyright (C) SchedMD LLC.
############################################################################
import re

import pytest

import atf


# Setup
@pytest.fixture(scope="module", autouse=True)
def setup():
    atf.require_slurm_running()


@pytest.fixture(scope="function", autouse=True)
def clean_core():
    yield
    coredumps = atf.get_coredumps()
    for coredump in coredumps:
        if "file_in" in coredump:
            atf.run_command(f"rm -f {coredump}")


def test_task_signal_abort_message():
    """Test of task signal abort message"""

    error_msg = r"Segmentation fault"
    file_in = atf.module_tmp_path / "file_in"
    atf.make_bash_script(file_in, "kill -11 $$")
    run_error = atf.run_command_error(f"srun {file_in}", xfail=True)
    assert (
        re.search(error_msg, run_error) is not None
    ), "Srun failed to send message after segfault"