def test_session_fail():
before_prompt()
print("PROMPT$ ", end='')
after_prompt()
print("command")
before_output()
print("command output 1")
print("command output 2")
print("command output 3", file=sys.stderr)
after_output(1)
def test_session_prompt_newline():
before_prompt()
print("PROMPT\n$ ", end='')
after_prompt()
print("command")
before_output()
print("command output 1")
print("command output 2")
print("command output 3", file=sys.stderr)
after_output(0)
def test_after_output_range():
after_output(0)
after_output(255)
raises(ValueError, after_output, 256)
raises(ValueError, after_output, -1)
def test_Output():
with Output() as o:
print("Output")
with Output() as o:
print("Output")
o.set_command_status(0)
with Output() as o:
print("Output")
o.set_command_status(1)
def test_readline_invisible():
assert readline_invisible(BEFORE_PROMPT) == "\001" + BEFORE_PROMPT + "\002"
if __name__ == '__main__':
test_session()
test_session_fail()
test_session_prompt_newline()
# In case this file is run in a shell that itself has shell integration
# enabled, some dummy output.
before_prompt()
print("DUMMY PROMPT, IGNORE$ ", end='')
after_prompt()
print("dummy command")
before_output()
print("dummy output")
after_output(0)
