diff options
author | Med Ismail Bennani <medismail.bennani@gmail.com> | 2022-05-18 21:35:20 +0300 |
---|---|---|
committer | Med Ismail Bennani <medismail.bennani@gmail.com> | 2022-05-19 04:22:46 +0300 |
commit | d71d1a947bee1247e952f22c13ad3ed3d041e36a (patch) | |
tree | 20e5da42e57180aef660b8661242995c681e0445 /lldb | |
parent | b3718bc4672bcad45f155c6d2f9fce4ad409bb8e (diff) |
[lldb/Test] Add `use_colors` argument to the PExpect.launch wrapper
This patch adds a new `use_colors` argument to the PExpect.launch
method.
As the name suggests, it allows the user to conditionally enable color
support in the debugger, which can be helpful to test functionalities that
rely on that, like progress reporting. It defaults to False.
Differential Revision: https://reviews.llvm.org/D125915
Signed-off-by: Med Ismail Bennani <medismail.bennani@gmail.com>
Diffstat (limited to 'lldb')
-rw-r--r-- | lldb/packages/Python/lldbsuite/test/lldbpexpect.py | 15 |
1 files changed, 10 insertions, 5 deletions
diff --git a/lldb/packages/Python/lldbsuite/test/lldbpexpect.py b/lldb/packages/Python/lldbsuite/test/lldbpexpect.py index de8f819e2552..fe44c86d14f2 100644 --- a/lldb/packages/Python/lldbsuite/test/lldbpexpect.py +++ b/lldb/packages/Python/lldbsuite/test/lldbpexpect.py @@ -24,16 +24,20 @@ class PExpectTest(TestBase): self.child.expect_exact(self.PROMPT) def launch(self, executable=None, extra_args=None, timeout=60, - dimensions=None, run_under=None, post_spawn=None): + dimensions=None, run_under=None, post_spawn=None, + use_colors=False): logfile = getattr(sys.stdout, 'buffer', sys.stdout) if self.TraceOn() else None args = [] if run_under is not None: args += run_under - args += [lldbtest_config.lldbExec, '--no-lldbinit', '--no-use-colors'] + args += [lldbtest_config.lldbExec, '--no-lldbinit'] + if not use_colors: + args += '--no-use-colors' for cmd in self.setUpCommands(): - args += ['-O', cmd] + if use_colors and "use-color false" not in cmd: + args += ['-O', cmd] if executable is not None: args += ['--file', executable] if extra_args is not None: @@ -54,8 +58,9 @@ class PExpectTest(TestBase): post_spawn() self.expect_prompt() for cmd in self.setUpCommands(): - self.child.expect_exact(cmd) - self.expect_prompt() + if use_colors and "use-color false" not in cmd: + self.child.expect_exact(cmd) + self.expect_prompt() if executable is not None: self.child.expect_exact("target create") self.child.expect_exact("Current executable set to") |