CLI.integration.hpp

Go to the documentation of this file.

#pragma once
 
#include <gtest/gtest.h>
 
#include "io/TempFile.hpp"
 
#include <cstdio>
#include <cstdlib>
#include <fstream>
#include <string>
#include <sstream>
#include <filesystem>
 
#include <nlohmann/json.hpp>
 
using json = nlohmann::json;
 
#ifndef PPFOREST2_BINARY_PATH
#error "PPFOREST2_BINARY_PATH must be defined"
#endif
 
#ifndef PPFOREST2_DATA_DIR
#error "PPFOREST2_DATA_DIR must be defined"
#endif
 
#ifndef PPFOREST2_GOLDEN_DIR
#error "PPFOREST2_GOLDEN_DIR must be defined"
#endif
 
inline const std::string BINARY     = PPFOREST2_BINARY_PATH;
inline std::string const DATA_DIR   = PPFOREST2_DATA_DIR;
inline std::string const GOLDEN_DIR = PPFOREST2_GOLDEN_DIR;
inline std::string const IRIS_CSV   = DATA_DIR + "/iris.csv";
inline std::string const CRAB_CSV   = DATA_DIR + "/crab.csv";
inline std::string const WINE_CSV   = DATA_DIR + "/wine.csv";
inline std::string const GLASS_CSV  = DATA_DIR + "/glass.csv";

struct ProcessResult {
  int exit_code;
  std::string stdout_output;
  std::string stderr_output;
};

inline ProcessResult run_ppforest2(std::string const& args) {
  ppforest2::io::TempFile stderr_file;
 
#ifdef _WIN32
  std::string cmd = BINARY + " " + args + " 2>\"" + stderr_file.path() + "\"";
  FILE* pipe      = _popen(cmd.c_str(), "r");
#else
  std::string cmd = BINARY + " " + args + " 2>\"" + stderr_file.path() + "\"";
  FILE* pipe      = popen(cmd.c_str(), "r");
#endif
 
  if (!pipe) {
    return {-1, "", ""};
  }
 
  std::string output;
  char buffer[4096];
 
  while (fgets(buffer, sizeof(buffer), pipe))
    output += buffer;
 
#ifdef _WIN32
  int exit_code = _pclose(pipe);
#else
  int status    = pclose(pipe);
  int exit_code = WIFEXITED(status) ? WEXITSTATUS(status) : -1;
#endif
 
  return {exit_code, output, stderr_file.read()};
}
 
using ppforest2::io::TempDir;
using ppforest2::io::TempFile;

class SavedModelTest : public ::testing::Test {
protected:
  void SetUp() override {
    model_.reset(new TempFile());
    model_->clear();
    auto result = run_ppforest2("-q train -d " + IRIS_CSV + " -n 5 -r 0 -s " + model_->path());
    ASSERT_EQ(result.exit_code, 0);
    model_json_ = json::parse(model_->read());
  }
 
  std::unique_ptr<TempFile> model_;
  json model_json_;
};