writer_tester.hpp

Go to the documentation of this file.

 
#ifndef WRITER_TESTS_HPP
#define WRITER_TESTS_HPP
 
#include <gtest/gtest.h>
 
#include <filesystem>
#include <fstream>
 
#include "ROOT/writer.hpp"
 
class WriterBaseTester : public ::testing::Test {
  public:
    template <typename T>
    void testBuildPrinter(const T& values, WritingMethod method,
                          std::unique_ptr<PrinterBase<Eigen::Vector2d>>& printer) {
        Writer<T> writer(values, method);
        writer.build_printer(printer);
 
        switch (method) {
            case WritingMethod::CONSOLE:
                EXPECT_NE(dynamic_cast<PrinterCLI<Eigen::Vector2d>*>(printer.get()), nullptr);
                break;
            case WritingMethod::CSV:
                EXPECT_NE(dynamic_cast<PrinterCSV<Eigen::Vector2d>*>(printer.get()), nullptr);
                break;
            case WritingMethod::DAT:
                EXPECT_NE(dynamic_cast<PrinterDAT<Eigen::Vector2d>*>(printer.get()), nullptr);
                break;
            case WritingMethod::GNUPLOT:
                EXPECT_NE(dynamic_cast<PrinterGNUPlot<Eigen::Vector2d>*>(printer.get()), nullptr);
                break;
            default:
                FAIL() << "Unknown WritingMethod";
        }
    }
 
    template <typename T>
    void testWrite(const T& values, WritingMethod method) {
        Writer<T> writer(values, method);
        writer.write();
        EXPECT_NO_THROW(writer.write());
        // check if the method is CLI it is written on CLI
        switch (method) {
            case WritingMethod::CONSOLE:
                // CLI output is not easily testable; assume success if no exceptions are thrown
                break;
            case WritingMethod::CSV:
                // check if the method is CSV file is created
                EXPECT_TRUE(std::filesystem::exists("output.csv"));
                std::filesystem::remove("output.csv");
                break;
            case WritingMethod::DAT:
                // check if the method is DAT file is created
                EXPECT_TRUE(std::filesystem::exists("output.dat"));
                std::filesystem::remove("output.dat");
                break;
            case WritingMethod::GNUPLOT:
                // check if the method is GNUPLOT file is created and gnuplot script is generated
                EXPECT_TRUE(std::filesystem::exists("output.dat"));
                std::filesystem::remove("output.dat");
                EXPECT_TRUE(std::filesystem::exists("output.plt"));
                std::filesystem::remove("output.plt");
                EXPECT_TRUE(std::filesystem::exists("output.png"));
                std::filesystem::remove("output.png");
                break;
            default:
                FAIL() << "Unknown WritingMethod";
        }
    }
};
 
#endif