flang/unittests/Runtime/testing.cpp

#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);
  }
}