/usr/share/pyshared/ttystatus/messager_tests.py is in python-ttystatus 0.15-1.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 | # Copyright 2010, 2011 Lars Wirzenius
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
import StringIO
import unittest
import ttystatus
class DummyTerminal(StringIO.StringIO):
def isatty(self):
return True
class MessagerTests(unittest.TestCase):
def setUp(self):
self.output = DummyTerminal()
self.messager = ttystatus.Messager(output=self.output)
def fast_time(self):
return self.messager._last_time + self.messager._period
def test_sets_output(self):
self.assertEqual(self.messager.output, self.output)
def test_handles_no_tty(self):
def notty():
raise IOError()
m = ttystatus.Messager(open_tty=notty)
self.assertEqual(m.output, None)
def test_raw_writes_nothing_if_output_is_not_a_terminal(self):
self.messager.output = StringIO.StringIO()
self.messager._raw_write('foo')
self.assertEqual(self.messager.output.getvalue(), '')
def test_raw_writes_something_if_output_is_not_a_terminal(self):
self.messager._raw_write('foo')
self.assertEqual(self.output.getvalue(), 'foo')
def test_knows_it_is_time_to_write_at_start(self):
self.assert_(self.messager.time_to_write())
def test_knows_it_is_not_time_to_write_right_after_previous_one(self):
self.messager._last_time = self.messager._now()
self.assertFalse(self.messager.time_to_write())
def test_knows_it_is_time_to_write_after_a_period(self):
self.messager._last_time = (self.messager._now() -
self.messager._period*2)
self.assert_(self.messager.time_to_write())
def test_cached_write_writes_first_thing(self):
self.messager.write('foo')
self.assertEqual(self.output.getvalue(), 'foo')
def test_cached_write_does_not_writes_first_thing_if_at_epoch(self):
self.messager._now = lambda: 0
self.messager.write('foo')
self.assertEqual(self.output.getvalue(), '')
def test_cached_write_writes_once_within_a_second(self):
self.messager._now = lambda: self.messager._period + 1
self.messager.write('foo')
self.messager.write('bar')
self.assertEqual(self.output.getvalue(), 'foo')
def test_write_removes_old_message(self):
self.messager._now = self.fast_time
self.messager.write('foo')
self.messager.write('bar')
self.assertEqual(self.output.getvalue(), 'foo\r \rbar')
def test_clear_removes_message(self):
self.messager._now = lambda: self.messager._period + 1
self.messager.write('foo')
self.messager.clear()
self.assertEqual(self.output.getvalue(), 'foo\r \r')
def test_notify_removes_message_and_puts_it_back_afterwards(self):
f = StringIO.StringIO()
self.messager.write('foo')
self.messager.notify('bar', f)
self.assertEqual(self.output.getvalue(), 'foo\r \rfoo')
self.assertEqual(f.getvalue(), 'bar\n')
def test_notify_does_not_mind_ioerror(self):
f = open('/dev/full', 'w')
self.messager.write('foo')
self.messager.notify('bar', f)
self.assertEqual(self.output.getvalue(), 'foo\r \rfoo')
f.close()
def test_finish_flushes_unwritten_message(self):
self.messager._now = lambda: 0
self.messager.write('foo')
self.messager.finish()
self.assertEqual(self.output.getvalue(), 'foo\n')
def test_has_width(self):
self.assertEqual(self.messager.width, 79)
def test_write_truncates_at_one_less_than_width(self):
self.messager.set_width(4)
self.messager.write('foobar')
self.assertEqual(self.output.getvalue(), 'foo')
def test_disables_output(self):
self.messager.disable()
self.messager.write('foo')
self.assertEqual(self.output.getvalue(), '')
def test_enables_output(self):
self.messager.disable()
self.messager.enable()
self.messager.write('foo')
self.assertEqual(self.output.getvalue(), 'foo')
|