← 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: 2009-07-02 03:48:25 UTC
  • mto: This revision was merged to the branch mainline in revision 4502.
  • Revision ID: mbp@sourcefrog.net-20090702034825-2330z69r93a701ws
iter(i) returns i so we don't need a check

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
20
20
import errno
21
21
import os
22
22
import re
23
 
import select
24
23
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,
31
30
    osutils,
32
 
    symbol_versioning,
33
31
    tests,
34
 
    trace,
35
32
    win32utils,
36
33
    )
37
34
from bzrlib.tests import (
38
 
    features,
39
35
    file_utils,
40
36
    test__walkdirs_win32,
41
37
    )
42
 
from bzrlib.tests.scenarios import load_tests_apply_scenarios
43
 
 
44
 
 
45
 
class _UTF8DirReaderFeature(features.Feature):
 
38
 
 
39
 
 
40
class _UTF8DirReaderFeature(tests.Feature):
46
41
 
47
42
    def _probe(self):
48
43
        try:
55
50
    def feature_name(self):
56
51
        return 'bzrlib._readdir_pyx'
57
52
 
58
 
UTF8DirReaderFeature = features.ModuleAvailableFeature('bzrlib._readdir_pyx')
59
 
 
60
 
term_ios_feature = features.ModuleAvailableFeature('termios')
 
53
UTF8DirReaderFeature = _UTF8DirReaderFeature()
61
54
 
62
55
 
63
56
def _already_unicode(s):
64
57
    return s
65
58
 
66
59
 
 
60
def _fs_enc_to_unicode(s):
 
61
    return s.decode(osutils._fs_enc)
 
62
 
 
63
 
67
64
def _utf8_to_unicode(s):
68
65
    return s.decode('UTF-8')
69
66
 
86
83
                          dict(_dir_reader_class=_readdir_pyx.UTF8DirReader,
87
84
                               _native_to_unicode=_utf8_to_unicode)))
88
85
 
89
 
    if test__walkdirs_win32.win32_readdir_feature.available():
 
86
    if test__walkdirs_win32.Win32ReadDirFeature.available():
90
87
        try:
91
88
            from bzrlib import _walkdirs_win32
 
89
            # TODO: check on windows, it may be that we need to use/add
 
90
            # safe_unicode instead of _fs_enc_to_unicode
92
91
            scenarios.append(
93
92
                ('win32',
94
93
                 dict(_dir_reader_class=_walkdirs_win32.Win32ReadDir,
95
 
                      _native_to_unicode=_already_unicode)))
 
94
                      _native_to_unicode=_fs_enc_to_unicode)))
96
95
        except ImportError:
97
96
            pass
98
97
    return scenarios
99
98
 
100
99
 
101
 
load_tests = load_tests_apply_scenarios
 
100
def load_tests(basic_tests, module, loader):
 
101
    suite = loader.suiteClass()
 
102
    dir_reader_tests, remaining_tests = tests.split_suite_by_condition(
 
103
        basic_tests, tests.condition_isinstance(TestDirReader))
 
104
    tests.multiply_tests(dir_reader_tests, dir_reader_scenarios(), suite)
 
105
    suite.addTest(remaining_tests)
 
106
    return suite
102
107
 
103
108
 
104
109
class TestContainsWhitespace(tests.TestCase):
105
110
 
106
111
    def test_contains_whitespace(self):
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'))
 
112
        self.failUnless(osutils.contains_whitespace(u' '))
 
113
        self.failUnless(osutils.contains_whitespace(u'hello there'))
 
114
        self.failUnless(osutils.contains_whitespace(u'hellothere\n'))
 
115
        self.failUnless(osutils.contains_whitespace(u'hello\nthere'))
 
116
        self.failUnless(osutils.contains_whitespace(u'hello\rthere'))
 
117
        self.failUnless(osutils.contains_whitespace(u'hello\tthere'))
113
118
 
114
119
        # \xa0 is "Non-breaking-space" which on some python locales thinks it
115
120
        # is whitespace, but we do not.
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
        self.failIf(osutils.contains_whitespace(u''))
 
122
        self.failIf(osutils.contains_whitespace(u'hellothere'))
 
123
        self.failIf(osutils.contains_whitespace(u'hello\xa0there'))
119
124
 
120
125
 
121
126
class TestRename(tests.TestCaseInTempDir):
122
127
 
123
 
    def create_file(self, filename, content):
124
 
        f = open(filename, 'wb')
125
 
        try:
126
 
            f.write(content)
127
 
        finally:
128
 
            f.close()
129
 
 
130
 
    def _fancy_rename(self, a, b):
131
 
        osutils.fancy_rename(a, b, rename_func=os.rename,
132
 
                             unlink_func=os.unlink)
133
 
 
134
128
    def test_fancy_rename(self):
135
129
        # This should work everywhere
136
 
        self.create_file('a', 'something in a\n')
137
 
        self._fancy_rename('a', 'b')
138
 
        self.assertPathDoesNotExist('a')
139
 
        self.assertPathExists('b')
 
130
        def rename(a, b):
 
131
            osutils.fancy_rename(a, b,
 
132
                    rename_func=os.rename,
 
133
                    unlink_func=os.unlink)
 
134
 
 
135
        open('a', 'wb').write('something in a\n')
 
136
        rename('a', 'b')
 
137
        self.failIfExists('a')
 
138
        self.failUnlessExists('b')
140
139
        self.check_file_contents('b', 'something in a\n')
141
140
 
142
 
        self.create_file('a', 'new something in a\n')
143
 
        self._fancy_rename('b', 'a')
 
141
        open('a', 'wb').write('new something in a\n')
 
142
        rename('b', 'a')
144
143
 
145
144
        self.check_file_contents('a', 'something in a\n')
146
145
 
147
 
    def test_fancy_rename_fails_source_missing(self):
148
 
        # An exception should be raised, and the target should be left in place
149
 
        self.create_file('target', 'data in target\n')
150
 
        self.assertRaises((IOError, OSError), self._fancy_rename,
151
 
                          'missingsource', 'target')
152
 
        self.assertPathExists('target')
153
 
        self.check_file_contents('target', 'data in target\n')
154
 
 
155
 
    def test_fancy_rename_fails_if_source_and_target_missing(self):
156
 
        self.assertRaises((IOError, OSError), self._fancy_rename,
157
 
                          'missingsource', 'missingtarget')
158
 
 
159
146
    def test_rename(self):
160
147
        # Rename should be semi-atomic on all platforms
161
 
        self.create_file('a', 'something in a\n')
 
148
        open('a', 'wb').write('something in a\n')
162
149
        osutils.rename('a', 'b')
163
 
        self.assertPathDoesNotExist('a')
164
 
        self.assertPathExists('b')
 
150
        self.failIfExists('a')
 
151
        self.failUnlessExists('b')
165
152
        self.check_file_contents('b', 'something in a\n')
166
153
 
167
 
        self.create_file('a', 'new something in a\n')
 
154
        open('a', 'wb').write('new something in a\n')
168
155
        osutils.rename('b', 'a')
169
156
 
170
157
        self.check_file_contents('a', 'something in a\n')
181
168
        shape = sorted(os.listdir('.'))
182
169
        self.assertEquals(['A', 'B'], shape)
183
170
 
184
 
    def test_rename_exception(self):
185
 
        try:
186
 
            osutils.rename('nonexistent_path', 'different_nonexistent_path')
187
 
        except OSError, e:
188
 
            self.assertEqual(e.old_filename, 'nonexistent_path')
189
 
            self.assertEqual(e.new_filename, 'different_nonexistent_path')
190
 
            self.assertTrue('nonexistent_path' in e.strerror)
191
 
            self.assertTrue('different_nonexistent_path' in e.strerror)
192
 
 
193
171
 
194
172
class TestRandChars(tests.TestCase):
195
173
 
241
219
            self.assertFalse(osutils.is_inside_or_parent_of_any(dirs, fn))
242
220
 
243
221
 
244
 
class TestLstat(tests.TestCaseInTempDir):
245
 
 
246
 
    def test_lstat_matches_fstat(self):
247
 
        # On Windows, lstat and fstat don't always agree, primarily in the
248
 
        # 'st_ino' and 'st_dev' fields. So we force them to be '0' in our
249
 
        # custom implementation.
250
 
        if sys.platform == 'win32':
251
 
            # We only have special lstat/fstat if we have the extension.
252
 
            # Without it, we may end up re-reading content when we don't have
253
 
            # to, but otherwise it doesn't effect correctness.
254
 
            self.requireFeature(test__walkdirs_win32.win32_readdir_feature)
255
 
        f = open('test-file.txt', 'wb')
256
 
        self.addCleanup(f.close)
257
 
        f.write('some content\n')
258
 
        f.flush()
259
 
        self.assertEqualStat(osutils.fstat(f.fileno()),
260
 
                             osutils.lstat('test-file.txt'))
261
 
 
262
 
 
263
222
class TestRmTree(tests.TestCaseInTempDir):
264
223
 
265
224
    def test_rmtree(self):
277
236
 
278
237
        osutils.rmtree('dir')
279
238
 
280
 
        self.assertPathDoesNotExist('dir/file')
281
 
        self.assertPathDoesNotExist('dir')
 
239
        self.failIfExists('dir/file')
 
240
        self.failIfExists('dir')
282
241
 
283
242
 
284
243
class TestDeleteAny(tests.TestCaseInTempDir):
335
294
        self.assertEqual("/", osutils.kind_marker(osutils._directory_kind))
336
295
        self.assertEqual("@", osutils.kind_marker("symlink"))
337
296
        self.assertEqual("+", osutils.kind_marker("tree-reference"))
338
 
        self.assertEqual("", osutils.kind_marker("fifo"))
339
 
        self.assertEqual("", osutils.kind_marker("socket"))
340
 
        self.assertEqual("", osutils.kind_marker("unknown"))
 
297
        self.assertRaises(errors.BzrError, osutils.kind_marker, "unknown")
341
298
 
342
299
 
343
300
class TestUmask(tests.TestCaseInTempDir):
408
365
        # Instead blackbox.test_locale should check for localized
409
366
        # dates once they do occur in output strings.
410
367
 
411
 
    def test_format_date_with_offset_in_original_timezone(self):
412
 
        self.assertEqual("Thu 1970-01-01 00:00:00 +0000",
413
 
            osutils.format_date_with_offset_in_original_timezone(0))
414
 
        self.assertEqual("Fri 1970-01-02 03:46:40 +0000",
415
 
            osutils.format_date_with_offset_in_original_timezone(100000))
416
 
        self.assertEqual("Fri 1970-01-02 05:46:40 +0200",
417
 
            osutils.format_date_with_offset_in_original_timezone(100000, 7200))
418
 
 
419
368
    def test_local_time_offset(self):
420
369
        """Test that local_time_offset() returns a sane value."""
421
370
        offset = osutils.local_time_offset()
439
388
class TestLinks(tests.TestCaseInTempDir):
440
389
 
441
390
    def test_dereference_path(self):
442
 
        self.requireFeature(features.SymlinkFeature)
 
391
        self.requireFeature(tests.SymlinkFeature)
443
392
        cwd = osutils.realpath('.')
444
393
        os.mkdir('bar')
445
394
        bar_path = osutils.pathjoin(cwd, 'bar')
492
441
 
493
442
class TestCanonicalRelPath(tests.TestCaseInTempDir):
494
443
 
495
 
    _test_needs_features = [features.CaseInsCasePresFilenameFeature]
 
444
    _test_needs_features = [tests.CaseInsCasePresFilenameFeature]
496
445
 
497
446
    def test_canonical_relpath_simple(self):
498
447
        f = file('MixedCaseName', 'w')
499
448
        f.close()
500
 
        actual = osutils.canonical_relpath(self.test_base_dir, 'mixedcasename')
501
 
        self.assertEqual('work/MixedCaseName', actual)
 
449
        # Watch out for tricky test dir (on OSX /tmp -> /private/tmp)
 
450
        real_base_dir = osutils.realpath(self.test_base_dir)
 
451
        actual = osutils.canonical_relpath(real_base_dir, 'mixedcasename')
 
452
        self.failUnlessEqual('work/MixedCaseName', actual)
502
453
 
503
454
    def test_canonical_relpath_missing_tail(self):
504
455
        os.mkdir('MixedCaseParent')
505
 
        actual = osutils.canonical_relpath(self.test_base_dir,
 
456
        # Watch out for tricky test dir (on OSX /tmp -> /private/tmp)
 
457
        real_base_dir = osutils.realpath(self.test_base_dir)
 
458
        actual = osutils.canonical_relpath(real_base_dir,
506
459
                                           'mixedcaseparent/nochild')
507
 
        self.assertEqual('work/MixedCaseParent/nochild', actual)
508
 
 
509
 
 
510
 
class Test_CICPCanonicalRelpath(tests.TestCaseWithTransport):
511
 
 
512
 
    def assertRelpath(self, expected, base, path):
513
 
        actual = osutils._cicp_canonical_relpath(base, path)
514
 
        self.assertEqual(expected, actual)
515
 
 
516
 
    def test_simple(self):
517
 
        self.build_tree(['MixedCaseName'])
518
 
        base = osutils.realpath(self.get_transport('.').local_abspath('.'))
519
 
        self.assertRelpath('MixedCaseName', base, 'mixedcAsename')
520
 
 
521
 
    def test_subdir_missing_tail(self):
522
 
        self.build_tree(['MixedCaseParent/', 'MixedCaseParent/a_child'])
523
 
        base = osutils.realpath(self.get_transport('.').local_abspath('.'))
524
 
        self.assertRelpath('MixedCaseParent/a_child', base,
525
 
                           'MixedCaseParent/a_child')
526
 
        self.assertRelpath('MixedCaseParent/a_child', base,
527
 
                           'MixedCaseParent/A_Child')
528
 
        self.assertRelpath('MixedCaseParent/not_child', base,
529
 
                           'MixedCaseParent/not_child')
530
 
 
531
 
    def test_at_root_slash(self):
532
 
        # We can't test this on Windows, because it has a 'MIN_ABS_PATHLENGTH'
533
 
        # check...
534
 
        if osutils.MIN_ABS_PATHLENGTH > 1:
535
 
            raise tests.TestSkipped('relpath requires %d chars'
536
 
                                    % osutils.MIN_ABS_PATHLENGTH)
537
 
        self.assertRelpath('foo', '/', '/foo')
538
 
 
539
 
    def test_at_root_drive(self):
540
 
        if sys.platform != 'win32':
541
 
            raise tests.TestNotApplicable('we can only test drive-letter relative'
542
 
                                          ' paths on Windows where we have drive'
543
 
                                          ' letters.')
544
 
        # see bug #322807
545
 
        # The specific issue is that when at the root of a drive, 'abspath'
546
 
        # returns "C:/" or just "/". However, the code assumes that abspath
547
 
        # always returns something like "C:/foo" or "/foo" (no trailing slash).
548
 
        self.assertRelpath('foo', 'C:/', 'C:/foo')
549
 
        self.assertRelpath('foo', 'X:/', 'X:/foo')
550
 
        self.assertRelpath('foo', 'X:/', 'X://foo')
 
460
        self.failUnlessEqual('work/MixedCaseParent/nochild', actual)
551
461
 
552
462
 
553
463
class TestPumpFile(tests.TestCase):
715
625
        self.assertEqual("1234", output.getvalue())
716
626
 
717
627
 
718
 
class TestRelpath(tests.TestCase):
719
 
 
720
 
    def test_simple_relpath(self):
721
 
        cwd = osutils.getcwd()
722
 
        subdir = cwd + '/subdir'
723
 
        self.assertEqual('subdir', osutils.relpath(cwd, subdir))
724
 
 
725
 
    def test_deep_relpath(self):
726
 
        cwd = osutils.getcwd()
727
 
        subdir = cwd + '/sub/subsubdir'
728
 
        self.assertEqual('sub/subsubdir', osutils.relpath(cwd, subdir))
729
 
 
730
 
    def test_not_relative(self):
731
 
        self.assertRaises(errors.PathNotChild,
732
 
                          osutils.relpath, 'C:/path', 'H:/path')
733
 
        self.assertRaises(errors.PathNotChild,
734
 
                          osutils.relpath, 'C:/', 'H:/path')
735
 
 
736
 
 
737
628
class TestSafeUnicode(tests.TestCase):
738
629
 
739
630
    def test_from_ascii_string(self):
828
719
        self.assertEqual(None, osutils.safe_file_id(None))
829
720
 
830
721
 
831
 
class TestPosixFuncs(tests.TestCase):
832
 
    """Test that the posix version of normpath returns an appropriate path
833
 
       when used with 2 leading slashes."""
834
 
 
835
 
    def test_normpath(self):
836
 
        self.assertEqual('/etc/shadow', osutils._posix_normpath('/etc/shadow'))
837
 
        self.assertEqual('/etc/shadow', osutils._posix_normpath('//etc/shadow'))
838
 
        self.assertEqual('/etc/shadow', osutils._posix_normpath('///etc/shadow'))
839
 
 
840
 
 
841
722
class TestWin32Funcs(tests.TestCase):
842
723
    """Test that _win32 versions of os utilities return appropriate paths."""
843
724
 
896
777
        self.assertEqual('//HOST/path', osutils._win98_abspath('//HOST/path'))
897
778
        # relative path
898
779
        cwd = osutils.getcwd().rstrip('/')
899
 
        drive = osutils.ntpath.splitdrive(cwd)[0]
 
780
        drive = osutils._nt_splitdrive(cwd)[0]
900
781
        self.assertEqual(cwd+'/path', osutils._win98_abspath('path'))
901
782
        self.assertEqual(drive+'/path', osutils._win98_abspath('/path'))
902
783
        # unicode path
908
789
    """Test win32 functions that create files."""
909
790
 
910
791
    def test_getcwd(self):
911
 
        self.requireFeature(features.UnicodeFilenameFeature)
 
792
        self.requireFeature(tests.UnicodeFilenameFeature)
912
793
        os.mkdir(u'mu-\xb5')
913
794
        os.chdir(u'mu-\xb5')
914
795
        # TODO: jam 20060427 This will probably fail on Mac OSX because
944
825
        b.close()
945
826
 
946
827
        osutils._win32_rename('b', 'a')
947
 
        self.assertPathExists('a')
948
 
        self.assertPathDoesNotExist('b')
 
828
        self.failUnlessExists('a')
 
829
        self.failIfExists('b')
949
830
        self.assertFileEqual('baz\n', 'a')
950
831
 
951
832
    def test_rename_missing_file(self):
1004
885
    """Test mac special functions that require directories."""
1005
886
 
1006
887
    def test_getcwd(self):
1007
 
        self.requireFeature(features.UnicodeFilenameFeature)
 
888
        self.requireFeature(tests.UnicodeFilenameFeature)
1008
889
        os.mkdir(u'B\xe5gfors')
1009
890
        os.chdir(u'B\xe5gfors')
1010
891
        self.assertEndsWith(osutils._mac_getcwd(), u'B\xe5gfors')
1011
892
 
1012
893
    def test_getcwd_nonnorm(self):
1013
 
        self.requireFeature(features.UnicodeFilenameFeature)
 
894
        self.requireFeature(tests.UnicodeFilenameFeature)
1014
895
        # Test that _mac_getcwd() will normalize this path
1015
896
        os.mkdir(u'Ba\u030agfors')
1016
897
        os.chdir(u'Ba\u030agfors')
1027
908
 
1028
909
    def test_osutils_binding(self):
1029
910
        from bzrlib.tests import test__chunks_to_lines
1030
 
        if test__chunks_to_lines.compiled_chunkstolines_feature.available():
 
911
        if test__chunks_to_lines.CompiledChunksToLinesFeature.available():
1031
912
            from bzrlib._chunks_to_lines_pyx import chunks_to_lines
1032
913
        else:
1033
914
            from bzrlib._chunks_to_lines_py import chunks_to_lines
1099
980
        self.assertExpectedBlocks(expected_dirblocks[1:], result)
1100
981
 
1101
982
    def test_walkdirs_os_error(self):
1102
 
        # <https://bugs.launchpad.net/bzr/+bug/338653>
 
983
        # <https://bugs.edge.launchpad.net/bzr/+bug/338653>
1103
984
        # Pyrex readdir didn't raise useful messages if it had an error
1104
985
        # reading the directory
1105
986
        if sys.platform == 'win32':
1106
987
            raise tests.TestNotApplicable(
1107
988
                "readdir IOError not tested on win32")
1108
 
        self.requireFeature(features.not_running_as_root)
1109
989
        os.mkdir("test-unreadable")
1110
990
        os.chmod("test-unreadable", 0000)
1111
991
        # must chmod it back so that it can be removed
1119
999
        # Ensure the message contains the file name
1120
1000
        self.assertContainsRe(str(e), "\./test-unreadable")
1121
1001
 
1122
 
 
1123
 
    def test_walkdirs_encoding_error(self):
1124
 
        # <https://bugs.launchpad.net/bzr/+bug/488519>
1125
 
        # walkdirs didn't raise a useful message when the filenames
1126
 
        # are not using the filesystem's encoding
1127
 
 
1128
 
        # require a bytestring based filesystem
1129
 
        self.requireFeature(features.ByteStringNamedFilesystem)
1130
 
 
1131
 
        tree = [
1132
 
            '.bzr',
1133
 
            '0file',
1134
 
            '1dir/',
1135
 
            '1dir/0file',
1136
 
            '1dir/1dir/',
1137
 
            '1file'
1138
 
            ]
1139
 
 
1140
 
        self.build_tree(tree)
1141
 
 
1142
 
        # rename the 1file to a latin-1 filename
1143
 
        os.rename("./1file", "\xe8file")
1144
 
        if "\xe8file" not in os.listdir("."):
1145
 
            self.skip("Lack filesystem that preserves arbitrary bytes")
1146
 
 
1147
 
        self._save_platform_info()
1148
 
        win32utils.winver = None # Avoid the win32 detection code
1149
 
        osutils._fs_enc = 'UTF-8'
1150
 
 
1151
 
        # this should raise on error
1152
 
        def attempt():
1153
 
            for dirdetail, dirblock in osutils.walkdirs('.'):
1154
 
                pass
1155
 
 
1156
 
        self.assertRaises(errors.BadFilenameEncoding, attempt)
1157
 
 
1158
1002
    def test__walkdirs_utf8(self):
1159
1003
        tree = [
1160
1004
            '.bzr',
1210
1054
            dirblock[:] = new_dirblock
1211
1055
 
1212
1056
    def _save_platform_info(self):
1213
 
        self.overrideAttr(win32utils, 'winver')
1214
 
        self.overrideAttr(osutils, '_fs_enc')
1215
 
        self.overrideAttr(osutils, '_selected_dir_reader')
 
1057
        cur_winver = win32utils.winver
 
1058
        cur_fs_enc = osutils._fs_enc
 
1059
        cur_dir_reader = osutils._selected_dir_reader
 
1060
        def restore():
 
1061
            win32utils.winver = cur_winver
 
1062
            osutils._fs_enc = cur_fs_enc
 
1063
            osutils._selected_dir_reader = cur_dir_reader
 
1064
        self.addCleanup(restore)
1216
1065
 
1217
1066
    def assertDirReaderIs(self, expected):
1218
1067
        """Assert the right implementation for _walkdirs_utf8 is chosen."""
1226
1075
        self.requireFeature(UTF8DirReaderFeature)
1227
1076
        self._save_platform_info()
1228
1077
        win32utils.winver = None # Avoid the win32 detection code
1229
 
        osutils._fs_enc = 'utf-8'
1230
 
        self.assertDirReaderIs(
1231
 
            UTF8DirReaderFeature.module.UTF8DirReader)
 
1078
        osutils._fs_enc = 'UTF-8'
 
1079
        self.assertDirReaderIs(UTF8DirReaderFeature.reader)
1232
1080
 
1233
1081
    def test_force_walkdirs_utf8_fs_ascii(self):
1234
1082
        self.requireFeature(UTF8DirReaderFeature)
1235
1083
        self._save_platform_info()
1236
1084
        win32utils.winver = None # Avoid the win32 detection code
1237
 
        osutils._fs_enc = 'ascii'
1238
 
        self.assertDirReaderIs(
1239
 
            UTF8DirReaderFeature.module.UTF8DirReader)
 
1085
        osutils._fs_enc = 'US-ASCII'
 
1086
        self.assertDirReaderIs(UTF8DirReaderFeature.reader)
 
1087
 
 
1088
    def test_force_walkdirs_utf8_fs_ANSI(self):
 
1089
        self.requireFeature(UTF8DirReaderFeature)
 
1090
        self._save_platform_info()
 
1091
        win32utils.winver = None # Avoid the win32 detection code
 
1092
        osutils._fs_enc = 'ANSI_X3.4-1968'
 
1093
        self.assertDirReaderIs(UTF8DirReaderFeature.reader)
1240
1094
 
1241
1095
    def test_force_walkdirs_utf8_fs_latin1(self):
1242
1096
        self._save_platform_info()
1243
1097
        win32utils.winver = None # Avoid the win32 detection code
1244
 
        osutils._fs_enc = 'iso-8859-1'
 
1098
        osutils._fs_enc = 'latin1'
1245
1099
        self.assertDirReaderIs(osutils.UnicodeDirReader)
1246
1100
 
1247
1101
    def test_force_walkdirs_utf8_nt(self):
1248
1102
        # Disabled because the thunk of the whole walkdirs api is disabled.
1249
 
        self.requireFeature(test__walkdirs_win32.win32_readdir_feature)
 
1103
        self.requireFeature(test__walkdirs_win32.Win32ReadDirFeature)
1250
1104
        self._save_platform_info()
1251
1105
        win32utils.winver = 'Windows NT'
1252
1106
        from bzrlib._walkdirs_win32 import Win32ReadDir
1253
1107
        self.assertDirReaderIs(Win32ReadDir)
1254
1108
 
1255
1109
    def test_force_walkdirs_utf8_98(self):
1256
 
        self.requireFeature(test__walkdirs_win32.win32_readdir_feature)
 
1110
        self.requireFeature(test__walkdirs_win32.Win32ReadDirFeature)
1257
1111
        self._save_platform_info()
1258
1112
        win32utils.winver = 'Windows 98'
1259
1113
        self.assertDirReaderIs(osutils.UnicodeDirReader)
1260
1114
 
1261
1115
    def test_unicode_walkdirs(self):
1262
1116
        """Walkdirs should always return unicode paths."""
1263
 
        self.requireFeature(features.UnicodeFilenameFeature)
 
1117
        self.requireFeature(tests.UnicodeFilenameFeature)
1264
1118
        name0 = u'0file-\xb6'
1265
1119
        name1 = u'1dir-\u062c\u0648'
1266
1120
        name2 = u'2file-\u0633'
1303
1157
 
1304
1158
        The abspath portion might be in unicode or utf-8
1305
1159
        """
1306
 
        self.requireFeature(features.UnicodeFilenameFeature)
 
1160
        self.requireFeature(tests.UnicodeFilenameFeature)
1307
1161
        name0 = u'0file-\xb6'
1308
1162
        name1 = u'1dir-\u062c\u0648'
1309
1163
        name2 = u'2file-\u0633'
1364
1218
 
1365
1219
        The abspath portion should be in unicode
1366
1220
        """
1367
 
        self.requireFeature(features.UnicodeFilenameFeature)
 
1221
        self.requireFeature(tests.UnicodeFilenameFeature)
1368
1222
        # Use the unicode reader. TODO: split into driver-and-driven unit
1369
1223
        # tests.
1370
1224
        self._save_platform_info()
1410
1264
        self.assertEqual(expected_dirblocks, result)
1411
1265
 
1412
1266
    def test__walkdirs_utf8_win32readdir(self):
1413
 
        self.requireFeature(test__walkdirs_win32.win32_readdir_feature)
1414
 
        self.requireFeature(features.UnicodeFilenameFeature)
 
1267
        self.requireFeature(test__walkdirs_win32.Win32ReadDirFeature)
 
1268
        self.requireFeature(tests.UnicodeFilenameFeature)
1415
1269
        from bzrlib._walkdirs_win32 import Win32ReadDir
1416
1270
        self._save_platform_info()
1417
1271
        osutils._selected_dir_reader = Win32ReadDir()
1467
1321
 
1468
1322
    def test__walkdirs_utf_win32_find_file_stat_file(self):
1469
1323
        """make sure our Stat values are valid"""
1470
 
        self.requireFeature(test__walkdirs_win32.win32_readdir_feature)
1471
 
        self.requireFeature(features.UnicodeFilenameFeature)
 
1324
        self.requireFeature(test__walkdirs_win32.Win32ReadDirFeature)
 
1325
        self.requireFeature(tests.UnicodeFilenameFeature)
1472
1326
        from bzrlib._walkdirs_win32 import Win32ReadDir
1473
1327
        name0u = u'0file-\xb6'
1474
1328
        name0 = name0u.encode('utf8')
1491
1345
 
1492
1346
    def test__walkdirs_utf_win32_find_file_stat_directory(self):
1493
1347
        """make sure our Stat values are valid"""
1494
 
        self.requireFeature(test__walkdirs_win32.win32_readdir_feature)
1495
 
        self.requireFeature(features.UnicodeFilenameFeature)
 
1348
        self.requireFeature(test__walkdirs_win32.Win32ReadDirFeature)
 
1349
        self.requireFeature(tests.UnicodeFilenameFeature)
1496
1350
        from bzrlib._walkdirs_win32 import Win32ReadDir
1497
1351
        name0u = u'0dir-\u062c\u0648'
1498
1352
        name0 = name0u.encode('utf8')
1598
1452
        self.assertEqual(['c'], os.listdir('target/b'))
1599
1453
 
1600
1454
    def test_copy_tree_symlinks(self):
1601
 
        self.requireFeature(features.SymlinkFeature)
 
1455
        self.requireFeature(tests.SymlinkFeature)
1602
1456
        self.build_tree(['source/'])
1603
1457
        os.symlink('a/generic/path', 'source/lnk')
1604
1458
        osutils.copy_tree('source', 'target')
1629
1483
                          ('d', 'source/b', 'target/b'),
1630
1484
                          ('f', 'source/b/c', 'target/b/c'),
1631
1485
                         ], processed_files)
1632
 
        self.assertPathDoesNotExist('target')
 
1486
        self.failIfExists('target')
1633
1487
        if osutils.has_symlinks():
1634
1488
            self.assertEqual([('source/lnk', 'target/lnk')], processed_links)
1635
1489
 
1646
1500
        def cleanup():
1647
1501
            if 'BZR_TEST_ENV_VAR' in os.environ:
1648
1502
                del os.environ['BZR_TEST_ENV_VAR']
 
1503
 
1649
1504
        self.addCleanup(cleanup)
1650
1505
 
1651
1506
    def test_set(self):
1681
1536
        old = osutils.set_or_unset_env('BZR_TEST_ENV_VAR', None)
1682
1537
        self.assertEqual('foo', old)
1683
1538
        self.assertEqual(None, os.environ.get('BZR_TEST_ENV_VAR'))
1684
 
        self.assertFalse('BZR_TEST_ENV_VAR' in os.environ)
 
1539
        self.failIf('BZR_TEST_ENV_VAR' in os.environ)
1685
1540
 
1686
1541
 
1687
1542
class TestSizeShaFile(tests.TestCaseInTempDir):
1699
1554
        text = 'test\r\nwith\nall\rpossible line endings\r\n'
1700
1555
        self.build_tree_contents([('foo', text)])
1701
1556
        expected_sha = osutils.sha_string(text)
1702
 
        f = open('foo', 'rb')
 
1557
        f = open('foo')
1703
1558
        self.addCleanup(f.close)
1704
1559
        size, sha = osutils.size_sha_file(f)
1705
1560
        self.assertEqual(38, size)
1738
1593
 
1739
1594
class TestReCompile(tests.TestCase):
1740
1595
 
1741
 
    def _deprecated_re_compile_checked(self, *args, **kwargs):
1742
 
        return self.applyDeprecated(symbol_versioning.deprecated_in((2, 2, 0)),
1743
 
            osutils.re_compile_checked, *args, **kwargs)
1744
 
 
1745
1596
    def test_re_compile_checked(self):
1746
 
        r = self._deprecated_re_compile_checked(r'A*', re.IGNORECASE)
 
1597
        r = osutils.re_compile_checked(r'A*', re.IGNORECASE)
1747
1598
        self.assertTrue(r.match('aaaa'))
1748
1599
        self.assertTrue(r.match('aAaA'))
1749
1600
 
1750
1601
    def test_re_compile_checked_error(self):
1751
1602
        # like https://bugs.launchpad.net/bzr/+bug/251352
1752
 
 
1753
 
        # Due to possible test isolation error, re.compile is not lazy at
1754
 
        # this point. We re-install lazy compile.
1755
 
        lazy_regex.install_lazy_compile()
1756
1603
        err = self.assertRaises(
1757
1604
            errors.BzrCommandError,
1758
 
            self._deprecated_re_compile_checked, '*', re.IGNORECASE, 'test case')
 
1605
            osutils.re_compile_checked, '*', re.IGNORECASE, 'test case')
1759
1606
        self.assertEqual(
1760
 
            'Invalid regular expression in test case: '
1761
 
            '"*" nothing to repeat',
 
1607
            "Invalid regular expression in test case: '*': "
 
1608
            "nothing to repeat",
1762
1609
            str(err))
1763
1610
 
1764
1611
 
1765
1612
class TestDirReader(tests.TestCaseInTempDir):
1766
1613
 
1767
 
    scenarios = dir_reader_scenarios()
1768
 
 
1769
1614
    # Set by load_tests
1770
1615
    _dir_reader_class = None
1771
1616
    _native_to_unicode = None
1772
1617
 
1773
1618
    def setUp(self):
1774
1619
        tests.TestCaseInTempDir.setUp(self)
1775
 
        self.overrideAttr(osutils,
1776
 
                          '_selected_dir_reader', self._dir_reader_class())
 
1620
 
 
1621
        # Save platform specific info and reset it
 
1622
        cur_dir_reader = osutils._selected_dir_reader
 
1623
 
 
1624
        def restore():
 
1625
            osutils._selected_dir_reader = cur_dir_reader
 
1626
        self.addCleanup(restore)
 
1627
 
 
1628
        osutils._selected_dir_reader = self._dir_reader_class()
1777
1629
 
1778
1630
    def _get_ascii_tree(self):
1779
1631
        tree = [
1871
1723
        return filtered_dirblocks
1872
1724
 
1873
1725
    def test_walk_unicode_tree(self):
1874
 
        self.requireFeature(features.UnicodeFilenameFeature)
 
1726
        self.requireFeature(tests.UnicodeFilenameFeature)
1875
1727
        tree, expected_dirblocks = self._get_unicode_tree()
1876
1728
        self.build_tree(tree)
1877
1729
        result = list(osutils._walkdirs_utf8('.'))
1878
1730
        self.assertEqual(expected_dirblocks, self._filter_out(result))
1879
1731
 
1880
1732
    def test_symlink(self):
1881
 
        self.requireFeature(features.SymlinkFeature)
1882
 
        self.requireFeature(features.UnicodeFilenameFeature)
 
1733
        self.requireFeature(tests.SymlinkFeature)
 
1734
        self.requireFeature(tests.UnicodeFilenameFeature)
1883
1735
        target = u'target\N{Euro Sign}'
1884
1736
        link_name = u'l\N{Euro Sign}nk'
1885
1737
        os.symlink(target, link_name)
1903
1755
    But prior python versions failed to properly encode the passed unicode
1904
1756
    string.
1905
1757
    """
1906
 
    _test_needs_features = [features.SymlinkFeature, features.UnicodeFilenameFeature]
 
1758
    _test_needs_features = [tests.SymlinkFeature, tests.UnicodeFilenameFeature]
1907
1759
 
1908
1760
    def setUp(self):
1909
1761
        super(tests.TestCaseInTempDir, self).setUp()
1912
1764
        os.symlink(self.target, self.link)
1913
1765
 
1914
1766
    def test_os_readlink_link_encoding(self):
1915
 
        self.assertEquals(self.target,  os.readlink(self.link))
 
1767
        if sys.version_info < (2, 6):
 
1768
            self.assertRaises(UnicodeEncodeError, os.readlink, self.link)
 
1769
        else:
 
1770
            self.assertEquals(self.target,  os.readlink(self.link))
1916
1771
 
1917
1772
    def test_os_readlink_link_decoding(self):
1918
1773
        self.assertEquals(self.target.encode(osutils._fs_enc),
1921
1776
 
1922
1777
class TestConcurrency(tests.TestCase):
1923
1778
 
1924
 
    def setUp(self):
1925
 
        super(TestConcurrency, self).setUp()
1926
 
        self.overrideAttr(osutils, '_cached_local_concurrency')
1927
 
 
1928
1779
    def test_local_concurrency(self):
1929
1780
        concurrency = osutils.local_concurrency()
1930
1781
        self.assertIsInstance(concurrency, int)
1931
 
 
1932
 
    def test_local_concurrency_environment_variable(self):
1933
 
        self.overrideEnv('BZR_CONCURRENCY', '2')
1934
 
        self.assertEqual(2, osutils.local_concurrency(use_cache=False))
1935
 
        self.overrideEnv('BZR_CONCURRENCY', '3')
1936
 
        self.assertEqual(3, osutils.local_concurrency(use_cache=False))
1937
 
        self.overrideEnv('BZR_CONCURRENCY', 'foo')
1938
 
        self.assertEqual(1, osutils.local_concurrency(use_cache=False))
1939
 
 
1940
 
    def test_option_concurrency(self):
1941
 
        self.overrideEnv('BZR_CONCURRENCY', '1')
1942
 
        self.run_bzr('rocks --concurrency 42')
1943
 
        # Command line overrides environment variable
1944
 
        self.assertEquals('42', os.environ['BZR_CONCURRENCY'])
1945
 
        self.assertEquals(42, osutils.local_concurrency(use_cache=False))
1946
 
 
1947
 
 
1948
 
class TestFailedToLoadExtension(tests.TestCase):
1949
 
 
1950
 
    def _try_loading(self):
1951
 
        try:
1952
 
            import bzrlib._fictional_extension_py
1953
 
        except ImportError, e:
1954
 
            osutils.failed_to_load_extension(e)
1955
 
            return True
1956
 
 
1957
 
    def setUp(self):
1958
 
        super(TestFailedToLoadExtension, self).setUp()
1959
 
        self.overrideAttr(osutils, '_extension_load_failures', [])
1960
 
 
1961
 
    def test_failure_to_load(self):
1962
 
        self._try_loading()
1963
 
        self.assertLength(1, osutils._extension_load_failures)
1964
 
        self.assertEquals(osutils._extension_load_failures[0],
1965
 
            "No module named _fictional_extension_py")
1966
 
 
1967
 
    def test_report_extension_load_failures_no_warning(self):
1968
 
        self.assertTrue(self._try_loading())
1969
 
        warnings, result = self.callCatchWarnings(osutils.report_extension_load_failures)
1970
 
        # it used to give a Python warning; it no longer does
1971
 
        self.assertLength(0, warnings)
1972
 
 
1973
 
    def test_report_extension_load_failures_message(self):
1974
 
        log = StringIO()
1975
 
        trace.push_log_file(log)
1976
 
        self.assertTrue(self._try_loading())
1977
 
        osutils.report_extension_load_failures()
1978
 
        self.assertContainsRe(
1979
 
            log.getvalue(),
1980
 
            r"bzr: warning: some compiled extensions could not be loaded; "
1981
 
            "see <https://answers\.launchpad\.net/bzr/\+faq/703>\n"
1982
 
            )
1983
 
 
1984
 
 
1985
 
class TestTerminalWidth(tests.TestCase):
1986
 
 
1987
 
    def setUp(self):
1988
 
        tests.TestCase.setUp(self)
1989
 
        self._orig_terminal_size_state = osutils._terminal_size_state
1990
 
        self._orig_first_terminal_size = osutils._first_terminal_size
1991
 
        self.addCleanup(self.restore_osutils_globals)
1992
 
        osutils._terminal_size_state = 'no_data'
1993
 
        osutils._first_terminal_size = None
1994
 
 
1995
 
    def restore_osutils_globals(self):
1996
 
        osutils._terminal_size_state = self._orig_terminal_size_state
1997
 
        osutils._first_terminal_size = self._orig_first_terminal_size
1998
 
 
1999
 
    def replace_stdout(self, new):
2000
 
        self.overrideAttr(sys, 'stdout', new)
2001
 
 
2002
 
    def replace__terminal_size(self, new):
2003
 
        self.overrideAttr(osutils, '_terminal_size', new)
2004
 
 
2005
 
    def set_fake_tty(self):
2006
 
 
2007
 
        class I_am_a_tty(object):
2008
 
            def isatty(self):
2009
 
                return True
2010
 
 
2011
 
        self.replace_stdout(I_am_a_tty())
2012
 
 
2013
 
    def test_default_values(self):
2014
 
        self.assertEqual(80, osutils.default_terminal_width)
2015
 
 
2016
 
    def test_defaults_to_BZR_COLUMNS(self):
2017
 
        # BZR_COLUMNS is set by the test framework
2018
 
        self.assertNotEqual('12', os.environ['BZR_COLUMNS'])
2019
 
        self.overrideEnv('BZR_COLUMNS', '12')
2020
 
        self.assertEqual(12, osutils.terminal_width())
2021
 
 
2022
 
    def test_BZR_COLUMNS_0_no_limit(self):
2023
 
        self.overrideEnv('BZR_COLUMNS', '0')
2024
 
        self.assertEqual(None, osutils.terminal_width())
2025
 
 
2026
 
    def test_falls_back_to_COLUMNS(self):
2027
 
        self.overrideEnv('BZR_COLUMNS', None)
2028
 
        self.assertNotEqual('42', os.environ['COLUMNS'])
2029
 
        self.set_fake_tty()
2030
 
        self.overrideEnv('COLUMNS', '42')
2031
 
        self.assertEqual(42, osutils.terminal_width())
2032
 
 
2033
 
    def test_tty_default_without_columns(self):
2034
 
        self.overrideEnv('BZR_COLUMNS', None)
2035
 
        self.overrideEnv('COLUMNS', None)
2036
 
 
2037
 
        def terminal_size(w, h):
2038
 
            return 42, 42
2039
 
 
2040
 
        self.set_fake_tty()
2041
 
        # We need to override the osutils definition as it depends on the
2042
 
        # running environment that we can't control (PQM running without a
2043
 
        # controlling terminal is one example).
2044
 
        self.replace__terminal_size(terminal_size)
2045
 
        self.assertEqual(42, osutils.terminal_width())
2046
 
 
2047
 
    def test_non_tty_default_without_columns(self):
2048
 
        self.overrideEnv('BZR_COLUMNS', None)
2049
 
        self.overrideEnv('COLUMNS', None)
2050
 
        self.replace_stdout(None)
2051
 
        self.assertEqual(None, osutils.terminal_width())
2052
 
 
2053
 
    def test_no_TIOCGWINSZ(self):
2054
 
        self.requireFeature(term_ios_feature)
2055
 
        termios = term_ios_feature.module
2056
 
        # bug 63539 is about a termios without TIOCGWINSZ attribute
2057
 
        try:
2058
 
            orig = termios.TIOCGWINSZ
2059
 
        except AttributeError:
2060
 
            # We won't remove TIOCGWINSZ, because it doesn't exist anyway :)
2061
 
            pass
2062
 
        else:
2063
 
            self.overrideAttr(termios, 'TIOCGWINSZ')
2064
 
            del termios.TIOCGWINSZ
2065
 
        self.overrideEnv('BZR_COLUMNS', None)
2066
 
        self.overrideEnv('COLUMNS', None)
2067
 
        # Whatever the result is, if we don't raise an exception, it's ok.
2068
 
        osutils.terminal_width()
2069
 
 
2070
 
 
2071
 
class TestCreationOps(tests.TestCaseInTempDir):
2072
 
    _test_needs_features = [features.chown_feature]
2073
 
 
2074
 
    def setUp(self):
2075
 
        tests.TestCaseInTempDir.setUp(self)
2076
 
        self.overrideAttr(os, 'chown', self._dummy_chown)
2077
 
 
2078
 
        # params set by call to _dummy_chown
2079
 
        self.path = self.uid = self.gid = None
2080
 
 
2081
 
    def _dummy_chown(self, path, uid, gid):
2082
 
        self.path, self.uid, self.gid = path, uid, gid
2083
 
 
2084
 
    def test_copy_ownership_from_path(self):
2085
 
        """copy_ownership_from_path test with specified src."""
2086
 
        ownsrc = '/'
2087
 
        f = open('test_file', 'wt')
2088
 
        osutils.copy_ownership_from_path('test_file', ownsrc)
2089
 
 
2090
 
        s = os.stat(ownsrc)
2091
 
        self.assertEquals(self.path, 'test_file')
2092
 
        self.assertEquals(self.uid, s.st_uid)
2093
 
        self.assertEquals(self.gid, s.st_gid)
2094
 
 
2095
 
    def test_copy_ownership_nonesrc(self):
2096
 
        """copy_ownership_from_path test with src=None."""
2097
 
        f = open('test_file', 'wt')
2098
 
        # should use parent dir for permissions
2099
 
        osutils.copy_ownership_from_path('test_file')
2100
 
 
2101
 
        s = os.stat('..')
2102
 
        self.assertEquals(self.path, 'test_file')
2103
 
        self.assertEquals(self.uid, s.st_uid)
2104
 
        self.assertEquals(self.gid, s.st_gid)
2105
 
 
2106
 
 
2107
 
class TestPathFromEnviron(tests.TestCase):
2108
 
 
2109
 
    def test_is_unicode(self):
2110
 
        self.overrideEnv('BZR_TEST_PATH', './anywhere at all/')
2111
 
        path = osutils.path_from_environ('BZR_TEST_PATH')
2112
 
        self.assertIsInstance(path, unicode)
2113
 
        self.assertEqual(u'./anywhere at all/', path)
2114
 
 
2115
 
    def test_posix_path_env_ascii(self):
2116
 
        self.overrideEnv('BZR_TEST_PATH', '/tmp')
2117
 
        home = osutils._posix_path_from_environ('BZR_TEST_PATH')
2118
 
        self.assertIsInstance(home, unicode)
2119
 
        self.assertEqual(u'/tmp', home)
2120
 
 
2121
 
    def test_posix_path_env_unicode(self):
2122
 
        self.requireFeature(features.ByteStringNamedFilesystem)
2123
 
        self.overrideEnv('BZR_TEST_PATH', '/home/\xa7test')
2124
 
        self.overrideAttr(osutils, "_fs_enc", "iso8859-1")
2125
 
        self.assertEqual(u'/home/\xa7test',
2126
 
            osutils._posix_path_from_environ('BZR_TEST_PATH'))
2127
 
        osutils._fs_enc = "iso8859-5"
2128
 
        self.assertEqual(u'/home/\u0407test',
2129
 
            osutils._posix_path_from_environ('BZR_TEST_PATH'))
2130
 
        osutils._fs_enc = "utf-8"
2131
 
        self.assertRaises(errors.BadFilenameEncoding,
2132
 
            osutils._posix_path_from_environ, 'BZR_TEST_PATH')
2133
 
 
2134
 
 
2135
 
class TestGetHomeDir(tests.TestCase):
2136
 
 
2137
 
    def test_is_unicode(self):
2138
 
        home = osutils._get_home_dir()
2139
 
        self.assertIsInstance(home, unicode)
2140
 
 
2141
 
    def test_posix_homeless(self):
2142
 
        self.overrideEnv('HOME', None)
2143
 
        home = osutils._get_home_dir()
2144
 
        self.assertIsInstance(home, unicode)
2145
 
 
2146
 
    def test_posix_home_ascii(self):
2147
 
        self.overrideEnv('HOME', '/home/test')
2148
 
        home = osutils._posix_get_home_dir()
2149
 
        self.assertIsInstance(home, unicode)
2150
 
        self.assertEqual(u'/home/test', home)
2151
 
 
2152
 
    def test_posix_home_unicode(self):
2153
 
        self.requireFeature(features.ByteStringNamedFilesystem)
2154
 
        self.overrideEnv('HOME', '/home/\xa7test')
2155
 
        self.overrideAttr(osutils, "_fs_enc", "iso8859-1")
2156
 
        self.assertEqual(u'/home/\xa7test', osutils._posix_get_home_dir())
2157
 
        osutils._fs_enc = "iso8859-5"
2158
 
        self.assertEqual(u'/home/\u0407test', osutils._posix_get_home_dir())
2159
 
        osutils._fs_enc = "utf-8"
2160
 
        self.assertRaises(errors.BadFilenameEncoding,
2161
 
            osutils._posix_get_home_dir)
2162
 
 
2163
 
 
2164
 
class TestGetuserUnicode(tests.TestCase):
2165
 
 
2166
 
    def test_is_unicode(self):
2167
 
        user = osutils.getuser_unicode()
2168
 
        self.assertIsInstance(user, unicode)
2169
 
 
2170
 
    def envvar_to_override(self):
2171
 
        if sys.platform == "win32":
2172
 
            # Disable use of platform calls on windows so envvar is used
2173
 
            self.overrideAttr(win32utils, 'has_ctypes', False)
2174
 
            return 'USERNAME' # only variable used on windows
2175
 
        return 'LOGNAME' # first variable checked by getpass.getuser()
2176
 
 
2177
 
    def test_ascii_user(self):
2178
 
        self.overrideEnv(self.envvar_to_override(), 'jrandom')
2179
 
        self.assertEqual(u'jrandom', osutils.getuser_unicode())
2180
 
 
2181
 
    def test_unicode_user(self):
2182
 
        ue = osutils.get_user_encoding()
2183
 
        uni_val, env_val = tests.probe_unicode_in_user_encoding()
2184
 
        if uni_val is None:
2185
 
            raise tests.TestSkipped(
2186
 
                'Cannot find a unicode character that works in encoding %s'
2187
 
                % (osutils.get_user_encoding(),))
2188
 
        uni_username = u'jrandom' + uni_val
2189
 
        encoded_username = uni_username.encode(ue)
2190
 
        self.overrideEnv(self.envvar_to_override(), encoded_username)
2191
 
        self.assertEqual(uni_username, osutils.getuser_unicode())
2192
 
 
2193
 
 
2194
 
class TestBackupNames(tests.TestCase):
2195
 
 
2196
 
    def setUp(self):
2197
 
        super(TestBackupNames, self).setUp()
2198
 
        self.backups = []
2199
 
 
2200
 
    def backup_exists(self, name):
2201
 
        return name in self.backups
2202
 
 
2203
 
    def available_backup_name(self, name):
2204
 
        backup_name = osutils.available_backup_name(name, self.backup_exists)
2205
 
        self.backups.append(backup_name)
2206
 
        return backup_name
2207
 
 
2208
 
    def assertBackupName(self, expected, name):
2209
 
        self.assertEqual(expected, self.available_backup_name(name))
2210
 
 
2211
 
    def test_empty(self):
2212
 
        self.assertBackupName('file.~1~', 'file')
2213
 
 
2214
 
    def test_existing(self):
2215
 
        self.available_backup_name('file')
2216
 
        self.available_backup_name('file')
2217
 
        self.assertBackupName('file.~3~', 'file')
2218
 
        # Empty slots are found, this is not a strict requirement and may be
2219
 
        # revisited if we test against all implementations.
2220
 
        self.backups.remove('file.~2~')
2221
 
        self.assertBackupName('file.~2~', 'file')
2222
 
 
2223
 
 
2224
 
class TestFindExecutableInPath(tests.TestCase):
2225
 
 
2226
 
    def test_windows(self):
2227
 
        if sys.platform != 'win32':
2228
 
            raise tests.TestSkipped('test requires win32')
2229
 
        self.assertTrue(osutils.find_executable_on_path('explorer') is not None)
2230
 
        self.assertTrue(
2231
 
            osutils.find_executable_on_path('explorer.exe') is not None)
2232
 
        self.assertTrue(
2233
 
            osutils.find_executable_on_path('EXPLORER.EXE') is not None)
2234
 
        self.assertTrue(
2235
 
            osutils.find_executable_on_path('THIS SHOULD NOT EXIST') is None)
2236
 
        self.assertTrue(osutils.find_executable_on_path('file.txt') is None)
2237
 
        
2238
 
    def test_windows_app_path(self):
2239
 
        if sys.platform != 'win32':
2240
 
            raise tests.TestSkipped('test requires win32')
2241
 
        # Override PATH env var so that exe can only be found on App Path
2242
 
        self.overrideEnv('PATH', '')
2243
 
        # Internt Explorer is always registered in the App Path
2244
 
        self.assertTrue(osutils.find_executable_on_path('iexplore') is not None)
2245
 
 
2246
 
    def test_other(self):
2247
 
        if sys.platform == 'win32':
2248
 
            raise tests.TestSkipped('test requires non-win32')
2249
 
        self.assertTrue(osutils.find_executable_on_path('sh') is not None)
2250
 
        self.assertTrue(
2251
 
            osutils.find_executable_on_path('THIS SHOULD NOT EXIST') is None)
2252
 
 
2253
 
 
2254
 
class TestEnvironmentErrors(tests.TestCase):
2255
 
    """Test handling of environmental errors"""
2256
 
 
2257
 
    def test_is_oserror(self):
2258
 
        self.assertTrue(osutils.is_environment_error(
2259
 
            OSError(errno.EINVAL, "Invalid parameter")))
2260
 
 
2261
 
    def test_is_ioerror(self):
2262
 
        self.assertTrue(osutils.is_environment_error(
2263
 
            IOError(errno.EINVAL, "Invalid parameter")))
2264
 
 
2265
 
    def test_is_socket_error(self):
2266
 
        self.assertTrue(osutils.is_environment_error(
2267
 
            socket.error(errno.EINVAL, "Invalid parameter")))
2268
 
 
2269
 
    def test_is_select_error(self):
2270
 
        self.assertTrue(osutils.is_environment_error(
2271
 
            select.error(errno.EINVAL, "Invalid parameter")))
2272
 
 
2273
 
    def test_is_pywintypes_error(self):
2274
 
        self.requireFeature(features.pywintypes)
2275
 
        import pywintypes
2276
 
        self.assertTrue(osutils.is_environment_error(
2277
 
            pywintypes.error(errno.EINVAL, "Invalid parameter", "Caller")))