forked from OSchip/llvm-project
140 lines
3.8 KiB
C++
140 lines
3.8 KiB
C++
// Tests basic FORMAT string traversal
|
|
|
|
#include "testing.h"
|
|
#include "../runtime/format-implementation.h"
|
|
#include "../runtime/io-error.h"
|
|
#include <cstdarg>
|
|
#include <cstring>
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
using namespace Fortran::runtime;
|
|
using namespace Fortran::runtime::io;
|
|
using namespace std::literals::string_literals;
|
|
|
|
using Results = std::vector<std::string>;
|
|
|
|
// A test harness context for testing FormatControl
|
|
class TestFormatContext : public IoErrorHandler {
|
|
public:
|
|
using CharType = char;
|
|
TestFormatContext() : IoErrorHandler{"format.cpp", 1} {}
|
|
bool Emit(const char *, std::size_t);
|
|
bool Emit(const char16_t *, std::size_t);
|
|
bool Emit(const char32_t *, std::size_t);
|
|
bool AdvanceRecord(int = 1);
|
|
void HandleRelativePosition(std::int64_t);
|
|
void HandleAbsolutePosition(std::int64_t);
|
|
void Report(const DataEdit &);
|
|
void Check(Results &);
|
|
Results results;
|
|
MutableModes &mutableModes() { return mutableModes_; }
|
|
|
|
private:
|
|
MutableModes mutableModes_;
|
|
};
|
|
|
|
bool TestFormatContext::Emit(const char *s, std::size_t len) {
|
|
std::string str{s, len};
|
|
results.push_back("'"s + str + '\'');
|
|
return true;
|
|
}
|
|
bool TestFormatContext::Emit(const char16_t *, std::size_t) {
|
|
Crash("TestFormatContext::Emit(const char16_t *) called");
|
|
return false;
|
|
}
|
|
bool TestFormatContext::Emit(const char32_t *, std::size_t) {
|
|
Crash("TestFormatContext::Emit(const char32_t *) called");
|
|
return false;
|
|
}
|
|
|
|
bool TestFormatContext::AdvanceRecord(int n) {
|
|
while (n-- > 0) {
|
|
results.emplace_back("/");
|
|
}
|
|
return true;
|
|
}
|
|
|
|
void TestFormatContext::HandleAbsolutePosition(std::int64_t n) {
|
|
results.push_back("T"s + std::to_string(n));
|
|
}
|
|
|
|
void TestFormatContext::HandleRelativePosition(std::int64_t n) {
|
|
if (n < 0) {
|
|
results.push_back("TL"s + std::to_string(-n));
|
|
} else {
|
|
results.push_back(std::to_string(n) + 'X');
|
|
}
|
|
}
|
|
|
|
void TestFormatContext::Report(const DataEdit &edit) {
|
|
std::string str{edit.descriptor};
|
|
if (edit.repeat != 1) {
|
|
str = std::to_string(edit.repeat) + '*' + str;
|
|
}
|
|
if (edit.variation) {
|
|
str += edit.variation;
|
|
}
|
|
if (edit.width) {
|
|
str += std::to_string(*edit.width);
|
|
}
|
|
if (edit.digits) {
|
|
str += "."s + std::to_string(*edit.digits);
|
|
}
|
|
if (edit.expoDigits) {
|
|
str += "E"s + std::to_string(*edit.expoDigits);
|
|
}
|
|
// modes?
|
|
results.push_back(str);
|
|
}
|
|
|
|
void TestFormatContext::Check(Results &expect) {
|
|
if (expect != results) {
|
|
Fail() << "expected:";
|
|
for (const std::string &s : expect) {
|
|
llvm::errs() << ' ' << s;
|
|
}
|
|
llvm::errs() << "\ngot:";
|
|
for (const std::string &s : results) {
|
|
llvm::errs() << ' ' << s;
|
|
}
|
|
llvm::errs() << '\n';
|
|
}
|
|
expect.clear();
|
|
results.clear();
|
|
}
|
|
|
|
static void Test(int n, const char *format, Results &&expect, int repeat = 1) {
|
|
TestFormatContext context;
|
|
FormatControl<TestFormatContext> control{
|
|
context, format, std::strlen(format)};
|
|
try {
|
|
for (int j{0}; j < n; ++j) {
|
|
context.Report(control.GetNextDataEdit(context, repeat));
|
|
}
|
|
control.Finish(context);
|
|
if (int iostat{context.GetIoStat()}) {
|
|
context.Crash("GetIoStat() == %d", iostat);
|
|
}
|
|
} catch (const std::string &crash) {
|
|
context.results.push_back("Crash:"s + crash);
|
|
}
|
|
context.Check(expect);
|
|
}
|
|
|
|
int main() {
|
|
StartTests();
|
|
Test(1, "('PI=',F9.7)", Results{"'PI='", "F9.7"});
|
|
Test(1, "(3HPI=F9.7)", Results{"'PI='", "F9.7"});
|
|
Test(1, "(3HPI=/F9.7)", Results{"'PI='", "/", "F9.7"});
|
|
Test(2, "('PI=',F9.7)", Results{"'PI='", "F9.7", "/", "'PI='", "F9.7"});
|
|
Test(2, "(2('PI=',F9.7),'done')",
|
|
Results{"'PI='", "F9.7", "'PI='", "F9.7", "'done'"});
|
|
Test(2, "(3('PI=',F9.7,:),'tooFar')",
|
|
Results{"'PI='", "F9.7", "'PI='", "F9.7"});
|
|
Test(2, "(*('PI=',F9.7,:),'tooFar')",
|
|
Results{"'PI='", "F9.7", "'PI='", "F9.7"});
|
|
Test(1, "(3F9.7)", Results{"2*F9.7"}, 2);
|
|
return EndTests();
|
|
}
|