mirror of
https://github.com/capstone-engine/llvm-capstone.git
synced 2025-02-11 04:06:20 +00:00
93ee588232
As per Steve Scalpone's suggestion, I've renamed the runtime routine to better evoke its purpose. I implemented a routine called "Crash" and added a test. Differential Revision: https://reviews.llvm.org/D118703
98 lines
3.8 KiB
C++
98 lines
3.8 KiB
C++
//===-- flang/unittests/Runtime/Stop.cpp ----------------------------------===//
|
|
//
|
|
// 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 "../../runtime/environment.h"
|
|
#include <cstdlib>
|
|
#include <gtest/gtest.h>
|
|
|
|
using namespace Fortran::runtime;
|
|
|
|
struct TestProgramEnd : CrashHandlerFixture {};
|
|
|
|
TEST(TestProgramEnd, StopTest) {
|
|
EXPECT_EXIT(RTNAME(StopStatement)(), testing::ExitedWithCode(EXIT_SUCCESS),
|
|
"Fortran STOP");
|
|
}
|
|
|
|
TEST(TestProgramEnd, StopTestNoStopMessage) {
|
|
putenv(const_cast<char *>("NO_STOP_MESSAGE=1"));
|
|
Fortran::runtime::executionEnvironment.Configure(0, nullptr, nullptr);
|
|
EXPECT_EXIT(
|
|
RTNAME(StopStatement)(), testing::ExitedWithCode(EXIT_SUCCESS), "");
|
|
}
|
|
|
|
TEST(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(TestProgramEnd, NoStopMessageTest) {
|
|
putenv(const_cast<char *>("NO_STOP_MESSAGE=1"));
|
|
Fortran::runtime::executionEnvironment.Configure(0, nullptr, nullptr);
|
|
static const char *message{"bye bye"};
|
|
EXPECT_EXIT(RTNAME(StopStatementText)(message, std::strlen(message),
|
|
/*isErrorStop=*/false, /*quiet=*/false),
|
|
testing::ExitedWithCode(EXIT_SUCCESS), "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(TestProgramEnd, FailImageTest) {
|
|
EXPECT_EXIT(
|
|
RTNAME(FailImageStatement)(), testing::ExitedWithCode(EXIT_FAILURE), "");
|
|
}
|
|
|
|
TEST(TestProgramEnd, ExitTest) {
|
|
EXPECT_EXIT(RTNAME(Exit)(), testing::ExitedWithCode(EXIT_SUCCESS), "");
|
|
EXPECT_EXIT(
|
|
RTNAME(Exit)(EXIT_FAILURE), testing::ExitedWithCode(EXIT_FAILURE), "");
|
|
}
|
|
|
|
TEST(TestProgramEnd, AbortTest) { EXPECT_DEATH(RTNAME(Abort)(), ""); }
|
|
|
|
TEST(TestProgramEnd, CrashTest) {
|
|
static const std::string crashMessage{"bad user code"};
|
|
static const std::string fileName{"file name"};
|
|
static const std::string headMessage{"fatal Fortran runtime error\\("};
|
|
static const std::string tailMessage{":343\\): "};
|
|
static const std::string fullMessage{
|
|
headMessage + fileName + tailMessage + crashMessage};
|
|
EXPECT_DEATH(
|
|
RTNAME(ReportFatalUserError)(crashMessage.c_str(), fileName.c_str(), 343),
|
|
fullMessage.c_str());
|
|
}
|