llvm-project/lldb/packages/Python/lldbsuite/test/terminal/TestSTTYBeforeAndAfter.py
Zachary Turner 19474e1801 Remove use_lldb_suite from the package, and don't import it anymore.
This module was originally intended to be imported by top-level
scripts to be able to find the LLDB packages and third party
libraries.  Packages themselves shouldn't need to import it,
because by the time it gets into the package, the top-level
script should have already done this.  Indeed, it was just
adding the same values to sys.path multiple times, so this
patch is essentially no functional change.

To make sure it doesn't get re-introduced, we also delete the
`use_lldb_suite` module from `lldbsuite/test`, although the
original copy still remains in `lldb/test`

llvm-svn: 251963
2015-11-03 19:20:39 +00:00

122 lines
4.1 KiB
Python

"""
Test that 'stty -a' displays the same output before and after running the lldb command.
"""
from __future__ import print_function
import os
import lldb
from lldbsuite.test.lldbtest import *
class TestSTTYBeforeAndAfter(TestBase):
mydir = TestBase.compute_mydir(__file__)
@classmethod
def classCleanup(cls):
"""Cleanup the test byproducts."""
cls.RemoveTempFile("child_send1.txt")
cls.RemoveTempFile("child_read1.txt")
cls.RemoveTempFile("child_send2.txt")
cls.RemoveTempFile("child_read2.txt")
@expectedFailureHostWindows("llvm.org/pr22274: need a pexpect replacement for windows")
@no_debug_info_test
def test_stty_dash_a_before_and_afetr_invoking_lldb_command(self):
"""Test that 'stty -a' displays the same output before and after running the lldb command."""
import pexpect
if not which('expect'):
self.skipTest("The 'expect' program cannot be located, skip the test")
# The expect prompt.
expect_prompt = "expect[0-9.]+> "
# The default lldb prompt.
lldb_prompt = "(lldb) "
# So that the child gets torn down after the test.
self.child = pexpect.spawn('expect')
child = self.child
child.expect(expect_prompt)
child.setecho(True)
if self.TraceOn():
child.logfile = sys.stdout
if self.platformIsDarwin():
child.sendline('set env(TERM) xterm')
else:
child.sendline('set env(TERM) vt100')
child.expect(expect_prompt)
child.sendline('puts $env(TERM)')
child.expect(expect_prompt)
# Turn on loggings for input/output to/from the child.
with open('child_send1.txt', 'w') as f_send1:
with open('child_read1.txt', 'w') as f_read1:
child.logfile_send = f_send1
child.logfile_read = f_read1
child.sendline('stty -a')
child.expect(expect_prompt)
# Now that the stage1 logging is done, restore logfile to None to
# stop further logging.
child.logfile_send = None
child.logfile_read = None
# Invoke the lldb command.
child.sendline('%s %s' % (lldbtest_config.lldbExec, self.lldbOption))
child.expect_exact(lldb_prompt)
# Immediately quit.
child.sendline('quit')
child.expect(expect_prompt)
with open('child_send2.txt', 'w') as f_send2:
with open('child_read2.txt', 'w') as f_read2:
child.logfile_send = f_send2
child.logfile_read = f_read2
child.sendline('stty -a')
child.expect(expect_prompt)
child.sendline('exit')
# Now that the stage2 logging is done, restore logfile to None to
# stop further logging.
child.logfile_send = None
child.logfile_read = None
with open('child_send1.txt', 'r') as fs:
if self.TraceOn():
print("\n\nContents of child_send1.txt:")
print(fs.read())
with open('child_read1.txt', 'r') as fr:
from_child1 = fr.read()
if self.TraceOn():
print("\n\nContents of child_read1.txt:")
print(from_child1)
with open('child_send2.txt', 'r') as fs:
if self.TraceOn():
print("\n\nContents of child_send2.txt:")
print(fs.read())
with open('child_read2.txt', 'r') as fr:
from_child2 = fr.read()
if self.TraceOn():
print("\n\nContents of child_read2.txt:")
print(from_child2)
stty_output1_lines = from_child1.splitlines()
stty_output2_lines = from_child2.splitlines()
zipped = list(zip(stty_output1_lines, stty_output2_lines))
for tuple in zipped:
if self.TraceOn():
print("tuple->%s" % str(tuple))
# Every line should compare equal until the first blank line.
if len(tuple[0]) == 0:
break
self.assertTrue(tuple[0] == tuple[1])