summaryrefslogtreecommitdiffstats
path: root/lldb/test/python_api/input_reader/TestInputReaderCallback.py
blob: 8fea84426f1ae839bfbc3207b2ba3a9cf39d8668 (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
"""Test the SBInputReader callbacks."""

import os
import unittest2
import lldb
from lldbtest import TestBase, python_api_test, dwarf_test


class InputReaderCallbackCase(TestBase):

    mydir = TestBase.compute_mydir(__file__)

    @python_api_test
    @dwarf_test
    def test_with_dwarf_and_good_callback(self):
        """Test the SBInputReader callbacks."""
        def callback(reader, notification, content):
            global succeeded
            if (notification == lldb.eInputReaderGotToken):
                self.succeeded = True
            return len(content)
        self.buildDwarf()
        self.input_reader_callback(callback)

    def setUp(self):
        # Call super's setUp().
        TestBase.setUp(self)

    def input_reader_callback(self, callback):
        """Test the SBInputReader callbacks."""
        self.succeeded = False

        input_reader = lldb.SBInputReader()
        input_reader.Initialize(self.dbg, callback, lldb.eInputReaderGranularityByte, "$", "^", False)

        self.dbg.PushInputReader(input_reader)
        self.dbg.DispatchInput("Hello!$")
        self.assertFalse(self.dbg.InputReaderIsTopReader(input_reader))
        self.assertTrue(self.succeeded)


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