69 lines · python
1"""Test the SBProgress API."""2 3import lldb4from lldbsuite.test.lldbtest import *5 6 7class SBProgressTestCase(TestBase):8 def test_with_external_bit_set(self):9 """Test SBProgress can handle null events."""10 11 progress = lldb.SBProgress("Test SBProgress", "Test progress", 3, self.dbg)12 listener = lldb.SBListener("Test listener")13 broadcaster = self.dbg.GetBroadcaster()14 broadcaster.AddListener(listener, lldb.eBroadcastBitExternalProgress)15 event = lldb.SBEvent()16 # Sample JSON we're expecting:17 # { id = 2, title = "Test SBProgress", details = "Test progress", type = update, progress = 1 of 3}18 # details remains the same as specified in the constructor of the progress19 # until we update it in the increment function, so we check for the Null and empty string case20 # that details hasn't changed, but progress x of 3 has.21 progress.Increment(1, None)22 self.assertTrue(listener.GetNextEvent(event))23 stream = lldb.SBStream()24 event.GetDescription(stream)25 self.assertIn("Test progress", stream.GetData())26 self.assertIn("1 of 3", stream.GetData())27 28 progress.Increment(1, "")29 self.assertTrue(listener.GetNextEvent(event))30 event.GetDescription(stream)31 self.assertIn("Test progress", stream.GetData())32 self.assertIn("2 of 3", stream.GetData())33 34 progress.Increment(1, "Step 3")35 self.assertTrue(listener.GetNextEvent(event))36 stream = lldb.SBStream()37 event.GetDescription(stream)38 self.assertIn("Step 3", stream.GetData())39 40 def test_progress_finalize_non_deterministic_progress(self):41 """Test SBProgress finalize sends the progressEnd event"""42 43 progress = lldb.SBProgress("Test SBProgress", "Test finalize", self.dbg)44 listener = lldb.SBListener("Test listener")45 broadcaster = self.dbg.GetBroadcaster()46 broadcaster.AddListener(listener, lldb.eBroadcastBitExternalProgress)47 event = lldb.SBEvent()48 progress.Finalize()49 self.assertTrue(listener.WaitForEvent(5, event))50 stream = lldb.SBStream()51 event.GetDescription(stream)52 self.assertIn("type = end", stream.GetData())53 54 def test_progress_finalize_deterministic_progress(self):55 """Test SBProgress finalize sends the progressEnd event"""56 57 progress = lldb.SBProgress("Test SBProgress", "Test finalize", 13, self.dbg)58 listener = lldb.SBListener("Test listener")59 broadcaster = self.dbg.GetBroadcaster()60 broadcaster.AddListener(listener, lldb.eBroadcastBitExternalProgress)61 event = lldb.SBEvent()62 progress.Finalize()63 self.assertTrue(listener.WaitForEvent(5, event))64 stream = lldb.SBStream()65 event.GetDescription(stream)66 # Note even for progresses with a total, the total isn't67 # sent in the end message.68 self.assertIn("type = end", stream.GetData())69