← 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: Martin Pool
  • Date: 2010-01-15 04:17:57 UTC
  • mto: This revision was merged to the branch mainline in revision 5019.
  • Revision ID: mbp@sourcefrog.net-20100115041757-cd8pu9o5a511jc8q
Rip out most remaining uses of DummyProgressBar

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test_osutils.py
1
 
# Copyright (C) 2005-2011 Canonical Ltd
 
1
# Copyright (C) 2005, 2006, 2007, 2008, 2009 Canonical Ltd
2
2
#
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
21
21
import os
22
22
import re
23
23
import socket
 
24
import stat
24
25
import sys
25
26
import time
26
27
 
27
28
from bzrlib import (
28
29
    errors,
29
 
    lazy_regex,
30
30
    osutils,
31
 
    symbol_versioning,
32
31
    tests,
33
32
    trace,
34
33
    win32utils,
35
34
    )
36
35
from bzrlib.tests import (
37
 
    features,
38
36
    file_utils,
39
37
    test__walkdirs_win32,
40
38
    )
41
 
from bzrlib.tests.scenarios import load_tests_apply_scenarios
42
 
 
43
 
 
44
 
class _UTF8DirReaderFeature(features.Feature):
 
39
 
 
40
 
 
41
class _UTF8DirReaderFeature(tests.Feature):
45
42
 
46
43
    def _probe(self):
47
44
        try:
54
51
    def feature_name(self):
55
52
        return 'bzrlib._readdir_pyx'
56
53
 
57
 
UTF8DirReaderFeature = features.ModuleAvailableFeature('bzrlib._readdir_pyx')
 
54
UTF8DirReaderFeature = _UTF8DirReaderFeature()
58
55
 
59
 
term_ios_feature = features.ModuleAvailableFeature('termios')
 
56
term_ios_feature = tests.ModuleAvailableFeature('termios')
60
57
 
61
58
 
62
59
def _already_unicode(s):
97
94
    return scenarios
98
95
 
99
96
 
100
 
load_tests = load_tests_apply_scenarios
 
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
104
 
102
105
 
103
106
class TestContainsWhitespace(tests.TestCase):
104
107
 
105
108
    def test_contains_whitespace(self):
106
 
        self.assertTrue(osutils.contains_whitespace(u' '))
107
 
        self.assertTrue(osutils.contains_whitespace(u'hello there'))
108
 
        self.assertTrue(osutils.contains_whitespace(u'hellothere\n'))
109
 
        self.assertTrue(osutils.contains_whitespace(u'hello\nthere'))
110
 
        self.assertTrue(osutils.contains_whitespace(u'hello\rthere'))
111
 
        self.assertTrue(osutils.contains_whitespace(u'hello\tthere'))
 
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'))
112
115
 
113
116
        # \xa0 is "Non-breaking-space" which on some python locales thinks it
114
117
        # is whitespace, but we do not.
115
 
        self.assertFalse(osutils.contains_whitespace(u''))
116
 
        self.assertFalse(osutils.contains_whitespace(u'hellothere'))
117
 
        self.assertFalse(osutils.contains_whitespace(u'hello\xa0there'))
 
118
        self.failIf(osutils.contains_whitespace(u''))
 
119
        self.failIf(osutils.contains_whitespace(u'hellothere'))
 
120
        self.failIf(osutils.contains_whitespace(u'hello\xa0there'))
118
121
 
119
122
 
120
123
class TestRename(tests.TestCaseInTempDir):
134
137
        # This should work everywhere
135
138
        self.create_file('a', 'something in a\n')
136
139
        self._fancy_rename('a', 'b')
137
 
        self.assertPathDoesNotExist('a')
138
 
        self.assertPathExists('b')
 
140
        self.failIfExists('a')
 
141
        self.failUnlessExists('b')
139
142
        self.check_file_contents('b', 'something in a\n')
140
143
 
141
144
        self.create_file('a', 'new something in a\n')
148
151
        self.create_file('target', 'data in target\n')
149
152
        self.assertRaises((IOError, OSError), self._fancy_rename,
150
153
                          'missingsource', 'target')
151
 
        self.assertPathExists('target')
 
154
        self.failUnlessExists('target')
152
155
        self.check_file_contents('target', 'data in target\n')
153
156
 
154
157
    def test_fancy_rename_fails_if_source_and_target_missing(self):
159
162
        # Rename should be semi-atomic on all platforms
160
163
        self.create_file('a', 'something in a\n')
161
164
        osutils.rename('a', 'b')
162
 
        self.assertPathDoesNotExist('a')
163
 
        self.assertPathExists('b')
 
165
        self.failIfExists('a')
 
166
        self.failUnlessExists('b')
164
167
        self.check_file_contents('b', 'something in a\n')
165
168
 
166
169
        self.create_file('a', 'new something in a\n')
231
234
            self.assertFalse(osutils.is_inside_or_parent_of_any(dirs, fn))
232
235
 
233
236
 
234
 
class TestLstat(tests.TestCaseInTempDir):
235
 
 
236
 
    def test_lstat_matches_fstat(self):
237
 
        # On Windows, lstat and fstat don't always agree, primarily in the
238
 
        # 'st_ino' and 'st_dev' fields. So we force them to be '0' in our
239
 
        # custom implementation.
240
 
        if sys.platform == 'win32':
241
 
            # We only have special lstat/fstat if we have the extension.
242
 
            # Without it, we may end up re-reading content when we don't have
243
 
            # to, but otherwise it doesn't effect correctness.
244
 
            self.requireFeature(test__walkdirs_win32.win32_readdir_feature)
245
 
        f = open('test-file.txt', 'wb')
246
 
        self.addCleanup(f.close)
247
 
        f.write('some content\n')
248
 
        f.flush()
249
 
        self.assertEqualStat(osutils.fstat(f.fileno()),
250
 
                             osutils.lstat('test-file.txt'))
251
 
 
252
 
 
253
237
class TestRmTree(tests.TestCaseInTempDir):
254
238
 
255
239
    def test_rmtree(self):
267
251
 
268
252
        osutils.rmtree('dir')
269
253
 
270
 
        self.assertPathDoesNotExist('dir/file')
271
 
        self.assertPathDoesNotExist('dir')
 
254
        self.failIfExists('dir/file')
 
255
        self.failIfExists('dir')
272
256
 
273
257
 
274
258
class TestDeleteAny(tests.TestCaseInTempDir):
325
309
        self.assertEqual("/", osutils.kind_marker(osutils._directory_kind))
326
310
        self.assertEqual("@", osutils.kind_marker("symlink"))
327
311
        self.assertEqual("+", osutils.kind_marker("tree-reference"))
328
 
        self.assertEqual("", osutils.kind_marker("fifo"))
329
 
        self.assertEqual("", osutils.kind_marker("socket"))
330
 
        self.assertEqual("", osutils.kind_marker("unknown"))
 
312
        self.assertRaises(errors.BzrError, osutils.kind_marker, "unknown")
331
313
 
332
314
 
333
315
class TestUmask(tests.TestCaseInTempDir):
429
411
class TestLinks(tests.TestCaseInTempDir):
430
412
 
431
413
    def test_dereference_path(self):
432
 
        self.requireFeature(features.SymlinkFeature)
 
414
        self.requireFeature(tests.SymlinkFeature)
433
415
        cwd = osutils.realpath('.')
434
416
        os.mkdir('bar')
435
417
        bar_path = osutils.pathjoin(cwd, 'bar')
482
464
 
483
465
class TestCanonicalRelPath(tests.TestCaseInTempDir):
484
466
 
485
 
    _test_needs_features = [features.CaseInsCasePresFilenameFeature]
 
467
    _test_needs_features = [tests.CaseInsCasePresFilenameFeature]
486
468
 
487
469
    def test_canonical_relpath_simple(self):
488
470
        f = file('MixedCaseName', 'w')
489
471
        f.close()
490
472
        actual = osutils.canonical_relpath(self.test_base_dir, 'mixedcasename')
491
 
        self.assertEqual('work/MixedCaseName', actual)
 
473
        self.failUnlessEqual('work/MixedCaseName', actual)
492
474
 
493
475
    def test_canonical_relpath_missing_tail(self):
494
476
        os.mkdir('MixedCaseParent')
495
477
        actual = osutils.canonical_relpath(self.test_base_dir,
496
478
                                           'mixedcaseparent/nochild')
497
 
        self.assertEqual('work/MixedCaseParent/nochild', actual)
 
479
        self.failUnlessEqual('work/MixedCaseParent/nochild', actual)
498
480
 
499
481
 
500
482
class Test_CICPCanonicalRelpath(tests.TestCaseWithTransport):
876
858
        self.assertEqual('//HOST/path', osutils._win98_abspath('//HOST/path'))
877
859
        # relative path
878
860
        cwd = osutils.getcwd().rstrip('/')
879
 
        drive = osutils.ntpath.splitdrive(cwd)[0]
 
861
        drive = osutils._nt_splitdrive(cwd)[0]
880
862
        self.assertEqual(cwd+'/path', osutils._win98_abspath('path'))
881
863
        self.assertEqual(drive+'/path', osutils._win98_abspath('/path'))
882
864
        # unicode path
888
870
    """Test win32 functions that create files."""
889
871
 
890
872
    def test_getcwd(self):
891
 
        self.requireFeature(features.UnicodeFilenameFeature)
 
873
        self.requireFeature(tests.UnicodeFilenameFeature)
892
874
        os.mkdir(u'mu-\xb5')
893
875
        os.chdir(u'mu-\xb5')
894
876
        # TODO: jam 20060427 This will probably fail on Mac OSX because
924
906
        b.close()
925
907
 
926
908
        osutils._win32_rename('b', 'a')
927
 
        self.assertPathExists('a')
928
 
        self.assertPathDoesNotExist('b')
 
909
        self.failUnlessExists('a')
 
910
        self.failIfExists('b')
929
911
        self.assertFileEqual('baz\n', 'a')
930
912
 
931
913
    def test_rename_missing_file(self):
984
966
    """Test mac special functions that require directories."""
985
967
 
986
968
    def test_getcwd(self):
987
 
        self.requireFeature(features.UnicodeFilenameFeature)
 
969
        self.requireFeature(tests.UnicodeFilenameFeature)
988
970
        os.mkdir(u'B\xe5gfors')
989
971
        os.chdir(u'B\xe5gfors')
990
972
        self.assertEndsWith(osutils._mac_getcwd(), u'B\xe5gfors')
991
973
 
992
974
    def test_getcwd_nonnorm(self):
993
 
        self.requireFeature(features.UnicodeFilenameFeature)
 
975
        self.requireFeature(tests.UnicodeFilenameFeature)
994
976
        # Test that _mac_getcwd() will normalize this path
995
977
        os.mkdir(u'Ba\u030agfors')
996
978
        os.chdir(u'Ba\u030agfors')
1079
1061
        self.assertExpectedBlocks(expected_dirblocks[1:], result)
1080
1062
 
1081
1063
    def test_walkdirs_os_error(self):
1082
 
        # <https://bugs.launchpad.net/bzr/+bug/338653>
 
1064
        # <https://bugs.edge.launchpad.net/bzr/+bug/338653>
1083
1065
        # Pyrex readdir didn't raise useful messages if it had an error
1084
1066
        # reading the directory
1085
1067
        if sys.platform == 'win32':
1086
1068
            raise tests.TestNotApplicable(
1087
1069
                "readdir IOError not tested on win32")
1088
 
        self.requireFeature(features.not_running_as_root)
1089
1070
        os.mkdir("test-unreadable")
1090
1071
        os.chmod("test-unreadable", 0000)
1091
1072
        # must chmod it back so that it can be removed
1099
1080
        # Ensure the message contains the file name
1100
1081
        self.assertContainsRe(str(e), "\./test-unreadable")
1101
1082
 
1102
 
 
1103
 
    def test_walkdirs_encoding_error(self):
1104
 
        # <https://bugs.launchpad.net/bzr/+bug/488519>
1105
 
        # walkdirs didn't raise a useful message when the filenames
1106
 
        # are not using the filesystem's encoding
1107
 
 
1108
 
        # require a bytestring based filesystem
1109
 
        self.requireFeature(features.ByteStringNamedFilesystem)
1110
 
 
1111
 
        tree = [
1112
 
            '.bzr',
1113
 
            '0file',
1114
 
            '1dir/',
1115
 
            '1dir/0file',
1116
 
            '1dir/1dir/',
1117
 
            '1file'
1118
 
            ]
1119
 
 
1120
 
        self.build_tree(tree)
1121
 
 
1122
 
        # rename the 1file to a latin-1 filename
1123
 
        os.rename("./1file", "\xe8file")
1124
 
        if "\xe8file" not in os.listdir("."):
1125
 
            self.skip("Lack filesystem that preserves arbitrary bytes")
1126
 
 
1127
 
        self._save_platform_info()
1128
 
        win32utils.winver = None # Avoid the win32 detection code
1129
 
        osutils._fs_enc = 'UTF-8'
1130
 
 
1131
 
        # this should raise on error
1132
 
        def attempt():
1133
 
            for dirdetail, dirblock in osutils.walkdirs('.'):
1134
 
                pass
1135
 
 
1136
 
        self.assertRaises(errors.BadFilenameEncoding, attempt)
1137
 
 
1138
1083
    def test__walkdirs_utf8(self):
1139
1084
        tree = [
1140
1085
            '.bzr',
1190
1135
            dirblock[:] = new_dirblock
1191
1136
 
1192
1137
    def _save_platform_info(self):
1193
 
        self.overrideAttr(win32utils, 'winver')
1194
 
        self.overrideAttr(osutils, '_fs_enc')
1195
 
        self.overrideAttr(osutils, '_selected_dir_reader')
 
1138
        cur_winver = win32utils.winver
 
1139
        cur_fs_enc = osutils._fs_enc
 
1140
        cur_dir_reader = osutils._selected_dir_reader
 
1141
        def restore():
 
1142
            win32utils.winver = cur_winver
 
1143
            osutils._fs_enc = cur_fs_enc
 
1144
            osutils._selected_dir_reader = cur_dir_reader
 
1145
        self.addCleanup(restore)
1196
1146
 
1197
1147
    def assertDirReaderIs(self, expected):
1198
1148
        """Assert the right implementation for _walkdirs_utf8 is chosen."""
1207
1157
        self._save_platform_info()
1208
1158
        win32utils.winver = None # Avoid the win32 detection code
1209
1159
        osutils._fs_enc = 'UTF-8'
1210
 
        self.assertDirReaderIs(
1211
 
            UTF8DirReaderFeature.module.UTF8DirReader)
 
1160
        self.assertDirReaderIs(UTF8DirReaderFeature.reader)
1212
1161
 
1213
1162
    def test_force_walkdirs_utf8_fs_ascii(self):
1214
1163
        self.requireFeature(UTF8DirReaderFeature)
1215
1164
        self._save_platform_info()
1216
1165
        win32utils.winver = None # Avoid the win32 detection code
1217
1166
        osutils._fs_enc = 'US-ASCII'
1218
 
        self.assertDirReaderIs(
1219
 
            UTF8DirReaderFeature.module.UTF8DirReader)
 
1167
        self.assertDirReaderIs(UTF8DirReaderFeature.reader)
1220
1168
 
1221
1169
    def test_force_walkdirs_utf8_fs_ANSI(self):
1222
1170
        self.requireFeature(UTF8DirReaderFeature)
1223
1171
        self._save_platform_info()
1224
1172
        win32utils.winver = None # Avoid the win32 detection code
1225
1173
        osutils._fs_enc = 'ANSI_X3.4-1968'
1226
 
        self.assertDirReaderIs(
1227
 
            UTF8DirReaderFeature.module.UTF8DirReader)
 
1174
        self.assertDirReaderIs(UTF8DirReaderFeature.reader)
1228
1175
 
1229
1176
    def test_force_walkdirs_utf8_fs_latin1(self):
1230
1177
        self._save_platform_info()
1248
1195
 
1249
1196
    def test_unicode_walkdirs(self):
1250
1197
        """Walkdirs should always return unicode paths."""
1251
 
        self.requireFeature(features.UnicodeFilenameFeature)
 
1198
        self.requireFeature(tests.UnicodeFilenameFeature)
1252
1199
        name0 = u'0file-\xb6'
1253
1200
        name1 = u'1dir-\u062c\u0648'
1254
1201
        name2 = u'2file-\u0633'
1291
1238
 
1292
1239
        The abspath portion might be in unicode or utf-8
1293
1240
        """
1294
 
        self.requireFeature(features.UnicodeFilenameFeature)
 
1241
        self.requireFeature(tests.UnicodeFilenameFeature)
1295
1242
        name0 = u'0file-\xb6'
1296
1243
        name1 = u'1dir-\u062c\u0648'
1297
1244
        name2 = u'2file-\u0633'
1352
1299
 
1353
1300
        The abspath portion should be in unicode
1354
1301
        """
1355
 
        self.requireFeature(features.UnicodeFilenameFeature)
 
1302
        self.requireFeature(tests.UnicodeFilenameFeature)
1356
1303
        # Use the unicode reader. TODO: split into driver-and-driven unit
1357
1304
        # tests.
1358
1305
        self._save_platform_info()
1399
1346
 
1400
1347
    def test__walkdirs_utf8_win32readdir(self):
1401
1348
        self.requireFeature(test__walkdirs_win32.win32_readdir_feature)
1402
 
        self.requireFeature(features.UnicodeFilenameFeature)
 
1349
        self.requireFeature(tests.UnicodeFilenameFeature)
1403
1350
        from bzrlib._walkdirs_win32 import Win32ReadDir
1404
1351
        self._save_platform_info()
1405
1352
        osutils._selected_dir_reader = Win32ReadDir()
1456
1403
    def test__walkdirs_utf_win32_find_file_stat_file(self):
1457
1404
        """make sure our Stat values are valid"""
1458
1405
        self.requireFeature(test__walkdirs_win32.win32_readdir_feature)
1459
 
        self.requireFeature(features.UnicodeFilenameFeature)
 
1406
        self.requireFeature(tests.UnicodeFilenameFeature)
1460
1407
        from bzrlib._walkdirs_win32 import Win32ReadDir
1461
1408
        name0u = u'0file-\xb6'
1462
1409
        name0 = name0u.encode('utf8')
1480
1427
    def test__walkdirs_utf_win32_find_file_stat_directory(self):
1481
1428
        """make sure our Stat values are valid"""
1482
1429
        self.requireFeature(test__walkdirs_win32.win32_readdir_feature)
1483
 
        self.requireFeature(features.UnicodeFilenameFeature)
 
1430
        self.requireFeature(tests.UnicodeFilenameFeature)
1484
1431
        from bzrlib._walkdirs_win32 import Win32ReadDir
1485
1432
        name0u = u'0dir-\u062c\u0648'
1486
1433
        name0 = name0u.encode('utf8')
1586
1533
        self.assertEqual(['c'], os.listdir('target/b'))
1587
1534
 
1588
1535
    def test_copy_tree_symlinks(self):
1589
 
        self.requireFeature(features.SymlinkFeature)
 
1536
        self.requireFeature(tests.SymlinkFeature)
1590
1537
        self.build_tree(['source/'])
1591
1538
        os.symlink('a/generic/path', 'source/lnk')
1592
1539
        osutils.copy_tree('source', 'target')
1617
1564
                          ('d', 'source/b', 'target/b'),
1618
1565
                          ('f', 'source/b/c', 'target/b/c'),
1619
1566
                         ], processed_files)
1620
 
        self.assertPathDoesNotExist('target')
 
1567
        self.failIfExists('target')
1621
1568
        if osutils.has_symlinks():
1622
1569
            self.assertEqual([('source/lnk', 'target/lnk')], processed_links)
1623
1570
 
1634
1581
        def cleanup():
1635
1582
            if 'BZR_TEST_ENV_VAR' in os.environ:
1636
1583
                del os.environ['BZR_TEST_ENV_VAR']
 
1584
 
1637
1585
        self.addCleanup(cleanup)
1638
1586
 
1639
1587
    def test_set(self):
1669
1617
        old = osutils.set_or_unset_env('BZR_TEST_ENV_VAR', None)
1670
1618
        self.assertEqual('foo', old)
1671
1619
        self.assertEqual(None, os.environ.get('BZR_TEST_ENV_VAR'))
1672
 
        self.assertFalse('BZR_TEST_ENV_VAR' in os.environ)
 
1620
        self.failIf('BZR_TEST_ENV_VAR' in os.environ)
1673
1621
 
1674
1622
 
1675
1623
class TestSizeShaFile(tests.TestCaseInTempDir):
1726
1674
 
1727
1675
class TestReCompile(tests.TestCase):
1728
1676
 
1729
 
    def _deprecated_re_compile_checked(self, *args, **kwargs):
1730
 
        return self.applyDeprecated(symbol_versioning.deprecated_in((2, 2, 0)),
1731
 
            osutils.re_compile_checked, *args, **kwargs)
1732
 
 
1733
1677
    def test_re_compile_checked(self):
1734
 
        r = self._deprecated_re_compile_checked(r'A*', re.IGNORECASE)
 
1678
        r = osutils.re_compile_checked(r'A*', re.IGNORECASE)
1735
1679
        self.assertTrue(r.match('aaaa'))
1736
1680
        self.assertTrue(r.match('aAaA'))
1737
1681
 
1738
1682
    def test_re_compile_checked_error(self):
1739
1683
        # like https://bugs.launchpad.net/bzr/+bug/251352
1740
 
 
1741
 
        # Due to possible test isolation error, re.compile is not lazy at
1742
 
        # this point. We re-install lazy compile.
1743
 
        lazy_regex.install_lazy_compile()
1744
1684
        err = self.assertRaises(
1745
1685
            errors.BzrCommandError,
1746
 
            self._deprecated_re_compile_checked, '*', re.IGNORECASE, 'test case')
 
1686
            osutils.re_compile_checked, '*', re.IGNORECASE, 'test case')
1747
1687
        self.assertEqual(
1748
 
            'Invalid regular expression in test case: '
1749
 
            '"*" nothing to repeat',
 
1688
            "Invalid regular expression in test case: '*': "
 
1689
            "nothing to repeat",
1750
1690
            str(err))
1751
1691
 
1752
1692
 
1753
1693
class TestDirReader(tests.TestCaseInTempDir):
1754
1694
 
1755
 
    scenarios = dir_reader_scenarios()
1756
 
 
1757
1695
    # Set by load_tests
1758
1696
    _dir_reader_class = None
1759
1697
    _native_to_unicode = None
1760
1698
 
1761
1699
    def setUp(self):
1762
1700
        tests.TestCaseInTempDir.setUp(self)
1763
 
        self.overrideAttr(osutils,
1764
 
                          '_selected_dir_reader', self._dir_reader_class())
 
1701
 
 
1702
        # Save platform specific info and reset it
 
1703
        cur_dir_reader = osutils._selected_dir_reader
 
1704
 
 
1705
        def restore():
 
1706
            osutils._selected_dir_reader = cur_dir_reader
 
1707
        self.addCleanup(restore)
 
1708
 
 
1709
        osutils._selected_dir_reader = self._dir_reader_class()
1765
1710
 
1766
1711
    def _get_ascii_tree(self):
1767
1712
        tree = [
1859
1804
        return filtered_dirblocks
1860
1805
 
1861
1806
    def test_walk_unicode_tree(self):
1862
 
        self.requireFeature(features.UnicodeFilenameFeature)
 
1807
        self.requireFeature(tests.UnicodeFilenameFeature)
1863
1808
        tree, expected_dirblocks = self._get_unicode_tree()
1864
1809
        self.build_tree(tree)
1865
1810
        result = list(osutils._walkdirs_utf8('.'))
1866
1811
        self.assertEqual(expected_dirblocks, self._filter_out(result))
1867
1812
 
1868
1813
    def test_symlink(self):
1869
 
        self.requireFeature(features.SymlinkFeature)
1870
 
        self.requireFeature(features.UnicodeFilenameFeature)
 
1814
        self.requireFeature(tests.SymlinkFeature)
 
1815
        self.requireFeature(tests.UnicodeFilenameFeature)
1871
1816
        target = u'target\N{Euro Sign}'
1872
1817
        link_name = u'l\N{Euro Sign}nk'
1873
1818
        os.symlink(target, link_name)
1891
1836
    But prior python versions failed to properly encode the passed unicode
1892
1837
    string.
1893
1838
    """
1894
 
    _test_needs_features = [features.SymlinkFeature, features.UnicodeFilenameFeature]
 
1839
    _test_needs_features = [tests.SymlinkFeature, tests.UnicodeFilenameFeature]
1895
1840
 
1896
1841
    def setUp(self):
1897
1842
        super(tests.TestCaseInTempDir, self).setUp()
1900
1845
        os.symlink(self.target, self.link)
1901
1846
 
1902
1847
    def test_os_readlink_link_encoding(self):
1903
 
        self.assertEquals(self.target,  os.readlink(self.link))
 
1848
        if sys.version_info < (2, 6):
 
1849
            self.assertRaises(UnicodeEncodeError, os.readlink, self.link)
 
1850
        else:
 
1851
            self.assertEquals(self.target,  os.readlink(self.link))
1904
1852
 
1905
1853
    def test_os_readlink_link_decoding(self):
1906
1854
        self.assertEquals(self.target.encode(osutils._fs_enc),
1911
1859
 
1912
1860
    def setUp(self):
1913
1861
        super(TestConcurrency, self).setUp()
1914
 
        self.overrideAttr(osutils, '_cached_local_concurrency')
 
1862
        orig = osutils._cached_local_concurrency
 
1863
        def restore():
 
1864
            osutils._cached_local_concurrency = orig
 
1865
        self.addCleanup(restore)
1915
1866
 
1916
1867
    def test_local_concurrency(self):
1917
1868
        concurrency = osutils.local_concurrency()
1918
1869
        self.assertIsInstance(concurrency, int)
1919
1870
 
1920
1871
    def test_local_concurrency_environment_variable(self):
1921
 
        self.overrideEnv('BZR_CONCURRENCY', '2')
 
1872
        os.environ['BZR_CONCURRENCY'] = '2'
1922
1873
        self.assertEqual(2, osutils.local_concurrency(use_cache=False))
1923
 
        self.overrideEnv('BZR_CONCURRENCY', '3')
 
1874
        os.environ['BZR_CONCURRENCY'] = '3'
1924
1875
        self.assertEqual(3, osutils.local_concurrency(use_cache=False))
1925
 
        self.overrideEnv('BZR_CONCURRENCY', 'foo')
 
1876
        os.environ['BZR_CONCURRENCY'] = 'foo'
1926
1877
        self.assertEqual(1, osutils.local_concurrency(use_cache=False))
1927
1878
 
1928
1879
    def test_option_concurrency(self):
1929
 
        self.overrideEnv('BZR_CONCURRENCY', '1')
 
1880
        os.environ['BZR_CONCURRENCY'] = '1'
1930
1881
        self.run_bzr('rocks --concurrency 42')
1931
 
        # Command line overrides environment variable
 
1882
        # Command line overrides envrionment variable
1932
1883
        self.assertEquals('42', os.environ['BZR_CONCURRENCY'])
1933
1884
        self.assertEquals(42, osutils.local_concurrency(use_cache=False))
1934
1885
 
1944
1895
 
1945
1896
    def setUp(self):
1946
1897
        super(TestFailedToLoadExtension, self).setUp()
1947
 
        self.overrideAttr(osutils, '_extension_load_failures', [])
 
1898
        self.saved_failures = osutils._extension_load_failures[:]
 
1899
        del osutils._extension_load_failures[:]
 
1900
        self.addCleanup(self.restore_failures)
 
1901
 
 
1902
    def restore_failures(self):
 
1903
        osutils._extension_load_failures = self.saved_failures
1948
1904
 
1949
1905
    def test_failure_to_load(self):
1950
1906
        self._try_loading()
1972
1928
 
1973
1929
class TestTerminalWidth(tests.TestCase):
1974
1930
 
1975
 
    def setUp(self):
1976
 
        tests.TestCase.setUp(self)
1977
 
        self._orig_terminal_size_state = osutils._terminal_size_state
1978
 
        self._orig_first_terminal_size = osutils._first_terminal_size
1979
 
        self.addCleanup(self.restore_osutils_globals)
1980
 
        osutils._terminal_size_state = 'no_data'
1981
 
        osutils._first_terminal_size = None
1982
 
 
1983
 
    def restore_osutils_globals(self):
1984
 
        osutils._terminal_size_state = self._orig_terminal_size_state
1985
 
        osutils._first_terminal_size = self._orig_first_terminal_size
1986
 
 
1987
1931
    def replace_stdout(self, new):
1988
 
        self.overrideAttr(sys, 'stdout', new)
 
1932
        orig_stdout = sys.stdout
 
1933
        def restore():
 
1934
            sys.stdout = orig_stdout
 
1935
        self.addCleanup(restore)
 
1936
        sys.stdout = new
1989
1937
 
1990
1938
    def replace__terminal_size(self, new):
1991
 
        self.overrideAttr(osutils, '_terminal_size', new)
 
1939
        orig__terminal_size = osutils._terminal_size
 
1940
        def restore():
 
1941
            osutils._terminal_size = orig__terminal_size
 
1942
        self.addCleanup(restore)
 
1943
        osutils._terminal_size = new
1992
1944
 
1993
1945
    def set_fake_tty(self):
1994
1946
 
2004
1956
    def test_defaults_to_BZR_COLUMNS(self):
2005
1957
        # BZR_COLUMNS is set by the test framework
2006
1958
        self.assertNotEqual('12', os.environ['BZR_COLUMNS'])
2007
 
        self.overrideEnv('BZR_COLUMNS', '12')
 
1959
        os.environ['BZR_COLUMNS'] = '12'
2008
1960
        self.assertEqual(12, osutils.terminal_width())
2009
1961
 
2010
 
    def test_BZR_COLUMNS_0_no_limit(self):
2011
 
        self.overrideEnv('BZR_COLUMNS', '0')
2012
 
        self.assertEqual(None, osutils.terminal_width())
2013
 
 
2014
1962
    def test_falls_back_to_COLUMNS(self):
2015
 
        self.overrideEnv('BZR_COLUMNS', None)
 
1963
        del os.environ['BZR_COLUMNS']
2016
1964
        self.assertNotEqual('42', os.environ['COLUMNS'])
2017
1965
        self.set_fake_tty()
2018
 
        self.overrideEnv('COLUMNS', '42')
 
1966
        os.environ['COLUMNS'] = '42'
2019
1967
        self.assertEqual(42, osutils.terminal_width())
2020
1968
 
2021
1969
    def test_tty_default_without_columns(self):
2022
 
        self.overrideEnv('BZR_COLUMNS', None)
2023
 
        self.overrideEnv('COLUMNS', None)
 
1970
        del os.environ['BZR_COLUMNS']
 
1971
        del os.environ['COLUMNS']
2024
1972
 
2025
1973
        def terminal_size(w, h):
2026
1974
            return 42, 42
2033
1981
        self.assertEqual(42, osutils.terminal_width())
2034
1982
 
2035
1983
    def test_non_tty_default_without_columns(self):
2036
 
        self.overrideEnv('BZR_COLUMNS', None)
2037
 
        self.overrideEnv('COLUMNS', None)
 
1984
        del os.environ['BZR_COLUMNS']
 
1985
        del os.environ['COLUMNS']
2038
1986
        self.replace_stdout(None)
2039
1987
        self.assertEqual(None, osutils.terminal_width())
2040
1988
 
2048
1996
            # We won't remove TIOCGWINSZ, because it doesn't exist anyway :)
2049
1997
            pass
2050
1998
        else:
2051
 
            self.overrideAttr(termios, 'TIOCGWINSZ')
 
1999
            def restore():
 
2000
                termios.TIOCGWINSZ = orig
 
2001
            self.addCleanup(restore)
2052
2002
            del termios.TIOCGWINSZ
2053
 
        self.overrideEnv('BZR_COLUMNS', None)
2054
 
        self.overrideEnv('COLUMNS', None)
 
2003
        del os.environ['BZR_COLUMNS']
 
2004
        del os.environ['COLUMNS']
2055
2005
        # Whatever the result is, if we don't raise an exception, it's ok.
2056
2006
        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)