Method rewriteTestFile

test/runner/e2e_test.cpp:127–274

Used when `REWRITE_TESTS` mode is enabled. Currently, the implementation of the rewrite functionality is inefficient. 1) The testStatements vector container does not contain all the tests specified in the file. Rather, it only contains tests for a single CASE. See `parseAndRegisterFileGroup()`. When rewriting the output, we need to first find this CASE. 2) This function will be called from `TearD

    //    parallel. The current code does not handle calls for the same test file in parallel, and
    //    rewrite mode is expected to be run in single-threaded mode using `TEST_JOBS=1`.
    void rewriteTestFile() const {
        std::fstream file;
        std::string newFile;
        std::string currLine;
        std::string testCaseName;
        file.open(testPath);

        for (auto& statement : testStatements) {
            if (statement.query.empty() || statement.isPartOfStatementBlock) {
                continue;
            }
            // Find `statement` in the file.
            while (getline(file, currLine)) {
                if (!currLine.starts_with("-STATEMENT")) {
                    newFile += currLine + '\n';
                    if (currLine.starts_with("-CASE")) {
                        static constexpr size_t caseNameOffset = std::string_view("-CASE ").size();
                        testCaseName = currLine.substr(caseNameOffset);
                    }
                    continue;
                }

                newFile += currLine + '\n';

                if (testCaseName != statement.testCaseName) {
                    // Not the CASE for the current `statement`.
                    continue;
                }

                std::string stmt = currLine;
                while (getline(file, currLine)) {
                    if (currLine.starts_with("----")) {
                        // "----" indicates the end of a statement.
                        break;
                    }
                    newFile += currLine + '\n';
                    // There may be other test options before the output is specified.
                    if (!currLine.starts_with("#") && !currLine.starts_with("-")) {
                        stmt += currLine;
                    }
                }
                // Manually add `connName` as `statement.query` does not retain it, if any.
                std::string connName;
                if (statement.connName.has_value() &&
                    statement.connName.value() != "conn_default") {
                    connName = "[" + statement.connName.value() + "] ";
                }

                if (removeAllSpaces(stmt) ==
                    removeAllSpaces("-STATEMENT " + connName + statement.originalQuery)) {
                    // Found the line containing `statement`.
                    break;
                }

                newFile += currLine + '\n';
            }

            // For all cases, ignore the specified expected output in the test file and append the