1 files changed, 34 insertions, 0 deletions
diff --git a/lldb/third_party/Python/module/pexpect-4.6/tests/test_FSM.py b/lldb/third_party/Python/module/pexpect-4.6/tests/test_FSM.py
new file mode 100644
index 00000000000..3425fd2110e
--- /dev/null
+++ b/lldb/third_party/Python/module/pexpect-4.6/tests/test_FSM.py
@@ -0,0 +1,34 @@
+import io
+import sys
+import unittest
+try:
+    import builtins
+except ImportError:
+    import __builtin__ as builtins
+
+PY3 = (sys.version_info[0] >= 3)
+input_name = 'input' if PY3 else 'raw_input'
+
+from pexpect import FSM
+
+class FSMTestCase(unittest.TestCase):
+    def test_run_fsm(self):
+        def _input(prompt):
+            return "167 3 2 2 * * * 1 - ="
+        orig_input = getattr(builtins, input_name)
+        orig_stdout = sys.stdout
+        setattr(builtins, input_name, _input)
+        sys.stdout = sio = (io.StringIO if PY3 else io.BytesIO)()
+        
+        try:
+            FSM.main()
+        finally:
+            setattr(builtins, input_name, orig_input)
+            sys.stdout = orig_stdout
+        
+        printed = sio.getvalue()
+        assert '2003' in printed, printed
+        
+        
+if __name__ == '__main__':
+    unittest.main()
+\ No newline at end of file