Method generateOutput

test/test_runner/test_runner.cpp:282–331

Appends `result` as the expected test output to `statement.newOutput` based on the given output type. Used in `EndToEndTest::rewriteTests` to rewrite outputs in test files.

// Appends `result` as the expected test output to `statement.newOutput` based on the given output
// type. Used in `EndToEndTest::rewriteTests` to rewrite outputs in test files.
void TestRunner::generateOutput(QueryResult* result, TestStatement& statement, size_t resultIdx) {
    TestQueryResult& testAnswer = statement.result[resultIdx];
    statement.testResultType = testAnswer.type;
    switch (testAnswer.type) {
    case ResultType::OK: {
        statement.newOutput +=
            "---- " + (result->isSuccess() ? std::string("ok") : std::string("error")) + '\n';
        if (!result->isSuccess()) {
            statement.newOutput += result->getErrorMessage() + '\n';
        }
    } break;
    case ResultType::HASH: {
        std::string resultHash =
            convertResultToMD5Hash(*result, statement.checkOutputOrder, statement.checkColumnNames);
        statement.newOutput += "---- hash\n" + std::to_string(result->getNumTuples()) +
                               " tuples hashed to " + resultHash + '\n';
    } break;
    case ResultType::TUPLES: {
        statement.newOutput +=
            "---- " + std::to_string(result->getNumTuples() + statement.checkColumnNames) + '\n';
        std::vector<std::string> resultTuples =
            convertResultToString(*result, statement.checkOutputOrder, statement.checkColumnNames);

        // Use the test file output if it otherwise matches the actual output. This allows
        // preserving the existing user-written order and avoids producing unnecessary diffs.
        auto orderedResults = testAnswer.expectedResult;
        if (!statement.checkOutputOrder) {
            std::ranges::sort(orderedResults);
        }
        auto& output = resultTuples == orderedResults ? testAnswer.expectedResult : resultTuples;

        for (auto& res : output) {
            statement.newOutput += res + '\n';
        }
    } break;
    case ResultType::CSV_FILE:
        // Not supported yet.
        return;
    case ResultType::ERROR_MSG: {
        statement.newOutput +=
            "---- " + (result->isSuccess() ? std::string("ok") : std::string("error")) + '\n';
        statement.newOutput += StringUtils::rtrim(result->getErrorMessage()) + '\n';
    } break;
    case ResultType::ERROR_REGEX: {
        statement.newOutput +=
            "---- " + (result->isSuccess() ? std::string("ok") : std::string("error(regex)")) +
            '\n';
    } break;
    }
}

void TestRunner::outputFailedPlan(Connection& conn, const TestStatement& statement) {
    spdlog::error("QUERY FAILED.");