refactored output stream for CoverageAndResultsStatisticsPrinter

Author: Lana243

server/src/printers/CoverageAndResultsStatisticsPrinter.cpp

@@ -38,9 +38,9 @@ namespace printer {
         return os;
     }
 
-    void CoverageAndResultsStatisticsPrinter::write(const utbot::ProjectContext &projectContext,
-                                                    const Coverage::TestResultMap &testsResultMap,
-                                                    const Coverage::CoverageMap &coverageMap) {
+    std::stringstream CoverageAndResultsStatisticsPrinter::write(const utbot::ProjectContext &projectContext,
+                                                                 const Coverage::TestResultMap &testsResultMap,
+                                                                 const Coverage::CoverageMap &coverageMap) {
         for (auto const &[testPath, testsResult]: testsResultMap) {
             fs::path sourcePath = Paths::testPathToSourcePath(projectContext, testPath);
             Coverage::FileCoverage fileCoverage = CollectionUtils::getOrDefault(coverageMap,
@@ -65,9 +65,7 @@ namespace printer {
             ss << statistics << '\n';
         }
         ss << "Total," << total << '\n';
-        fs::path resultsFilePath = resultsDirectory / "coverage-and-results.csv";
-        FileSystemUtils::writeToFile(resultsFilePath, ss.str());
-        LOG_S(INFO) << StringUtils::stringFormat("See statistics info here: %s", resultsFilePath);
+        return ss;
     }
 
     FileCoverageAndResultsStatistics::FileCoverageAndResultsStatistics(

server/src/printers/CoverageAndResultsStatisticsPrinter.h

@@ -36,9 +36,10 @@ namespace printer {
 
     class CoverageAndResultsStatisticsPrinter : CollectionUtils::MapFileTo<FileCoverageAndResultsStatistics> {
     public:
-        explicit CoverageAndResultsStatisticsPrinter(fs::path resultsDirectory) :
-            resultsDirectory(std::move(resultsDirectory)) {};
-        void write(const utbot::ProjectContext &, const Coverage::TestResultMap &, const Coverage::CoverageMap &);
+        CoverageAndResultsStatisticsPrinter() = default;
+        std::stringstream write(const utbot::ProjectContext &projectContext,
+                                const Coverage::TestResultMap &testsResultMap,
+                                const Coverage::CoverageMap &coverageMap);
 
     private:
         fs::path resultsDirectory;

server/src/streams/coverage/CLICoverageAndResultsWriter.cpp

@@ -7,6 +7,7 @@
 #include "loguru.h"
 
 #include <fstream>
+#include <utils/StringFormat.h>
 
 CLICoverageAndResultsWriter::CLICoverageAndResultsWriter(const fs::path &resultsDirectory)
     : resultsDirectory(resultsDirectory), CoverageAndResultsWriter(nullptr) {
@@ -57,6 +58,10 @@ void CLICoverageAndResultsWriter::writeResponse(const utbot::ProjectContext &pro
     fs::path resultsFilePath = resultsDirectory / "tests-result.log";
     FileSystemUtils::writeToFile(resultsFilePath, ss.str());
     LOG_S(INFO) << ss.str();
-    printer::CoverageAndResultsStatisticsPrinter statsPrinter = printer::CoverageAndResultsStatisticsPrinter(resultsDirectory);
-    statsPrinter.write(projectContext, testsResultMap, coverageMap);
+    
+    printer::CoverageAndResultsStatisticsPrinter statsPrinter = printer::CoverageAndResultsStatisticsPrinter();
+    std::stringstream statsStringStream = statsPrinter.write(projectContext, testsResultMap, coverageMap);
+    fs::path statsResultsFilePath = resultsDirectory / "coverage-and-results.csv";
+    FileSystemUtils::writeToFile(statsResultsFilePath, statsStringStream.str());
+    LOG_S(INFO) << StringUtils::stringFormat("See coverage and results statistics info here: %s", statsResultsFilePath);
 }