Add Messager.finish.

2 files changed, 8 insertions, 0 deletions

diff --git a/ttystatus/messager.py b/ttystatus/messager.py
index 42a202d..a54f0ce 100644
--- a/ttystatus/messager.py
+++ b/ttystatus/messager.py
@@ -69,3 +69,7 @@ class Messager(object):
         self.clear()
         self.output.write('%s\n' % string)
         self._raw_write(old)
+        
+    def finish(self):
+        '''Finalize output.'''
+        self.clear()
diff --git a/ttystatus/messager_tests.py b/ttystatus/messager_tests.py
index 95448ea..c086fc1 100644
--- a/ttystatus/messager_tests.py
+++ b/ttystatus/messager_tests.py
@@ -74,3 +74,7 @@ class MessagerTests(unittest.TestCase):
         self.messager.notify('bar')
         self.assertEqual(self.output.getvalue(), 'foo\r   \rbar\nfoo')
 
+    def test_finish_removes_message(self):
+        self.messager.write('foo')
+        self.messager.finish()
+        self.assertEqual(self.output.getvalue(), 'foo\r   \r')