test_utils.h

Go to the documentation of this file.

// Copyright (c) Space-ng, inc. All rights reserved.
 
#pragma once
 
#include "sol3/cpp/filesystem.h"
 
#include <cstdlib>
#include <optional>
#include <random>
#include <stdexcept>
#include <string>
#include <system_error>
 
namespace sol3::core::test {
inline cpp::fs::path createUniqueTempDirName() {
  cpp::fs::path temp_dir = cpp::fs::temp_directory_path();
  std::random_device rd;
  std::mt19937 gen(rd());
  std::uniform_int_distribution<> distrib(0, 999999);
 
  for (int i = 0; i < 100; ++i) {
    cpp::fs::path full_path =
        temp_dir / ("sol3_tmp_" + std::to_string(distrib(gen)));
    if (!cpp::fs::exists(full_path)) {
      if (cpp::fs::create_directory(full_path)) {
        return full_path;
      }
    }
  }
  throw std::runtime_error(
      "Failed to create a unique temporary directory after several attempts.");
}
 
class UniqueTempOutputDir {
 public:
  UniqueTempOutputDir() : temp_dir_(createUniqueTempDirName()) {}
  ~UniqueTempOutputDir() {
    if (temp_dir_.empty()) {
      return;
    }
    std::error_code ec;
    cpp::fs::remove_all(temp_dir_, ec);
  }
 
  UniqueTempOutputDir(UniqueTempOutputDir const&) = delete;
  UniqueTempOutputDir(UniqueTempOutputDir&&) = delete;
  UniqueTempOutputDir& operator=(UniqueTempOutputDir const&) = delete;
  UniqueTempOutputDir& operator=(UniqueTempOutputDir&&) = delete;
 
  cpp::fs::path path() const { return temp_dir_; }
 
 private:
  cpp::fs::path temp_dir_;
};
 
class ScopedEnvVar {
 public:
  ScopedEnvVar(std::string name, std::string value) : name_(std::move(name)) {
    char const* original = std::getenv(name_.c_str());
    if (original != nullptr) {
      original_value_ = std::string(original);
    }
    ::setenv(name_.c_str(), value.c_str(), 1);
  }
 
  ~ScopedEnvVar() {
    if (original_value_.has_value()) {
      ::setenv(name_.c_str(), original_value_->c_str(), 1);
      return;
    }
    ::unsetenv(name_.c_str());
  }
 
  ScopedEnvVar(ScopedEnvVar const&) = delete;
  ScopedEnvVar(ScopedEnvVar&&) = delete;
  ScopedEnvVar& operator=(ScopedEnvVar const&) = delete;
  ScopedEnvVar& operator=(ScopedEnvVar&&) = delete;
 
 private:
  std::string name_;
  std::optional<std::string> original_value_;
};
}  // namespace sol3::core::test