← 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: Jelmer Vernooij
  • Date: 2011-12-05 14:12:23 UTC
  • mto: This revision was merged to the branch mainline in revision 6348.
  • Revision ID: jelmer@samba.org-20111205141223-8qxae4h37satlzgq
Move more functionality to vf_search.

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-2011 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
25
24
import sys
26
25
import time
27
26
 
28
27
from bzrlib import (
29
28
    errors,
 
29
    lazy_regex,
30
30
    osutils,
 
31
    symbol_versioning,
31
32
    tests,
32
33
    trace,
33
34
    win32utils,
37
38
    file_utils,
38
39
    test__walkdirs_win32,
39
40
    )
40
 
 
41
 
 
42
 
class _UTF8DirReaderFeature(tests.Feature):
 
41
from bzrlib.tests.scenarios import load_tests_apply_scenarios
 
42
 
 
43
 
 
44
class _UTF8DirReaderFeature(features.Feature):
43
45
 
44
46
    def _probe(self):
45
47
        try:
52
54
    def feature_name(self):
53
55
        return 'bzrlib._readdir_pyx'
54
56
 
55
 
UTF8DirReaderFeature = _UTF8DirReaderFeature()
 
57
UTF8DirReaderFeature = features.ModuleAvailableFeature('bzrlib._readdir_pyx')
56
58
 
57
 
term_ios_feature = tests.ModuleAvailableFeature('termios')
 
59
term_ios_feature = features.ModuleAvailableFeature('termios')
58
60
 
59
61
 
60
62
def _already_unicode(s):
95
97
    return scenarios
96
98
 
97
99
 
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
 
100
load_tests = load_tests_apply_scenarios
105
101
 
106
102
 
107
103
class TestContainsWhitespace(tests.TestCase):
108
104
 
109
105
    def test_contains_whitespace(self):
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'))
 
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'))
116
112
 
117
113
        # \xa0 is "Non-breaking-space" which on some python locales thinks it
118
114
        # is whitespace, but we do not.
119
 
        self.failIf(osutils.contains_whitespace(u''))
120
 
        self.failIf(osutils.contains_whitespace(u'hellothere'))
121
 
        self.failIf(osutils.contains_whitespace(u'hello\xa0there'))
 
115
        self.assertFalse(osutils.contains_whitespace(u''))
 
116
        self.assertFalse(osutils.contains_whitespace(u'hellothere'))
 
117
        self.assertFalse(osutils.contains_whitespace(u'hello\xa0there'))
122
118
 
123
119
 
124
120
class TestRename(tests.TestCaseInTempDir):
138
134
        # This should work everywhere
139
135
        self.create_file('a', 'something in a\n')
140
136
        self._fancy_rename('a', 'b')
141
 
        self.failIfExists('a')
142
 
        self.failUnlessExists('b')
 
137
        self.assertPathDoesNotExist('a')
 
138
        self.assertPathExists('b')
143
139
        self.check_file_contents('b', 'something in a\n')
144
140
 
145
141
        self.create_file('a', 'new something in a\n')
152
148
        self.create_file('target', 'data in target\n')
153
149
        self.assertRaises((IOError, OSError), self._fancy_rename,
154
150
                          'missingsource', 'target')
155
 
        self.failUnlessExists('target')
 
151
        self.assertPathExists('target')
156
152
        self.check_file_contents('target', 'data in target\n')
157
153
 
158
154
    def test_fancy_rename_fails_if_source_and_target_missing(self):
163
159
        # Rename should be semi-atomic on all platforms
164
160
        self.create_file('a', 'something in a\n')
165
161
        osutils.rename('a', 'b')
166
 
        self.failIfExists('a')
167
 
        self.failUnlessExists('b')
 
162
        self.assertPathDoesNotExist('a')
 
163
        self.assertPathExists('b')
168
164
        self.check_file_contents('b', 'something in a\n')
169
165
 
170
166
        self.create_file('a', 'new something in a\n')
184
180
        shape = sorted(os.listdir('.'))
185
181
        self.assertEquals(['A', 'B'], shape)
186
182
 
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
183
 
195
184
class TestRandChars(tests.TestCase):
196
185
 
242
231
            self.assertFalse(osutils.is_inside_or_parent_of_any(dirs, fn))
243
232
 
244
233
 
 
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
 
245
253
class TestRmTree(tests.TestCaseInTempDir):
246
254
 
247
255
    def test_rmtree(self):
259
267
 
260
268
        osutils.rmtree('dir')
261
269
 
262
 
        self.failIfExists('dir/file')
263
 
        self.failIfExists('dir')
 
270
        self.assertPathDoesNotExist('dir/file')
 
271
        self.assertPathDoesNotExist('dir')
264
272
 
265
273
 
266
274
class TestDeleteAny(tests.TestCaseInTempDir):
421
429
class TestLinks(tests.TestCaseInTempDir):
422
430
 
423
431
    def test_dereference_path(self):
424
 
        self.requireFeature(tests.SymlinkFeature)
 
432
        self.requireFeature(features.SymlinkFeature)
425
433
        cwd = osutils.realpath('.')
426
434
        os.mkdir('bar')
427
435
        bar_path = osutils.pathjoin(cwd, 'bar')
474
482
 
475
483
class TestCanonicalRelPath(tests.TestCaseInTempDir):
476
484
 
477
 
    _test_needs_features = [tests.CaseInsCasePresFilenameFeature]
 
485
    _test_needs_features = [features.CaseInsCasePresFilenameFeature]
478
486
 
479
487
    def test_canonical_relpath_simple(self):
480
488
        f = file('MixedCaseName', 'w')
481
489
        f.close()
482
490
        actual = osutils.canonical_relpath(self.test_base_dir, 'mixedcasename')
483
 
        self.failUnlessEqual('work/MixedCaseName', actual)
 
491
        self.assertEqual('work/MixedCaseName', actual)
484
492
 
485
493
    def test_canonical_relpath_missing_tail(self):
486
494
        os.mkdir('MixedCaseParent')
487
495
        actual = osutils.canonical_relpath(self.test_base_dir,
488
496
                                           'mixedcaseparent/nochild')
489
 
        self.failUnlessEqual('work/MixedCaseParent/nochild', actual)
 
497
        self.assertEqual('work/MixedCaseParent/nochild', actual)
490
498
 
491
499
 
492
500
class Test_CICPCanonicalRelpath(tests.TestCaseWithTransport):
810
818
        self.assertEqual(None, osutils.safe_file_id(None))
811
819
 
812
820
 
 
821
class TestPosixFuncs(tests.TestCase):
 
822
    """Test that the posix version of normpath returns an appropriate path
 
823
       when used with 2 leading slashes."""
 
824
 
 
825
    def test_normpath(self):
 
826
        self.assertEqual('/etc/shadow', osutils._posix_normpath('/etc/shadow'))
 
827
        self.assertEqual('/etc/shadow', osutils._posix_normpath('//etc/shadow'))
 
828
        self.assertEqual('/etc/shadow', osutils._posix_normpath('///etc/shadow'))
 
829
 
 
830
 
813
831
class TestWin32Funcs(tests.TestCase):
814
832
    """Test that _win32 versions of os utilities return appropriate paths."""
815
833
 
868
886
        self.assertEqual('//HOST/path', osutils._win98_abspath('//HOST/path'))
869
887
        # relative path
870
888
        cwd = osutils.getcwd().rstrip('/')
871
 
        drive = osutils._nt_splitdrive(cwd)[0]
 
889
        drive = osutils.ntpath.splitdrive(cwd)[0]
872
890
        self.assertEqual(cwd+'/path', osutils._win98_abspath('path'))
873
891
        self.assertEqual(drive+'/path', osutils._win98_abspath('/path'))
874
892
        # unicode path
880
898
    """Test win32 functions that create files."""
881
899
 
882
900
    def test_getcwd(self):
883
 
        self.requireFeature(tests.UnicodeFilenameFeature)
 
901
        self.requireFeature(features.UnicodeFilenameFeature)
884
902
        os.mkdir(u'mu-\xb5')
885
903
        os.chdir(u'mu-\xb5')
886
904
        # TODO: jam 20060427 This will probably fail on Mac OSX because
916
934
        b.close()
917
935
 
918
936
        osutils._win32_rename('b', 'a')
919
 
        self.failUnlessExists('a')
920
 
        self.failIfExists('b')
 
937
        self.assertPathExists('a')
 
938
        self.assertPathDoesNotExist('b')
921
939
        self.assertFileEqual('baz\n', 'a')
922
940
 
923
941
    def test_rename_missing_file(self):
976
994
    """Test mac special functions that require directories."""
977
995
 
978
996
    def test_getcwd(self):
979
 
        self.requireFeature(tests.UnicodeFilenameFeature)
 
997
        self.requireFeature(features.UnicodeFilenameFeature)
980
998
        os.mkdir(u'B\xe5gfors')
981
999
        os.chdir(u'B\xe5gfors')
982
1000
        self.assertEndsWith(osutils._mac_getcwd(), u'B\xe5gfors')
983
1001
 
984
1002
    def test_getcwd_nonnorm(self):
985
 
        self.requireFeature(tests.UnicodeFilenameFeature)
 
1003
        self.requireFeature(features.UnicodeFilenameFeature)
986
1004
        # Test that _mac_getcwd() will normalize this path
987
1005
        os.mkdir(u'Ba\u030agfors')
988
1006
        os.chdir(u'Ba\u030agfors')
1071
1089
        self.assertExpectedBlocks(expected_dirblocks[1:], result)
1072
1090
 
1073
1091
    def test_walkdirs_os_error(self):
1074
 
        # <https://bugs.edge.launchpad.net/bzr/+bug/338653>
 
1092
        # <https://bugs.launchpad.net/bzr/+bug/338653>
1075
1093
        # Pyrex readdir didn't raise useful messages if it had an error
1076
1094
        # reading the directory
1077
1095
        if sys.platform == 'win32':
1078
1096
            raise tests.TestNotApplicable(
1079
1097
                "readdir IOError not tested on win32")
 
1098
        self.requireFeature(features.not_running_as_root)
1080
1099
        os.mkdir("test-unreadable")
1081
1100
        os.chmod("test-unreadable", 0000)
1082
1101
        # must chmod it back so that it can be removed
1090
1109
        # Ensure the message contains the file name
1091
1110
        self.assertContainsRe(str(e), "\./test-unreadable")
1092
1111
 
 
1112
 
 
1113
    def test_walkdirs_encoding_error(self):
 
1114
        # <https://bugs.launchpad.net/bzr/+bug/488519>
 
1115
        # walkdirs didn't raise a useful message when the filenames
 
1116
        # are not using the filesystem's encoding
 
1117
 
 
1118
        # require a bytestring based filesystem
 
1119
        self.requireFeature(features.ByteStringNamedFilesystem)
 
1120
 
 
1121
        tree = [
 
1122
            '.bzr',
 
1123
            '0file',
 
1124
            '1dir/',
 
1125
            '1dir/0file',
 
1126
            '1dir/1dir/',
 
1127
            '1file'
 
1128
            ]
 
1129
 
 
1130
        self.build_tree(tree)
 
1131
 
 
1132
        # rename the 1file to a latin-1 filename
 
1133
        os.rename("./1file", "\xe8file")
 
1134
        if "\xe8file" not in os.listdir("."):
 
1135
            self.skip("Lack filesystem that preserves arbitrary bytes")
 
1136
 
 
1137
        self._save_platform_info()
 
1138
        win32utils.winver = None # Avoid the win32 detection code
 
1139
        osutils._fs_enc = 'UTF-8'
 
1140
 
 
1141
        # this should raise on error
 
1142
        def attempt():
 
1143
            for dirdetail, dirblock in osutils.walkdirs('.'):
 
1144
                pass
 
1145
 
 
1146
        self.assertRaises(errors.BadFilenameEncoding, attempt)
 
1147
 
1093
1148
    def test__walkdirs_utf8(self):
1094
1149
        tree = [
1095
1150
            '.bzr',
1162
1217
        self._save_platform_info()
1163
1218
        win32utils.winver = None # Avoid the win32 detection code
1164
1219
        osutils._fs_enc = 'UTF-8'
1165
 
        self.assertDirReaderIs(UTF8DirReaderFeature.reader)
 
1220
        self.assertDirReaderIs(
 
1221
            UTF8DirReaderFeature.module.UTF8DirReader)
1166
1222
 
1167
1223
    def test_force_walkdirs_utf8_fs_ascii(self):
1168
1224
        self.requireFeature(UTF8DirReaderFeature)
1169
1225
        self._save_platform_info()
1170
1226
        win32utils.winver = None # Avoid the win32 detection code
1171
1227
        osutils._fs_enc = 'US-ASCII'
1172
 
        self.assertDirReaderIs(UTF8DirReaderFeature.reader)
 
1228
        self.assertDirReaderIs(
 
1229
            UTF8DirReaderFeature.module.UTF8DirReader)
1173
1230
 
1174
1231
    def test_force_walkdirs_utf8_fs_ANSI(self):
1175
1232
        self.requireFeature(UTF8DirReaderFeature)
1176
1233
        self._save_platform_info()
1177
1234
        win32utils.winver = None # Avoid the win32 detection code
1178
1235
        osutils._fs_enc = 'ANSI_X3.4-1968'
1179
 
        self.assertDirReaderIs(UTF8DirReaderFeature.reader)
 
1236
        self.assertDirReaderIs(
 
1237
            UTF8DirReaderFeature.module.UTF8DirReader)
1180
1238
 
1181
1239
    def test_force_walkdirs_utf8_fs_latin1(self):
1182
1240
        self._save_platform_info()
1200
1258
 
1201
1259
    def test_unicode_walkdirs(self):
1202
1260
        """Walkdirs should always return unicode paths."""
1203
 
        self.requireFeature(tests.UnicodeFilenameFeature)
 
1261
        self.requireFeature(features.UnicodeFilenameFeature)
1204
1262
        name0 = u'0file-\xb6'
1205
1263
        name1 = u'1dir-\u062c\u0648'
1206
1264
        name2 = u'2file-\u0633'
1243
1301
 
1244
1302
        The abspath portion might be in unicode or utf-8
1245
1303
        """
1246
 
        self.requireFeature(tests.UnicodeFilenameFeature)
 
1304
        self.requireFeature(features.UnicodeFilenameFeature)
1247
1305
        name0 = u'0file-\xb6'
1248
1306
        name1 = u'1dir-\u062c\u0648'
1249
1307
        name2 = u'2file-\u0633'
1304
1362
 
1305
1363
        The abspath portion should be in unicode
1306
1364
        """
1307
 
        self.requireFeature(tests.UnicodeFilenameFeature)
 
1365
        self.requireFeature(features.UnicodeFilenameFeature)
1308
1366
        # Use the unicode reader. TODO: split into driver-and-driven unit
1309
1367
        # tests.
1310
1368
        self._save_platform_info()
1351
1409
 
1352
1410
    def test__walkdirs_utf8_win32readdir(self):
1353
1411
        self.requireFeature(test__walkdirs_win32.win32_readdir_feature)
1354
 
        self.requireFeature(tests.UnicodeFilenameFeature)
 
1412
        self.requireFeature(features.UnicodeFilenameFeature)
1355
1413
        from bzrlib._walkdirs_win32 import Win32ReadDir
1356
1414
        self._save_platform_info()
1357
1415
        osutils._selected_dir_reader = Win32ReadDir()
1408
1466
    def test__walkdirs_utf_win32_find_file_stat_file(self):
1409
1467
        """make sure our Stat values are valid"""
1410
1468
        self.requireFeature(test__walkdirs_win32.win32_readdir_feature)
1411
 
        self.requireFeature(tests.UnicodeFilenameFeature)
 
1469
        self.requireFeature(features.UnicodeFilenameFeature)
1412
1470
        from bzrlib._walkdirs_win32 import Win32ReadDir
1413
1471
        name0u = u'0file-\xb6'
1414
1472
        name0 = name0u.encode('utf8')
1432
1490
    def test__walkdirs_utf_win32_find_file_stat_directory(self):
1433
1491
        """make sure our Stat values are valid"""
1434
1492
        self.requireFeature(test__walkdirs_win32.win32_readdir_feature)
1435
 
        self.requireFeature(tests.UnicodeFilenameFeature)
 
1493
        self.requireFeature(features.UnicodeFilenameFeature)
1436
1494
        from bzrlib._walkdirs_win32 import Win32ReadDir
1437
1495
        name0u = u'0dir-\u062c\u0648'
1438
1496
        name0 = name0u.encode('utf8')
1538
1596
        self.assertEqual(['c'], os.listdir('target/b'))
1539
1597
 
1540
1598
    def test_copy_tree_symlinks(self):
1541
 
        self.requireFeature(tests.SymlinkFeature)
 
1599
        self.requireFeature(features.SymlinkFeature)
1542
1600
        self.build_tree(['source/'])
1543
1601
        os.symlink('a/generic/path', 'source/lnk')
1544
1602
        osutils.copy_tree('source', 'target')
1569
1627
                          ('d', 'source/b', 'target/b'),
1570
1628
                          ('f', 'source/b/c', 'target/b/c'),
1571
1629
                         ], processed_files)
1572
 
        self.failIfExists('target')
 
1630
        self.assertPathDoesNotExist('target')
1573
1631
        if osutils.has_symlinks():
1574
1632
            self.assertEqual([('source/lnk', 'target/lnk')], processed_links)
1575
1633
 
1621
1679
        old = osutils.set_or_unset_env('BZR_TEST_ENV_VAR', None)
1622
1680
        self.assertEqual('foo', old)
1623
1681
        self.assertEqual(None, os.environ.get('BZR_TEST_ENV_VAR'))
1624
 
        self.failIf('BZR_TEST_ENV_VAR' in os.environ)
 
1682
        self.assertFalse('BZR_TEST_ENV_VAR' in os.environ)
1625
1683
 
1626
1684
 
1627
1685
class TestSizeShaFile(tests.TestCaseInTempDir):
1678
1736
 
1679
1737
class TestReCompile(tests.TestCase):
1680
1738
 
 
1739
    def _deprecated_re_compile_checked(self, *args, **kwargs):
 
1740
        return self.applyDeprecated(symbol_versioning.deprecated_in((2, 2, 0)),
 
1741
            osutils.re_compile_checked, *args, **kwargs)
 
1742
 
1681
1743
    def test_re_compile_checked(self):
1682
 
        r = osutils.re_compile_checked(r'A*', re.IGNORECASE)
 
1744
        r = self._deprecated_re_compile_checked(r'A*', re.IGNORECASE)
1683
1745
        self.assertTrue(r.match('aaaa'))
1684
1746
        self.assertTrue(r.match('aAaA'))
1685
1747
 
1686
1748
    def test_re_compile_checked_error(self):
1687
1749
        # like https://bugs.launchpad.net/bzr/+bug/251352
 
1750
 
 
1751
        # Due to possible test isolation error, re.compile is not lazy at
 
1752
        # this point. We re-install lazy compile.
 
1753
        lazy_regex.install_lazy_compile()
1688
1754
        err = self.assertRaises(
1689
1755
            errors.BzrCommandError,
1690
 
            osutils.re_compile_checked, '*', re.IGNORECASE, 'test case')
 
1756
            self._deprecated_re_compile_checked, '*', re.IGNORECASE, 'test case')
1691
1757
        self.assertEqual(
1692
 
            "Invalid regular expression in test case: '*': "
1693
 
            "nothing to repeat",
 
1758
            'Invalid regular expression in test case: '
 
1759
            '"*" nothing to repeat',
1694
1760
            str(err))
1695
1761
 
1696
1762
 
1697
1763
class TestDirReader(tests.TestCaseInTempDir):
1698
1764
 
 
1765
    scenarios = dir_reader_scenarios()
 
1766
 
1699
1767
    # Set by load_tests
1700
1768
    _dir_reader_class = None
1701
1769
    _native_to_unicode = None
1801
1869
        return filtered_dirblocks
1802
1870
 
1803
1871
    def test_walk_unicode_tree(self):
1804
 
        self.requireFeature(tests.UnicodeFilenameFeature)
 
1872
        self.requireFeature(features.UnicodeFilenameFeature)
1805
1873
        tree, expected_dirblocks = self._get_unicode_tree()
1806
1874
        self.build_tree(tree)
1807
1875
        result = list(osutils._walkdirs_utf8('.'))
1808
1876
        self.assertEqual(expected_dirblocks, self._filter_out(result))
1809
1877
 
1810
1878
    def test_symlink(self):
1811
 
        self.requireFeature(tests.SymlinkFeature)
1812
 
        self.requireFeature(tests.UnicodeFilenameFeature)
 
1879
        self.requireFeature(features.SymlinkFeature)
 
1880
        self.requireFeature(features.UnicodeFilenameFeature)
1813
1881
        target = u'target\N{Euro Sign}'
1814
1882
        link_name = u'l\N{Euro Sign}nk'
1815
1883
        os.symlink(target, link_name)
1833
1901
    But prior python versions failed to properly encode the passed unicode
1834
1902
    string.
1835
1903
    """
1836
 
    _test_needs_features = [tests.SymlinkFeature, tests.UnicodeFilenameFeature]
 
1904
    _test_needs_features = [features.SymlinkFeature, features.UnicodeFilenameFeature]
1837
1905
 
1838
1906
    def setUp(self):
1839
1907
        super(tests.TestCaseInTempDir, self).setUp()
1842
1910
        os.symlink(self.target, self.link)
1843
1911
 
1844
1912
    def test_os_readlink_link_encoding(self):
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))
 
1913
        self.assertEquals(self.target,  os.readlink(self.link))
1849
1914
 
1850
1915
    def test_os_readlink_link_decoding(self):
1851
1916
        self.assertEquals(self.target.encode(osutils._fs_enc),
1863
1928
        self.assertIsInstance(concurrency, int)
1864
1929
 
1865
1930
    def test_local_concurrency_environment_variable(self):
1866
 
        os.environ['BZR_CONCURRENCY'] = '2'
 
1931
        self.overrideEnv('BZR_CONCURRENCY', '2')
1867
1932
        self.assertEqual(2, osutils.local_concurrency(use_cache=False))
1868
 
        os.environ['BZR_CONCURRENCY'] = '3'
 
1933
        self.overrideEnv('BZR_CONCURRENCY', '3')
1869
1934
        self.assertEqual(3, osutils.local_concurrency(use_cache=False))
1870
 
        os.environ['BZR_CONCURRENCY'] = 'foo'
 
1935
        self.overrideEnv('BZR_CONCURRENCY', 'foo')
1871
1936
        self.assertEqual(1, osutils.local_concurrency(use_cache=False))
1872
1937
 
1873
1938
    def test_option_concurrency(self):
1874
 
        os.environ['BZR_CONCURRENCY'] = '1'
 
1939
        self.overrideEnv('BZR_CONCURRENCY', '1')
1875
1940
        self.run_bzr('rocks --concurrency 42')
1876
 
        # Command line overrides envrionment variable
 
1941
        # Command line overrides environment variable
1877
1942
        self.assertEquals('42', os.environ['BZR_CONCURRENCY'])
1878
1943
        self.assertEquals(42, osutils.local_concurrency(use_cache=False))
1879
1944
 
1917
1982
 
1918
1983
class TestTerminalWidth(tests.TestCase):
1919
1984
 
 
1985
    def setUp(self):
 
1986
        tests.TestCase.setUp(self)
 
1987
        self._orig_terminal_size_state = osutils._terminal_size_state
 
1988
        self._orig_first_terminal_size = osutils._first_terminal_size
 
1989
        self.addCleanup(self.restore_osutils_globals)
 
1990
        osutils._terminal_size_state = 'no_data'
 
1991
        osutils._first_terminal_size = None
 
1992
 
 
1993
    def restore_osutils_globals(self):
 
1994
        osutils._terminal_size_state = self._orig_terminal_size_state
 
1995
        osutils._first_terminal_size = self._orig_first_terminal_size
 
1996
 
1920
1997
    def replace_stdout(self, new):
1921
1998
        self.overrideAttr(sys, 'stdout', new)
1922
1999
 
1937
2014
    def test_defaults_to_BZR_COLUMNS(self):
1938
2015
        # BZR_COLUMNS is set by the test framework
1939
2016
        self.assertNotEqual('12', os.environ['BZR_COLUMNS'])
1940
 
        os.environ['BZR_COLUMNS'] = '12'
 
2017
        self.overrideEnv('BZR_COLUMNS', '12')
1941
2018
        self.assertEqual(12, osutils.terminal_width())
1942
2019
 
 
2020
    def test_BZR_COLUMNS_0_no_limit(self):
 
2021
        self.overrideEnv('BZR_COLUMNS', '0')
 
2022
        self.assertEqual(None, osutils.terminal_width())
 
2023
 
1943
2024
    def test_falls_back_to_COLUMNS(self):
1944
 
        del os.environ['BZR_COLUMNS']
 
2025
        self.overrideEnv('BZR_COLUMNS', None)
1945
2026
        self.assertNotEqual('42', os.environ['COLUMNS'])
1946
2027
        self.set_fake_tty()
1947
 
        os.environ['COLUMNS'] = '42'
 
2028
        self.overrideEnv('COLUMNS', '42')
1948
2029
        self.assertEqual(42, osutils.terminal_width())
1949
2030
 
1950
2031
    def test_tty_default_without_columns(self):
1951
 
        del os.environ['BZR_COLUMNS']
1952
 
        del os.environ['COLUMNS']
 
2032
        self.overrideEnv('BZR_COLUMNS', None)
 
2033
        self.overrideEnv('COLUMNS', None)
1953
2034
 
1954
2035
        def terminal_size(w, h):
1955
2036
            return 42, 42
1962
2043
        self.assertEqual(42, osutils.terminal_width())
1963
2044
 
1964
2045
    def test_non_tty_default_without_columns(self):
1965
 
        del os.environ['BZR_COLUMNS']
1966
 
        del os.environ['COLUMNS']
 
2046
        self.overrideEnv('BZR_COLUMNS', None)
 
2047
        self.overrideEnv('COLUMNS', None)
1967
2048
        self.replace_stdout(None)
1968
2049
        self.assertEqual(None, osutils.terminal_width())
1969
2050
 
1979
2060
        else:
1980
2061
            self.overrideAttr(termios, 'TIOCGWINSZ')
1981
2062
            del termios.TIOCGWINSZ
1982
 
        del os.environ['BZR_COLUMNS']
1983
 
        del os.environ['COLUMNS']
 
2063
        self.overrideEnv('BZR_COLUMNS', None)
 
2064
        self.overrideEnv('COLUMNS', None)
1984
2065
        # Whatever the result is, if we don't raise an exception, it's ok.
1985
2066
        osutils.terminal_width()
1986
2067
 
 
2068
 
1987
2069
class TestCreationOps(tests.TestCaseInTempDir):
1988
2070
    _test_needs_features = [features.chown_feature]
1989
2071
 
2018
2100
        self.assertEquals(self.path, 'test_file')
2019
2101
        self.assertEquals(self.uid, s.st_uid)
2020
2102
        self.assertEquals(self.gid, s.st_gid)
 
2103
 
 
2104
 
 
2105
class TestGetuserUnicode(tests.TestCase):
 
2106
 
 
2107
    def test_ascii_user(self):
 
2108
        self.overrideEnv('LOGNAME', 'jrandom')
 
2109
        self.assertEqual(u'jrandom', osutils.getuser_unicode())
 
2110
 
 
2111
    def test_unicode_user(self):
 
2112
        ue = osutils.get_user_encoding()
 
2113
        uni_val, env_val = tests.probe_unicode_in_user_encoding()
 
2114
        if uni_val is None:
 
2115
            raise tests.TestSkipped(
 
2116
                'Cannot find a unicode character that works in encoding %s'
 
2117
                % (osutils.get_user_encoding(),))
 
2118
        uni_username = u'jrandom' + uni_val
 
2119
        encoded_username = uni_username.encode(ue)
 
2120
        self.overrideEnv('LOGNAME', encoded_username)
 
2121
        self.assertEqual(uni_username, osutils.getuser_unicode())
 
2122
        self.overrideEnv('LOGNAME', u'jrandom\xb6'.encode(ue))
 
2123
        self.assertEqual(u'jrandom\xb6', osutils.getuser_unicode())
 
2124
 
 
2125
    def test_no_username_bug_660174(self):
 
2126
        self.requireFeature(features.win32_feature)
 
2127
        for name in ('LOGNAME', 'USER', 'LNAME', 'USERNAME'):
 
2128
            self.overrideEnv(name, None)
 
2129
        self.assertEqual(u'UNKNOWN', osutils.getuser_unicode())
 
2130
 
 
2131
 
 
2132
class TestBackupNames(tests.TestCase):
 
2133
 
 
2134
    def setUp(self):
 
2135
        super(TestBackupNames, self).setUp()
 
2136
        self.backups = []
 
2137
 
 
2138
    def backup_exists(self, name):
 
2139
        return name in self.backups
 
2140
 
 
2141
    def available_backup_name(self, name):
 
2142
        backup_name = osutils.available_backup_name(name, self.backup_exists)
 
2143
        self.backups.append(backup_name)
 
2144
        return backup_name
 
2145
 
 
2146
    def assertBackupName(self, expected, name):
 
2147
        self.assertEqual(expected, self.available_backup_name(name))
 
2148
 
 
2149
    def test_empty(self):
 
2150
        self.assertBackupName('file.~1~', 'file')
 
2151
 
 
2152
    def test_existing(self):
 
2153
        self.available_backup_name('file')
 
2154
        self.available_backup_name('file')
 
2155
        self.assertBackupName('file.~3~', 'file')
 
2156
        # Empty slots are found, this is not a strict requirement and may be
 
2157
        # revisited if we test against all implementations.
 
2158
        self.backups.remove('file.~2~')
 
2159
        self.assertBackupName('file.~2~', 'file')
 
2160
 
 
2161
 
 
2162
class TestFindExecutableInPath(tests.TestCase):
 
2163
 
 
2164
    def test_windows(self):
 
2165
        if sys.platform != 'win32':
 
2166
            raise tests.TestSkipped('test requires win32')
 
2167
        self.assertTrue(osutils.find_executable_on_path('explorer') is not None)
 
2168
        self.assertTrue(
 
2169
            osutils.find_executable_on_path('explorer.exe') is not None)
 
2170
        self.assertTrue(
 
2171
            osutils.find_executable_on_path('EXPLORER.EXE') is not None)
 
2172
        self.assertTrue(
 
2173
            osutils.find_executable_on_path('THIS SHOULD NOT EXIST') is None)
 
2174
        self.assertTrue(osutils.find_executable_on_path('file.txt') is None)
 
2175
 
 
2176
    def test_other(self):
 
2177
        if sys.platform == 'win32':
 
2178
            raise tests.TestSkipped('test requires non-win32')
 
2179
        self.assertTrue(osutils.find_executable_on_path('sh') is not None)
 
2180
        self.assertTrue(
 
2181
            osutils.find_executable_on_path('THIS SHOULD NOT EXIST') is None)