Fix rewrite to not write past the edge of the screen

This commit is contained in:
Bruno Oliveira 2017-11-21 22:04:30 -02:00
parent 2455f8670e
commit f05333ab75
2 changed files with 2 additions and 2 deletions

View File

@ -214,7 +214,7 @@ class TerminalReporter:
"""
erase = markup.pop('erase', False)
if erase:
fill_count = self.writer.fullwidth - len(line)
fill_count = self.writer.fullwidth - len(line) - 1
fill = ' ' * fill_count
else:
fill = ''

View File

@ -222,7 +222,7 @@ class TestTerminal(object):
tr.writer.fullwidth = 10
tr.write('hello')
tr.rewrite('hey', erase=True)
assert f.getvalue() == 'hello' + '\r' + 'hey' + (7 * ' ')
assert f.getvalue() == 'hello' + '\r' + 'hey' + (6 * ' ')
class TestCollectonly(object):