Zabbix

/*

** Copyright (C) 2001-2025 Zabbix SIA

**

** This program is free software: you can redistribute it and/or modify it under the terms of

** the GNU Affero General Public License as published by the Free Software Foundation, version 3.

**

** This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;

** without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.

** See the GNU Affero General Public License for more details.

**

** You should have received a copy of the GNU Affero General Public License along with this program.

** If not, see <https://www.gnu.org/licenses/>.

**/

​

#define exit    __real_exit

#include <stdlib.h>

#undef exit

​

#include "zbxmocktest.h"

#include "zbxmockdata.h"

​

#include "zbxcommon.h"

​

void    __wrap_exit(int status);

​

void    __wrap_exit(int status)

{

    zbx_mock_error_t    error;

    int         expected_status;

​

    if (ZBX_MOCK_NO_EXIT_CODE == (error = zbx_mock_exit_code(&expected_status)))

        expected_status = EXIT_SUCCESS;

    else

    {

        if (ZBX_MOCK_SUCCESS != error)

            fail_msg("Cannot get exit code from test case data: %s", zbx_mock_error_string(error));

    }

​

    switch (status)

    {

        case EXIT_SUCCESS:

        case EXIT_FAILURE:

            if (status != expected_status)

                fail_msg("exit() called with status %d, expected %d.", status, expected_status);

            __real_exit(EXIT_SUCCESS);

        default:

            fail_msg("exit() called with status %d that is neither EXIT_SUCCESS nor EXIT_FAILURE.", status);

    }

}