llvm-project/flang/unittests/Runtime/testing.cpp

47 lines
1.1 KiB
C++

#include "testing.h"
#include "../../runtime/terminator.h"
#include <algorithm>
#include <cstdarg>
#include <cstdio>
#include <cstring>
#include <string>
static int failures{0};
// Override the Fortran runtime's Crash() for testing purposes
[[noreturn]] static void CatchCrash(
const char *sourceFile, int sourceLine, const char *message, va_list &ap) {
char buffer[1000];
std::vsnprintf(buffer, sizeof buffer, message, ap);
va_end(ap);
llvm::errs() << (sourceFile ? sourceFile : "unknown source file") << '('
<< sourceLine << "): CRASH: " << buffer << '\n';
throw std::string{buffer};
}
void StartTests() {
Fortran::runtime::Terminator::RegisterCrashHandler(CatchCrash);
}
llvm::raw_ostream &Fail() {
++failures;
return llvm::errs();
}
int EndTests() {
if (failures == 0) {
llvm::outs() << "PASS\n";
} else {
llvm::outs() << "FAIL " << failures << " tests\n";
}
return failures != 0;
}
void SetCharacter(char *to, std::size_t n, const char *from) {
auto len{std::strlen(from)};
std::memcpy(to, from, std::min(len, n));
if (len < n) {
std::memset(to + len, ' ', n - len);
}
}