/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_osutils.py

• browse files at revision 4880.1.1
• compare with another revision
• download diff
• download tarball
• view history from revision 4880.1.1

• reverse the comparison (4880.1.1 to 5200.3.2)
• stop comparing with revision 5200.3.2

Show diffs side-by-side

added

removed

bzrlib/tests/test_osutils.py

1

 

# Copyright (C) 2005-2010 Canonical Ltd

 

1

# Copyright (C) 2005, 2006, 2007, 2008, 2009 Canonical Ltd

2

2

#

3

3

# This program is free software; you can redistribute it and/or modify

4

4

# it under the terms of the GNU General Public License as published by

33

33

    win32utils,

34

34

    )

35

35

from bzrlib.tests import (

36

 

    features,

37

36

    file_utils,

38

37

    test__walkdirs_win32,

39

38

    )

54

53

 

55

54

UTF8DirReaderFeature = _UTF8DirReaderFeature()

56

55

 

57

 

term_ios_feature = tests.ModuleAvailableFeature('termios')

 

56

TermIOSFeature = tests.ModuleAvailableFeature('termios')

58

57

 

59

58

 

60

59

def _already_unicode(s):

83

82

                          dict(_dir_reader_class=_readdir_pyx.UTF8DirReader,

84

83

                               _native_to_unicode=_utf8_to_unicode)))

85

84

 

86

 

    if test__walkdirs_win32.win32_readdir_feature.available():

 

85

    if test__walkdirs_win32.Win32ReadDirFeature.available():

87

86

        try:

88

87

            from bzrlib import _walkdirs_win32

89

88

            scenarios.append(

184

183

        shape = sorted(os.listdir('.'))

185

184

        self.assertEquals(['A', 'B'], shape)

186

185

 

187

 

    def test_rename_error(self):

188

 

        # We wrap os.rename to make it give an error including the filenames

189

 

        # https://bugs.launchpad.net/bzr/+bug/491763

190

 

        err = self.assertRaises(OSError, osutils.rename,

191

 

            'nonexistent', 'target')

192

 

        self.assertContainsString(str(err), 'nonexistent')

193

 

 

194

186

 

195

187

class TestRandChars(tests.TestCase):

196

188

 

317

309

        self.assertEqual("/", osutils.kind_marker(osutils._directory_kind))

318

310

        self.assertEqual("@", osutils.kind_marker("symlink"))

319

311

        self.assertEqual("+", osutils.kind_marker("tree-reference"))

320

 

        self.assertEqual("", osutils.kind_marker("fifo"))

321

 

        self.assertEqual("", osutils.kind_marker("socket"))

322

 

        self.assertEqual("", osutils.kind_marker("unknown"))

 

312

        self.assertRaises(errors.BzrError, osutils.kind_marker, "unknown")

323

313

 

324

314

 

325

315

class TestUmask(tests.TestCaseInTempDir):

999

989

 

1000

990

    def test_osutils_binding(self):

1001

991

        from bzrlib.tests import test__chunks_to_lines

1002

 

        if test__chunks_to_lines.compiled_chunkstolines_feature.available():

 

992

        if test__chunks_to_lines.CompiledChunksToLinesFeature.available():

1003

993

            from bzrlib._chunks_to_lines_pyx import chunks_to_lines

1004

994

        else:

1005

995

            from bzrlib._chunks_to_lines_py import chunks_to_lines

1145

1135

            dirblock[:] = new_dirblock

1146

1136

 

1147

1137

    def _save_platform_info(self):

1148

 

        self.overrideAttr(win32utils, 'winver')

1149

 

        self.overrideAttr(osutils, '_fs_enc')

1150

 

        self.overrideAttr(osutils, '_selected_dir_reader')

 

1138

        cur_winver = win32utils.winver

 

1139

        cur_fs_enc = osutils._fs_enc

 

1140

        cur_dir_reader = osutils._selected_dir_reader

 

1141

        def restore():

 

1142

            win32utils.winver = cur_winver

 

1143

            osutils._fs_enc = cur_fs_enc

 

1144

            osutils._selected_dir_reader = cur_dir_reader

 

1145

        self.addCleanup(restore)

1151

1146

 

1152

1147

    def assertDirReaderIs(self, expected):

1153

1148

        """Assert the right implementation for _walkdirs_utf8 is chosen."""

1186

1181

 

1187

1182

    def test_force_walkdirs_utf8_nt(self):

1188

1183

        # Disabled because the thunk of the whole walkdirs api is disabled.

1189

 

        self.requireFeature(test__walkdirs_win32.win32_readdir_feature)

 

1184

        self.requireFeature(test__walkdirs_win32.Win32ReadDirFeature)

1190

1185

        self._save_platform_info()

1191

1186

        win32utils.winver = 'Windows NT'

1192

1187

        from bzrlib._walkdirs_win32 import Win32ReadDir

1193

1188

        self.assertDirReaderIs(Win32ReadDir)

1194

1189

 

1195

1190

    def test_force_walkdirs_utf8_98(self):

1196

 

        self.requireFeature(test__walkdirs_win32.win32_readdir_feature)

 

1191

        self.requireFeature(test__walkdirs_win32.Win32ReadDirFeature)

1197

1192

        self._save_platform_info()

1198

1193

        win32utils.winver = 'Windows 98'

1199

1194

        self.assertDirReaderIs(osutils.UnicodeDirReader)

1350

1345

        self.assertEqual(expected_dirblocks, result)

1351

1346

 

1352

1347

    def test__walkdirs_utf8_win32readdir(self):

1353

 

        self.requireFeature(test__walkdirs_win32.win32_readdir_feature)

 

1348

        self.requireFeature(test__walkdirs_win32.Win32ReadDirFeature)

1354

1349

        self.requireFeature(tests.UnicodeFilenameFeature)

1355

1350

        from bzrlib._walkdirs_win32 import Win32ReadDir

1356

1351

        self._save_platform_info()

1407

1402

 

1408

1403

    def test__walkdirs_utf_win32_find_file_stat_file(self):

1409

1404

        """make sure our Stat values are valid"""

1410

 

        self.requireFeature(test__walkdirs_win32.win32_readdir_feature)

 

1405

        self.requireFeature(test__walkdirs_win32.Win32ReadDirFeature)

1411

1406

        self.requireFeature(tests.UnicodeFilenameFeature)

1412

1407

        from bzrlib._walkdirs_win32 import Win32ReadDir

1413

1408

        name0u = u'0file-\xb6'

1431

1426

 

1432

1427

    def test__walkdirs_utf_win32_find_file_stat_directory(self):

1433

1428

        """make sure our Stat values are valid"""

1434

 

        self.requireFeature(test__walkdirs_win32.win32_readdir_feature)

 

1429

        self.requireFeature(test__walkdirs_win32.Win32ReadDirFeature)

1435

1430

        self.requireFeature(tests.UnicodeFilenameFeature)

1436

1431

        from bzrlib._walkdirs_win32 import Win32ReadDir

1437

1432

        name0u = u'0dir-\u062c\u0648'

1586

1581

        def cleanup():

1587

1582

            if 'BZR_TEST_ENV_VAR' in os.environ:

1588

1583

                del os.environ['BZR_TEST_ENV_VAR']

 

1584

 

1589

1585

        self.addCleanup(cleanup)

1590

1586

 

1591

1587

    def test_set(self):

1702

1698

 

1703

1699

    def setUp(self):

1704

1700

        tests.TestCaseInTempDir.setUp(self)

1705

 

        self.overrideAttr(osutils,

1706

 

                          '_selected_dir_reader', self._dir_reader_class())

 

1701

 

 

1702

        # Save platform specific info and reset it

 

1703

        cur_dir_reader = osutils._selected_dir_reader

 

1704

 

 

1705

        def restore():

 

1706

            osutils._selected_dir_reader = cur_dir_reader

 

1707

        self.addCleanup(restore)

 

1708

 

 

1709

        osutils._selected_dir_reader = self._dir_reader_class()

1707

1710

 

1708

1711

    def _get_ascii_tree(self):

1709

1712

        tree = [

1856

1859

 

1857

1860

    def setUp(self):

1858

1861

        super(TestConcurrency, self).setUp()

1859

 

        self.overrideAttr(osutils, '_cached_local_concurrency')

 

1862

        orig = osutils._cached_local_concurrency

 

1863

        def restore():

 

1864

            osutils._cached_local_concurrency = orig

 

1865

        self.addCleanup(restore)

1860

1866

 

1861

1867

    def test_local_concurrency(self):

1862

1868

        concurrency = osutils.local_concurrency()

1889

1895

 

1890

1896

    def setUp(self):

1891

1897

        super(TestFailedToLoadExtension, self).setUp()

1892

 

        self.overrideAttr(osutils, '_extension_load_failures', [])

 

1898

        self.saved_failures = osutils._extension_load_failures[:]

 

1899

        del osutils._extension_load_failures[:]

 

1900

        self.addCleanup(self.restore_failures)

 

1901

 

 

1902

    def restore_failures(self):

 

1903

        osutils._extension_load_failures = self.saved_failures

1893

1904

 

1894

1905

    def test_failure_to_load(self):

1895

1906

        self._try_loading()

1917

1928

 

1918

1929

class TestTerminalWidth(tests.TestCase):

1919

1930

 

1920

 

    def replace_stdout(self, new):

1921

 

        self.overrideAttr(sys, 'stdout', new)

1922

 

 

1923

 

    def replace__terminal_size(self, new):

1924

 

        self.overrideAttr(osutils, '_terminal_size', new)

1925

 

 

1926

 

    def set_fake_tty(self):

1927

 

 

1928

 

        class I_am_a_tty(object):

1929

 

            def isatty(self):

1930

 

                return True

1931

 

 

1932

 

        self.replace_stdout(I_am_a_tty())

1933

 

 

1934

1931

    def test_default_values(self):

1935

 

        self.assertEqual(80, osutils.default_terminal_width)

 

1932

        self.assertEquals(80, osutils.default_terminal_width)

1936

1933

 

1937

1934

    def test_defaults_to_BZR_COLUMNS(self):

1938

1935

        # BZR_COLUMNS is set by the test framework

1939

 

        self.assertNotEqual('12', os.environ['BZR_COLUMNS'])

 

1936

        self.assertEquals('80', os.environ['BZR_COLUMNS'])

1940

1937

        os.environ['BZR_COLUMNS'] = '12'

1941

 

        self.assertEqual(12, osutils.terminal_width())

1942

 

 

1943

 

    def test_falls_back_to_COLUMNS(self):

1944

 

        del os.environ['BZR_COLUMNS']

1945

 

        self.assertNotEqual('42', os.environ['COLUMNS'])

1946

 

        self.set_fake_tty()

1947

 

        os.environ['COLUMNS'] = '42'

1948

 

        self.assertEqual(42, osutils.terminal_width())

 

1938

        self.assertEquals(12, osutils.terminal_width())

1949

1939

 

1950

1940

    def test_tty_default_without_columns(self):

1951

1941

        del os.environ['BZR_COLUMNS']

1952

1942

        del os.environ['COLUMNS']

1953

 

 

1954

 

        def terminal_size(w, h):

1955

 

            return 42, 42

1956

 

 

1957

 

        self.set_fake_tty()

1958

 

        # We need to override the osutils definition as it depends on the

1959

 

        # running environment that we can't control (PQM running without a

1960

 

        # controlling terminal is one example).

1961

 

        self.replace__terminal_size(terminal_size)

1962

 

        self.assertEqual(42, osutils.terminal_width())

 

1943

        orig_stdout = sys.stdout

 

1944

        def restore():

 

1945

            sys.stdout = orig_stdout

 

1946

        self.addCleanup(restore)

 

1947

 

 

1948

        class I_am_a_tty(object):

 

1949

            def isatty(self):

 

1950

                return True

 

1951

 

 

1952

        sys.stdout = I_am_a_tty()

 

1953

        self.assertEquals(None, osutils.terminal_width())

1963

1954

 

1964

1955

    def test_non_tty_default_without_columns(self):

1965

1956

        del os.environ['BZR_COLUMNS']

1966

1957

        del os.environ['COLUMNS']

1967

 

        self.replace_stdout(None)

1968

 

        self.assertEqual(None, osutils.terminal_width())

 

1958

        orig_stdout = sys.stdout

 

1959

        def restore():

 

1960

            sys.stdout = orig_stdout

 

1961

        self.addCleanup(restore)

 

1962

        sys.stdout = None

 

1963

        self.assertEquals(None, osutils.terminal_width())

1969

1964

 

1970

1965

    def test_no_TIOCGWINSZ(self):

1971

 

        self.requireFeature(term_ios_feature)

1972

 

        termios = term_ios_feature.module

 

1966

        self.requireFeature(TermIOSFeature)

 

1967

        termios = TermIOSFeature.module

1973

1968

        # bug 63539 is about a termios without TIOCGWINSZ attribute

1974

1969

        try:

1975

1970

            orig = termios.TIOCGWINSZ

1977

1972

            # We won't remove TIOCGWINSZ, because it doesn't exist anyway :)

1978

1973

            pass

1979

1974

        else:

1980

 

            self.overrideAttr(termios, 'TIOCGWINSZ')

 

1975

            def restore():

 

1976

                termios.TIOCGWINSZ = orig

 

1977

            self.addCleanup(restore)

1981

1978

            del termios.TIOCGWINSZ

1982

1979

        del os.environ['BZR_COLUMNS']

1983

1980

        del os.environ['COLUMNS']

1984

 

        # Whatever the result is, if we don't raise an exception, it's ok.

1985

 

        osutils.terminal_width()

1986

 

 

1987

 

class TestCreationOps(tests.TestCaseInTempDir):

1988

 

    _test_needs_features = [features.chown_feature]

1989

 

 

1990

 

    def setUp(self):

1991

 

        tests.TestCaseInTempDir.setUp(self)

1992

 

        self.overrideAttr(os, 'chown', self._dummy_chown)

1993

 

 

1994

 

        # params set by call to _dummy_chown

1995

 

        self.path = self.uid = self.gid = None

1996

 

 

1997

 

    def _dummy_chown(self, path, uid, gid):

1998

 

        self.path, self.uid, self.gid = path, uid, gid

1999

 

 

2000

 

    def test_copy_ownership_from_path(self):

2001

 

        """copy_ownership_from_path test with specified src."""

2002

 

        ownsrc = '/'

2003

 

        f = open('test_file', 'wt')

2004

 

        osutils.copy_ownership_from_path('test_file', ownsrc)

2005

 

 

2006

 

        s = os.stat(ownsrc)

2007

 

        self.assertEquals(self.path, 'test_file')

2008

 

        self.assertEquals(self.uid, s.st_uid)

2009

 

        self.assertEquals(self.gid, s.st_gid)

2010

 

 

2011

 

    def test_copy_ownership_nonesrc(self):

2012

 

        """copy_ownership_from_path test with src=None."""

2013

 

        f = open('test_file', 'wt')

2014

 

        # should use parent dir for permissions

2015

 

        osutils.copy_ownership_from_path('test_file')

2016

 

 

2017

 

        s = os.stat('..')

2018

 

        self.assertEquals(self.path, 'test_file')

2019

 

        self.assertEquals(self.uid, s.st_uid)

2020

 

        self.assertEquals(self.gid, s.st_gid)

 

1981

        self.assertIs(None, osutils.terminal_width())

• Older »

Loggerhead is a web-based interface for Breezy
Version: 2.0.2.dev0