app_helper.hpp
1.15 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
#pragma once
#ifdef CLI_SINGLE_FILE
#include "CLI11.hpp"
#else
#include "CLI/CLI.hpp"
#endif
#include "gtest/gtest.h"
#include <iostream>
typedef std::vector<std::string> input_t;
struct TApp : public ::testing::Test {
CLI::App app{"My Test Program"};
input_t args;
std::vector<std::string> run() {
input_t newargs = args;
std::reverse(std::begin(newargs), std::end(newargs));
return app.parse(newargs);
}
};
class TempFile {
std::string _name;
public:
TempFile(std::string name) : _name(name) {
if(!CLI::NonexistentPath(_name))
throw std::runtime_error(_name);
}
~TempFile() {
std::remove(_name.c_str()); // Doesn't matter if returns 0 or not
}
operator const std::string &() const { return _name; }
const char *c_str() const { return _name.c_str(); }
};
inline void put_env(std::string name, std::string value) {
#ifdef _MSC_VER
_putenv_s(name.c_str(), value.c_str());
#else
setenv(name.c_str(), value.c_str(), 1);
#endif
}
inline void unset_env(std::string name) {
#ifdef _MSC_VER
_putenv_s(name.c_str(), "");
#else
unsetenv(name.c_str());
#endif
}