Added two test cases to TestSettings.py which exercise the lldb's:

(lldb) settings set process.run-args A B C
(lldb) settings set process.env-vars ["MY_ENV_VAR"]=YES

commands.  The main.cpp checks whether A, B, C is passed to main and whether
the $MY_ENV_VAR env variable is defined and outputs the findings to a file.

llvm-svn: 114031
This commit is contained in:
Johnny Chen
2010-09-15 22:27:29 +00:00
parent cb54b960b8
commit d5e111c2c1
3 changed files with 79 additions and 0 deletions

View File

@@ -0,0 +1,5 @@
LEVEL = ../make
CXX_SOURCES := main.cpp
include $(LEVEL)/Makefile.rules

View File

@@ -39,6 +39,36 @@ class SettingsCommandTestCase(TestBase):
self.expect("settings show",
startstr = "term-width (int) = '70'")
@unittest2.skipUnless(sys.platform.startswith("darwin"), "requires Darwin")
def test_with_dsym(self):
"""Test that run-args and env-vars are passed to the launched process."""
self.buildDsym()
self.pass_run_args_and_env_vars()
def test_with_dwarf(self):
"""Test that run-args and env-vars are passed to the launched process."""
self.buildDwarf()
self.pass_run_args_and_env_vars()
def pass_run_args_and_env_vars(self):
"""Test that run-args and env-vars are passed to the launched process."""
exe = os.path.join(os.getcwd(), "a.out")
self.runCmd("file " + exe, CURRENT_EXECUTABLE_SET)
# Set the run-args and the env-vars.
self.runCmd('settings set process.run-args A B C')
self.runCmd('settings set process.env-vars ["MY_ENV_VAR"]=YES')
self.runCmd("run", RUN_SUCCEEDED)
# Read the output file produced by running the program.
output = open('/tmp/output.txt', 'r').read()
self.assertTrue(output.startswith("argv[1] matches") and
output.find("argv[2] matches") > 0 and
output.find("argv[3] matches") > 0 and
output.find("Environment variable 'MY_ENV_VAR' successfully passed.") > 0)
if __name__ == '__main__':
import atexit

View File

@@ -0,0 +1,44 @@
//===-- main.cpp ------------------------------------------------*- C++ -*-===//
//
// The LLVM Compiler Infrastructure
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//
#include <cstdlib>
#include <string>
#include <fstream>
int
main(int argc, char const *argv[])
{
char const *cptr = NULL;
// The program writes its output to the "output.txt" file.
std::ofstream outfile("output.txt");
for (unsigned i = 0, e = sizeof(argv); i < e; ++i) {
if ((cptr = argv[i]) == NULL)
break;
std::string str(cptr);
if (i == 1 && "A" == str)
outfile << "argv[1] matches\n";
if (i == 2 && "B" == str)
outfile << "argv[2] matches\n";
if (i == 3 && "C" == str)
outfile << "argv[3] matches\n";
}
if (::getenv("MY_ENV_VAR")) {
std::string MY_ENV_VAR(getenv("MY_ENV_VAR"));
if ("YES" == MY_ENV_VAR) {
outfile << "Environment variable 'MY_ENV_VAR' successfully passed.\n";
}
}
return 0;
}