~paul-mcspadden/computer-janitor/bug-726616

« back to all changes in this revision

Viewing changes to computerjanitorapp/terminalsize_tests.py

  • Committer: Bazaar Package Importer
  • Author(s): Barry Warsaw
  • Date: 2010-05-21 16:55:46 UTC
  • mfrom: (1.1.7 upstream)
  • Revision ID: james.westby@ubuntu.com-20100521165546-53kn5xq1irmkjazx
Tags: 2.0.1-0ubuntu1
BumpingĀ forĀ Ubuntu.

Show diffs side-by-side

added added

removed removed

Lines of Context:
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)