home *** CD-ROM | disk | FTP | other *** search
/ Mac Easy 2010 May / Mac Life Ubuntu.iso / casper / filesystem.squashfs / usr / share / pyshared / computerjanitorapp / terminalsize_tests.py < prev    next >
Encoding:
Python Source  |  2009-03-04  |  1.4 KB  |  40 lines
  1. # terminalsize_tests.py - unit tests for terminalsize.py
  2. # Copyright (C) 2008  Canonical, Ltd.
  3. #
  4. # This program is free software: you can redistribute it and/or modify
  5. # it under the terms of the GNU General Public License as published by
  6. # the Free Software Foundation, version 3 of the License.
  7. #
  8. # This program is distributed in the hope that it will be useful,
  9. # but WITHOUT ANY WARRANTY; without even the implied warranty of
  10. # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  11. # GNU General Public License for more details.
  12. #
  13. # You should have received a copy of the GNU General Public License
  14. # along with this program.  If not, see <http://www.gnu.org/licenses/>.
  15.  
  16.  
  17. import os
  18. import unittest
  19.  
  20. import terminalsize
  21.  
  22.  
  23. class GetTerminalSizeTests(unittest.TestCase):
  24.  
  25.     def testReturnsUnknownWhenQueryingDevNull(self):
  26.         fd = os.open("/dev/null", os.O_RDONLY)
  27.         rows, cols = terminalsize.get_terminal_size(fd)
  28.         os.close(fd)
  29.         self.assertEqual(rows, None)
  30.         self.assertEqual(cols, None)
  31.  
  32.     def testReturnsTwoIntegersWhenStdoutIsATerminal(self):
  33.         # We only run this check if stdout is a terminal.
  34.         # Unfortunately, there is no sensible way of checking the values.
  35.         # But that's OK, they're lumberjacks.
  36.         if os.isatty(1):
  37.             rows, cols = terminalsize.get_terminal_size(1)
  38.             self.assertEqual(type(rows), int)
  39.             self.assertEqual(type(cols), int)
  40.