← 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: Canonical.com Patch Queue Manager
  • Date: 2011-08-17 08:40:16 UTC
  • mfrom: (5642.4.6 712474-module-available)
  • Revision ID: pqm@pqm.ubuntu.com-20110817084016-600z65qzqmmt44w7
(vila) ModuleAvailableFeature don't try to imported already imported
 modules. (Vincent Ladeuil)

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,
34
35
    )
35
36
from bzrlib.tests import (
 
37
    features,
36
38
    file_utils,
37
39
    test__walkdirs_win32,
38
40
    )
39
 
 
40
 
 
41
 
class _UTF8DirReaderFeature(tests.Feature):
 
41
from bzrlib.tests.scenarios import load_tests_apply_scenarios
 
42
 
 
43
 
 
44
class _UTF8DirReaderFeature(features.Feature):
42
45
 
43
46
    def _probe(self):
44
47
        try:
51
54
    def feature_name(self):
52
55
        return 'bzrlib._readdir_pyx'
53
56
 
54
 
UTF8DirReaderFeature = _UTF8DirReaderFeature()
 
57
UTF8DirReaderFeature = features.ModuleAvailableFeature('bzrlib._readdir_pyx')
55
58
 
56
 
term_ios_feature = tests.ModuleAvailableFeature('termios')
 
59
term_ios_feature = features.ModuleAvailableFeature('termios')
57
60
 
58
61
 
59
62
def _already_unicode(s):
94
97
    return scenarios
95
98
 
96
99
 
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
 
100
load_tests = load_tests_apply_scenarios
104
101
 
105
102
 
106
103
class TestContainsWhitespace(tests.TestCase):
107
104
 
108
105
    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'))
 
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'))
115
112
 
116
113
        # \xa0 is "Non-breaking-space" which on some python locales thinks it
117
114
        # 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'))
 
115
        self.assertFalse(osutils.contains_whitespace(u''))
 
116
        self.assertFalse(osutils.contains_whitespace(u'hellothere'))
 
117
        self.assertFalse(osutils.contains_whitespace(u'hello\xa0there'))
121
118
 
122
119
 
123
120
class TestRename(tests.TestCaseInTempDir):
137
134
        # This should work everywhere
138
135
        self.create_file('a', 'something in a\n')
139
136
        self._fancy_rename('a', 'b')
140
 
        self.failIfExists('a')
141
 
        self.failUnlessExists('b')
 
137
        self.assertPathDoesNotExist('a')
 
138
        self.assertPathExists('b')
142
139
        self.check_file_contents('b', 'something in a\n')
143
140
 
144
141
        self.create_file('a', 'new something in a\n')
151
148
        self.create_file('target', 'data in target\n')
152
149
        self.assertRaises((IOError, OSError), self._fancy_rename,
153
150
                          'missingsource', 'target')
154
 
        self.failUnlessExists('target')
 
151
        self.assertPathExists('target')
155
152
        self.check_file_contents('target', 'data in target\n')
156
153
 
157
154
    def test_fancy_rename_fails_if_source_and_target_missing(self):
162
159
        # Rename should be semi-atomic on all platforms
163
160
        self.create_file('a', 'something in a\n')
164
161
        osutils.rename('a', 'b')
165
 
        self.failIfExists('a')
166
 
        self.failUnlessExists('b')
 
162
        self.assertPathDoesNotExist('a')
 
163
        self.assertPathExists('b')
167
164
        self.check_file_contents('b', 'something in a\n')
168
165
 
169
166
        self.create_file('a', 'new something in a\n')
234
231
            self.assertFalse(osutils.is_inside_or_parent_of_any(dirs, fn))
235
232
 
236
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
 
237
253
class TestRmTree(tests.TestCaseInTempDir):
238
254
 
239
255
    def test_rmtree(self):
251
267
 
252
268
        osutils.rmtree('dir')
253
269
 
254
 
        self.failIfExists('dir/file')
255
 
        self.failIfExists('dir')
 
270
        self.assertPathDoesNotExist('dir/file')
 
271
        self.assertPathDoesNotExist('dir')
256
272
 
257
273
 
258
274
class TestDeleteAny(tests.TestCaseInTempDir):
413
429
class TestLinks(tests.TestCaseInTempDir):
414
430
 
415
431
    def test_dereference_path(self):
416
 
        self.requireFeature(tests.SymlinkFeature)
 
432
        self.requireFeature(features.SymlinkFeature)
417
433
        cwd = osutils.realpath('.')
418
434
        os.mkdir('bar')
419
435
        bar_path = osutils.pathjoin(cwd, 'bar')
466
482
 
467
483
class TestCanonicalRelPath(tests.TestCaseInTempDir):
468
484
 
469
 
    _test_needs_features = [tests.CaseInsCasePresFilenameFeature]
 
485
    _test_needs_features = [features.CaseInsCasePresFilenameFeature]
470
486
 
471
487
    def test_canonical_relpath_simple(self):
472
488
        f = file('MixedCaseName', 'w')
473
489
        f.close()
474
490
        actual = osutils.canonical_relpath(self.test_base_dir, 'mixedcasename')
475
 
        self.failUnlessEqual('work/MixedCaseName', actual)
 
491
        self.assertEqual('work/MixedCaseName', actual)
476
492
 
477
493
    def test_canonical_relpath_missing_tail(self):
478
494
        os.mkdir('MixedCaseParent')
479
495
        actual = osutils.canonical_relpath(self.test_base_dir,
480
496
                                           'mixedcaseparent/nochild')
481
 
        self.failUnlessEqual('work/MixedCaseParent/nochild', actual)
 
497
        self.assertEqual('work/MixedCaseParent/nochild', actual)
482
498
 
483
499
 
484
500
class Test_CICPCanonicalRelpath(tests.TestCaseWithTransport):
860
876
        self.assertEqual('//HOST/path', osutils._win98_abspath('//HOST/path'))
861
877
        # relative path
862
878
        cwd = osutils.getcwd().rstrip('/')
863
 
        drive = osutils._nt_splitdrive(cwd)[0]
 
879
        drive = osutils.ntpath.splitdrive(cwd)[0]
864
880
        self.assertEqual(cwd+'/path', osutils._win98_abspath('path'))
865
881
        self.assertEqual(drive+'/path', osutils._win98_abspath('/path'))
866
882
        # unicode path
872
888
    """Test win32 functions that create files."""
873
889
 
874
890
    def test_getcwd(self):
875
 
        self.requireFeature(tests.UnicodeFilenameFeature)
 
891
        self.requireFeature(features.UnicodeFilenameFeature)
876
892
        os.mkdir(u'mu-\xb5')
877
893
        os.chdir(u'mu-\xb5')
878
894
        # TODO: jam 20060427 This will probably fail on Mac OSX because
908
924
        b.close()
909
925
 
910
926
        osutils._win32_rename('b', 'a')
911
 
        self.failUnlessExists('a')
912
 
        self.failIfExists('b')
 
927
        self.assertPathExists('a')
 
928
        self.assertPathDoesNotExist('b')
913
929
        self.assertFileEqual('baz\n', 'a')
914
930
 
915
931
    def test_rename_missing_file(self):
968
984
    """Test mac special functions that require directories."""
969
985
 
970
986
    def test_getcwd(self):
971
 
        self.requireFeature(tests.UnicodeFilenameFeature)
 
987
        self.requireFeature(features.UnicodeFilenameFeature)
972
988
        os.mkdir(u'B\xe5gfors')
973
989
        os.chdir(u'B\xe5gfors')
974
990
        self.assertEndsWith(osutils._mac_getcwd(), u'B\xe5gfors')
975
991
 
976
992
    def test_getcwd_nonnorm(self):
977
 
        self.requireFeature(tests.UnicodeFilenameFeature)
 
993
        self.requireFeature(features.UnicodeFilenameFeature)
978
994
        # Test that _mac_getcwd() will normalize this path
979
995
        os.mkdir(u'Ba\u030agfors')
980
996
        os.chdir(u'Ba\u030agfors')
1063
1079
        self.assertExpectedBlocks(expected_dirblocks[1:], result)
1064
1080
 
1065
1081
    def test_walkdirs_os_error(self):
1066
 
        # <https://bugs.edge.launchpad.net/bzr/+bug/338653>
 
1082
        # <https://bugs.launchpad.net/bzr/+bug/338653>
1067
1083
        # Pyrex readdir didn't raise useful messages if it had an error
1068
1084
        # reading the directory
1069
1085
        if sys.platform == 'win32':
1070
1086
            raise tests.TestNotApplicable(
1071
1087
                "readdir IOError not tested on win32")
 
1088
        self.requireFeature(features.not_running_as_root)
1072
1089
        os.mkdir("test-unreadable")
1073
1090
        os.chmod("test-unreadable", 0000)
1074
1091
        # must chmod it back so that it can be removed
1082
1099
        # Ensure the message contains the file name
1083
1100
        self.assertContainsRe(str(e), "\./test-unreadable")
1084
1101
 
 
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
 
1085
1138
    def test__walkdirs_utf8(self):
1086
1139
        tree = [
1087
1140
            '.bzr',
1154
1207
        self._save_platform_info()
1155
1208
        win32utils.winver = None # Avoid the win32 detection code
1156
1209
        osutils._fs_enc = 'UTF-8'
1157
 
        self.assertDirReaderIs(UTF8DirReaderFeature.reader)
 
1210
        self.assertDirReaderIs(
 
1211
            UTF8DirReaderFeature.module.UTF8DirReader)
1158
1212
 
1159
1213
    def test_force_walkdirs_utf8_fs_ascii(self):
1160
1214
        self.requireFeature(UTF8DirReaderFeature)
1161
1215
        self._save_platform_info()
1162
1216
        win32utils.winver = None # Avoid the win32 detection code
1163
1217
        osutils._fs_enc = 'US-ASCII'
1164
 
        self.assertDirReaderIs(UTF8DirReaderFeature.reader)
 
1218
        self.assertDirReaderIs(
 
1219
            UTF8DirReaderFeature.module.UTF8DirReader)
1165
1220
 
1166
1221
    def test_force_walkdirs_utf8_fs_ANSI(self):
1167
1222
        self.requireFeature(UTF8DirReaderFeature)
1168
1223
        self._save_platform_info()
1169
1224
        win32utils.winver = None # Avoid the win32 detection code
1170
1225
        osutils._fs_enc = 'ANSI_X3.4-1968'
1171
 
        self.assertDirReaderIs(UTF8DirReaderFeature.reader)
 
1226
        self.assertDirReaderIs(
 
1227
            UTF8DirReaderFeature.module.UTF8DirReader)
1172
1228
 
1173
1229
    def test_force_walkdirs_utf8_fs_latin1(self):
1174
1230
        self._save_platform_info()
1192
1248
 
1193
1249
    def test_unicode_walkdirs(self):
1194
1250
        """Walkdirs should always return unicode paths."""
1195
 
        self.requireFeature(tests.UnicodeFilenameFeature)
 
1251
        self.requireFeature(features.UnicodeFilenameFeature)
1196
1252
        name0 = u'0file-\xb6'
1197
1253
        name1 = u'1dir-\u062c\u0648'
1198
1254
        name2 = u'2file-\u0633'
1235
1291
 
1236
1292
        The abspath portion might be in unicode or utf-8
1237
1293
        """
1238
 
        self.requireFeature(tests.UnicodeFilenameFeature)
 
1294
        self.requireFeature(features.UnicodeFilenameFeature)
1239
1295
        name0 = u'0file-\xb6'
1240
1296
        name1 = u'1dir-\u062c\u0648'
1241
1297
        name2 = u'2file-\u0633'
1296
1352
 
1297
1353
        The abspath portion should be in unicode
1298
1354
        """
1299
 
        self.requireFeature(tests.UnicodeFilenameFeature)
 
1355
        self.requireFeature(features.UnicodeFilenameFeature)
1300
1356
        # Use the unicode reader. TODO: split into driver-and-driven unit
1301
1357
        # tests.
1302
1358
        self._save_platform_info()
1343
1399
 
1344
1400
    def test__walkdirs_utf8_win32readdir(self):
1345
1401
        self.requireFeature(test__walkdirs_win32.win32_readdir_feature)
1346
 
        self.requireFeature(tests.UnicodeFilenameFeature)
 
1402
        self.requireFeature(features.UnicodeFilenameFeature)
1347
1403
        from bzrlib._walkdirs_win32 import Win32ReadDir
1348
1404
        self._save_platform_info()
1349
1405
        osutils._selected_dir_reader = Win32ReadDir()
1400
1456
    def test__walkdirs_utf_win32_find_file_stat_file(self):
1401
1457
        """make sure our Stat values are valid"""
1402
1458
        self.requireFeature(test__walkdirs_win32.win32_readdir_feature)
1403
 
        self.requireFeature(tests.UnicodeFilenameFeature)
 
1459
        self.requireFeature(features.UnicodeFilenameFeature)
1404
1460
        from bzrlib._walkdirs_win32 import Win32ReadDir
1405
1461
        name0u = u'0file-\xb6'
1406
1462
        name0 = name0u.encode('utf8')
1424
1480
    def test__walkdirs_utf_win32_find_file_stat_directory(self):
1425
1481
        """make sure our Stat values are valid"""
1426
1482
        self.requireFeature(test__walkdirs_win32.win32_readdir_feature)
1427
 
        self.requireFeature(tests.UnicodeFilenameFeature)
 
1483
        self.requireFeature(features.UnicodeFilenameFeature)
1428
1484
        from bzrlib._walkdirs_win32 import Win32ReadDir
1429
1485
        name0u = u'0dir-\u062c\u0648'
1430
1486
        name0 = name0u.encode('utf8')
1530
1586
        self.assertEqual(['c'], os.listdir('target/b'))
1531
1587
 
1532
1588
    def test_copy_tree_symlinks(self):
1533
 
        self.requireFeature(tests.SymlinkFeature)
 
1589
        self.requireFeature(features.SymlinkFeature)
1534
1590
        self.build_tree(['source/'])
1535
1591
        os.symlink('a/generic/path', 'source/lnk')
1536
1592
        osutils.copy_tree('source', 'target')
1561
1617
                          ('d', 'source/b', 'target/b'),
1562
1618
                          ('f', 'source/b/c', 'target/b/c'),
1563
1619
                         ], processed_files)
1564
 
        self.failIfExists('target')
 
1620
        self.assertPathDoesNotExist('target')
1565
1621
        if osutils.has_symlinks():
1566
1622
            self.assertEqual([('source/lnk', 'target/lnk')], processed_links)
1567
1623
 
1613
1669
        old = osutils.set_or_unset_env('BZR_TEST_ENV_VAR', None)
1614
1670
        self.assertEqual('foo', old)
1615
1671
        self.assertEqual(None, os.environ.get('BZR_TEST_ENV_VAR'))
1616
 
        self.failIf('BZR_TEST_ENV_VAR' in os.environ)
 
1672
        self.assertFalse('BZR_TEST_ENV_VAR' in os.environ)
1617
1673
 
1618
1674
 
1619
1675
class TestSizeShaFile(tests.TestCaseInTempDir):
1670
1726
 
1671
1727
class TestReCompile(tests.TestCase):
1672
1728
 
 
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
 
1673
1733
    def test_re_compile_checked(self):
1674
 
        r = osutils.re_compile_checked(r'A*', re.IGNORECASE)
 
1734
        r = self._deprecated_re_compile_checked(r'A*', re.IGNORECASE)
1675
1735
        self.assertTrue(r.match('aaaa'))
1676
1736
        self.assertTrue(r.match('aAaA'))
1677
1737
 
1678
1738
    def test_re_compile_checked_error(self):
1679
1739
        # 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()
1680
1744
        err = self.assertRaises(
1681
1745
            errors.BzrCommandError,
1682
 
            osutils.re_compile_checked, '*', re.IGNORECASE, 'test case')
 
1746
            self._deprecated_re_compile_checked, '*', re.IGNORECASE, 'test case')
1683
1747
        self.assertEqual(
1684
 
            "Invalid regular expression in test case: '*': "
1685
 
            "nothing to repeat",
 
1748
            'Invalid regular expression in test case: '
 
1749
            '"*" nothing to repeat',
1686
1750
            str(err))
1687
1751
 
1688
1752
 
1689
1753
class TestDirReader(tests.TestCaseInTempDir):
1690
1754
 
 
1755
    scenarios = dir_reader_scenarios()
 
1756
 
1691
1757
    # Set by load_tests
1692
1758
    _dir_reader_class = None
1693
1759
    _native_to_unicode = None
1793
1859
        return filtered_dirblocks
1794
1860
 
1795
1861
    def test_walk_unicode_tree(self):
1796
 
        self.requireFeature(tests.UnicodeFilenameFeature)
 
1862
        self.requireFeature(features.UnicodeFilenameFeature)
1797
1863
        tree, expected_dirblocks = self._get_unicode_tree()
1798
1864
        self.build_tree(tree)
1799
1865
        result = list(osutils._walkdirs_utf8('.'))
1800
1866
        self.assertEqual(expected_dirblocks, self._filter_out(result))
1801
1867
 
1802
1868
    def test_symlink(self):
1803
 
        self.requireFeature(tests.SymlinkFeature)
1804
 
        self.requireFeature(tests.UnicodeFilenameFeature)
 
1869
        self.requireFeature(features.SymlinkFeature)
 
1870
        self.requireFeature(features.UnicodeFilenameFeature)
1805
1871
        target = u'target\N{Euro Sign}'
1806
1872
        link_name = u'l\N{Euro Sign}nk'
1807
1873
        os.symlink(target, link_name)
1825
1891
    But prior python versions failed to properly encode the passed unicode
1826
1892
    string.
1827
1893
    """
1828
 
    _test_needs_features = [tests.SymlinkFeature, tests.UnicodeFilenameFeature]
 
1894
    _test_needs_features = [features.SymlinkFeature, features.UnicodeFilenameFeature]
1829
1895
 
1830
1896
    def setUp(self):
1831
1897
        super(tests.TestCaseInTempDir, self).setUp()
1834
1900
        os.symlink(self.target, self.link)
1835
1901
 
1836
1902
    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))
 
1903
        self.assertEquals(self.target,  os.readlink(self.link))
1841
1904
 
1842
1905
    def test_os_readlink_link_decoding(self):
1843
1906
        self.assertEquals(self.target.encode(osutils._fs_enc),
1855
1918
        self.assertIsInstance(concurrency, int)
1856
1919
 
1857
1920
    def test_local_concurrency_environment_variable(self):
1858
 
        os.environ['BZR_CONCURRENCY'] = '2'
 
1921
        self.overrideEnv('BZR_CONCURRENCY', '2')
1859
1922
        self.assertEqual(2, osutils.local_concurrency(use_cache=False))
1860
 
        os.environ['BZR_CONCURRENCY'] = '3'
 
1923
        self.overrideEnv('BZR_CONCURRENCY', '3')
1861
1924
        self.assertEqual(3, osutils.local_concurrency(use_cache=False))
1862
 
        os.environ['BZR_CONCURRENCY'] = 'foo'
 
1925
        self.overrideEnv('BZR_CONCURRENCY', 'foo')
1863
1926
        self.assertEqual(1, osutils.local_concurrency(use_cache=False))
1864
1927
 
1865
1928
    def test_option_concurrency(self):
1866
 
        os.environ['BZR_CONCURRENCY'] = '1'
 
1929
        self.overrideEnv('BZR_CONCURRENCY', '1')
1867
1930
        self.run_bzr('rocks --concurrency 42')
1868
 
        # Command line overrides envrionment variable
 
1931
        # Command line overrides environment variable
1869
1932
        self.assertEquals('42', os.environ['BZR_CONCURRENCY'])
1870
1933
        self.assertEquals(42, osutils.local_concurrency(use_cache=False))
1871
1934
 
1909
1972
 
1910
1973
class TestTerminalWidth(tests.TestCase):
1911
1974
 
 
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
 
1912
1987
    def replace_stdout(self, new):
1913
1988
        self.overrideAttr(sys, 'stdout', new)
1914
1989
 
1929
2004
    def test_defaults_to_BZR_COLUMNS(self):
1930
2005
        # BZR_COLUMNS is set by the test framework
1931
2006
        self.assertNotEqual('12', os.environ['BZR_COLUMNS'])
1932
 
        os.environ['BZR_COLUMNS'] = '12'
 
2007
        self.overrideEnv('BZR_COLUMNS', '12')
1933
2008
        self.assertEqual(12, osutils.terminal_width())
1934
2009
 
 
2010
    def test_BZR_COLUMNS_0_no_limit(self):
 
2011
        self.overrideEnv('BZR_COLUMNS', '0')
 
2012
        self.assertEqual(None, osutils.terminal_width())
 
2013
 
1935
2014
    def test_falls_back_to_COLUMNS(self):
1936
 
        del os.environ['BZR_COLUMNS']
 
2015
        self.overrideEnv('BZR_COLUMNS', None)
1937
2016
        self.assertNotEqual('42', os.environ['COLUMNS'])
1938
2017
        self.set_fake_tty()
1939
 
        os.environ['COLUMNS'] = '42'
 
2018
        self.overrideEnv('COLUMNS', '42')
1940
2019
        self.assertEqual(42, osutils.terminal_width())
1941
2020
 
1942
2021
    def test_tty_default_without_columns(self):
1943
 
        del os.environ['BZR_COLUMNS']
1944
 
        del os.environ['COLUMNS']
 
2022
        self.overrideEnv('BZR_COLUMNS', None)
 
2023
        self.overrideEnv('COLUMNS', None)
1945
2024
 
1946
2025
        def terminal_size(w, h):
1947
2026
            return 42, 42
1954
2033
        self.assertEqual(42, osutils.terminal_width())
1955
2034
 
1956
2035
    def test_non_tty_default_without_columns(self):
1957
 
        del os.environ['BZR_COLUMNS']
1958
 
        del os.environ['COLUMNS']
 
2036
        self.overrideEnv('BZR_COLUMNS', None)
 
2037
        self.overrideEnv('COLUMNS', None)
1959
2038
        self.replace_stdout(None)
1960
2039
        self.assertEqual(None, osutils.terminal_width())
1961
2040
 
1971
2050
        else:
1972
2051
            self.overrideAttr(termios, 'TIOCGWINSZ')
1973
2052
            del termios.TIOCGWINSZ
1974
 
        del os.environ['BZR_COLUMNS']
1975
 
        del os.environ['COLUMNS']
 
2053
        self.overrideEnv('BZR_COLUMNS', None)
 
2054
        self.overrideEnv('COLUMNS', None)
1976
2055
        # Whatever the result is, if we don't raise an exception, it's ok.
1977
2056
        osutils.terminal_width()
 
2057
 
 
2058
 
 
2059
class TestCreationOps(tests.TestCaseInTempDir):
 
2060
    _test_needs_features = [features.chown_feature]
 
2061
 
 
2062
    def setUp(self):
 
2063
        tests.TestCaseInTempDir.setUp(self)
 
2064
        self.overrideAttr(os, 'chown', self._dummy_chown)
 
2065
 
 
2066
        # params set by call to _dummy_chown
 
2067
        self.path = self.uid = self.gid = None
 
2068
 
 
2069
    def _dummy_chown(self, path, uid, gid):
 
2070
        self.path, self.uid, self.gid = path, uid, gid
 
2071
 
 
2072
    def test_copy_ownership_from_path(self):
 
2073
        """copy_ownership_from_path test with specified src."""
 
2074
        ownsrc = '/'
 
2075
        f = open('test_file', 'wt')
 
2076
        osutils.copy_ownership_from_path('test_file', ownsrc)
 
2077
 
 
2078
        s = os.stat(ownsrc)
 
2079
        self.assertEquals(self.path, 'test_file')
 
2080
        self.assertEquals(self.uid, s.st_uid)
 
2081
        self.assertEquals(self.gid, s.st_gid)
 
2082
 
 
2083
    def test_copy_ownership_nonesrc(self):
 
2084
        """copy_ownership_from_path test with src=None."""
 
2085
        f = open('test_file', 'wt')
 
2086
        # should use parent dir for permissions
 
2087
        osutils.copy_ownership_from_path('test_file')
 
2088
 
 
2089
        s = os.stat('..')
 
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
 
 
2095
class TestGetuserUnicode(tests.TestCase):
 
2096
 
 
2097
    def test_ascii_user(self):
 
2098
        self.overrideEnv('LOGNAME', 'jrandom')
 
2099
        self.assertEqual(u'jrandom', osutils.getuser_unicode())
 
2100
 
 
2101
    def test_unicode_user(self):
 
2102
        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))
 
2113
        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)