1 //===- llvm/unittest/Support/CommandLineTest.cpp - CommandLine tests ------===//
2 //
3 // The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9
10 #include "llvm/Support/CommandLine.h"
11 #include "llvm/Config/config.h"
12
13 #include "gtest/gtest.h"
14
15 #include <string>
16 #include <stdlib.h>
17
18 using namespace llvm;
19
20 namespace {
21
22 class TempEnvVar {
23 public:
TempEnvVar(const char * name,const char * value)24 TempEnvVar(const char *name, const char *value)
25 : name(name) {
26 const char *old_value = getenv(name);
27 EXPECT_EQ(NULL, old_value) << old_value;
28 #if HAVE_SETENV
29 setenv(name, value, true);
30 #else
31 # define SKIP_ENVIRONMENT_TESTS
32 #endif
33 }
34
~TempEnvVar()35 ~TempEnvVar() {
36 #if HAVE_SETENV
37 // Assume setenv and unsetenv come together.
38 unsetenv(name);
39 #endif
40 }
41
42 private:
43 const char *const name;
44 };
45
46 #ifndef SKIP_ENVIRONMENT_TESTS
47
48 const char test_env_var[] = "LLVM_TEST_COMMAND_LINE_FLAGS";
49
50 cl::opt<std::string> EnvironmentTestOption("env-test-opt");
TEST(CommandLineTest,ParseEnvironment)51 TEST(CommandLineTest, ParseEnvironment) {
52 TempEnvVar TEV(test_env_var, "-env-test-opt=hello");
53 EXPECT_EQ("", EnvironmentTestOption);
54 cl::ParseEnvironmentOptions("CommandLineTest", test_env_var);
55 EXPECT_EQ("hello", EnvironmentTestOption);
56 }
57
58 #endif // SKIP_ENVIRONMENT_TESTS
59
60 } // anonymous namespace
61