1a5d176359
- Use TEST_F instead of TEST so CrashHandlerFixture::SetUp() is actually
called, registering the custom crash handler for death tests.
- Move putenv/executionEnvironment.Configure calls inside EXPECT_EXIT
blocks so they run in the forked child process, preventing the
NO_STOP_MESSAGE environment variable and configured global state from
leaking into subsequent tests.
- Replace const_cast<char *>("NO_STOP_MESSAGE=1") with a mutable static
char array, to avoid casting away constness of a string literal.
- Update CrashTest's expected pattern to match the output format of the
custom crash handler installed by CrashHandlerFixture, which was
previously never invoked due to the TEST vs TEST_F bug. (Note: there was
a buildbot failure related to this:
https://lab.llvm.org/buildbot/#/builders/130/builds/18413 )
Assisted-by: AI
131 lines
4.7 KiB
C++
131 lines
4.7 KiB
C++
//===-- unittests/Runtime/Stop.cpp ------------------------------*- C++ -*-===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
/// Test runtime API for STOP statement and runtime API to kill the program.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "flang/Runtime/stop.h"
|
|
#include "CrashHandlerFixture.h"
|
|
#include "flang-rt/runtime/environment.h"
|
|
#include <cstdlib>
|
|
#include <cstring>
|
|
#include <gtest/gtest.h>
|
|
|
|
using namespace Fortran::runtime;
|
|
|
|
struct TestProgramEnd : CrashHandlerFixture {};
|
|
|
|
TEST_F(TestProgramEnd, StopTest) {
|
|
EXPECT_EXIT(RTNAME(StopStatement)(), testing::ExitedWithCode(EXIT_SUCCESS),
|
|
"Fortran STOP");
|
|
}
|
|
|
|
TEST_F(TestProgramEnd, StopTestNoStopMessage) {
|
|
EXPECT_EXIT(
|
|
{
|
|
static char noStopMessage[] = "NO_STOP_MESSAGE=1";
|
|
putenv(noStopMessage);
|
|
Fortran::runtime::executionEnvironment.Configure(
|
|
0, nullptr, nullptr, nullptr);
|
|
RTNAME(StopStatement)();
|
|
},
|
|
testing::ExitedWithCode(EXIT_SUCCESS), "");
|
|
}
|
|
|
|
TEST_F(TestProgramEnd, StopMessageTest) {
|
|
static const char *message{"bye bye"};
|
|
EXPECT_EXIT(RTNAME(StopStatementText)(message, std::strlen(message),
|
|
/*isErrorStop=*/false, /*quiet=*/false),
|
|
testing::ExitedWithCode(EXIT_SUCCESS), "Fortran STOP: bye bye");
|
|
|
|
EXPECT_EXIT(RTNAME(StopStatementText)(message, std::strlen(message),
|
|
/*isErrorStop=*/false, /*quiet=*/true),
|
|
testing::ExitedWithCode(EXIT_SUCCESS), "");
|
|
|
|
EXPECT_EXIT(RTNAME(StopStatementText)(message, std::strlen(message),
|
|
/*isErrorStop=*/true, /*quiet=*/false),
|
|
testing::ExitedWithCode(EXIT_FAILURE), "Fortran ERROR STOP: bye bye");
|
|
|
|
EXPECT_EXIT(RTNAME(StopStatementText)(message, std::strlen(message),
|
|
/*isErrorStop=*/true, /*quiet=*/true),
|
|
testing::ExitedWithCode(EXIT_FAILURE), "");
|
|
}
|
|
|
|
TEST_F(TestProgramEnd, NoStopMessageTest) {
|
|
static const char *message{"bye bye"};
|
|
EXPECT_EXIT(
|
|
{
|
|
static char noStopMessage[] = "NO_STOP_MESSAGE=1";
|
|
putenv(noStopMessage);
|
|
Fortran::runtime::executionEnvironment.Configure(
|
|
0, nullptr, nullptr, nullptr);
|
|
RTNAME(StopStatementText)(message, std::strlen(message),
|
|
/*isErrorStop=*/false, /*quiet=*/false);
|
|
},
|
|
testing::ExitedWithCode(EXIT_SUCCESS), "bye bye");
|
|
|
|
EXPECT_EXIT(
|
|
{
|
|
static char noStopMessage[] = "NO_STOP_MESSAGE=1";
|
|
putenv(noStopMessage);
|
|
Fortran::runtime::executionEnvironment.Configure(
|
|
0, nullptr, nullptr, nullptr);
|
|
RTNAME(StopStatementText)(message, std::strlen(message),
|
|
/*isErrorStop=*/false, /*quiet=*/true);
|
|
},
|
|
testing::ExitedWithCode(EXIT_SUCCESS), "");
|
|
|
|
EXPECT_EXIT(
|
|
{
|
|
static char noStopMessage[] = "NO_STOP_MESSAGE=1";
|
|
putenv(noStopMessage);
|
|
Fortran::runtime::executionEnvironment.Configure(
|
|
0, nullptr, nullptr, nullptr);
|
|
RTNAME(StopStatementText)(message, std::strlen(message),
|
|
/*isErrorStop=*/true, /*quiet=*/false);
|
|
},
|
|
testing::ExitedWithCode(EXIT_FAILURE), "Fortran ERROR STOP: bye bye");
|
|
|
|
EXPECT_EXIT(
|
|
{
|
|
static char noStopMessage[] = "NO_STOP_MESSAGE=1";
|
|
putenv(noStopMessage);
|
|
Fortran::runtime::executionEnvironment.Configure(
|
|
0, nullptr, nullptr, nullptr);
|
|
RTNAME(StopStatementText)(message, std::strlen(message),
|
|
/*isErrorStop=*/true, /*quiet=*/true);
|
|
},
|
|
testing::ExitedWithCode(EXIT_FAILURE), "");
|
|
}
|
|
|
|
TEST_F(TestProgramEnd, FailImageTest) {
|
|
EXPECT_EXIT(
|
|
RTNAME(FailImageStatement)(), testing::ExitedWithCode(EXIT_FAILURE), "");
|
|
}
|
|
|
|
TEST_F(TestProgramEnd, ExitTest) {
|
|
EXPECT_EXIT(RTNAME(Exit)(), testing::ExitedWithCode(EXIT_SUCCESS), "");
|
|
EXPECT_EXIT(
|
|
RTNAME(Exit)(EXIT_FAILURE), testing::ExitedWithCode(EXIT_FAILURE), "");
|
|
}
|
|
|
|
TEST_F(TestProgramEnd, AbortTest) { EXPECT_DEATH(RTNAME(Abort)(), ""); }
|
|
|
|
TEST_F(TestProgramEnd, CrashTest) {
|
|
static const std::string crashMessage{"bad user code"};
|
|
static const std::string fileName{"file name"};
|
|
// CrashHandlerFixture installs a custom crash handler that formats messages
|
|
// as: "Test <name> crashed in file <file>(<line>): <message>"
|
|
static const std::string fullMessage{"Test CrashTest crashed in file " +
|
|
fileName + "\\(343\\): " + crashMessage};
|
|
EXPECT_DEATH(
|
|
RTNAME(ReportFatalUserError)(crashMessage.c_str(), fileName.c_str(), 343),
|
|
fullMessage.c_str());
|
|
}
|