← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_osutils.py

  • Committer: Joe Julian
  • Date: 2010-01-10 02:25:31 UTC
  • mto: (4634.119.7 2.0)
  • mto: This revision was merged to the branch mainline in revision 4959.
  • Revision ID: joe@julianfamily.org-20100110022531-wqk61rsagz8xsiga
https://launchpad.net/bugs/175839
Added MANIFEST.in to allow bdist_rpm to have all the required include files and tools. bdist_rpm will still fail to build correctly on some distributions due to a disttools bug http://bugs.python.org/issue644744

Show diffs side-by-side

added added

removed removed

Lines of Context:
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
21
21
import os
22
22
import re
23
23
import socket
 
24
import stat
24
25
import sys
25
26
import time
26
27
 
27
28
from bzrlib import (
28
29
    errors,
29
 
    lazy_regex,
30
30
    osutils,
31
 
    symbol_versioning,
32
31
    tests,
33
 
    trace,
34
32
    win32utils,
35
33
    )
36
34
from bzrlib.tests import (
37
 
    features,
38
35
    file_utils,
39
36
    test__walkdirs_win32,
40
37
    )
55
52
 
56
53
UTF8DirReaderFeature = _UTF8DirReaderFeature()
57
54
 
58
 
term_ios_feature = tests.ModuleAvailableFeature('termios')
59
 
 
60
55
 
61
56
def _already_unicode(s):
62
57
    return s
63
58
 
64
59
 
 
60
def _fs_enc_to_unicode(s):
 
61
    return s.decode(osutils._fs_enc)
 
62
 
 
63
 
65
64
def _utf8_to_unicode(s):
66
65
    return s.decode('UTF-8')
67
66
 
84
83
                          dict(_dir_reader_class=_readdir_pyx.UTF8DirReader,
85
84
                               _native_to_unicode=_utf8_to_unicode)))
86
85
 
87
 
    if test__walkdirs_win32.win32_readdir_feature.available():
 
86
    if test__walkdirs_win32.Win32ReadDirFeature.available():
88
87
        try:
89
88
            from bzrlib import _walkdirs_win32
 
89
            # TODO: check on windows, it may be that we need to use/add
 
90
            # safe_unicode instead of _fs_enc_to_unicode
90
91
            scenarios.append(
91
92
                ('win32',
92
93
                 dict(_dir_reader_class=_walkdirs_win32.Win32ReadDir,
93
 
                      _native_to_unicode=_already_unicode)))
 
94
                      _native_to_unicode=_fs_enc_to_unicode)))
94
95
        except ImportError:
95
96
            pass
96
97
    return scenarios
124
125
 
125
126
class TestRename(tests.TestCaseInTempDir):
126
127
 
127
 
    def create_file(self, filename, content):
128
 
        f = open(filename, 'wb')
129
 
        try:
130
 
            f.write(content)
131
 
        finally:
132
 
            f.close()
133
 
 
134
 
    def _fancy_rename(self, a, b):
135
 
        osutils.fancy_rename(a, b, rename_func=os.rename,
136
 
                             unlink_func=os.unlink)
137
 
 
138
128
    def test_fancy_rename(self):
139
129
        # This should work everywhere
140
 
        self.create_file('a', 'something in a\n')
141
 
        self._fancy_rename('a', 'b')
 
130
        def rename(a, b):
 
131
            osutils.fancy_rename(a, b,
 
132
                    rename_func=os.rename,
 
133
                    unlink_func=os.unlink)
 
134
 
 
135
        open('a', 'wb').write('something in a\n')
 
136
        rename('a', 'b')
142
137
        self.failIfExists('a')
143
138
        self.failUnlessExists('b')
144
139
        self.check_file_contents('b', 'something in a\n')
145
140
 
146
 
        self.create_file('a', 'new something in a\n')
147
 
        self._fancy_rename('b', 'a')
 
141
        open('a', 'wb').write('new something in a\n')
 
142
        rename('b', 'a')
148
143
 
149
144
        self.check_file_contents('a', 'something in a\n')
150
145
 
151
 
    def test_fancy_rename_fails_source_missing(self):
152
 
        # An exception should be raised, and the target should be left in place
153
 
        self.create_file('target', 'data in target\n')
154
 
        self.assertRaises((IOError, OSError), self._fancy_rename,
155
 
                          'missingsource', 'target')
156
 
        self.failUnlessExists('target')
157
 
        self.check_file_contents('target', 'data in target\n')
158
 
 
159
 
    def test_fancy_rename_fails_if_source_and_target_missing(self):
160
 
        self.assertRaises((IOError, OSError), self._fancy_rename,
161
 
                          'missingsource', 'missingtarget')
162
 
 
163
146
    def test_rename(self):
164
147
        # Rename should be semi-atomic on all platforms
165
 
        self.create_file('a', 'something in a\n')
 
148
        open('a', 'wb').write('something in a\n')
166
149
        osutils.rename('a', 'b')
167
150
        self.failIfExists('a')
168
151
        self.failUnlessExists('b')
169
152
        self.check_file_contents('b', 'something in a\n')
170
153
 
171
 
        self.create_file('a', 'new something in a\n')
 
154
        open('a', 'wb').write('new something in a\n')
172
155
        osutils.rename('b', 'a')
173
156
 
174
157
        self.check_file_contents('a', 'something in a\n')
311
294
        self.assertEqual("/", osutils.kind_marker(osutils._directory_kind))
312
295
        self.assertEqual("@", osutils.kind_marker("symlink"))
313
296
        self.assertEqual("+", osutils.kind_marker("tree-reference"))
314
 
        self.assertEqual("", osutils.kind_marker("fifo"))
315
 
        self.assertEqual("", osutils.kind_marker("socket"))
316
 
        self.assertEqual("", osutils.kind_marker("unknown"))
 
297
        self.assertRaises(errors.BzrError, osutils.kind_marker, "unknown")
317
298
 
318
299
 
319
300
class TestUmask(tests.TestCaseInTempDir):
384
365
        # Instead blackbox.test_locale should check for localized
385
366
        # dates once they do occur in output strings.
386
367
 
387
 
    def test_format_date_with_offset_in_original_timezone(self):
388
 
        self.assertEqual("Thu 1970-01-01 00:00:00 +0000",
389
 
            osutils.format_date_with_offset_in_original_timezone(0))
390
 
        self.assertEqual("Fri 1970-01-02 03:46:40 +0000",
391
 
            osutils.format_date_with_offset_in_original_timezone(100000))
392
 
        self.assertEqual("Fri 1970-01-02 05:46:40 +0200",
393
 
            osutils.format_date_with_offset_in_original_timezone(100000, 7200))
394
 
 
395
368
    def test_local_time_offset(self):
396
369
        """Test that local_time_offset() returns a sane value."""
397
370
        offset = osutils.local_time_offset()
473
446
    def test_canonical_relpath_simple(self):
474
447
        f = file('MixedCaseName', 'w')
475
448
        f.close()
476
 
        actual = osutils.canonical_relpath(self.test_base_dir, 'mixedcasename')
 
449
        # Watch out for tricky test dir (on OSX /tmp -> /private/tmp)
 
450
        real_base_dir = osutils.realpath(self.test_base_dir)
 
451
        actual = osutils.canonical_relpath(real_base_dir, 'mixedcasename')
477
452
        self.failUnlessEqual('work/MixedCaseName', actual)
478
453
 
479
454
    def test_canonical_relpath_missing_tail(self):
480
455
        os.mkdir('MixedCaseParent')
481
 
        actual = osutils.canonical_relpath(self.test_base_dir,
 
456
        # Watch out for tricky test dir (on OSX /tmp -> /private/tmp)
 
457
        real_base_dir = osutils.realpath(self.test_base_dir)
 
458
        actual = osutils.canonical_relpath(real_base_dir,
482
459
                                           'mixedcaseparent/nochild')
483
460
        self.failUnlessEqual('work/MixedCaseParent/nochild', actual)
484
461
 
862
839
        self.assertEqual('//HOST/path', osutils._win98_abspath('//HOST/path'))
863
840
        # relative path
864
841
        cwd = osutils.getcwd().rstrip('/')
865
 
        drive = osutils.ntpath.splitdrive(cwd)[0]
 
842
        drive = osutils._nt_splitdrive(cwd)[0]
866
843
        self.assertEqual(cwd+'/path', osutils._win98_abspath('path'))
867
844
        self.assertEqual(drive+'/path', osutils._win98_abspath('/path'))
868
845
        # unicode path
993
970
 
994
971
    def test_osutils_binding(self):
995
972
        from bzrlib.tests import test__chunks_to_lines
996
 
        if test__chunks_to_lines.compiled_chunkstolines_feature.available():
 
973
        if test__chunks_to_lines.CompiledChunksToLinesFeature.available():
997
974
            from bzrlib._chunks_to_lines_pyx import chunks_to_lines
998
975
        else:
999
976
            from bzrlib._chunks_to_lines_py import chunks_to_lines
1065
1042
        self.assertExpectedBlocks(expected_dirblocks[1:], result)
1066
1043
 
1067
1044
    def test_walkdirs_os_error(self):
1068
 
        # <https://bugs.launchpad.net/bzr/+bug/338653>
 
1045
        # <https://bugs.edge.launchpad.net/bzr/+bug/338653>
1069
1046
        # Pyrex readdir didn't raise useful messages if it had an error
1070
1047
        # reading the directory
1071
1048
        if sys.platform == 'win32':
1072
1049
            raise tests.TestNotApplicable(
1073
1050
                "readdir IOError not tested on win32")
1074
 
        self.requireFeature(features.not_running_as_root)
1075
1051
        os.mkdir("test-unreadable")
1076
1052
        os.chmod("test-unreadable", 0000)
1077
1053
        # must chmod it back so that it can be removed
1085
1061
        # Ensure the message contains the file name
1086
1062
        self.assertContainsRe(str(e), "\./test-unreadable")
1087
1063
 
1088
 
 
1089
 
    def test_walkdirs_encoding_error(self):
1090
 
        # <https://bugs.launchpad.net/bzr/+bug/488519>
1091
 
        # walkdirs didn't raise a useful message when the filenames
1092
 
        # are not using the filesystem's encoding
1093
 
 
1094
 
        # require a bytestring based filesystem
1095
 
        self.requireFeature(tests.ByteStringNamedFilesystem)
1096
 
 
1097
 
        tree = [
1098
 
            '.bzr',
1099
 
            '0file',
1100
 
            '1dir/',
1101
 
            '1dir/0file',
1102
 
            '1dir/1dir/',
1103
 
            '1file'
1104
 
            ]
1105
 
 
1106
 
        self.build_tree(tree)
1107
 
 
1108
 
        # rename the 1file to a latin-1 filename
1109
 
        os.rename("./1file", "\xe8file")
1110
 
 
1111
 
        self._save_platform_info()
1112
 
        win32utils.winver = None # Avoid the win32 detection code
1113
 
        osutils._fs_enc = 'UTF-8'
1114
 
 
1115
 
        # this should raise on error
1116
 
        def attempt():
1117
 
            for dirdetail, dirblock in osutils.walkdirs('.'):
1118
 
                pass
1119
 
 
1120
 
        self.assertRaises(errors.BadFilenameEncoding, attempt)
1121
 
 
1122
1064
    def test__walkdirs_utf8(self):
1123
1065
        tree = [
1124
1066
            '.bzr',
1174
1116
            dirblock[:] = new_dirblock
1175
1117
 
1176
1118
    def _save_platform_info(self):
1177
 
        self.overrideAttr(win32utils, 'winver')
1178
 
        self.overrideAttr(osutils, '_fs_enc')
1179
 
        self.overrideAttr(osutils, '_selected_dir_reader')
 
1119
        cur_winver = win32utils.winver
 
1120
        cur_fs_enc = osutils._fs_enc
 
1121
        cur_dir_reader = osutils._selected_dir_reader
 
1122
        def restore():
 
1123
            win32utils.winver = cur_winver
 
1124
            osutils._fs_enc = cur_fs_enc
 
1125
            osutils._selected_dir_reader = cur_dir_reader
 
1126
        self.addCleanup(restore)
1180
1127
 
1181
1128
    def assertDirReaderIs(self, expected):
1182
1129
        """Assert the right implementation for _walkdirs_utf8 is chosen."""
1215
1162
 
1216
1163
    def test_force_walkdirs_utf8_nt(self):
1217
1164
        # Disabled because the thunk of the whole walkdirs api is disabled.
1218
 
        self.requireFeature(test__walkdirs_win32.win32_readdir_feature)
 
1165
        self.requireFeature(test__walkdirs_win32.Win32ReadDirFeature)
1219
1166
        self._save_platform_info()
1220
1167
        win32utils.winver = 'Windows NT'
1221
1168
        from bzrlib._walkdirs_win32 import Win32ReadDir
1222
1169
        self.assertDirReaderIs(Win32ReadDir)
1223
1170
 
1224
1171
    def test_force_walkdirs_utf8_98(self):
1225
 
        self.requireFeature(test__walkdirs_win32.win32_readdir_feature)
 
1172
        self.requireFeature(test__walkdirs_win32.Win32ReadDirFeature)
1226
1173
        self._save_platform_info()
1227
1174
        win32utils.winver = 'Windows 98'
1228
1175
        self.assertDirReaderIs(osutils.UnicodeDirReader)
1379
1326
        self.assertEqual(expected_dirblocks, result)
1380
1327
 
1381
1328
    def test__walkdirs_utf8_win32readdir(self):
1382
 
        self.requireFeature(test__walkdirs_win32.win32_readdir_feature)
 
1329
        self.requireFeature(test__walkdirs_win32.Win32ReadDirFeature)
1383
1330
        self.requireFeature(tests.UnicodeFilenameFeature)
1384
1331
        from bzrlib._walkdirs_win32 import Win32ReadDir
1385
1332
        self._save_platform_info()
1436
1383
 
1437
1384
    def test__walkdirs_utf_win32_find_file_stat_file(self):
1438
1385
        """make sure our Stat values are valid"""
1439
 
        self.requireFeature(test__walkdirs_win32.win32_readdir_feature)
 
1386
        self.requireFeature(test__walkdirs_win32.Win32ReadDirFeature)
1440
1387
        self.requireFeature(tests.UnicodeFilenameFeature)
1441
1388
        from bzrlib._walkdirs_win32 import Win32ReadDir
1442
1389
        name0u = u'0file-\xb6'
1460
1407
 
1461
1408
    def test__walkdirs_utf_win32_find_file_stat_directory(self):
1462
1409
        """make sure our Stat values are valid"""
1463
 
        self.requireFeature(test__walkdirs_win32.win32_readdir_feature)
 
1410
        self.requireFeature(test__walkdirs_win32.Win32ReadDirFeature)
1464
1411
        self.requireFeature(tests.UnicodeFilenameFeature)
1465
1412
        from bzrlib._walkdirs_win32 import Win32ReadDir
1466
1413
        name0u = u'0dir-\u062c\u0648'
1615
1562
        def cleanup():
1616
1563
            if 'BZR_TEST_ENV_VAR' in os.environ:
1617
1564
                del os.environ['BZR_TEST_ENV_VAR']
 
1565
 
1618
1566
        self.addCleanup(cleanup)
1619
1567
 
1620
1568
    def test_set(self):
1668
1616
        text = 'test\r\nwith\nall\rpossible line endings\r\n'
1669
1617
        self.build_tree_contents([('foo', text)])
1670
1618
        expected_sha = osutils.sha_string(text)
1671
 
        f = open('foo', 'rb')
 
1619
        f = open('foo')
1672
1620
        self.addCleanup(f.close)
1673
1621
        size, sha = osutils.size_sha_file(f)
1674
1622
        self.assertEqual(38, size)
1707
1655
 
1708
1656
class TestReCompile(tests.TestCase):
1709
1657
 
1710
 
    def _deprecated_re_compile_checked(self, *args, **kwargs):
1711
 
        return self.applyDeprecated(symbol_versioning.deprecated_in((2, 2, 0)),
1712
 
            osutils.re_compile_checked, *args, **kwargs)
1713
 
 
1714
1658
    def test_re_compile_checked(self):
1715
 
        r = self._deprecated_re_compile_checked(r'A*', re.IGNORECASE)
 
1659
        r = osutils.re_compile_checked(r'A*', re.IGNORECASE)
1716
1660
        self.assertTrue(r.match('aaaa'))
1717
1661
        self.assertTrue(r.match('aAaA'))
1718
1662
 
1719
1663
    def test_re_compile_checked_error(self):
1720
1664
        # like https://bugs.launchpad.net/bzr/+bug/251352
1721
 
 
1722
 
        # Due to possible test isolation error, re.compile is not lazy at
1723
 
        # this point. We re-install lazy compile.
1724
 
        lazy_regex.install_lazy_compile()
1725
1665
        err = self.assertRaises(
1726
1666
            errors.BzrCommandError,
1727
 
            self._deprecated_re_compile_checked, '*', re.IGNORECASE, 'test case')
 
1667
            osutils.re_compile_checked, '*', re.IGNORECASE, 'test case')
1728
1668
        self.assertEqual(
1729
 
            'Invalid regular expression in test case: '
1730
 
            '"*" nothing to repeat',
 
1669
            "Invalid regular expression in test case: '*': "
 
1670
            "nothing to repeat",
1731
1671
            str(err))
1732
1672
 
1733
1673
 
1739
1679
 
1740
1680
    def setUp(self):
1741
1681
        tests.TestCaseInTempDir.setUp(self)
1742
 
        self.overrideAttr(osutils,
1743
 
                          '_selected_dir_reader', self._dir_reader_class())
 
1682
 
 
1683
        # Save platform specific info and reset it
 
1684
        cur_dir_reader = osutils._selected_dir_reader
 
1685
 
 
1686
        def restore():
 
1687
            osutils._selected_dir_reader = cur_dir_reader
 
1688
        self.addCleanup(restore)
 
1689
 
 
1690
        osutils._selected_dir_reader = self._dir_reader_class()
1744
1691
 
1745
1692
    def _get_ascii_tree(self):
1746
1693
        tree = [
1891
1838
 
1892
1839
class TestConcurrency(tests.TestCase):
1893
1840
 
1894
 
    def setUp(self):
1895
 
        super(TestConcurrency, self).setUp()
1896
 
        self.overrideAttr(osutils, '_cached_local_concurrency')
1897
 
 
1898
1841
    def test_local_concurrency(self):
1899
1842
        concurrency = osutils.local_concurrency()
1900
1843
        self.assertIsInstance(concurrency, int)
1901
 
 
1902
 
    def test_local_concurrency_environment_variable(self):
1903
 
        os.environ['BZR_CONCURRENCY'] = '2'
1904
 
        self.assertEqual(2, osutils.local_concurrency(use_cache=False))
1905
 
        os.environ['BZR_CONCURRENCY'] = '3'
1906
 
        self.assertEqual(3, osutils.local_concurrency(use_cache=False))
1907
 
        os.environ['BZR_CONCURRENCY'] = 'foo'
1908
 
        self.assertEqual(1, osutils.local_concurrency(use_cache=False))
1909
 
 
1910
 
    def test_option_concurrency(self):
1911
 
        os.environ['BZR_CONCURRENCY'] = '1'
1912
 
        self.run_bzr('rocks --concurrency 42')
1913
 
        # Command line overrides envrionment variable
1914
 
        self.assertEquals('42', os.environ['BZR_CONCURRENCY'])
1915
 
        self.assertEquals(42, osutils.local_concurrency(use_cache=False))
1916
 
 
1917
 
 
1918
 
class TestFailedToLoadExtension(tests.TestCase):
1919
 
 
1920
 
    def _try_loading(self):
1921
 
        try:
1922
 
            import bzrlib._fictional_extension_py
1923
 
        except ImportError, e:
1924
 
            osutils.failed_to_load_extension(e)
1925
 
            return True
1926
 
 
1927
 
    def setUp(self):
1928
 
        super(TestFailedToLoadExtension, self).setUp()
1929
 
        self.overrideAttr(osutils, '_extension_load_failures', [])
1930
 
 
1931
 
    def test_failure_to_load(self):
1932
 
        self._try_loading()
1933
 
        self.assertLength(1, osutils._extension_load_failures)
1934
 
        self.assertEquals(osutils._extension_load_failures[0],
1935
 
            "No module named _fictional_extension_py")
1936
 
 
1937
 
    def test_report_extension_load_failures_no_warning(self):
1938
 
        self.assertTrue(self._try_loading())
1939
 
        warnings, result = self.callCatchWarnings(osutils.report_extension_load_failures)
1940
 
        # it used to give a Python warning; it no longer does
1941
 
        self.assertLength(0, warnings)
1942
 
 
1943
 
    def test_report_extension_load_failures_message(self):
1944
 
        log = StringIO()
1945
 
        trace.push_log_file(log)
1946
 
        self.assertTrue(self._try_loading())
1947
 
        osutils.report_extension_load_failures()
1948
 
        self.assertContainsRe(
1949
 
            log.getvalue(),
1950
 
            r"bzr: warning: some compiled extensions could not be loaded; "
1951
 
            "see <https://answers\.launchpad\.net/bzr/\+faq/703>\n"
1952
 
            )
1953
 
 
1954
 
 
1955
 
class TestTerminalWidth(tests.TestCase):
1956
 
 
1957
 
    def setUp(self):
1958
 
        tests.TestCase.setUp(self)
1959
 
        self._orig_terminal_size_state = osutils._terminal_size_state
1960
 
        self._orig_first_terminal_size = osutils._first_terminal_size
1961
 
        self.addCleanup(self.restore_osutils_globals)
1962
 
        osutils._terminal_size_state = 'no_data'
1963
 
        osutils._first_terminal_size = None
1964
 
 
1965
 
    def restore_osutils_globals(self):
1966
 
        osutils._terminal_size_state = self._orig_terminal_size_state
1967
 
        osutils._first_terminal_size = self._orig_first_terminal_size
1968
 
 
1969
 
    def replace_stdout(self, new):
1970
 
        self.overrideAttr(sys, 'stdout', new)
1971
 
 
1972
 
    def replace__terminal_size(self, new):
1973
 
        self.overrideAttr(osutils, '_terminal_size', new)
1974
 
 
1975
 
    def set_fake_tty(self):
1976
 
 
1977
 
        class I_am_a_tty(object):
1978
 
            def isatty(self):
1979
 
                return True
1980
 
 
1981
 
        self.replace_stdout(I_am_a_tty())
1982
 
 
1983
 
    def test_default_values(self):
1984
 
        self.assertEqual(80, osutils.default_terminal_width)
1985
 
 
1986
 
    def test_defaults_to_BZR_COLUMNS(self):
1987
 
        # BZR_COLUMNS is set by the test framework
1988
 
        self.assertNotEqual('12', os.environ['BZR_COLUMNS'])
1989
 
        os.environ['BZR_COLUMNS'] = '12'
1990
 
        self.assertEqual(12, osutils.terminal_width())
1991
 
 
1992
 
    def test_falls_back_to_COLUMNS(self):
1993
 
        del os.environ['BZR_COLUMNS']
1994
 
        self.assertNotEqual('42', os.environ['COLUMNS'])
1995
 
        self.set_fake_tty()
1996
 
        os.environ['COLUMNS'] = '42'
1997
 
        self.assertEqual(42, osutils.terminal_width())
1998
 
 
1999
 
    def test_tty_default_without_columns(self):
2000
 
        del os.environ['BZR_COLUMNS']
2001
 
        del os.environ['COLUMNS']
2002
 
 
2003
 
        def terminal_size(w, h):
2004
 
            return 42, 42
2005
 
 
2006
 
        self.set_fake_tty()
2007
 
        # We need to override the osutils definition as it depends on the
2008
 
        # running environment that we can't control (PQM running without a
2009
 
        # controlling terminal is one example).
2010
 
        self.replace__terminal_size(terminal_size)
2011
 
        self.assertEqual(42, osutils.terminal_width())
2012
 
 
2013
 
    def test_non_tty_default_without_columns(self):
2014
 
        del os.environ['BZR_COLUMNS']
2015
 
        del os.environ['COLUMNS']
2016
 
        self.replace_stdout(None)
2017
 
        self.assertEqual(None, osutils.terminal_width())
2018
 
 
2019
 
    def test_no_TIOCGWINSZ(self):
2020
 
        self.requireFeature(term_ios_feature)
2021
 
        termios = term_ios_feature.module
2022
 
        # bug 63539 is about a termios without TIOCGWINSZ attribute
2023
 
        try:
2024
 
            orig = termios.TIOCGWINSZ
2025
 
        except AttributeError:
2026
 
            # We won't remove TIOCGWINSZ, because it doesn't exist anyway :)
2027
 
            pass
2028
 
        else:
2029
 
            self.overrideAttr(termios, 'TIOCGWINSZ')
2030
 
            del termios.TIOCGWINSZ
2031
 
        del os.environ['BZR_COLUMNS']
2032
 
        del os.environ['COLUMNS']
2033
 
        # Whatever the result is, if we don't raise an exception, it's ok.
2034
 
        osutils.terminal_width()
2035
 
 
2036
 
class TestCreationOps(tests.TestCaseInTempDir):
2037
 
    _test_needs_features = [features.chown_feature]
2038
 
 
2039
 
    def setUp(self):
2040
 
        tests.TestCaseInTempDir.setUp(self)
2041
 
        self.overrideAttr(os, 'chown', self._dummy_chown)
2042
 
 
2043
 
        # params set by call to _dummy_chown
2044
 
        self.path = self.uid = self.gid = None
2045
 
 
2046
 
    def _dummy_chown(self, path, uid, gid):
2047
 
        self.path, self.uid, self.gid = path, uid, gid
2048
 
 
2049
 
    def test_copy_ownership_from_path(self):
2050
 
        """copy_ownership_from_path test with specified src."""
2051
 
        ownsrc = '/'
2052
 
        f = open('test_file', 'wt')
2053
 
        osutils.copy_ownership_from_path('test_file', ownsrc)
2054
 
 
2055
 
        s = os.stat(ownsrc)
2056
 
        self.assertEquals(self.path, 'test_file')
2057
 
        self.assertEquals(self.uid, s.st_uid)
2058
 
        self.assertEquals(self.gid, s.st_gid)
2059
 
 
2060
 
    def test_copy_ownership_nonesrc(self):
2061
 
        """copy_ownership_from_path test with src=None."""
2062
 
        f = open('test_file', 'wt')
2063
 
        # should use parent dir for permissions
2064
 
        osutils.copy_ownership_from_path('test_file')
2065
 
 
2066
 
        s = os.stat('..')
2067
 
        self.assertEquals(self.path, 'test_file')
2068
 
        self.assertEquals(self.uid, s.st_uid)
2069
 
        self.assertEquals(self.gid, s.st_gid)
2070
 
 
2071
 
class TestGetuserUnicode(tests.TestCase):
2072
 
 
2073
 
    def test_ascii_user(self):
2074
 
        osutils.set_or_unset_env('LOGNAME', 'jrandom')
2075
 
        self.assertEqual(u'jrandom', osutils.getuser_unicode())
2076
 
 
2077
 
    def test_unicode_user(self):
2078
 
        ue = osutils.get_user_encoding()
2079
 
        uni_val, env_val = tests.probe_unicode_in_user_encoding()
2080
 
        if uni_val is None:
2081
 
            raise tests.TestSkipped(
2082
 
                'Cannot find a unicode character that works in encoding %s'
2083
 
                % (osutils.get_user_encoding(),))
2084
 
        uni_username = u'jrandom' + uni_val
2085
 
        encoded_username = uni_username.encode(ue)
2086
 
        osutils.set_or_unset_env('LOGNAME', encoded_username)
2087
 
        self.assertEqual(uni_username, osutils.getuser_unicode())