mirror of
https://github.com/google/benchmark.git
synced 2024-12-28 05:20:14 +08:00
79 lines
2.0 KiB
C++
79 lines
2.0 KiB
C++
#include <cstdlib>
|
|
|
|
#include "../src/commandlineflags.h"
|
|
#include "../src/internal_macros.h"
|
|
#include "gtest/gtest.h"
|
|
|
|
namespace benchmark {
|
|
namespace {
|
|
|
|
#if defined(BENCHMARK_OS_WINDOWS)
|
|
int setenv(const char* name, const char* value, int overwrite) {
|
|
if (!overwrite) {
|
|
// NOTE: getenv_s is far superior but not available under mingw.
|
|
char* env_value = getenv(name);
|
|
if (env_value == nullptr) {
|
|
return -1;
|
|
}
|
|
}
|
|
return _putenv_s(name, value);
|
|
}
|
|
|
|
int unsetenv(const char* name) {
|
|
return _putenv_s(name, "");
|
|
}
|
|
|
|
#endif // BENCHMARK_OS_WINDOWS
|
|
|
|
TEST(BoolFromEnv, Default) {
|
|
ASSERT_EQ(unsetenv("BENCHMARK_NOT_IN_ENV"), 0);
|
|
EXPECT_EQ(BoolFromEnv("not_in_env", true), true);
|
|
}
|
|
|
|
TEST(BoolFromEnv, False) {
|
|
ASSERT_EQ(setenv("BENCHMARK_IN_ENV", "0", 1), 0);
|
|
EXPECT_EQ(BoolFromEnv("in_env", true), false);
|
|
unsetenv("BENCHMARK_IN_ENV");
|
|
}
|
|
|
|
TEST(BoolFromEnv, True) {
|
|
ASSERT_EQ(setenv("BENCHMARK_IN_ENV", "1", 1), 0);
|
|
EXPECT_EQ(BoolFromEnv("in_env", false), true);
|
|
unsetenv("BENCHMARK_IN_ENV");
|
|
|
|
ASSERT_EQ(setenv("BENCHMARK_IN_ENV", "foo", 1), 0);
|
|
EXPECT_EQ(BoolFromEnv("in_env", false), true);
|
|
unsetenv("BENCHMARK_IN_ENV");
|
|
}
|
|
|
|
TEST(Int32FromEnv, NotInEnv) {
|
|
ASSERT_EQ(unsetenv("BENCHMARK_NOT_IN_ENV"), 0);
|
|
EXPECT_EQ(Int32FromEnv("not_in_env", 42), 42);
|
|
}
|
|
|
|
TEST(Int32FromEnv, InvalidInteger) {
|
|
ASSERT_EQ(setenv("BENCHMARK_IN_ENV", "foo", 1), 0);
|
|
EXPECT_EQ(Int32FromEnv("in_env", 42), 42);
|
|
ASSERT_EQ(unsetenv("BENCHMARK_IN_ENV"), 0);
|
|
}
|
|
|
|
TEST(Int32FromEnv, ValidInteger) {
|
|
ASSERT_EQ(setenv("BENCHMARK_IN_ENV", "42", 1), 0);
|
|
EXPECT_EQ(Int32FromEnv("in_env", 64), 42);
|
|
unsetenv("BENCHMARK_IN_ENV");
|
|
}
|
|
|
|
TEST(StringFromEnv, Default) {
|
|
ASSERT_EQ(unsetenv("BENCHMARK_NOT_IN_ENV"), 0);
|
|
EXPECT_STREQ(StringFromEnv("not_in_env", "foo"), "foo");
|
|
}
|
|
|
|
TEST(StringFromEnv, Valid) {
|
|
ASSERT_EQ(setenv("BENCHMARK_IN_ENV", "foo", 1), 0);
|
|
EXPECT_STREQ(StringFromEnv("in_env", "bar"), "foo");
|
|
unsetenv("BENCHMARK_IN_ENV");
|
|
}
|
|
|
|
} // namespace
|
|
} // namespace benchmark
|