packages/Python/lldbsuite/test/python_api/lldbutil/iter/TestLLDBIterator.py

   1 """
   2 Test the iteration protocol for some lldb container objects.
   3 """
   4
   5 from __future__ import print_function
   6
   7
   8
   9 import os, time
  10 import re
  11 import lldb
  12 from lldbsuite.test.decorators import *
  13 from lldbsuite.test.lldbtest import *
  14 from lldbsuite.test import lldbutil
  15
  16 class LLDBIteratorTestCase(TestBase):
  17
  18     mydir = TestBase.compute_mydir(__file__)
  19
  20     def setUp(self):
  21         # Call super's setUp().
  22         TestBase.setUp(self)
  23         # Find the line numbers to break inside main().
  24         self.line1 = line_number('main.cpp', '// Set break point at this line.')
  25         self.line2 = line_number('main.cpp', '// And that line.')
  26
  27     @add_test_categories(['pyapi'])
  28     def test_lldb_iter_module(self):
  29         """Test module_iter works correctly for SBTarget -> SBModule."""
  30         self.build()
  31         exe = os.path.join(os.getcwd(), "a.out")
  32
  33         target = self.dbg.CreateTarget(exe)
  34         self.assertTrue(target, VALID_TARGET)
  35
  36         breakpoint = target.BreakpointCreateByLocation("main.cpp", self.line1)
  37         self.assertTrue(breakpoint, VALID_BREAKPOINT)
  38
  39         # Now launch the process, and do not stop at entry point.
  40         process = target.LaunchSimple (None, None, self.get_process_working_directory())
  41
  42         if not process:
  43             self.fail("SBTarget.LaunchProcess() failed")
  44
  45         from lldbsuite.test.lldbutil import get_description
  46         yours = []
  47         for i in range(target.GetNumModules()):
  48             yours.append(target.GetModuleAtIndex(i))
  49         mine = []
  50         for m in target.module_iter():
  51             mine.append(m)
  52
  53         self.assertTrue(len(yours) == len(mine))
  54         for i in range(len(yours)):
  55             if self.TraceOn():
  56                 print("yours[%d]='%s'" % (i, get_description(yours[i])))
  57                 print("mine[%d]='%s'" % (i, get_description(mine[i])))
  58             self.assertTrue(yours[i] == mine[i],
  59                             "UUID+FileSpec of yours[{0}] and mine[{0}] matches".format(i))
  60
  61     @add_test_categories(['pyapi'])
  62     def test_lldb_iter_breakpoint(self):
  63         """Test breakpoint_iter works correctly for SBTarget -> SBBreakpoint."""
  64         self.build()
  65         exe = os.path.join(os.getcwd(), "a.out")
  66
  67         target = self.dbg.CreateTarget(exe)
  68         self.assertTrue(target, VALID_TARGET)
  69
  70         breakpoint = target.BreakpointCreateByLocation("main.cpp", self.line1)
  71         self.assertTrue(breakpoint, VALID_BREAKPOINT)
  72         breakpoint = target.BreakpointCreateByLocation("main.cpp", self.line2)
  73         self.assertTrue(breakpoint, VALID_BREAKPOINT)
  74
  75         self.assertTrue(target.GetNumBreakpoints() == 2)
  76
  77         from lldbsuite.test.lldbutil import get_description
  78         yours = []
  79         for i in range(target.GetNumBreakpoints()):
  80             yours.append(target.GetBreakpointAtIndex(i))
  81         mine = []
  82         for b in target.breakpoint_iter():
  83             mine.append(b)
  84
  85         self.assertTrue(len(yours) == len(mine))
  86         for i in range(len(yours)):
  87             if self.TraceOn():
  88                 print("yours[%d]='%s'" % (i, get_description(yours[i])))
  89                 print("mine[%d]='%s'" % (i, get_description(mine[i])))
  90             self.assertTrue(yours[i] == mine[i],
  91                             "ID of yours[{0}] and mine[{0}] matches".format(i))
  92
  93     @add_test_categories(['pyapi'])
  94     def test_lldb_iter_frame(self):
  95         """Test iterator works correctly for SBProcess->SBThread->SBFrame."""
  96         self.build()
  97         exe = os.path.join(os.getcwd(), "a.out")
  98
  99         target = self.dbg.CreateTarget(exe)
 100         self.assertTrue(target, VALID_TARGET)
 101
 102         breakpoint = target.BreakpointCreateByLocation("main.cpp", self.line1)
 103         self.assertTrue(breakpoint, VALID_BREAKPOINT)
 104
 105         # Now launch the process, and do not stop at entry point.
 106         process = target.LaunchSimple (None, None, self.get_process_working_directory())
 107
 108         if not process:
 109             self.fail("SBTarget.LaunchProcess() failed")
 110
 111         from lldbsuite.test.lldbutil import print_stacktrace
 112         stopped_due_to_breakpoint = False
 113         for thread in process:
 114             if self.TraceOn():
 115                 print_stacktrace(thread)
 116             ID = thread.GetThreadID()
 117             if thread.GetStopReason() == lldb.eStopReasonBreakpoint:
 118                 stopped_due_to_breakpoint = True
 119             for frame in thread:
 120                 self.assertTrue(frame.GetThread().GetThreadID() == ID)
 121                 if self.TraceOn():
 122                     print(frame)
 123
 124         self.assertTrue(stopped_due_to_breakpoint)