54 lines
1.4 KiB
Python
54 lines
1.4 KiB
Python
import contextlib
|
|
import sys
|
|
|
|
from hecate import Runner
|
|
|
|
import babi
|
|
|
|
|
|
@contextlib.contextmanager
|
|
def run(*args, **kwargs):
|
|
cmd = (sys.executable, '-mcoverage', 'run', '-m', 'babi', *args)
|
|
with Runner(*cmd, **kwargs) as h:
|
|
h.await_text(babi.VERSION_STR)
|
|
yield h
|
|
|
|
|
|
def await_text_missing(h, text):
|
|
"""largely based on await_text"""
|
|
for _ in h.poll_until_timeout():
|
|
screen = h.screenshot()
|
|
munged = screen.replace('\n', '')
|
|
if text not in munged: # pragma: no branch
|
|
return
|
|
raise AssertionError(f'Timeout while waiting for text {text!r} to appear')
|
|
|
|
|
|
def test_window_bounds(tmpdir):
|
|
f = tmpdir.join('f.txt')
|
|
f.write(f'{"x" * 40}\n' * 40)
|
|
|
|
with run(str(f), width=30, height=30) as h:
|
|
# make sure we don't go off the top left of the screen
|
|
h.press('LEFT')
|
|
h.press('UP')
|
|
# make sure we don't go off the bottom of the screen
|
|
for i in range(32):
|
|
h.press('RIGHT')
|
|
h.press('DOWN')
|
|
h.press('C-x')
|
|
h.await_exit()
|
|
|
|
|
|
def test_status_clearing_behaviour():
|
|
with run() as h:
|
|
h.press('C-j')
|
|
h.await_text('unknown key')
|
|
for i in range(24):
|
|
h.press('LEFT')
|
|
h.await_text('unknown key')
|
|
h.press('LEFT')
|
|
await_text_missing(h, 'unknown key')
|
|
h.press('C-x')
|
|
h.await_exit()
|