1*99451b44SJordan Rupprecht"""
2*99451b44SJordan RupprechtTest SBLaunchInfo
3*99451b44SJordan Rupprecht"""
4*99451b44SJordan Rupprecht
5*99451b44SJordan Rupprecht
6*99451b44SJordan Rupprecht
7*99451b44SJordan Rupprechtfrom lldbsuite.test.lldbtest import *
8*99451b44SJordan Rupprecht
9*99451b44SJordan Rupprecht
10*99451b44SJordan Rupprechtdef lookup(info, key):
11*99451b44SJordan Rupprecht    for i in range(info.GetNumEnvironmentEntries()):
12*99451b44SJordan Rupprecht        KeyEqValue = info.GetEnvironmentEntryAtIndex(i)
13*99451b44SJordan Rupprecht        Key, Value = KeyEqValue.split("=")
14*99451b44SJordan Rupprecht        if Key == key:
15*99451b44SJordan Rupprecht            return Value
16*99451b44SJordan Rupprecht    return ""
17*99451b44SJordan Rupprecht
18*99451b44SJordan Rupprechtclass TestSBLaunchInfo(TestBase):
19*99451b44SJordan Rupprecht    NO_DEBUG_INFO_TESTCASE = True
20*99451b44SJordan Rupprecht
21*99451b44SJordan Rupprecht    def test_environment_getset(self):
22*99451b44SJordan Rupprecht        info = lldb.SBLaunchInfo(None)
23*99451b44SJordan Rupprecht        info.SetEnvironmentEntries(["FOO=BAR"], False)
24*99451b44SJordan Rupprecht        self.assertEquals(1, info.GetNumEnvironmentEntries())
25*99451b44SJordan Rupprecht        info.SetEnvironmentEntries(["BAR=BAZ"], True)
26*99451b44SJordan Rupprecht        self.assertEquals(2, info.GetNumEnvironmentEntries())
27*99451b44SJordan Rupprecht        self.assertEquals("BAR", lookup(info, "FOO"))
28*99451b44SJordan Rupprecht        self.assertEquals("BAZ", lookup(info, "BAR"))
29