← 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-14 12:15:44 UTC
  • mto: This revision was merged to the branch mainline in revision 6365.
  • Revision ID: jelmer@samba.org-20111214121544-v07cbvmi30re6q7w
s/NoVfsCalls/ContainsNoVfsCalls/.

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
20
20
import errno
21
21
import os
22
22
import re
 
23
import select
23
24
import socket
24
 
import stat
25
25
import sys
26
26
import time
27
27
 
28
28
from bzrlib import (
29
29
    errors,
 
30
    lazy_regex,
30
31
    osutils,
 
32
    symbol_versioning,
31
33
    tests,
32
34
    trace,
33
35
    win32utils,
34
36
    )
35
37
from bzrlib.tests import (
 
38
    features,
36
39
    file_utils,
37
40
    test__walkdirs_win32,
38
41
    )
39
 
 
40
 
 
41
 
class _UTF8DirReaderFeature(tests.Feature):
 
42
from bzrlib.tests.scenarios import load_tests_apply_scenarios
 
43
 
 
44
 
 
45
class _UTF8DirReaderFeature(features.Feature):
42
46
 
43
47
    def _probe(self):
44
48
        try:
51
55
    def feature_name(self):
52
56
        return 'bzrlib._readdir_pyx'
53
57
 
54
 
UTF8DirReaderFeature = _UTF8DirReaderFeature()
 
58
UTF8DirReaderFeature = features.ModuleAvailableFeature('bzrlib._readdir_pyx')
55
59
 
56
 
term_ios_feature = tests.ModuleAvailableFeature('termios')
 
60
term_ios_feature = features.ModuleAvailableFeature('termios')
57
61
 
58
62
 
59
63
def _already_unicode(s):
94
98
    return scenarios
95
99
 
96
100
 
97
 
def load_tests(basic_tests, module, loader):
98
 
    suite = loader.suiteClass()
99
 
    dir_reader_tests, remaining_tests = tests.split_suite_by_condition(
100
 
        basic_tests, tests.condition_isinstance(TestDirReader))
101
 
    tests.multiply_tests(dir_reader_tests, dir_reader_scenarios(), suite)
102
 
    suite.addTest(remaining_tests)
103
 
    return suite
 
101
load_tests = load_tests_apply_scenarios
104
102
 
105
103
 
106
104
class TestContainsWhitespace(tests.TestCase):
107
105
 
108
106
    def test_contains_whitespace(self):
109
 
        self.failUnless(osutils.contains_whitespace(u' '))
110
 
        self.failUnless(osutils.contains_whitespace(u'hello there'))
111
 
        self.failUnless(osutils.contains_whitespace(u'hellothere\n'))
112
 
        self.failUnless(osutils.contains_whitespace(u'hello\nthere'))
113
 
        self.failUnless(osutils.contains_whitespace(u'hello\rthere'))
114
 
        self.failUnless(osutils.contains_whitespace(u'hello\tthere'))
 
107
        self.assertTrue(osutils.contains_whitespace(u' '))
 
108
        self.assertTrue(osutils.contains_whitespace(u'hello there'))
 
109
        self.assertTrue(osutils.contains_whitespace(u'hellothere\n'))
 
110
        self.assertTrue(osutils.contains_whitespace(u'hello\nthere'))
 
111
        self.assertTrue(osutils.contains_whitespace(u'hello\rthere'))
 
112
        self.assertTrue(osutils.contains_whitespace(u'hello\tthere'))
115
113
 
116
114
        # \xa0 is "Non-breaking-space" which on some python locales thinks it
117
115
        # is whitespace, but we do not.
118
 
        self.failIf(osutils.contains_whitespace(u''))
119
 
        self.failIf(osutils.contains_whitespace(u'hellothere'))
120
 
        self.failIf(osutils.contains_whitespace(u'hello\xa0there'))
 
116
        self.assertFalse(osutils.contains_whitespace(u''))
 
117
        self.assertFalse(osutils.contains_whitespace(u'hellothere'))
 
118
        self.assertFalse(osutils.contains_whitespace(u'hello\xa0there'))
121
119
 
122
120
 
123
121
class TestRename(tests.TestCaseInTempDir):
137
135
        # This should work everywhere
138
136
        self.create_file('a', 'something in a\n')
139
137
        self._fancy_rename('a', 'b')
140
 
        self.failIfExists('a')
141
 
        self.failUnlessExists('b')
 
138
        self.assertPathDoesNotExist('a')
 
139
        self.assertPathExists('b')
142
140
        self.check_file_contents('b', 'something in a\n')
143
141
 
144
142
        self.create_file('a', 'new something in a\n')
151
149
        self.create_file('target', 'data in target\n')
152
150
        self.assertRaises((IOError, OSError), self._fancy_rename,
153
151
                          'missingsource', 'target')
154
 
        self.failUnlessExists('target')
 
152
        self.assertPathExists('target')
155
153
        self.check_file_contents('target', 'data in target\n')
156
154
 
157
155
    def test_fancy_rename_fails_if_source_and_target_missing(self):
162
160
        # Rename should be semi-atomic on all platforms
163
161
        self.create_file('a', 'something in a\n')
164
162
        osutils.rename('a', 'b')
165
 
        self.failIfExists('a')
166
 
        self.failUnlessExists('b')
 
163
        self.assertPathDoesNotExist('a')
 
164
        self.assertPathExists('b')
167
165
        self.check_file_contents('b', 'something in a\n')
168
166
 
169
167
        self.create_file('a', 'new something in a\n')
234
232
            self.assertFalse(osutils.is_inside_or_parent_of_any(dirs, fn))
235
233
 
236
234
 
 
235
class TestLstat(tests.TestCaseInTempDir):
 
236
 
 
237
    def test_lstat_matches_fstat(self):
 
238
        # On Windows, lstat and fstat don't always agree, primarily in the
 
239
        # 'st_ino' and 'st_dev' fields. So we force them to be '0' in our
 
240
        # custom implementation.
 
241
        if sys.platform == 'win32':
 
242
            # We only have special lstat/fstat if we have the extension.
 
243
            # Without it, we may end up re-reading content when we don't have
 
244
            # to, but otherwise it doesn't effect correctness.
 
245
            self.requireFeature(test__walkdirs_win32.win32_readdir_feature)
 
246
        f = open('test-file.txt', 'wb')
 
247
        self.addCleanup(f.close)
 
248
        f.write('some content\n')
 
249
        f.flush()
 
250
        self.assertEqualStat(osutils.fstat(f.fileno()),
 
251
                             osutils.lstat('test-file.txt'))
 
252
 
 
253
 
237
254
class TestRmTree(tests.TestCaseInTempDir):
238
255
 
239
256
    def test_rmtree(self):
251
268
 
252
269
        osutils.rmtree('dir')
253
270
 
254
 
        self.failIfExists('dir/file')
255
 
        self.failIfExists('dir')
 
271
        self.assertPathDoesNotExist('dir/file')
 
272
        self.assertPathDoesNotExist('dir')
256
273
 
257
274
 
258
275
class TestDeleteAny(tests.TestCaseInTempDir):
413
430
class TestLinks(tests.TestCaseInTempDir):
414
431
 
415
432
    def test_dereference_path(self):
416
 
        self.requireFeature(tests.SymlinkFeature)
 
433
        self.requireFeature(features.SymlinkFeature)
417
434
        cwd = osutils.realpath('.')
418
435
        os.mkdir('bar')
419
436
        bar_path = osutils.pathjoin(cwd, 'bar')
466
483
 
467
484
class TestCanonicalRelPath(tests.TestCaseInTempDir):
468
485
 
469
 
    _test_needs_features = [tests.CaseInsCasePresFilenameFeature]
 
486
    _test_needs_features = [features.CaseInsCasePresFilenameFeature]
470
487
 
471
488
    def test_canonical_relpath_simple(self):
472
489
        f = file('MixedCaseName', 'w')
473
490
        f.close()
474
491
        actual = osutils.canonical_relpath(self.test_base_dir, 'mixedcasename')
475
 
        self.failUnlessEqual('work/MixedCaseName', actual)
 
492
        self.assertEqual('work/MixedCaseName', actual)
476
493
 
477
494
    def test_canonical_relpath_missing_tail(self):
478
495
        os.mkdir('MixedCaseParent')
479
496
        actual = osutils.canonical_relpath(self.test_base_dir,
480
497
                                           'mixedcaseparent/nochild')
481
 
        self.failUnlessEqual('work/MixedCaseParent/nochild', actual)
 
498
        self.assertEqual('work/MixedCaseParent/nochild', actual)
482
499
 
483
500
 
484
501
class Test_CICPCanonicalRelpath(tests.TestCaseWithTransport):
802
819
        self.assertEqual(None, osutils.safe_file_id(None))
803
820
 
804
821
 
 
822
class TestPosixFuncs(tests.TestCase):
 
823
    """Test that the posix version of normpath returns an appropriate path
 
824
       when used with 2 leading slashes."""
 
825
 
 
826
    def test_normpath(self):
 
827
        self.assertEqual('/etc/shadow', osutils._posix_normpath('/etc/shadow'))
 
828
        self.assertEqual('/etc/shadow', osutils._posix_normpath('//etc/shadow'))
 
829
        self.assertEqual('/etc/shadow', osutils._posix_normpath('///etc/shadow'))
 
830
 
 
831
 
805
832
class TestWin32Funcs(tests.TestCase):
806
833
    """Test that _win32 versions of os utilities return appropriate paths."""
807
834
 
860
887
        self.assertEqual('//HOST/path', osutils._win98_abspath('//HOST/path'))
861
888
        # relative path
862
889
        cwd = osutils.getcwd().rstrip('/')
863
 
        drive = osutils._nt_splitdrive(cwd)[0]
 
890
        drive = osutils.ntpath.splitdrive(cwd)[0]
864
891
        self.assertEqual(cwd+'/path', osutils._win98_abspath('path'))
865
892
        self.assertEqual(drive+'/path', osutils._win98_abspath('/path'))
866
893
        # unicode path
872
899
    """Test win32 functions that create files."""
873
900
 
874
901
    def test_getcwd(self):
875
 
        self.requireFeature(tests.UnicodeFilenameFeature)
 
902
        self.requireFeature(features.UnicodeFilenameFeature)
876
903
        os.mkdir(u'mu-\xb5')
877
904
        os.chdir(u'mu-\xb5')
878
905
        # TODO: jam 20060427 This will probably fail on Mac OSX because
908
935
        b.close()
909
936
 
910
937
        osutils._win32_rename('b', 'a')
911
 
        self.failUnlessExists('a')
912
 
        self.failIfExists('b')
 
938
        self.assertPathExists('a')
 
939
        self.assertPathDoesNotExist('b')
913
940
        self.assertFileEqual('baz\n', 'a')
914
941
 
915
942
    def test_rename_missing_file(self):
968
995
    """Test mac special functions that require directories."""
969
996
 
970
997
    def test_getcwd(self):
971
 
        self.requireFeature(tests.UnicodeFilenameFeature)
 
998
        self.requireFeature(features.UnicodeFilenameFeature)
972
999
        os.mkdir(u'B\xe5gfors')
973
1000
        os.chdir(u'B\xe5gfors')
974
1001
        self.assertEndsWith(osutils._mac_getcwd(), u'B\xe5gfors')
975
1002
 
976
1003
    def test_getcwd_nonnorm(self):
977
 
        self.requireFeature(tests.UnicodeFilenameFeature)
 
1004
        self.requireFeature(features.UnicodeFilenameFeature)
978
1005
        # Test that _mac_getcwd() will normalize this path
979
1006
        os.mkdir(u'Ba\u030agfors')
980
1007
        os.chdir(u'Ba\u030agfors')
1063
1090
        self.assertExpectedBlocks(expected_dirblocks[1:], result)
1064
1091
 
1065
1092
    def test_walkdirs_os_error(self):
1066
 
        # <https://bugs.edge.launchpad.net/bzr/+bug/338653>
 
1093
        # <https://bugs.launchpad.net/bzr/+bug/338653>
1067
1094
        # Pyrex readdir didn't raise useful messages if it had an error
1068
1095
        # reading the directory
1069
1096
        if sys.platform == 'win32':
1070
1097
            raise tests.TestNotApplicable(
1071
1098
                "readdir IOError not tested on win32")
 
1099
        self.requireFeature(features.not_running_as_root)
1072
1100
        os.mkdir("test-unreadable")
1073
1101
        os.chmod("test-unreadable", 0000)
1074
1102
        # must chmod it back so that it can be removed
1082
1110
        # Ensure the message contains the file name
1083
1111
        self.assertContainsRe(str(e), "\./test-unreadable")
1084
1112
 
 
1113
 
 
1114
    def test_walkdirs_encoding_error(self):
 
1115
        # <https://bugs.launchpad.net/bzr/+bug/488519>
 
1116
        # walkdirs didn't raise a useful message when the filenames
 
1117
        # are not using the filesystem's encoding
 
1118
 
 
1119
        # require a bytestring based filesystem
 
1120
        self.requireFeature(features.ByteStringNamedFilesystem)
 
1121
 
 
1122
        tree = [
 
1123
            '.bzr',
 
1124
            '0file',
 
1125
            '1dir/',
 
1126
            '1dir/0file',
 
1127
            '1dir/1dir/',
 
1128
            '1file'
 
1129
            ]
 
1130
 
 
1131
        self.build_tree(tree)
 
1132
 
 
1133
        # rename the 1file to a latin-1 filename
 
1134
        os.rename("./1file", "\xe8file")
 
1135
        if "\xe8file" not in os.listdir("."):
 
1136
            self.skip("Lack filesystem that preserves arbitrary bytes")
 
1137
 
 
1138
        self._save_platform_info()
 
1139
        win32utils.winver = None # Avoid the win32 detection code
 
1140
        osutils._fs_enc = 'UTF-8'
 
1141
 
 
1142
        # this should raise on error
 
1143
        def attempt():
 
1144
            for dirdetail, dirblock in osutils.walkdirs('.'):
 
1145
                pass
 
1146
 
 
1147
        self.assertRaises(errors.BadFilenameEncoding, attempt)
 
1148
 
1085
1149
    def test__walkdirs_utf8(self):
1086
1150
        tree = [
1087
1151
            '.bzr',
1154
1218
        self._save_platform_info()
1155
1219
        win32utils.winver = None # Avoid the win32 detection code
1156
1220
        osutils._fs_enc = 'UTF-8'
1157
 
        self.assertDirReaderIs(UTF8DirReaderFeature.reader)
 
1221
        self.assertDirReaderIs(
 
1222
            UTF8DirReaderFeature.module.UTF8DirReader)
1158
1223
 
1159
1224
    def test_force_walkdirs_utf8_fs_ascii(self):
1160
1225
        self.requireFeature(UTF8DirReaderFeature)
1161
1226
        self._save_platform_info()
1162
1227
        win32utils.winver = None # Avoid the win32 detection code
1163
1228
        osutils._fs_enc = 'US-ASCII'
1164
 
        self.assertDirReaderIs(UTF8DirReaderFeature.reader)
 
1229
        self.assertDirReaderIs(
 
1230
            UTF8DirReaderFeature.module.UTF8DirReader)
1165
1231
 
1166
1232
    def test_force_walkdirs_utf8_fs_ANSI(self):
1167
1233
        self.requireFeature(UTF8DirReaderFeature)
1168
1234
        self._save_platform_info()
1169
1235
        win32utils.winver = None # Avoid the win32 detection code
1170
1236
        osutils._fs_enc = 'ANSI_X3.4-1968'
1171
 
        self.assertDirReaderIs(UTF8DirReaderFeature.reader)
 
1237
        self.assertDirReaderIs(
 
1238
            UTF8DirReaderFeature.module.UTF8DirReader)
1172
1239
 
1173
1240
    def test_force_walkdirs_utf8_fs_latin1(self):
1174
1241
        self._save_platform_info()
1192
1259
 
1193
1260
    def test_unicode_walkdirs(self):
1194
1261
        """Walkdirs should always return unicode paths."""
1195
 
        self.requireFeature(tests.UnicodeFilenameFeature)
 
1262
        self.requireFeature(features.UnicodeFilenameFeature)
1196
1263
        name0 = u'0file-\xb6'
1197
1264
        name1 = u'1dir-\u062c\u0648'
1198
1265
        name2 = u'2file-\u0633'
1235
1302
 
1236
1303
        The abspath portion might be in unicode or utf-8
1237
1304
        """
1238
 
        self.requireFeature(tests.UnicodeFilenameFeature)
 
1305
        self.requireFeature(features.UnicodeFilenameFeature)
1239
1306
        name0 = u'0file-\xb6'
1240
1307
        name1 = u'1dir-\u062c\u0648'
1241
1308
        name2 = u'2file-\u0633'
1296
1363
 
1297
1364
        The abspath portion should be in unicode
1298
1365
        """
1299
 
        self.requireFeature(tests.UnicodeFilenameFeature)
 
1366
        self.requireFeature(features.UnicodeFilenameFeature)
1300
1367
        # Use the unicode reader. TODO: split into driver-and-driven unit
1301
1368
        # tests.
1302
1369
        self._save_platform_info()
1343
1410
 
1344
1411
    def test__walkdirs_utf8_win32readdir(self):
1345
1412
        self.requireFeature(test__walkdirs_win32.win32_readdir_feature)
1346
 
        self.requireFeature(tests.UnicodeFilenameFeature)
 
1413
        self.requireFeature(features.UnicodeFilenameFeature)
1347
1414
        from bzrlib._walkdirs_win32 import Win32ReadDir
1348
1415
        self._save_platform_info()
1349
1416
        osutils._selected_dir_reader = Win32ReadDir()
1400
1467
    def test__walkdirs_utf_win32_find_file_stat_file(self):
1401
1468
        """make sure our Stat values are valid"""
1402
1469
        self.requireFeature(test__walkdirs_win32.win32_readdir_feature)
1403
 
        self.requireFeature(tests.UnicodeFilenameFeature)
 
1470
        self.requireFeature(features.UnicodeFilenameFeature)
1404
1471
        from bzrlib._walkdirs_win32 import Win32ReadDir
1405
1472
        name0u = u'0file-\xb6'
1406
1473
        name0 = name0u.encode('utf8')
1424
1491
    def test__walkdirs_utf_win32_find_file_stat_directory(self):
1425
1492
        """make sure our Stat values are valid"""
1426
1493
        self.requireFeature(test__walkdirs_win32.win32_readdir_feature)
1427
 
        self.requireFeature(tests.UnicodeFilenameFeature)
 
1494
        self.requireFeature(features.UnicodeFilenameFeature)
1428
1495
        from bzrlib._walkdirs_win32 import Win32ReadDir
1429
1496
        name0u = u'0dir-\u062c\u0648'
1430
1497
        name0 = name0u.encode('utf8')
1530
1597
        self.assertEqual(['c'], os.listdir('target/b'))
1531
1598
 
1532
1599
    def test_copy_tree_symlinks(self):
1533
 
        self.requireFeature(tests.SymlinkFeature)
 
1600
        self.requireFeature(features.SymlinkFeature)
1534
1601
        self.build_tree(['source/'])
1535
1602
        os.symlink('a/generic/path', 'source/lnk')
1536
1603
        osutils.copy_tree('source', 'target')
1561
1628
                          ('d', 'source/b', 'target/b'),
1562
1629
                          ('f', 'source/b/c', 'target/b/c'),
1563
1630
                         ], processed_files)
1564
 
        self.failIfExists('target')
 
1631
        self.assertPathDoesNotExist('target')
1565
1632
        if osutils.has_symlinks():
1566
1633
            self.assertEqual([('source/lnk', 'target/lnk')], processed_links)
1567
1634
 
1613
1680
        old = osutils.set_or_unset_env('BZR_TEST_ENV_VAR', None)
1614
1681
        self.assertEqual('foo', old)
1615
1682
        self.assertEqual(None, os.environ.get('BZR_TEST_ENV_VAR'))
1616
 
        self.failIf('BZR_TEST_ENV_VAR' in os.environ)
 
1683
        self.assertFalse('BZR_TEST_ENV_VAR' in os.environ)
1617
1684
 
1618
1685
 
1619
1686
class TestSizeShaFile(tests.TestCaseInTempDir):
1670
1737
 
1671
1738
class TestReCompile(tests.TestCase):
1672
1739
 
 
1740
    def _deprecated_re_compile_checked(self, *args, **kwargs):
 
1741
        return self.applyDeprecated(symbol_versioning.deprecated_in((2, 2, 0)),
 
1742
            osutils.re_compile_checked, *args, **kwargs)
 
1743
 
1673
1744
    def test_re_compile_checked(self):
1674
 
        r = osutils.re_compile_checked(r'A*', re.IGNORECASE)
 
1745
        r = self._deprecated_re_compile_checked(r'A*', re.IGNORECASE)
1675
1746
        self.assertTrue(r.match('aaaa'))
1676
1747
        self.assertTrue(r.match('aAaA'))
1677
1748
 
1678
1749
    def test_re_compile_checked_error(self):
1679
1750
        # like https://bugs.launchpad.net/bzr/+bug/251352
 
1751
 
 
1752
        # Due to possible test isolation error, re.compile is not lazy at
 
1753
        # this point. We re-install lazy compile.
 
1754
        lazy_regex.install_lazy_compile()
1680
1755
        err = self.assertRaises(
1681
1756
            errors.BzrCommandError,
1682
 
            osutils.re_compile_checked, '*', re.IGNORECASE, 'test case')
 
1757
            self._deprecated_re_compile_checked, '*', re.IGNORECASE, 'test case')
1683
1758
        self.assertEqual(
1684
 
            "Invalid regular expression in test case: '*': "
1685
 
            "nothing to repeat",
 
1759
            'Invalid regular expression in test case: '
 
1760
            '"*" nothing to repeat',
1686
1761
            str(err))
1687
1762
 
1688
1763
 
1689
1764
class TestDirReader(tests.TestCaseInTempDir):
1690
1765
 
 
1766
    scenarios = dir_reader_scenarios()
 
1767
 
1691
1768
    # Set by load_tests
1692
1769
    _dir_reader_class = None
1693
1770
    _native_to_unicode = None
1793
1870
        return filtered_dirblocks
1794
1871
 
1795
1872
    def test_walk_unicode_tree(self):
1796
 
        self.requireFeature(tests.UnicodeFilenameFeature)
 
1873
        self.requireFeature(features.UnicodeFilenameFeature)
1797
1874
        tree, expected_dirblocks = self._get_unicode_tree()
1798
1875
        self.build_tree(tree)
1799
1876
        result = list(osutils._walkdirs_utf8('.'))
1800
1877
        self.assertEqual(expected_dirblocks, self._filter_out(result))
1801
1878
 
1802
1879
    def test_symlink(self):
1803
 
        self.requireFeature(tests.SymlinkFeature)
1804
 
        self.requireFeature(tests.UnicodeFilenameFeature)
 
1880
        self.requireFeature(features.SymlinkFeature)
 
1881
        self.requireFeature(features.UnicodeFilenameFeature)
1805
1882
        target = u'target\N{Euro Sign}'
1806
1883
        link_name = u'l\N{Euro Sign}nk'
1807
1884
        os.symlink(target, link_name)
1825
1902
    But prior python versions failed to properly encode the passed unicode
1826
1903
    string.
1827
1904
    """
1828
 
    _test_needs_features = [tests.SymlinkFeature, tests.UnicodeFilenameFeature]
 
1905
    _test_needs_features = [features.SymlinkFeature, features.UnicodeFilenameFeature]
1829
1906
 
1830
1907
    def setUp(self):
1831
1908
        super(tests.TestCaseInTempDir, self).setUp()
1834
1911
        os.symlink(self.target, self.link)
1835
1912
 
1836
1913
    def test_os_readlink_link_encoding(self):
1837
 
        if sys.version_info < (2, 6):
1838
 
            self.assertRaises(UnicodeEncodeError, os.readlink, self.link)
1839
 
        else:
1840
 
            self.assertEquals(self.target,  os.readlink(self.link))
 
1914
        self.assertEquals(self.target,  os.readlink(self.link))
1841
1915
 
1842
1916
    def test_os_readlink_link_decoding(self):
1843
1917
        self.assertEquals(self.target.encode(osutils._fs_enc),
1855
1929
        self.assertIsInstance(concurrency, int)
1856
1930
 
1857
1931
    def test_local_concurrency_environment_variable(self):
1858
 
        os.environ['BZR_CONCURRENCY'] = '2'
 
1932
        self.overrideEnv('BZR_CONCURRENCY', '2')
1859
1933
        self.assertEqual(2, osutils.local_concurrency(use_cache=False))
1860
 
        os.environ['BZR_CONCURRENCY'] = '3'
 
1934
        self.overrideEnv('BZR_CONCURRENCY', '3')
1861
1935
        self.assertEqual(3, osutils.local_concurrency(use_cache=False))
1862
 
        os.environ['BZR_CONCURRENCY'] = 'foo'
 
1936
        self.overrideEnv('BZR_CONCURRENCY', 'foo')
1863
1937
        self.assertEqual(1, osutils.local_concurrency(use_cache=False))
1864
1938
 
1865
1939
    def test_option_concurrency(self):
1866
 
        os.environ['BZR_CONCURRENCY'] = '1'
 
1940
        self.overrideEnv('BZR_CONCURRENCY', '1')
1867
1941
        self.run_bzr('rocks --concurrency 42')
1868
 
        # Command line overrides envrionment variable
 
1942
        # Command line overrides environment variable
1869
1943
        self.assertEquals('42', os.environ['BZR_CONCURRENCY'])
1870
1944
        self.assertEquals(42, osutils.local_concurrency(use_cache=False))
1871
1945
 
1909
1983
 
1910
1984
class TestTerminalWidth(tests.TestCase):
1911
1985
 
 
1986
    def setUp(self):
 
1987
        tests.TestCase.setUp(self)
 
1988
        self._orig_terminal_size_state = osutils._terminal_size_state
 
1989
        self._orig_first_terminal_size = osutils._first_terminal_size
 
1990
        self.addCleanup(self.restore_osutils_globals)
 
1991
        osutils._terminal_size_state = 'no_data'
 
1992
        osutils._first_terminal_size = None
 
1993
 
 
1994
    def restore_osutils_globals(self):
 
1995
        osutils._terminal_size_state = self._orig_terminal_size_state
 
1996
        osutils._first_terminal_size = self._orig_first_terminal_size
 
1997
 
1912
1998
    def replace_stdout(self, new):
1913
1999
        self.overrideAttr(sys, 'stdout', new)
1914
2000
 
1929
2015
    def test_defaults_to_BZR_COLUMNS(self):
1930
2016
        # BZR_COLUMNS is set by the test framework
1931
2017
        self.assertNotEqual('12', os.environ['BZR_COLUMNS'])
1932
 
        os.environ['BZR_COLUMNS'] = '12'
 
2018
        self.overrideEnv('BZR_COLUMNS', '12')
1933
2019
        self.assertEqual(12, osutils.terminal_width())
1934
2020
 
 
2021
    def test_BZR_COLUMNS_0_no_limit(self):
 
2022
        self.overrideEnv('BZR_COLUMNS', '0')
 
2023
        self.assertEqual(None, osutils.terminal_width())
 
2024
 
1935
2025
    def test_falls_back_to_COLUMNS(self):
1936
 
        del os.environ['BZR_COLUMNS']
 
2026
        self.overrideEnv('BZR_COLUMNS', None)
1937
2027
        self.assertNotEqual('42', os.environ['COLUMNS'])
1938
2028
        self.set_fake_tty()
1939
 
        os.environ['COLUMNS'] = '42'
 
2029
        self.overrideEnv('COLUMNS', '42')
1940
2030
        self.assertEqual(42, osutils.terminal_width())
1941
2031
 
1942
2032
    def test_tty_default_without_columns(self):
1943
 
        del os.environ['BZR_COLUMNS']
1944
 
        del os.environ['COLUMNS']
 
2033
        self.overrideEnv('BZR_COLUMNS', None)
 
2034
        self.overrideEnv('COLUMNS', None)
1945
2035
 
1946
2036
        def terminal_size(w, h):
1947
2037
            return 42, 42
1954
2044
        self.assertEqual(42, osutils.terminal_width())
1955
2045
 
1956
2046
    def test_non_tty_default_without_columns(self):
1957
 
        del os.environ['BZR_COLUMNS']
1958
 
        del os.environ['COLUMNS']
 
2047
        self.overrideEnv('BZR_COLUMNS', None)
 
2048
        self.overrideEnv('COLUMNS', None)
1959
2049
        self.replace_stdout(None)
1960
2050
        self.assertEqual(None, osutils.terminal_width())
1961
2051
 
1971
2061
        else:
1972
2062
            self.overrideAttr(termios, 'TIOCGWINSZ')
1973
2063
            del termios.TIOCGWINSZ
1974
 
        del os.environ['BZR_COLUMNS']
1975
 
        del os.environ['COLUMNS']
 
2064
        self.overrideEnv('BZR_COLUMNS', None)
 
2065
        self.overrideEnv('COLUMNS', None)
1976
2066
        # Whatever the result is, if we don't raise an exception, it's ok.
1977
2067
        osutils.terminal_width()
 
2068
 
 
2069
 
 
2070
class TestCreationOps(tests.TestCaseInTempDir):
 
2071
    _test_needs_features = [features.chown_feature]
 
2072
 
 
2073
    def setUp(self):
 
2074
        tests.TestCaseInTempDir.setUp(self)
 
2075
        self.overrideAttr(os, 'chown', self._dummy_chown)
 
2076
 
 
2077
        # params set by call to _dummy_chown
 
2078
        self.path = self.uid = self.gid = None
 
2079
 
 
2080
    def _dummy_chown(self, path, uid, gid):
 
2081
        self.path, self.uid, self.gid = path, uid, gid
 
2082
 
 
2083
    def test_copy_ownership_from_path(self):
 
2084
        """copy_ownership_from_path test with specified src."""
 
2085
        ownsrc = '/'
 
2086
        f = open('test_file', 'wt')
 
2087
        osutils.copy_ownership_from_path('test_file', ownsrc)
 
2088
 
 
2089
        s = os.stat(ownsrc)
 
2090
        self.assertEquals(self.path, 'test_file')
 
2091
        self.assertEquals(self.uid, s.st_uid)
 
2092
        self.assertEquals(self.gid, s.st_gid)
 
2093
 
 
2094
    def test_copy_ownership_nonesrc(self):
 
2095
        """copy_ownership_from_path test with src=None."""
 
2096
        f = open('test_file', 'wt')
 
2097
        # should use parent dir for permissions
 
2098
        osutils.copy_ownership_from_path('test_file')
 
2099
 
 
2100
        s = os.stat('..')
 
2101
        self.assertEquals(self.path, 'test_file')
 
2102
        self.assertEquals(self.uid, s.st_uid)
 
2103
        self.assertEquals(self.gid, s.st_gid)
 
2104
 
 
2105
 
 
2106
class TestGetuserUnicode(tests.TestCase):
 
2107
 
 
2108
    def test_ascii_user(self):
 
2109
        self.overrideEnv('LOGNAME', 'jrandom')
 
2110
        self.assertEqual(u'jrandom', osutils.getuser_unicode())
 
2111
 
 
2112
    def test_unicode_user(self):
 
2113
        ue = osutils.get_user_encoding()
 
2114
        uni_val, env_val = tests.probe_unicode_in_user_encoding()
 
2115
        if uni_val is None:
 
2116
            raise tests.TestSkipped(
 
2117
                'Cannot find a unicode character that works in encoding %s'
 
2118
                % (osutils.get_user_encoding(),))
 
2119
        uni_username = u'jrandom' + uni_val
 
2120
        encoded_username = uni_username.encode(ue)
 
2121
        self.overrideEnv('LOGNAME', encoded_username)
 
2122
        self.assertEqual(uni_username, osutils.getuser_unicode())
 
2123
        self.overrideEnv('LOGNAME', u'jrandom\xb6'.encode(ue))
 
2124
        self.assertEqual(u'jrandom\xb6', osutils.getuser_unicode())
 
2125
 
 
2126
    def test_no_username_bug_660174(self):
 
2127
        self.requireFeature(features.win32_feature)
 
2128
        for name in ('LOGNAME', 'USER', 'LNAME', 'USERNAME'):
 
2129
            self.overrideEnv(name, None)
 
2130
        self.assertEqual(u'UNKNOWN', osutils.getuser_unicode())
 
2131
 
 
2132
 
 
2133
class TestBackupNames(tests.TestCase):
 
2134
 
 
2135
    def setUp(self):
 
2136
        super(TestBackupNames, self).setUp()
 
2137
        self.backups = []
 
2138
 
 
2139
    def backup_exists(self, name):
 
2140
        return name in self.backups
 
2141
 
 
2142
    def available_backup_name(self, name):
 
2143
        backup_name = osutils.available_backup_name(name, self.backup_exists)
 
2144
        self.backups.append(backup_name)
 
2145
        return backup_name
 
2146
 
 
2147
    def assertBackupName(self, expected, name):
 
2148
        self.assertEqual(expected, self.available_backup_name(name))
 
2149
 
 
2150
    def test_empty(self):
 
2151
        self.assertBackupName('file.~1~', 'file')
 
2152
 
 
2153
    def test_existing(self):
 
2154
        self.available_backup_name('file')
 
2155
        self.available_backup_name('file')
 
2156
        self.assertBackupName('file.~3~', 'file')
 
2157
        # Empty slots are found, this is not a strict requirement and may be
 
2158
        # revisited if we test against all implementations.
 
2159
        self.backups.remove('file.~2~')
 
2160
        self.assertBackupName('file.~2~', 'file')
 
2161
 
 
2162
 
 
2163
class TestFindExecutableInPath(tests.TestCase):
 
2164
 
 
2165
    def test_windows(self):
 
2166
        if sys.platform != 'win32':
 
2167
            raise tests.TestSkipped('test requires win32')
 
2168
        self.assertTrue(osutils.find_executable_on_path('explorer') is not None)
 
2169
        self.assertTrue(
 
2170
            osutils.find_executable_on_path('explorer.exe') is not None)
 
2171
        self.assertTrue(
 
2172
            osutils.find_executable_on_path('EXPLORER.EXE') is not None)
 
2173
        self.assertTrue(
 
2174
            osutils.find_executable_on_path('THIS SHOULD NOT EXIST') is None)
 
2175
        self.assertTrue(osutils.find_executable_on_path('file.txt') is None)
 
2176
 
 
2177
    def test_other(self):
 
2178
        if sys.platform == 'win32':
 
2179
            raise tests.TestSkipped('test requires non-win32')
 
2180
        self.assertTrue(osutils.find_executable_on_path('sh') is not None)
 
2181
        self.assertTrue(
 
2182
            osutils.find_executable_on_path('THIS SHOULD NOT EXIST') is None)
 
2183
 
 
2184
 
 
2185
class TestEnvironmentErrors(tests.TestCase):
 
2186
    """Test handling of environmental errors"""
 
2187
 
 
2188
    def test_is_oserror(self):
 
2189
        self.assertTrue(osutils.is_environment_error(
 
2190
            OSError(errno.EINVAL, "Invalid parameter")))
 
2191
 
 
2192
    def test_is_ioerror(self):
 
2193
        self.assertTrue(osutils.is_environment_error(
 
2194
            IOError(errno.EINVAL, "Invalid parameter")))
 
2195
 
 
2196
    def test_is_socket_error(self):
 
2197
        self.assertTrue(osutils.is_environment_error(
 
2198
            socket.error(errno.EINVAL, "Invalid parameter")))
 
2199
 
 
2200
    def test_is_select_error(self):
 
2201
        self.assertTrue(osutils.is_environment_error(
 
2202
            select.error(errno.EINVAL, "Invalid parameter")))
 
2203
 
 
2204
    def test_is_pywintypes_error(self):
 
2205
        self.requireFeature(features.pywintypes)
 
2206
        import pywintypes
 
2207
        self.assertTrue(osutils.is_environment_error(
 
2208
            pywintypes.error(errno.EINVAL, "Invalid parameter", "Caller")))