llvm-project/lldb/packages/Python/lldbsuite/test/terminal/TestSTTYBeforeAndAfter.py
Zachary Turner 95c453a221 Tighten up sys.path, and use absolute imports everywhere.
For convenience, we had added the folder that dotest.py was in
to sys.path, so that we could easily write things like
`import lldbutil` from anywhere and any test.  This introduces
a subtle problem when using Python's package system, because when
unittest2 imports a particular test suite, the test suite is detached
from the package.  Thus, writing "import lldbutil" from dotest imports
it as part of the package, and writing the same line from a test
does a fresh import since the importing module was not part of
the same package.

The real way to fix this is to use absolute imports everywhere.  Instead
of writing "import lldbutil", we need to write "import
lldbsuite.test.util".  This patch fixes up that and all other similar
cases, and additionally removes the script directory from sys.path
to ensure that this can't happen again.

llvm-svn: 251886
2015-11-03 02:06:18 +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 use_lldb_suite
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])