aboutsummaryrefslogtreecommitdiff
path: root/lldb/test/API/python_api/sbprogress/TestSBProgress.py
blob: b64c4b9c7206e6f94eab79838a45ac33db3aba22 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
"""Test the SBProgress API."""

import lldb
from lldbsuite.test.lldbtest import *


class SBProgressTestCase(TestBase):
    def test_with_external_bit_set(self):
        """Test SBProgress can handle null events."""

        progress = lldb.SBProgress("Test SBProgress", "Test progress", 3, self.dbg)
        listener = lldb.SBListener("Test listener")
        broadcaster = self.dbg.GetBroadcaster()
        broadcaster.AddListener(listener, lldb.eBroadcastBitExternalProgress)
        event = lldb.SBEvent()
        # Sample JSON we're expecting:
        # { id = 2, title = "Test SBProgress", details = "Test progress", type = update, progress = 1 of 3}
        # details remains the same as specified in the constructor of the progress
        # until we update it in the increment function, so we check for the Null and empty string case
        # that details hasn't changed, but progress x of 3 has.
        progress.Increment(1, None)
        self.assertTrue(listener.GetNextEvent(event))
        stream = lldb.SBStream()
        event.GetDescription(stream)
        self.assertIn("Test progress", stream.GetData())
        self.assertIn("1 of 3", stream.GetData())

        progress.Increment(1, "")
        self.assertTrue(listener.GetNextEvent(event))
        event.GetDescription(stream)
        self.assertIn("Test progress", stream.GetData())
        self.assertIn("2 of 3", stream.GetData())

        progress.Increment(1, "Step 3")
        self.assertTrue(listener.GetNextEvent(event))
        stream = lldb.SBStream()
        event.GetDescription(stream)
        self.assertIn("Step 3", stream.GetData())

    def test_progress_finalize_non_deterministic_progress(self):
        """Test SBProgress finalize sends the progressEnd event"""

        progress = lldb.SBProgress("Test SBProgress", "Test finalize", self.dbg)
        listener = lldb.SBListener("Test listener")
        broadcaster = self.dbg.GetBroadcaster()
        broadcaster.AddListener(listener, lldb.eBroadcastBitExternalProgress)
        event = lldb.SBEvent()
        progress.Finalize()
        self.assertTrue(listener.WaitForEvent(5, event))
        stream = lldb.SBStream()
        event.GetDescription(stream)
        self.assertIn("type = end", stream.GetData())

    def test_progress_finalize_deterministic_progress(self):
        """Test SBProgress finalize sends the progressEnd event"""

        progress = lldb.SBProgress("Test SBProgress", "Test finalize", 13, self.dbg)
        listener = lldb.SBListener("Test listener")
        broadcaster = self.dbg.GetBroadcaster()
        broadcaster.AddListener(listener, lldb.eBroadcastBitExternalProgress)
        event = lldb.SBEvent()
        progress.Finalize()
        self.assertTrue(listener.WaitForEvent(5, event))
        stream = lldb.SBStream()
        event.GetDescription(stream)
        # Note even for progresses with a total, the total isn't
        # sent in the end message.
        self.assertIn("type = end", stream.GetData())