llvm-project/lldb/test/terminal/TestSTTYBeforeAndAfter.py

125 lines
4.1 KiB
Python

"""
Test that 'stty -a' displays the same output before and after running the lldb command.
"""
import os
import unittest2
import lldb
from lldbtest import *
class CommandLineCompletionTestCase(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")
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 = 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])
if __name__ == '__main__':
import atexit
lldb.SBDebugger.Initialize()
atexit.register(lambda: lldb.SBDebugger.Terminate())
unittest2.main()