"""
Test lldbutil.lldb_iter() which returns an iterator object for lldb's aggregate
data structures.
"""

import os, time
import re
import unittest2
import lldb
from lldbtest import *

class LLDBIteratorTestCase(TestBase):

    mydir = "python_api/lldbutil"

    def setUp(self):
        # Call super's setUp().
        TestBase.setUp(self)
        # Find the line numbers to break inside main().
        self.line1 = line_number('main.cpp', '// Set break point at this line.')
        self.line2 = line_number('main.cpp', '// And that line.')

    def test_lldb_iter_1(self):
        """Test lldb_iter works correctly for SBTarget -> SBModule."""
        self.buildDefault()
        self.lldb_iter_1()

    def test_lldb_iter_2(self):
        """Test lldb_iter works correctly for SBTarget -> SBBreakpoint."""
        self.buildDefault()
        self.lldb_iter_2()

    def lldb_iter_1(self):
        exe = os.path.join(os.getcwd(), "a.out")

        target = self.dbg.CreateTarget(exe)
        self.assertTrue(target.IsValid(), VALID_TARGET)

        breakpoint = target.BreakpointCreateByLocation("main.cpp", self.line1)
        self.assertTrue(breakpoint.IsValid(), VALID_BREAKPOINT)

        # Now launch the process, and do not stop at entry point.
        rc = lldb.SBError()
        self.process = target.Launch (self.dbg.GetListener(), None, None, os.ctermid(), os.ctermid(), os.ctermid(), None, 0, False, rc)

        if not rc.Success() or not self.process.IsValid():
            self.fail("SBTarget.LaunchProcess() failed")

        from lldbutil import lldb_iter, get_description
        yours = []
        for i in range(target.GetNumModules()):
            yours.append(target.GetModuleAtIndex(i))
        mine = []
        for m in lldb_iter(target, 'GetNumModules', 'GetModuleAtIndex'):
            mine.append(m)

        self.assertTrue(len(yours) == len(mine))
        for i in range(len(yours)):
            if self.TraceOn():
                print "yours[%d]='%s'" % (i, get_description(yours[i]))
                print "mine[%d]='%s'" % (i, get_description(mine[i]))
            self.assertTrue(yours[i].GetUUIDString() == mine[i].GetUUIDString(),
                            "UUID of yours[%d] and mine[%d] matches" % (i, i))

    def lldb_iter_2(self):
        exe = os.path.join(os.getcwd(), "a.out")

        target = self.dbg.CreateTarget(exe)
        self.assertTrue(target.IsValid(), VALID_TARGET)

        breakpoint = target.BreakpointCreateByLocation("main.cpp", self.line1)
        self.assertTrue(breakpoint.IsValid(), VALID_BREAKPOINT)
        breakpoint = target.BreakpointCreateByLocation("main.cpp", self.line2)
        self.assertTrue(breakpoint.IsValid(), VALID_BREAKPOINT)

        self.assertTrue(target.GetNumBreakpoints() == 2)

        from lldbutil import lldb_iter, get_description
        yours = []
        for i in range(target.GetNumBreakpoints()):
            yours.append(target.GetBreakpointAtIndex(i))
        mine = []
        for m in lldb_iter(target, 'GetNumBreakpoints', 'GetBreakpointAtIndex'):
            mine.append(m)

        self.assertTrue(len(yours) == len(mine))
        for i in range(len(yours)):
            if self.TraceOn():
                print "yours[%d]='%s'" % (i, get_description(yours[i]))
                print "mine[%d]='%s'" % (i, get_description(mine[i]))
            self.assertTrue(yours[i].GetID() == mine[i].GetID(),
                            "ID of yours[%d] and mine[%d] matches" % (i, i))


if __name__ == '__main__':
    import atexit
    lldb.SBDebugger.Initialize()
    atexit.register(lambda: lldb.SBDebugger.Terminate())
    unittest2.main()

@@ -2199,34 +2199,35 @@
  %22UUID of yours%5B
-%25d
+%7B0%7D
 %5D and mine%5B%25d%5D m
@@ -2213,34 +2213,35 @@
 s%5B%7B0%7D%5D and mine%5B
-%25d
+%7B0%7D
 %5D matches%22 %25 (i,
@@ -2230,31 +2230,32 @@
 0%7D%5D matches%22
- %25 (i, 
+.format(
 i))%0A%0A    def
@@ -3445,18 +3445,19 @@
 f yours%5B
-%25d
+%7B0%7D
 %5D and mi
@@ -3459,18 +3459,19 @@
 nd mine%5B
-%25d
+%7B0%7D
 %5D matche
@@ -3472,23 +3472,24 @@
 matches%22
- %25 (i, 
+.format(
 i))%0A%0A%0Aif

Use new string formatting operations in assert messages to simplify things.