~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_osutils.py

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

• reverse the comparison (4819.2.3 to 5200.3.5)
• stop comparing with revision 5200.3.5

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

390

378

        # Instead blackbox.test_locale should check for localized

391

379

        # dates once they do occur in output strings.

392

380

 

393

 

    def test_format_date_with_offset_in_original_timezone(self):

394

 

        self.assertEqual("Thu 1970-01-01 00:00:00 +0000",

395

 

            osutils.format_date_with_offset_in_original_timezone(0))

396

 

        self.assertEqual("Fri 1970-01-02 03:46:40 +0000",

397

 

            osutils.format_date_with_offset_in_original_timezone(100000))

398

 

        self.assertEqual("Fri 1970-01-02 05:46:40 +0200",

399

 

            osutils.format_date_with_offset_in_original_timezone(100000, 7200))

400

 

 

401

381

    def test_local_time_offset(self):

402

382

        """Test that local_time_offset() returns a sane value."""

403

383

        offset = osutils.local_time_offset()

999

979

 

1000

980

    def test_osutils_binding(self):

1001

981

        from bzrlib.tests import test__chunks_to_lines

1002

 

        if test__chunks_to_lines.compiled_chunkstolines_feature.available():

 

982

        if test__chunks_to_lines.CompiledChunksToLinesFeature.available():

1003

983

            from bzrlib._chunks_to_lines_pyx import chunks_to_lines

1004

984

        else:

1005

985

            from bzrlib._chunks_to_lines_py import chunks_to_lines

1145

1125

            dirblock[:] = new_dirblock

1146

1126

 

1147

1127

    def _save_platform_info(self):

1148

 

        self.overrideAttr(win32utils, 'winver')

1149

 

        self.overrideAttr(osutils, '_fs_enc')

1150

 

        self.overrideAttr(osutils, '_selected_dir_reader')

 

1128

        cur_winver = win32utils.winver

 

1129

        cur_fs_enc = osutils._fs_enc

 

1130

        cur_dir_reader = osutils._selected_dir_reader

 

1131

        def restore():

 

1132

            win32utils.winver = cur_winver

 

1133

            osutils._fs_enc = cur_fs_enc

 

1134

            osutils._selected_dir_reader = cur_dir_reader

 

1135

        self.addCleanup(restore)

1151

1136

 

1152

1137

    def assertDirReaderIs(self, expected):

1153

1138

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

1186

1171

 

1187

1172

    def test_force_walkdirs_utf8_nt(self):

1188

1173

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

1189

 

        self.requireFeature(test__walkdirs_win32.win32_readdir_feature)

 

1174

        self.requireFeature(test__walkdirs_win32.Win32ReadDirFeature)

1190

1175

        self._save_platform_info()

1191

1176

        win32utils.winver = 'Windows NT'

1192

1177

        from bzrlib._walkdirs_win32 import Win32ReadDir

1193

1178

        self.assertDirReaderIs(Win32ReadDir)

1194

1179

 

1195

1180

    def test_force_walkdirs_utf8_98(self):

1196

 

        self.requireFeature(test__walkdirs_win32.win32_readdir_feature)

 

1181

        self.requireFeature(test__walkdirs_win32.Win32ReadDirFeature)

1197

1182

        self._save_platform_info()

1198

1183

        win32utils.winver = 'Windows 98'

1199

1184

        self.assertDirReaderIs(osutils.UnicodeDirReader)

1350

1335

        self.assertEqual(expected_dirblocks, result)

1351

1336

 

1352

1337

    def test__walkdirs_utf8_win32readdir(self):

1353

 

        self.requireFeature(test__walkdirs_win32.win32_readdir_feature)

 

1338

        self.requireFeature(test__walkdirs_win32.Win32ReadDirFeature)

1354

1339

        self.requireFeature(tests.UnicodeFilenameFeature)

1355

1340

        from bzrlib._walkdirs_win32 import Win32ReadDir

1356

1341

        self._save_platform_info()

1407

1392

 

1408

1393

    def test__walkdirs_utf_win32_find_file_stat_file(self):

1409

1394

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

1410

 

        self.requireFeature(test__walkdirs_win32.win32_readdir_feature)

 

1395

        self.requireFeature(test__walkdirs_win32.Win32ReadDirFeature)

1411

1396

        self.requireFeature(tests.UnicodeFilenameFeature)

1412

1397

        from bzrlib._walkdirs_win32 import Win32ReadDir

1413

1398

        name0u = u'0file-\xb6'

1431

1416

 

1432

1417

    def test__walkdirs_utf_win32_find_file_stat_directory(self):

1433

1418

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

1434

 

        self.requireFeature(test__walkdirs_win32.win32_readdir_feature)

 

1419

        self.requireFeature(test__walkdirs_win32.Win32ReadDirFeature)

1435

1420

        self.requireFeature(tests.UnicodeFilenameFeature)

1436

1421

        from bzrlib._walkdirs_win32 import Win32ReadDir

1437

1422

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

1586

1571

        def cleanup():

1587

1572

            if 'BZR_TEST_ENV_VAR' in os.environ:

1588

1573

                del os.environ['BZR_TEST_ENV_VAR']

 

1574

 

1589

1575

        self.addCleanup(cleanup)

1590

1576

 

1591

1577

    def test_set(self):

1702

1688

 

1703

1689

    def setUp(self):

1704

1690

        tests.TestCaseInTempDir.setUp(self)

1705

 

        self.overrideAttr(osutils,

1706

 

                          '_selected_dir_reader', self._dir_reader_class())

 

1691

 

 

1692

        # Save platform specific info and reset it

 

1693

        cur_dir_reader = osutils._selected_dir_reader

 

1694

 

 

1695

        def restore():

 

1696

            osutils._selected_dir_reader = cur_dir_reader

 

1697

        self.addCleanup(restore)

 

1698

 

 

1699

        osutils._selected_dir_reader = self._dir_reader_class()

1707

1700

 

1708

1701

    def _get_ascii_tree(self):

1709

1702

        tree = [

1854

1847

 

1855

1848

class TestConcurrency(tests.TestCase):

1856

1849

 

1857

 

    def setUp(self):

1858

 

        super(TestConcurrency, self).setUp()

1859

 

        self.overrideAttr(osutils, '_cached_local_concurrency')

1860

 

 

1861

1850

    def test_local_concurrency(self):

1862

1851

        concurrency = osutils.local_concurrency()

1863

1852

        self.assertIsInstance(concurrency, int)

1864

1853

 

1865

 

    def test_local_concurrency_environment_variable(self):

1866

 

        os.environ['BZR_CONCURRENCY'] = '2'

1867

 

        self.assertEqual(2, osutils.local_concurrency(use_cache=False))

1868

 

        os.environ['BZR_CONCURRENCY'] = '3'

1869

 

        self.assertEqual(3, osutils.local_concurrency(use_cache=False))

1870

 

        os.environ['BZR_CONCURRENCY'] = 'foo'

1871

 

        self.assertEqual(1, osutils.local_concurrency(use_cache=False))

1872

 

 

1873

 

    def test_option_concurrency(self):

1874

 

        os.environ['BZR_CONCURRENCY'] = '1'

1875

 

        self.run_bzr('rocks --concurrency 42')

1876

 

        # Command line overrides envrionment variable

1877

 

        self.assertEquals('42', os.environ['BZR_CONCURRENCY'])

1878

 

        self.assertEquals(42, osutils.local_concurrency(use_cache=False))

1879

 

 

1880

1854

 

1881

1855

class TestFailedToLoadExtension(tests.TestCase):

1882

1856

 

1889

1863

 

1890

1864

    def setUp(self):

1891

1865

        super(TestFailedToLoadExtension, self).setUp()

1892

 

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

 

1866

        self.saved_failures = osutils._extension_load_failures[:]

 

1867

        del osutils._extension_load_failures[:]

 

1868

        self.addCleanup(self.restore_failures)

 

1869

 

 

1870

    def restore_failures(self):

 

1871

        osutils._extension_load_failures = self.saved_failures

1893

1872

 

1894

1873

    def test_failure_to_load(self):

1895

1874

        self._try_loading()

1913

1892

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

1914

1893

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

1915

1894

            )

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