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 <memory>
#include <string>
 
#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;
// Classification datasets live under `${DATA_DIR}/classification/`, regression
// under `${DATA_DIR}/regression/`. Path constants pin the subfolder explicitly
// so tests don't have to know the layout convention.
inline std::string const IRIS_CSV    = DATA_DIR + "/classification/iris.csv";
inline std::string const CRAB_CSV    = DATA_DIR + "/classification/crab.csv";
inline std::string const WINE_CSV    = DATA_DIR + "/classification/wine.csv";
inline std::string const GLASS_CSV   = DATA_DIR + "/classification/glass.csv";
inline std::string const MTCARS_CSV  = DATA_DIR + "/regression/mtcars.csv";
inline std::string const HOUSING_CSV = DATA_DIR + "/regression/california_housing.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;
 
  // clang-format off
  #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
  // clang-format on
  if (pipe == nullptr) {
    return {-1, "", ""};
  }
 
  std::string output;
  char buffer[4096];
 
  while (fgets(buffer, sizeof(buffer), pipe) != nullptr) {
    output += buffer;
  }
 
  // clang-format off
  #ifdef _WIN32
  int exit_code = _pclose(pipe);
  #else
  int status    = pclose(pipe);
  int exit_code = WIFEXITED(status) ? WEXITSTATUS(status) : -1;
  #endif
  // clang-format on
 
  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_ = std::make_unique<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_;
};