← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_osutils.py

(robertc) Fix lp: urls behind an https proxy.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test_osutils.py
1
 
# Copyright (C) 2005-2011 Canonical Ltd
 
1
# Copyright (C) 2005-2010 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
32
    trace,
34
33
    win32utils,
38
37
    file_utils,
39
38
    test__walkdirs_win32,
40
39
    )
41
 
from bzrlib.tests.scenarios import load_tests_apply_scenarios
42
 
 
43
 
 
44
 
class _UTF8DirReaderFeature(features.Feature):
 
40
 
 
41
 
 
42
class _UTF8DirReaderFeature(tests.Feature):
45
43
 
46
44
    def _probe(self):
47
45
        try:
54
52
    def feature_name(self):
55
53
        return 'bzrlib._readdir_pyx'
56
54
 
57
 
UTF8DirReaderFeature = features.ModuleAvailableFeature('bzrlib._readdir_pyx')
 
55
UTF8DirReaderFeature = _UTF8DirReaderFeature()
58
56
 
59
 
term_ios_feature = features.ModuleAvailableFeature('termios')
 
57
term_ios_feature = tests.ModuleAvailableFeature('termios')
60
58
 
61
59
 
62
60
def _already_unicode(s):
97
95
    return scenarios
98
96
 
99
97
 
100
 
load_tests = load_tests_apply_scenarios
 
98
def load_tests(basic_tests, module, loader):
 
99
    suite = loader.suiteClass()
 
100
    dir_reader_tests, remaining_tests = tests.split_suite_by_condition(
 
101
        basic_tests, tests.condition_isinstance(TestDirReader))
 
102
    tests.multiply_tests(dir_reader_tests, dir_reader_scenarios(), suite)
 
103
    suite.addTest(remaining_tests)
 
104
    return suite
101
105
 
102
106
 
103
107
class TestContainsWhitespace(tests.TestCase):
104
108
 
105
109
    def test_contains_whitespace(self):
106
 
        self.assertTrue(osutils.contains_whitespace(u' '))
107
 
        self.assertTrue(osutils.contains_whitespace(u'hello there'))
108
 
        self.assertTrue(osutils.contains_whitespace(u'hellothere\n'))
109
 
        self.assertTrue(osutils.contains_whitespace(u'hello\nthere'))
110
 
        self.assertTrue(osutils.contains_whitespace(u'hello\rthere'))
111
 
        self.assertTrue(osutils.contains_whitespace(u'hello\tthere'))
 
110
        self.failUnless(osutils.contains_whitespace(u' '))
 
111
        self.failUnless(osutils.contains_whitespace(u'hello there'))
 
112
        self.failUnless(osutils.contains_whitespace(u'hellothere\n'))
 
113
        self.failUnless(osutils.contains_whitespace(u'hello\nthere'))
 
114
        self.failUnless(osutils.contains_whitespace(u'hello\rthere'))
 
115
        self.failUnless(osutils.contains_whitespace(u'hello\tthere'))
112
116
 
113
117
        # \xa0 is "Non-breaking-space" which on some python locales thinks it
114
118
        # is whitespace, but we do not.
115
 
        self.assertFalse(osutils.contains_whitespace(u''))
116
 
        self.assertFalse(osutils.contains_whitespace(u'hellothere'))
117
 
        self.assertFalse(osutils.contains_whitespace(u'hello\xa0there'))
 
119
        self.failIf(osutils.contains_whitespace(u''))
 
120
        self.failIf(osutils.contains_whitespace(u'hellothere'))
 
121
        self.failIf(osutils.contains_whitespace(u'hello\xa0there'))
118
122
 
119
123
 
120
124
class TestRename(tests.TestCaseInTempDir):
134
138
        # This should work everywhere
135
139
        self.create_file('a', 'something in a\n')
136
140
        self._fancy_rename('a', 'b')
137
 
        self.assertPathDoesNotExist('a')
138
 
        self.assertPathExists('b')
 
141
        self.failIfExists('a')
 
142
        self.failUnlessExists('b')
139
143
        self.check_file_contents('b', 'something in a\n')
140
144
 
141
145
        self.create_file('a', 'new something in a\n')
148
152
        self.create_file('target', 'data in target\n')
149
153
        self.assertRaises((IOError, OSError), self._fancy_rename,
150
154
                          'missingsource', 'target')
151
 
        self.assertPathExists('target')
 
155
        self.failUnlessExists('target')
152
156
        self.check_file_contents('target', 'data in target\n')
153
157
 
154
158
    def test_fancy_rename_fails_if_source_and_target_missing(self):
159
163
        # Rename should be semi-atomic on all platforms
160
164
        self.create_file('a', 'something in a\n')
161
165
        osutils.rename('a', 'b')
162
 
        self.assertPathDoesNotExist('a')
163
 
        self.assertPathExists('b')
 
166
        self.failIfExists('a')
 
167
        self.failUnlessExists('b')
164
168
        self.check_file_contents('b', 'something in a\n')
165
169
 
166
170
        self.create_file('a', 'new something in a\n')
180
184
        shape = sorted(os.listdir('.'))
181
185
        self.assertEquals(['A', 'B'], shape)
182
186
 
 
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
 
183
194
 
184
195
class TestRandChars(tests.TestCase):
185
196
 
231
242
            self.assertFalse(osutils.is_inside_or_parent_of_any(dirs, fn))
232
243
 
233
244
 
234
 
class TestLstat(tests.TestCaseInTempDir):
235
 
 
236
 
    def test_lstat_matches_fstat(self):
237
 
        # On Windows, lstat and fstat don't always agree, primarily in the
238
 
        # 'st_ino' and 'st_dev' fields. So we force them to be '0' in our
239
 
        # custom implementation.
240
 
        if sys.platform == 'win32':
241
 
            # We only have special lstat/fstat if we have the extension.
242
 
            # Without it, we may end up re-reading content when we don't have
243
 
            # to, but otherwise it doesn't effect correctness.
244
 
            self.requireFeature(test__walkdirs_win32.win32_readdir_feature)
245
 
        f = open('test-file.txt', 'wb')
246
 
        self.addCleanup(f.close)
247
 
        f.write('some content\n')
248
 
        f.flush()
249
 
        self.assertEqualStat(osutils.fstat(f.fileno()),
250
 
                             osutils.lstat('test-file.txt'))
251
 
 
252
 
 
253
245
class TestRmTree(tests.TestCaseInTempDir):
254
246
 
255
247
    def test_rmtree(self):
267
259
 
268
260
        osutils.rmtree('dir')
269
261
 
270
 
        self.assertPathDoesNotExist('dir/file')
271
 
        self.assertPathDoesNotExist('dir')
 
262
        self.failIfExists('dir/file')
 
263
        self.failIfExists('dir')
272
264
 
273
265
 
274
266
class TestDeleteAny(tests.TestCaseInTempDir):
429
421
class TestLinks(tests.TestCaseInTempDir):
430
422
 
431
423
    def test_dereference_path(self):
432
 
        self.requireFeature(features.SymlinkFeature)
 
424
        self.requireFeature(tests.SymlinkFeature)
433
425
        cwd = osutils.realpath('.')
434
426
        os.mkdir('bar')
435
427
        bar_path = osutils.pathjoin(cwd, 'bar')
482
474
 
483
475
class TestCanonicalRelPath(tests.TestCaseInTempDir):
484
476
 
485
 
    _test_needs_features = [features.CaseInsCasePresFilenameFeature]
 
477
    _test_needs_features = [tests.CaseInsCasePresFilenameFeature]
486
478
 
487
479
    def test_canonical_relpath_simple(self):
488
480
        f = file('MixedCaseName', 'w')
489
481
        f.close()
490
482
        actual = osutils.canonical_relpath(self.test_base_dir, 'mixedcasename')
491
 
        self.assertEqual('work/MixedCaseName', actual)
 
483
        self.failUnlessEqual('work/MixedCaseName', actual)
492
484
 
493
485
    def test_canonical_relpath_missing_tail(self):
494
486
        os.mkdir('MixedCaseParent')
495
487
        actual = osutils.canonical_relpath(self.test_base_dir,
496
488
                                           'mixedcaseparent/nochild')
497
 
        self.assertEqual('work/MixedCaseParent/nochild', actual)
 
489
        self.failUnlessEqual('work/MixedCaseParent/nochild', actual)
498
490
 
499
491
 
500
492
class Test_CICPCanonicalRelpath(tests.TestCaseWithTransport):
876
868
        self.assertEqual('//HOST/path', osutils._win98_abspath('//HOST/path'))
877
869
        # relative path
878
870
        cwd = osutils.getcwd().rstrip('/')
879
 
        drive = osutils.ntpath.splitdrive(cwd)[0]
 
871
        drive = osutils._nt_splitdrive(cwd)[0]
880
872
        self.assertEqual(cwd+'/path', osutils._win98_abspath('path'))
881
873
        self.assertEqual(drive+'/path', osutils._win98_abspath('/path'))
882
874
        # unicode path
888
880
    """Test win32 functions that create files."""
889
881
 
890
882
    def test_getcwd(self):
891
 
        self.requireFeature(features.UnicodeFilenameFeature)
 
883
        self.requireFeature(tests.UnicodeFilenameFeature)
892
884
        os.mkdir(u'mu-\xb5')
893
885
        os.chdir(u'mu-\xb5')
894
886
        # TODO: jam 20060427 This will probably fail on Mac OSX because
924
916
        b.close()
925
917
 
926
918
        osutils._win32_rename('b', 'a')
927
 
        self.assertPathExists('a')
928
 
        self.assertPathDoesNotExist('b')
 
919
        self.failUnlessExists('a')
 
920
        self.failIfExists('b')
929
921
        self.assertFileEqual('baz\n', 'a')
930
922
 
931
923
    def test_rename_missing_file(self):
984
976
    """Test mac special functions that require directories."""
985
977
 
986
978
    def test_getcwd(self):
987
 
        self.requireFeature(features.UnicodeFilenameFeature)
 
979
        self.requireFeature(tests.UnicodeFilenameFeature)
988
980
        os.mkdir(u'B\xe5gfors')
989
981
        os.chdir(u'B\xe5gfors')
990
982
        self.assertEndsWith(osutils._mac_getcwd(), u'B\xe5gfors')
991
983
 
992
984
    def test_getcwd_nonnorm(self):
993
 
        self.requireFeature(features.UnicodeFilenameFeature)
 
985
        self.requireFeature(tests.UnicodeFilenameFeature)
994
986
        # Test that _mac_getcwd() will normalize this path
995
987
        os.mkdir(u'Ba\u030agfors')
996
988
        os.chdir(u'Ba\u030agfors')
1079
1071
        self.assertExpectedBlocks(expected_dirblocks[1:], result)
1080
1072
 
1081
1073
    def test_walkdirs_os_error(self):
1082
 
        # <https://bugs.launchpad.net/bzr/+bug/338653>
 
1074
        # <https://bugs.edge.launchpad.net/bzr/+bug/338653>
1083
1075
        # Pyrex readdir didn't raise useful messages if it had an error
1084
1076
        # reading the directory
1085
1077
        if sys.platform == 'win32':
1086
1078
            raise tests.TestNotApplicable(
1087
1079
                "readdir IOError not tested on win32")
1088
 
        self.requireFeature(features.not_running_as_root)
1089
1080
        os.mkdir("test-unreadable")
1090
1081
        os.chmod("test-unreadable", 0000)
1091
1082
        # must chmod it back so that it can be removed
1099
1090
        # Ensure the message contains the file name
1100
1091
        self.assertContainsRe(str(e), "\./test-unreadable")
1101
1092
 
1102
 
 
1103
 
    def test_walkdirs_encoding_error(self):
1104
 
        # <https://bugs.launchpad.net/bzr/+bug/488519>
1105
 
        # walkdirs didn't raise a useful message when the filenames
1106
 
        # are not using the filesystem's encoding
1107
 
 
1108
 
        # require a bytestring based filesystem
1109
 
        self.requireFeature(features.ByteStringNamedFilesystem)
1110
 
 
1111
 
        tree = [
1112
 
            '.bzr',
1113
 
            '0file',
1114
 
            '1dir/',
1115
 
            '1dir/0file',
1116
 
            '1dir/1dir/',
1117
 
            '1file'
1118
 
            ]
1119
 
 
1120
 
        self.build_tree(tree)
1121
 
 
1122
 
        # rename the 1file to a latin-1 filename
1123
 
        os.rename("./1file", "\xe8file")
1124
 
        if "\xe8file" not in os.listdir("."):
1125
 
            self.skip("Lack filesystem that preserves arbitrary bytes")
1126
 
 
1127
 
        self._save_platform_info()
1128
 
        win32utils.winver = None # Avoid the win32 detection code
1129
 
        osutils._fs_enc = 'UTF-8'
1130
 
 
1131
 
        # this should raise on error
1132
 
        def attempt():
1133
 
            for dirdetail, dirblock in osutils.walkdirs('.'):
1134
 
                pass
1135
 
 
1136
 
        self.assertRaises(errors.BadFilenameEncoding, attempt)
1137
 
 
1138
1093
    def test__walkdirs_utf8(self):
1139
1094
        tree = [
1140
1095
            '.bzr',
1207
1162
        self._save_platform_info()
1208
1163
        win32utils.winver = None # Avoid the win32 detection code
1209
1164
        osutils._fs_enc = 'UTF-8'
1210
 
        self.assertDirReaderIs(
1211
 
            UTF8DirReaderFeature.module.UTF8DirReader)
 
1165
        self.assertDirReaderIs(UTF8DirReaderFeature.reader)
1212
1166
 
1213
1167
    def test_force_walkdirs_utf8_fs_ascii(self):
1214
1168
        self.requireFeature(UTF8DirReaderFeature)
1215
1169
        self._save_platform_info()
1216
1170
        win32utils.winver = None # Avoid the win32 detection code
1217
1171
        osutils._fs_enc = 'US-ASCII'
1218
 
        self.assertDirReaderIs(
1219
 
            UTF8DirReaderFeature.module.UTF8DirReader)
 
1172
        self.assertDirReaderIs(UTF8DirReaderFeature.reader)
1220
1173
 
1221
1174
    def test_force_walkdirs_utf8_fs_ANSI(self):
1222
1175
        self.requireFeature(UTF8DirReaderFeature)
1223
1176
        self._save_platform_info()
1224
1177
        win32utils.winver = None # Avoid the win32 detection code
1225
1178
        osutils._fs_enc = 'ANSI_X3.4-1968'
1226
 
        self.assertDirReaderIs(
1227
 
            UTF8DirReaderFeature.module.UTF8DirReader)
 
1179
        self.assertDirReaderIs(UTF8DirReaderFeature.reader)
1228
1180
 
1229
1181
    def test_force_walkdirs_utf8_fs_latin1(self):
1230
1182
        self._save_platform_info()
1248
1200
 
1249
1201
    def test_unicode_walkdirs(self):
1250
1202
        """Walkdirs should always return unicode paths."""
1251
 
        self.requireFeature(features.UnicodeFilenameFeature)
 
1203
        self.requireFeature(tests.UnicodeFilenameFeature)
1252
1204
        name0 = u'0file-\xb6'
1253
1205
        name1 = u'1dir-\u062c\u0648'
1254
1206
        name2 = u'2file-\u0633'
1291
1243
 
1292
1244
        The abspath portion might be in unicode or utf-8
1293
1245
        """
1294
 
        self.requireFeature(features.UnicodeFilenameFeature)
 
1246
        self.requireFeature(tests.UnicodeFilenameFeature)
1295
1247
        name0 = u'0file-\xb6'
1296
1248
        name1 = u'1dir-\u062c\u0648'
1297
1249
        name2 = u'2file-\u0633'
1352
1304
 
1353
1305
        The abspath portion should be in unicode
1354
1306
        """
1355
 
        self.requireFeature(features.UnicodeFilenameFeature)
 
1307
        self.requireFeature(tests.UnicodeFilenameFeature)
1356
1308
        # Use the unicode reader. TODO: split into driver-and-driven unit
1357
1309
        # tests.
1358
1310
        self._save_platform_info()
1399
1351
 
1400
1352
    def test__walkdirs_utf8_win32readdir(self):
1401
1353
        self.requireFeature(test__walkdirs_win32.win32_readdir_feature)
1402
 
        self.requireFeature(features.UnicodeFilenameFeature)
 
1354
        self.requireFeature(tests.UnicodeFilenameFeature)
1403
1355
        from bzrlib._walkdirs_win32 import Win32ReadDir
1404
1356
        self._save_platform_info()
1405
1357
        osutils._selected_dir_reader = Win32ReadDir()
1456
1408
    def test__walkdirs_utf_win32_find_file_stat_file(self):
1457
1409
        """make sure our Stat values are valid"""
1458
1410
        self.requireFeature(test__walkdirs_win32.win32_readdir_feature)
1459
 
        self.requireFeature(features.UnicodeFilenameFeature)
 
1411
        self.requireFeature(tests.UnicodeFilenameFeature)
1460
1412
        from bzrlib._walkdirs_win32 import Win32ReadDir
1461
1413
        name0u = u'0file-\xb6'
1462
1414
        name0 = name0u.encode('utf8')
1480
1432
    def test__walkdirs_utf_win32_find_file_stat_directory(self):
1481
1433
        """make sure our Stat values are valid"""
1482
1434
        self.requireFeature(test__walkdirs_win32.win32_readdir_feature)
1483
 
        self.requireFeature(features.UnicodeFilenameFeature)
 
1435
        self.requireFeature(tests.UnicodeFilenameFeature)
1484
1436
        from bzrlib._walkdirs_win32 import Win32ReadDir
1485
1437
        name0u = u'0dir-\u062c\u0648'
1486
1438
        name0 = name0u.encode('utf8')
1586
1538
        self.assertEqual(['c'], os.listdir('target/b'))
1587
1539
 
1588
1540
    def test_copy_tree_symlinks(self):
1589
 
        self.requireFeature(features.SymlinkFeature)
 
1541
        self.requireFeature(tests.SymlinkFeature)
1590
1542
        self.build_tree(['source/'])
1591
1543
        os.symlink('a/generic/path', 'source/lnk')
1592
1544
        osutils.copy_tree('source', 'target')
1617
1569
                          ('d', 'source/b', 'target/b'),
1618
1570
                          ('f', 'source/b/c', 'target/b/c'),
1619
1571
                         ], processed_files)
1620
 
        self.assertPathDoesNotExist('target')
 
1572
        self.failIfExists('target')
1621
1573
        if osutils.has_symlinks():
1622
1574
            self.assertEqual([('source/lnk', 'target/lnk')], processed_links)
1623
1575
 
1669
1621
        old = osutils.set_or_unset_env('BZR_TEST_ENV_VAR', None)
1670
1622
        self.assertEqual('foo', old)
1671
1623
        self.assertEqual(None, os.environ.get('BZR_TEST_ENV_VAR'))
1672
 
        self.assertFalse('BZR_TEST_ENV_VAR' in os.environ)
 
1624
        self.failIf('BZR_TEST_ENV_VAR' in os.environ)
1673
1625
 
1674
1626
 
1675
1627
class TestSizeShaFile(tests.TestCaseInTempDir):
1726
1678
 
1727
1679
class TestReCompile(tests.TestCase):
1728
1680
 
1729
 
    def _deprecated_re_compile_checked(self, *args, **kwargs):
1730
 
        return self.applyDeprecated(symbol_versioning.deprecated_in((2, 2, 0)),
1731
 
            osutils.re_compile_checked, *args, **kwargs)
1732
 
 
1733
1681
    def test_re_compile_checked(self):
1734
 
        r = self._deprecated_re_compile_checked(r'A*', re.IGNORECASE)
 
1682
        r = osutils.re_compile_checked(r'A*', re.IGNORECASE)
1735
1683
        self.assertTrue(r.match('aaaa'))
1736
1684
        self.assertTrue(r.match('aAaA'))
1737
1685
 
1738
1686
    def test_re_compile_checked_error(self):
1739
1687
        # like https://bugs.launchpad.net/bzr/+bug/251352
1740
 
 
1741
 
        # Due to possible test isolation error, re.compile is not lazy at
1742
 
        # this point. We re-install lazy compile.
1743
 
        lazy_regex.install_lazy_compile()
1744
1688
        err = self.assertRaises(
1745
1689
            errors.BzrCommandError,
1746
 
            self._deprecated_re_compile_checked, '*', re.IGNORECASE, 'test case')
 
1690
            osutils.re_compile_checked, '*', re.IGNORECASE, 'test case')
1747
1691
        self.assertEqual(
1748
 
            'Invalid regular expression in test case: '
1749
 
            '"*" nothing to repeat',
 
1692
            "Invalid regular expression in test case: '*': "
 
1693
            "nothing to repeat",
1750
1694
            str(err))
1751
1695
 
1752
1696
 
1753
1697
class TestDirReader(tests.TestCaseInTempDir):
1754
1698
 
1755
 
    scenarios = dir_reader_scenarios()
1756
 
 
1757
1699
    # Set by load_tests
1758
1700
    _dir_reader_class = None
1759
1701
    _native_to_unicode = None
1859
1801
        return filtered_dirblocks
1860
1802
 
1861
1803
    def test_walk_unicode_tree(self):
1862
 
        self.requireFeature(features.UnicodeFilenameFeature)
 
1804
        self.requireFeature(tests.UnicodeFilenameFeature)
1863
1805
        tree, expected_dirblocks = self._get_unicode_tree()
1864
1806
        self.build_tree(tree)
1865
1807
        result = list(osutils._walkdirs_utf8('.'))
1866
1808
        self.assertEqual(expected_dirblocks, self._filter_out(result))
1867
1809
 
1868
1810
    def test_symlink(self):
1869
 
        self.requireFeature(features.SymlinkFeature)
1870
 
        self.requireFeature(features.UnicodeFilenameFeature)
 
1811
        self.requireFeature(tests.SymlinkFeature)
 
1812
        self.requireFeature(tests.UnicodeFilenameFeature)
1871
1813
        target = u'target\N{Euro Sign}'
1872
1814
        link_name = u'l\N{Euro Sign}nk'
1873
1815
        os.symlink(target, link_name)
1891
1833
    But prior python versions failed to properly encode the passed unicode
1892
1834
    string.
1893
1835
    """
1894
 
    _test_needs_features = [features.SymlinkFeature, features.UnicodeFilenameFeature]
 
1836
    _test_needs_features = [tests.SymlinkFeature, tests.UnicodeFilenameFeature]
1895
1837
 
1896
1838
    def setUp(self):
1897
1839
        super(tests.TestCaseInTempDir, self).setUp()
1900
1842
        os.symlink(self.target, self.link)
1901
1843
 
1902
1844
    def test_os_readlink_link_encoding(self):
1903
 
        self.assertEquals(self.target,  os.readlink(self.link))
 
1845
        if sys.version_info < (2, 6):
 
1846
            self.assertRaises(UnicodeEncodeError, os.readlink, self.link)
 
1847
        else:
 
1848
            self.assertEquals(self.target,  os.readlink(self.link))
1904
1849
 
1905
1850
    def test_os_readlink_link_decoding(self):
1906
1851
        self.assertEquals(self.target.encode(osutils._fs_enc),
1918
1863
        self.assertIsInstance(concurrency, int)
1919
1864
 
1920
1865
    def test_local_concurrency_environment_variable(self):
1921
 
        self.overrideEnv('BZR_CONCURRENCY', '2')
 
1866
        os.environ['BZR_CONCURRENCY'] = '2'
1922
1867
        self.assertEqual(2, osutils.local_concurrency(use_cache=False))
1923
 
        self.overrideEnv('BZR_CONCURRENCY', '3')
 
1868
        os.environ['BZR_CONCURRENCY'] = '3'
1924
1869
        self.assertEqual(3, osutils.local_concurrency(use_cache=False))
1925
 
        self.overrideEnv('BZR_CONCURRENCY', 'foo')
 
1870
        os.environ['BZR_CONCURRENCY'] = 'foo'
1926
1871
        self.assertEqual(1, osutils.local_concurrency(use_cache=False))
1927
1872
 
1928
1873
    def test_option_concurrency(self):
1929
 
        self.overrideEnv('BZR_CONCURRENCY', '1')
 
1874
        os.environ['BZR_CONCURRENCY'] = '1'
1930
1875
        self.run_bzr('rocks --concurrency 42')
1931
 
        # Command line overrides environment variable
 
1876
        # Command line overrides envrionment variable
1932
1877
        self.assertEquals('42', os.environ['BZR_CONCURRENCY'])
1933
1878
        self.assertEquals(42, osutils.local_concurrency(use_cache=False))
1934
1879
 
1972
1917
 
1973
1918
class TestTerminalWidth(tests.TestCase):
1974
1919
 
1975
 
    def setUp(self):
1976
 
        tests.TestCase.setUp(self)
1977
 
        self._orig_terminal_size_state = osutils._terminal_size_state
1978
 
        self._orig_first_terminal_size = osutils._first_terminal_size
1979
 
        self.addCleanup(self.restore_osutils_globals)
1980
 
        osutils._terminal_size_state = 'no_data'
1981
 
        osutils._first_terminal_size = None
1982
 
 
1983
 
    def restore_osutils_globals(self):
1984
 
        osutils._terminal_size_state = self._orig_terminal_size_state
1985
 
        osutils._first_terminal_size = self._orig_first_terminal_size
1986
 
 
1987
1920
    def replace_stdout(self, new):
1988
1921
        self.overrideAttr(sys, 'stdout', new)
1989
1922
 
2004
1937
    def test_defaults_to_BZR_COLUMNS(self):
2005
1938
        # BZR_COLUMNS is set by the test framework
2006
1939
        self.assertNotEqual('12', os.environ['BZR_COLUMNS'])
2007
 
        self.overrideEnv('BZR_COLUMNS', '12')
 
1940
        os.environ['BZR_COLUMNS'] = '12'
2008
1941
        self.assertEqual(12, osutils.terminal_width())
2009
1942
 
2010
 
    def test_BZR_COLUMNS_0_no_limit(self):
2011
 
        self.overrideEnv('BZR_COLUMNS', '0')
2012
 
        self.assertEqual(None, osutils.terminal_width())
2013
 
 
2014
1943
    def test_falls_back_to_COLUMNS(self):
2015
 
        self.overrideEnv('BZR_COLUMNS', None)
 
1944
        del os.environ['BZR_COLUMNS']
2016
1945
        self.assertNotEqual('42', os.environ['COLUMNS'])
2017
1946
        self.set_fake_tty()
2018
 
        self.overrideEnv('COLUMNS', '42')
 
1947
        os.environ['COLUMNS'] = '42'
2019
1948
        self.assertEqual(42, osutils.terminal_width())
2020
1949
 
2021
1950
    def test_tty_default_without_columns(self):
2022
 
        self.overrideEnv('BZR_COLUMNS', None)
2023
 
        self.overrideEnv('COLUMNS', None)
 
1951
        del os.environ['BZR_COLUMNS']
 
1952
        del os.environ['COLUMNS']
2024
1953
 
2025
1954
        def terminal_size(w, h):
2026
1955
            return 42, 42
2033
1962
        self.assertEqual(42, osutils.terminal_width())
2034
1963
 
2035
1964
    def test_non_tty_default_without_columns(self):
2036
 
        self.overrideEnv('BZR_COLUMNS', None)
2037
 
        self.overrideEnv('COLUMNS', None)
 
1965
        del os.environ['BZR_COLUMNS']
 
1966
        del os.environ['COLUMNS']
2038
1967
        self.replace_stdout(None)
2039
1968
        self.assertEqual(None, osutils.terminal_width())
2040
1969
 
2050
1979
        else:
2051
1980
            self.overrideAttr(termios, 'TIOCGWINSZ')
2052
1981
            del termios.TIOCGWINSZ
2053
 
        self.overrideEnv('BZR_COLUMNS', None)
2054
 
        self.overrideEnv('COLUMNS', None)
 
1982
        del os.environ['BZR_COLUMNS']
 
1983
        del os.environ['COLUMNS']
2055
1984
        # Whatever the result is, if we don't raise an exception, it's ok.
2056
1985
        osutils.terminal_width()
2057
1986
 
2058
 
 
2059
1987
class TestCreationOps(tests.TestCaseInTempDir):
2060
1988
    _test_needs_features = [features.chown_feature]
2061
1989
 
2091
2019
        self.assertEquals(self.uid, s.st_uid)
2092
2020
        self.assertEquals(self.gid, s.st_gid)
2093
2021
 
2094
 
 
2095
2022
class TestGetuserUnicode(tests.TestCase):
2096
2023
 
2097
2024
    def test_ascii_user(self):
2098
 
        self.overrideEnv('LOGNAME', 'jrandom')
 
2025
        osutils.set_or_unset_env('LOGNAME', 'jrandom')
2099
2026
        self.assertEqual(u'jrandom', osutils.getuser_unicode())
2100
2027
 
2101
2028
    def test_unicode_user(self):
2102
2029
        ue = osutils.get_user_encoding()
2103
 
        uni_val, env_val = tests.probe_unicode_in_user_encoding()
2104
 
        if uni_val is None:
2105
 
            raise tests.TestSkipped(
2106
 
                'Cannot find a unicode character that works in encoding %s'
2107
 
                % (osutils.get_user_encoding(),))
2108
 
        uni_username = u'jrandom' + uni_val
2109
 
        encoded_username = uni_username.encode(ue)
2110
 
        self.overrideEnv('LOGNAME', encoded_username)
2111
 
        self.assertEqual(uni_username, osutils.getuser_unicode())
2112
 
        self.overrideEnv('LOGNAME', u'jrandom\xb6'.encode(ue))
 
2030
        osutils.set_or_unset_env('LOGNAME', u'jrandom\xb6'.encode(ue))
2113
2031
        self.assertEqual(u'jrandom\xb6', osutils.getuser_unicode())
2114
 
 
2115
 
    def test_no_username_bug_660174(self):
2116
 
        self.requireFeature(features.win32_feature)
2117
 
        for name in ('LOGNAME', 'USER', 'LNAME', 'USERNAME'):
2118
 
            self.overrideEnv(name, None)
2119
 
        self.assertEqual(u'UNKNOWN', osutils.getuser_unicode())
2120
 
 
2121
 
 
2122
 
class TestBackupNames(tests.TestCase):
2123
 
 
2124
 
    def setUp(self):
2125
 
        super(TestBackupNames, self).setUp()
2126
 
        self.backups = []
2127
 
 
2128
 
    def backup_exists(self, name):
2129
 
        return name in self.backups
2130
 
 
2131
 
    def available_backup_name(self, name):
2132
 
        backup_name = osutils.available_backup_name(name, self.backup_exists)
2133
 
        self.backups.append(backup_name)
2134
 
        return backup_name
2135
 
 
2136
 
    def assertBackupName(self, expected, name):
2137
 
        self.assertEqual(expected, self.available_backup_name(name))
2138
 
 
2139
 
    def test_empty(self):
2140
 
        self.assertBackupName('file.~1~', 'file')
2141
 
 
2142
 
    def test_existing(self):
2143
 
        self.available_backup_name('file')
2144
 
        self.available_backup_name('file')
2145
 
        self.assertBackupName('file.~3~', 'file')
2146
 
        # Empty slots are found, this is not a strict requirement and may be
2147
 
        # revisited if we test against all implementations.
2148
 
        self.backups.remove('file.~2~')
2149
 
        self.assertBackupName('file.~2~', 'file')
2150
 
 
2151
 
 
2152
 
class TestFindExecutableInPath(tests.TestCase):
2153
 
 
2154
 
    def test_windows(self):
2155
 
        if sys.platform != 'win32':
2156
 
            raise tests.TestSkipped('test requires win32')
2157
 
        self.assertTrue(osutils.find_executable_on_path('explorer') is not None)
2158
 
        self.assertTrue(
2159
 
            osutils.find_executable_on_path('explorer.exe') is not None)
2160
 
        self.assertTrue(
2161
 
            osutils.find_executable_on_path('EXPLORER.EXE') is not None)
2162
 
        self.assertTrue(
2163
 
            osutils.find_executable_on_path('THIS SHOULD NOT EXIST') is None)
2164
 
        self.assertTrue(osutils.find_executable_on_path('file.txt') is None)
2165
 
 
2166
 
    def test_other(self):
2167
 
        if sys.platform == 'win32':
2168
 
            raise tests.TestSkipped('test requires non-win32')
2169
 
        self.assertTrue(osutils.find_executable_on_path('sh') is not None)
2170
 
        self.assertTrue(
2171
 
            osutils.find_executable_on_path('THIS SHOULD NOT EXIST') is None)