/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_osutils.py

  • Committer: Andrew Bennetts
  • Date: 2009-12-18 08:22:42 UTC
  • mfrom: (4906 +trunk)
  • mto: This revision was merged to the branch mainline in revision 4909.
  • Revision ID: andrew.bennetts@canonical.com-20091218082242-f7wy9tlk0yxvkkju
Merge lp:bzr.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test_osutils.py
23
23
import socket
24
24
import stat
25
25
import sys
26
 
import termios
27
26
import time
28
27
 
29
28
from bzrlib import (
54
53
 
55
54
UTF8DirReaderFeature = _UTF8DirReaderFeature()
56
55
 
 
56
TermIOSFeature = tests.ModuleAvailableFeature('termios')
 
57
 
57
58
 
58
59
def _already_unicode(s):
59
60
    return s
1927
1928
 
1928
1929
class TestTerminalWidth(tests.TestCase):
1929
1930
 
 
1931
    def replace_stdout(self, new):
 
1932
        orig_stdout = sys.stdout
 
1933
        def restore():
 
1934
            sys.stdout = orig_stdout
 
1935
        self.addCleanup(restore)
 
1936
        sys.stdout = new
 
1937
 
 
1938
    def replace__terminal_size(self, new):
 
1939
        orig__terminal_size = osutils._terminal_size
 
1940
        def restore():
 
1941
            osutils._terminal_size = orig__terminal_size
 
1942
        self.addCleanup(restore)
 
1943
        osutils._terminal_size = new
 
1944
 
 
1945
    def set_fake_tty(self):
 
1946
 
 
1947
        class I_am_a_tty(object):
 
1948
            def isatty(self):
 
1949
                return True
 
1950
 
 
1951
        self.replace_stdout(I_am_a_tty())
 
1952
 
1930
1953
    def test_default_values(self):
1931
 
        self.assertEquals(80, osutils.default_terminal_width)
 
1954
        self.assertEqual(80, osutils.default_terminal_width)
1932
1955
 
1933
1956
    def test_defaults_to_BZR_COLUMNS(self):
1934
1957
        # BZR_COLUMNS is set by the test framework
1935
 
        self.assertEquals('80', os.environ['BZR_COLUMNS'])
 
1958
        self.assertNotEqual('12', os.environ['BZR_COLUMNS'])
1936
1959
        os.environ['BZR_COLUMNS'] = '12'
1937
 
        self.assertEquals(12, osutils.terminal_width())
 
1960
        self.assertEqual(12, osutils.terminal_width())
 
1961
 
 
1962
    def test_falls_back_to_COLUMNS(self):
 
1963
        del os.environ['BZR_COLUMNS']
 
1964
        self.assertNotEqual('42', os.environ['COLUMNS'])
 
1965
        self.set_fake_tty()
 
1966
        os.environ['COLUMNS'] = '42'
 
1967
        self.assertEqual(42, osutils.terminal_width())
1938
1968
 
1939
1969
    def test_tty_default_without_columns(self):
1940
1970
        del os.environ['BZR_COLUMNS']
1941
1971
        del os.environ['COLUMNS']
1942
 
        orig_stdout = sys.stdout
1943
 
        def restore():
1944
 
            sys.stdout = orig_stdout
1945
 
        self.addCleanup(restore)
1946
 
 
1947
 
        class I_am_a_tty(object):
1948
 
            def isatty(self):
1949
 
                return True
1950
 
 
1951
 
        sys.stdout = I_am_a_tty()
1952
 
        self.assertEquals(None, osutils.terminal_width())
 
1972
 
 
1973
        def terminal_size(w, h):
 
1974
            return 42, 42
 
1975
 
 
1976
        self.set_fake_tty()
 
1977
        # We need to override the osutils definition as it depends on the
 
1978
        # running environment that we can't control (PQM running without a
 
1979
        # controlling terminal is one example).
 
1980
        self.replace__terminal_size(terminal_size)
 
1981
        self.assertEqual(42, osutils.terminal_width())
1953
1982
 
1954
1983
    def test_non_tty_default_without_columns(self):
1955
1984
        del os.environ['BZR_COLUMNS']
1956
1985
        del os.environ['COLUMNS']
1957
 
        orig_stdout = sys.stdout
1958
 
        def restore():
1959
 
            sys.stdout = orig_stdout
1960
 
        self.addCleanup(restore)
1961
 
        sys.stdout = None
1962
 
        self.assertEquals(None, osutils.terminal_width())
 
1986
        self.replace_stdout(None)
 
1987
        self.assertEqual(None, osutils.terminal_width())
1963
1988
 
1964
 
    def test_TIOCGWINSZ(self):
 
1989
    def test_no_TIOCGWINSZ(self):
 
1990
        self.requireFeature(TermIOSFeature)
 
1991
        termios = TermIOSFeature.module
1965
1992
        # bug 63539 is about a termios without TIOCGWINSZ attribute
1966
 
        exist = True
1967
1993
        try:
1968
1994
            orig = termios.TIOCGWINSZ
1969
1995
        except AttributeError:
1970
 
            exist = False
1971
 
 
1972
 
        def restore():
1973
 
            if exist:
 
1996
            # We won't remove TIOCGWINSZ, because it doesn't exist anyway :)
 
1997
            pass
 
1998
        else:
 
1999
            def restore():
1974
2000
                termios.TIOCGWINSZ = orig
1975
 
        self.addCleanup(restore)
1976
 
 
1977
 
        del termios.TIOCGWINSZ
 
2001
            self.addCleanup(restore)
 
2002
            del termios.TIOCGWINSZ
1978
2003
        del os.environ['BZR_COLUMNS']
1979
2004
        del os.environ['COLUMNS']
1980
 
        self.assertEquals(None, osutils.terminal_width())
 
2005
        # Whatever the result is, if we don't raise an exception, it's ok.
 
2006
        osutils.terminal_width()