xref: /llvm-project/lldb/test/API/python_api/sblaunchinfo/TestSBLaunchInfo.py (revision 80fcecb13c388ff087a27a4b0e7ca3dd8c98eaa4) 
1"""
2Test SBLaunchInfo
3"""
4
5
6from lldbsuite.test.lldbtest import *
7
8
9def lookup(info, key):
10    for i in range(info.GetNumEnvironmentEntries()):
11        KeyEqValue = info.GetEnvironmentEntryAtIndex(i)
12        Key, Value = KeyEqValue.split("=")
13        if Key == key:
14            return Value
15    return ""
16
17
18class TestSBLaunchInfo(TestBase):
19    NO_DEBUG_INFO_TESTCASE = True
20
21    def test_environment_getset(self):
22        info = lldb.SBLaunchInfo(None)
23        info.SetEnvironmentEntries(["FOO=BAR"], False)
24        self.assertEqual(1, info.GetNumEnvironmentEntries())
25        info.SetEnvironmentEntries(["BAR=BAZ"], True)
26        self.assertEqual(2, info.GetNumEnvironmentEntries())
27        self.assertEqual("BAR", lookup(info, "FOO"))
28        self.assertEqual("BAZ", lookup(info, "BAR"))
29