~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_osutils.py

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

• reverse the comparison (4859.1.1 to 5200.3.3)
• stop comparing with revision 5200.3.3

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')

58

 

 

59

56

 

60

57

def _already_unicode(s):

61

58

    return s

83

80

                          dict(_dir_reader_class=_readdir_pyx.UTF8DirReader,

84

81

                               _native_to_unicode=_utf8_to_unicode)))

85

82

 

86

 

    if test__walkdirs_win32.win32_readdir_feature.available():

 

83

    if test__walkdirs_win32.Win32ReadDirFeature.available():

87

84

        try:

88

85

            from bzrlib import _walkdirs_win32

89

86

            scenarios.append(

184

181

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

185

182

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

186

183

 

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

184

 

195

185

class TestRandChars(tests.TestCase):

196

186

 

317

307

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

318

308

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

319

309

        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"))

 

310

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

323

311

 

324

312

 

325

313

class TestUmask(tests.TestCaseInTempDir):

999

987

 

1000

988

    def test_osutils_binding(self):

1001

989

        from bzrlib.tests import test__chunks_to_lines

1002

 

        if test__chunks_to_lines.compiled_chunkstolines_feature.available():

 

990

        if test__chunks_to_lines.CompiledChunksToLinesFeature.available():

1003

991

            from bzrlib._chunks_to_lines_pyx import chunks_to_lines

1004

992

        else:

1005

993

            from bzrlib._chunks_to_lines_py import chunks_to_lines

1145

1133

            dirblock[:] = new_dirblock

1146

1134

 

1147

1135

    def _save_platform_info(self):

1148

 

        self.overrideAttr(win32utils, 'winver')

1149

 

        self.overrideAttr(osutils, '_fs_enc')

1150

 

        self.overrideAttr(osutils, '_selected_dir_reader')

 

1136

        cur_winver = win32utils.winver

 

1137

        cur_fs_enc = osutils._fs_enc

 

1138

        cur_dir_reader = osutils._selected_dir_reader

 

1139

        def restore():

 

1140

            win32utils.winver = cur_winver

 

1141

            osutils._fs_enc = cur_fs_enc

 

1142

            osutils._selected_dir_reader = cur_dir_reader

 

1143

        self.addCleanup(restore)

1151

1144

 

1152

1145

    def assertDirReaderIs(self, expected):

1153

1146

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

1186

1179

 

1187

1180

    def test_force_walkdirs_utf8_nt(self):

1188

1181

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

1189

 

        self.requireFeature(test__walkdirs_win32.win32_readdir_feature)

 

1182

        self.requireFeature(test__walkdirs_win32.Win32ReadDirFeature)

1190

1183

        self._save_platform_info()

1191

1184

        win32utils.winver = 'Windows NT'

1192

1185

        from bzrlib._walkdirs_win32 import Win32ReadDir

1193

1186

        self.assertDirReaderIs(Win32ReadDir)

1194

1187

 

1195

1188

    def test_force_walkdirs_utf8_98(self):

1196

 

        self.requireFeature(test__walkdirs_win32.win32_readdir_feature)

 

1189

        self.requireFeature(test__walkdirs_win32.Win32ReadDirFeature)

1197

1190

        self._save_platform_info()

1198

1191

        win32utils.winver = 'Windows 98'

1199

1192

        self.assertDirReaderIs(osutils.UnicodeDirReader)

1350

1343

        self.assertEqual(expected_dirblocks, result)

1351

1344

 

1352

1345

    def test__walkdirs_utf8_win32readdir(self):

1353

 

        self.requireFeature(test__walkdirs_win32.win32_readdir_feature)

 

1346

        self.requireFeature(test__walkdirs_win32.Win32ReadDirFeature)

1354

1347

        self.requireFeature(tests.UnicodeFilenameFeature)

1355

1348

        from bzrlib._walkdirs_win32 import Win32ReadDir

1356

1349

        self._save_platform_info()

1407

1400

 

1408

1401

    def test__walkdirs_utf_win32_find_file_stat_file(self):

1409

1402

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

1410

 

        self.requireFeature(test__walkdirs_win32.win32_readdir_feature)

 

1403

        self.requireFeature(test__walkdirs_win32.Win32ReadDirFeature)

1411

1404

        self.requireFeature(tests.UnicodeFilenameFeature)

1412

1405

        from bzrlib._walkdirs_win32 import Win32ReadDir

1413

1406

        name0u = u'0file-\xb6'

1431

1424

 

1432

1425

    def test__walkdirs_utf_win32_find_file_stat_directory(self):

1433

1426

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

1434

 

        self.requireFeature(test__walkdirs_win32.win32_readdir_feature)

 

1427

        self.requireFeature(test__walkdirs_win32.Win32ReadDirFeature)

1435

1428

        self.requireFeature(tests.UnicodeFilenameFeature)

1436

1429

        from bzrlib._walkdirs_win32 import Win32ReadDir

1437

1430

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

1586

1579

        def cleanup():

1587

1580

            if 'BZR_TEST_ENV_VAR' in os.environ:

1588

1581

                del os.environ['BZR_TEST_ENV_VAR']

 

1582

 

1589

1583

        self.addCleanup(cleanup)

1590

1584

 

1591

1585

    def test_set(self):

1702

1696

 

1703

1697

    def setUp(self):

1704

1698

        tests.TestCaseInTempDir.setUp(self)

1705

 

        self.overrideAttr(osutils,

1706

 

                          '_selected_dir_reader', self._dir_reader_class())

 

1699

 

 

1700

        # Save platform specific info and reset it

 

1701

        cur_dir_reader = osutils._selected_dir_reader

 

1702

 

 

1703

        def restore():

 

1704

            osutils._selected_dir_reader = cur_dir_reader

 

1705

        self.addCleanup(restore)

 

1706

 

 

1707

        osutils._selected_dir_reader = self._dir_reader_class()

1707

1708

 

1708

1709

    def _get_ascii_tree(self):

1709

1710

        tree = [

1856

1857

 

1857

1858

    def setUp(self):

1858

1859

        super(TestConcurrency, self).setUp()

1859

 

        self.overrideAttr(osutils, '_cached_local_concurrency')

 

1860

        orig = osutils._cached_local_concurrency

 

1861

        def restore():

 

1862

            osutils._cached_local_concurrency = orig

 

1863

        self.addCleanup(restore)

1860

1864

 

1861

1865

    def test_local_concurrency(self):

1862

1866

        concurrency = osutils.local_concurrency()

1889

1893

 

1890

1894

    def setUp(self):

1891

1895

        super(TestFailedToLoadExtension, self).setUp()

1892

 

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

 

1896

        self.saved_failures = osutils._extension_load_failures[:]

 

1897

        del osutils._extension_load_failures[:]

 

1898

        self.addCleanup(self.restore_failures)

 

1899

 

 

1900

    def restore_failures(self):

 

1901

        osutils._extension_load_failures = self.saved_failures

1893

1902

 

1894

1903

    def test_failure_to_load(self):

1895

1904

        self._try_loading()

1913

1922

            r"bzr: warning: some compiled extensions could not be loaded; "

1914

1923

            "see <https://answers\.launchpad\.net/bzr/\+faq/703>\n"

1915

1924

            )

1916

 

 

1917

 

 

1918

 

class TestTerminalWidth(tests.TestCase):

1919

 

 

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

 

    def test_default_values(self):

1935

 

        self.assertEqual(80, osutils.default_terminal_width)

1936

 

 

1937

 

    def test_defaults_to_BZR_COLUMNS(self):

1938

 

        # BZR_COLUMNS is set by the test framework

1939

 

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

1940

 

        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())

1949

 

 

1950

 

    def test_tty_default_without_columns(self):

1951

 

        del os.environ['BZR_COLUMNS']

1952

 

        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())

1963

 

 

1964

 

    def test_non_tty_default_without_columns(self):

1965

 

        del os.environ['BZR_COLUMNS']

1966

 

        del os.environ['COLUMNS']

1967

 

        self.replace_stdout(None)

1968

 

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

1969

 

 

1970

 

    def test_no_TIOCGWINSZ(self):

1971

 

        self.requireFeature(term_ios_feature)

1972

 

        termios = term_ios_feature.module

1973

 

        # bug 63539 is about a termios without TIOCGWINSZ attribute

1974

 

        try:

1975

 

            orig = termios.TIOCGWINSZ

1976

 

        except AttributeError:

1977

 

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

1978

 

            pass

1979

 

        else:

1980

 

            self.overrideAttr(termios, 'TIOCGWINSZ')

1981

 

            del termios.TIOCGWINSZ

1982

 

        del os.environ['BZR_COLUMNS']

1983

 

        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)

• Older »

Loggerhead is a web-based interface for Breezy
Version: 2.0.1