/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

(vila) Provides a better isolation from os.environ for tests. (Vincent
 Ladeuil)

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test_osutils.py
1899
1899
        self.assertIsInstance(concurrency, int)
1900
1900
 
1901
1901
    def test_local_concurrency_environment_variable(self):
1902
 
        os.environ['BZR_CONCURRENCY'] = '2'
 
1902
        self.overrideEnv('BZR_CONCURRENCY', '2')
1903
1903
        self.assertEqual(2, osutils.local_concurrency(use_cache=False))
1904
 
        os.environ['BZR_CONCURRENCY'] = '3'
 
1904
        self.overrideEnv('BZR_CONCURRENCY', '3')
1905
1905
        self.assertEqual(3, osutils.local_concurrency(use_cache=False))
1906
 
        os.environ['BZR_CONCURRENCY'] = 'foo'
 
1906
        self.overrideEnv('BZR_CONCURRENCY', 'foo')
1907
1907
        self.assertEqual(1, osutils.local_concurrency(use_cache=False))
1908
1908
 
1909
1909
    def test_option_concurrency(self):
1910
 
        os.environ['BZR_CONCURRENCY'] = '1'
 
1910
        self.overrideEnv('BZR_CONCURRENCY', '1')
1911
1911
        self.run_bzr('rocks --concurrency 42')
1912
 
        # Command line overrides envrionment variable
 
1912
        # Command line overrides environment variable
1913
1913
        self.assertEquals('42', os.environ['BZR_CONCURRENCY'])
1914
1914
        self.assertEquals(42, osutils.local_concurrency(use_cache=False))
1915
1915
 
1985
1985
    def test_defaults_to_BZR_COLUMNS(self):
1986
1986
        # BZR_COLUMNS is set by the test framework
1987
1987
        self.assertNotEqual('12', os.environ['BZR_COLUMNS'])
1988
 
        os.environ['BZR_COLUMNS'] = '12'
 
1988
        self.overrideEnv('BZR_COLUMNS', '12')
1989
1989
        self.assertEqual(12, osutils.terminal_width())
1990
1990
 
1991
1991
    def test_falls_back_to_COLUMNS(self):
1992
 
        del os.environ['BZR_COLUMNS']
 
1992
        self.overrideEnv('BZR_COLUMNS', None)
1993
1993
        self.assertNotEqual('42', os.environ['COLUMNS'])
1994
1994
        self.set_fake_tty()
1995
 
        os.environ['COLUMNS'] = '42'
 
1995
        self.overrideEnv('COLUMNS', '42')
1996
1996
        self.assertEqual(42, osutils.terminal_width())
1997
1997
 
1998
1998
    def test_tty_default_without_columns(self):
1999
 
        del os.environ['BZR_COLUMNS']
2000
 
        del os.environ['COLUMNS']
 
1999
        self.overrideEnv('BZR_COLUMNS', None)
 
2000
        self.overrideEnv('COLUMNS', None)
2001
2001
 
2002
2002
        def terminal_size(w, h):
2003
2003
            return 42, 42
2010
2010
        self.assertEqual(42, osutils.terminal_width())
2011
2011
 
2012
2012
    def test_non_tty_default_without_columns(self):
2013
 
        del os.environ['BZR_COLUMNS']
2014
 
        del os.environ['COLUMNS']
 
2013
        self.overrideEnv('BZR_COLUMNS', None)
 
2014
        self.overrideEnv('COLUMNS', None)
2015
2015
        self.replace_stdout(None)
2016
2016
        self.assertEqual(None, osutils.terminal_width())
2017
2017
 
2027
2027
        else:
2028
2028
            self.overrideAttr(termios, 'TIOCGWINSZ')
2029
2029
            del termios.TIOCGWINSZ
2030
 
        del os.environ['BZR_COLUMNS']
2031
 
        del os.environ['COLUMNS']
 
2030
        self.overrideEnv('BZR_COLUMNS', None)
 
2031
        self.overrideEnv('COLUMNS', None)
2032
2032
        # Whatever the result is, if we don't raise an exception, it's ok.
2033
2033
        osutils.terminal_width()
2034
2034
 
2070
2070
class TestGetuserUnicode(tests.TestCase):
2071
2071
 
2072
2072
    def test_ascii_user(self):
2073
 
        osutils.set_or_unset_env('LOGNAME', 'jrandom')
 
2073
        self.overrideEnv('LOGNAME', 'jrandom')
2074
2074
        self.assertEqual(u'jrandom', osutils.getuser_unicode())
2075
2075
 
2076
2076
    def test_unicode_user(self):
2082
2082
                % (osutils.get_user_encoding(),))
2083
2083
        uni_username = u'jrandom' + uni_val
2084
2084
        encoded_username = uni_username.encode(ue)
2085
 
        osutils.set_or_unset_env('LOGNAME', encoded_username)
 
2085
        self.overrideEnv('LOGNAME', encoded_username)
2086
2086
        self.assertEqual(uni_username, osutils.getuser_unicode())
2087
 
        osutils.set_or_unset_env('LOGNAME', u'jrandom\xb6'.encode(ue))
 
2087
        self.overrideEnv('LOGNAME', u'jrandom\xb6'.encode(ue))
2088
2088
        self.assertEqual(u'jrandom\xb6', osutils.getuser_unicode())
2089
2089
 
2090
2090
class TestBackupNames(tests.TestCase):