← 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: Patch Queue Manager
  • Date: 2011-10-14 16:54:26 UTC
  • mfrom: (6216.1.1 remove-this-file)
  • Revision ID: pqm@pqm.ubuntu.com-20111014165426-tjix4e6idryf1r2z
(jelmer) Remove an accidentally committed .THIS file. (Jelmer Vernooij)

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
 
39
38
    file_utils,
40
39
    test__walkdirs_win32,
41
40
    )
42
 
 
43
 
 
44
 
class _UTF8DirReaderFeature(tests.Feature):
 
41
from bzrlib.tests.scenarios import load_tests_apply_scenarios
 
42
 
 
43
 
 
44
class _UTF8DirReaderFeature(features.Feature):
45
45
 
46
46
    def _probe(self):
47
47
        try:
54
54
    def feature_name(self):
55
55
        return 'bzrlib._readdir_pyx'
56
56
 
57
 
UTF8DirReaderFeature = _UTF8DirReaderFeature()
 
57
UTF8DirReaderFeature = features.ModuleAvailableFeature('bzrlib._readdir_pyx')
58
58
 
59
 
term_ios_feature = tests.ModuleAvailableFeature('termios')
 
59
term_ios_feature = features.ModuleAvailableFeature('termios')
60
60
 
61
61
 
62
62
def _already_unicode(s):
97
97
    return scenarios
98
98
 
99
99
 
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
 
100
load_tests = load_tests_apply_scenarios
107
101
 
108
102
 
109
103
class TestContainsWhitespace(tests.TestCase):
110
104
 
111
105
    def test_contains_whitespace(self):
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'))
 
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'))
118
112
 
119
113
        # \xa0 is "Non-breaking-space" which on some python locales thinks it
120
114
        # is whitespace, but we do not.
121
 
        self.failIf(osutils.contains_whitespace(u''))
122
 
        self.failIf(osutils.contains_whitespace(u'hellothere'))
123
 
        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'))
124
118
 
125
119
 
126
120
class TestRename(tests.TestCaseInTempDir):
140
134
        # This should work everywhere
141
135
        self.create_file('a', 'something in a\n')
142
136
        self._fancy_rename('a', 'b')
143
 
        self.failIfExists('a')
144
 
        self.failUnlessExists('b')
 
137
        self.assertPathDoesNotExist('a')
 
138
        self.assertPathExists('b')
145
139
        self.check_file_contents('b', 'something in a\n')
146
140
 
147
141
        self.create_file('a', 'new something in a\n')
154
148
        self.create_file('target', 'data in target\n')
155
149
        self.assertRaises((IOError, OSError), self._fancy_rename,
156
150
                          'missingsource', 'target')
157
 
        self.failUnlessExists('target')
 
151
        self.assertPathExists('target')
158
152
        self.check_file_contents('target', 'data in target\n')
159
153
 
160
154
    def test_fancy_rename_fails_if_source_and_target_missing(self):
165
159
        # Rename should be semi-atomic on all platforms
166
160
        self.create_file('a', 'something in a\n')
167
161
        osutils.rename('a', 'b')
168
 
        self.failIfExists('a')
169
 
        self.failUnlessExists('b')
 
162
        self.assertPathDoesNotExist('a')
 
163
        self.assertPathExists('b')
170
164
        self.check_file_contents('b', 'something in a\n')
171
165
 
172
166
        self.create_file('a', 'new something in a\n')
237
231
            self.assertFalse(osutils.is_inside_or_parent_of_any(dirs, fn))
238
232
 
239
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
 
240
253
class TestRmTree(tests.TestCaseInTempDir):
241
254
 
242
255
    def test_rmtree(self):
254
267
 
255
268
        osutils.rmtree('dir')
256
269
 
257
 
        self.failIfExists('dir/file')
258
 
        self.failIfExists('dir')
 
270
        self.assertPathDoesNotExist('dir/file')
 
271
        self.assertPathDoesNotExist('dir')
259
272
 
260
273
 
261
274
class TestDeleteAny(tests.TestCaseInTempDir):
416
429
class TestLinks(tests.TestCaseInTempDir):
417
430
 
418
431
    def test_dereference_path(self):
419
 
        self.requireFeature(tests.SymlinkFeature)
 
432
        self.requireFeature(features.SymlinkFeature)
420
433
        cwd = osutils.realpath('.')
421
434
        os.mkdir('bar')
422
435
        bar_path = osutils.pathjoin(cwd, 'bar')
469
482
 
470
483
class TestCanonicalRelPath(tests.TestCaseInTempDir):
471
484
 
472
 
    _test_needs_features = [tests.CaseInsCasePresFilenameFeature]
 
485
    _test_needs_features = [features.CaseInsCasePresFilenameFeature]
473
486
 
474
487
    def test_canonical_relpath_simple(self):
475
488
        f = file('MixedCaseName', 'w')
476
489
        f.close()
477
490
        actual = osutils.canonical_relpath(self.test_base_dir, 'mixedcasename')
478
 
        self.failUnlessEqual('work/MixedCaseName', actual)
 
491
        self.assertEqual('work/MixedCaseName', actual)
479
492
 
480
493
    def test_canonical_relpath_missing_tail(self):
481
494
        os.mkdir('MixedCaseParent')
482
495
        actual = osutils.canonical_relpath(self.test_base_dir,
483
496
                                           'mixedcaseparent/nochild')
484
 
        self.failUnlessEqual('work/MixedCaseParent/nochild', actual)
 
497
        self.assertEqual('work/MixedCaseParent/nochild', actual)
485
498
 
486
499
 
487
500
class Test_CICPCanonicalRelpath(tests.TestCaseWithTransport):
805
818
        self.assertEqual(None, osutils.safe_file_id(None))
806
819
 
807
820
 
 
821
class TestPosixFuncs(tests.TestCase):
 
822
    """Test that the posix version of normpath returns an appropriate path
 
823
       when used with 2 leading slashes."""
 
824
 
 
825
    def test_normpath(self):
 
826
        self.assertEqual('/etc/shadow', osutils._posix_normpath('/etc/shadow'))
 
827
        self.assertEqual('/etc/shadow', osutils._posix_normpath('//etc/shadow'))
 
828
        self.assertEqual('/etc/shadow', osutils._posix_normpath('///etc/shadow'))
 
829
 
 
830
 
808
831
class TestWin32Funcs(tests.TestCase):
809
832
    """Test that _win32 versions of os utilities return appropriate paths."""
810
833
 
875
898
    """Test win32 functions that create files."""
876
899
 
877
900
    def test_getcwd(self):
878
 
        self.requireFeature(tests.UnicodeFilenameFeature)
 
901
        self.requireFeature(features.UnicodeFilenameFeature)
879
902
        os.mkdir(u'mu-\xb5')
880
903
        os.chdir(u'mu-\xb5')
881
904
        # TODO: jam 20060427 This will probably fail on Mac OSX because
911
934
        b.close()
912
935
 
913
936
        osutils._win32_rename('b', 'a')
914
 
        self.failUnlessExists('a')
915
 
        self.failIfExists('b')
 
937
        self.assertPathExists('a')
 
938
        self.assertPathDoesNotExist('b')
916
939
        self.assertFileEqual('baz\n', 'a')
917
940
 
918
941
    def test_rename_missing_file(self):
971
994
    """Test mac special functions that require directories."""
972
995
 
973
996
    def test_getcwd(self):
974
 
        self.requireFeature(tests.UnicodeFilenameFeature)
 
997
        self.requireFeature(features.UnicodeFilenameFeature)
975
998
        os.mkdir(u'B\xe5gfors')
976
999
        os.chdir(u'B\xe5gfors')
977
1000
        self.assertEndsWith(osutils._mac_getcwd(), u'B\xe5gfors')
978
1001
 
979
1002
    def test_getcwd_nonnorm(self):
980
 
        self.requireFeature(tests.UnicodeFilenameFeature)
 
1003
        self.requireFeature(features.UnicodeFilenameFeature)
981
1004
        # Test that _mac_getcwd() will normalize this path
982
1005
        os.mkdir(u'Ba\u030agfors')
983
1006
        os.chdir(u'Ba\u030agfors')
1072
1095
        if sys.platform == 'win32':
1073
1096
            raise tests.TestNotApplicable(
1074
1097
                "readdir IOError not tested on win32")
 
1098
        self.requireFeature(features.not_running_as_root)
1075
1099
        os.mkdir("test-unreadable")
1076
1100
        os.chmod("test-unreadable", 0000)
1077
1101
        # must chmod it back so that it can be removed
1092
1116
        # are not using the filesystem's encoding
1093
1117
 
1094
1118
        # require a bytestring based filesystem
1095
 
        self.requireFeature(tests.ByteStringNamedFilesystem)
 
1119
        self.requireFeature(features.ByteStringNamedFilesystem)
1096
1120
 
1097
1121
        tree = [
1098
1122
            '.bzr',
1107
1131
 
1108
1132
        # rename the 1file to a latin-1 filename
1109
1133
        os.rename("./1file", "\xe8file")
 
1134
        if "\xe8file" not in os.listdir("."):
 
1135
            self.skip("Lack filesystem that preserves arbitrary bytes")
1110
1136
 
1111
1137
        self._save_platform_info()
1112
1138
        win32utils.winver = None # Avoid the win32 detection code
1191
1217
        self._save_platform_info()
1192
1218
        win32utils.winver = None # Avoid the win32 detection code
1193
1219
        osutils._fs_enc = 'UTF-8'
1194
 
        self.assertDirReaderIs(UTF8DirReaderFeature.reader)
 
1220
        self.assertDirReaderIs(
 
1221
            UTF8DirReaderFeature.module.UTF8DirReader)
1195
1222
 
1196
1223
    def test_force_walkdirs_utf8_fs_ascii(self):
1197
1224
        self.requireFeature(UTF8DirReaderFeature)
1198
1225
        self._save_platform_info()
1199
1226
        win32utils.winver = None # Avoid the win32 detection code
1200
1227
        osutils._fs_enc = 'US-ASCII'
1201
 
        self.assertDirReaderIs(UTF8DirReaderFeature.reader)
 
1228
        self.assertDirReaderIs(
 
1229
            UTF8DirReaderFeature.module.UTF8DirReader)
1202
1230
 
1203
1231
    def test_force_walkdirs_utf8_fs_ANSI(self):
1204
1232
        self.requireFeature(UTF8DirReaderFeature)
1205
1233
        self._save_platform_info()
1206
1234
        win32utils.winver = None # Avoid the win32 detection code
1207
1235
        osutils._fs_enc = 'ANSI_X3.4-1968'
1208
 
        self.assertDirReaderIs(UTF8DirReaderFeature.reader)
 
1236
        self.assertDirReaderIs(
 
1237
            UTF8DirReaderFeature.module.UTF8DirReader)
1209
1238
 
1210
1239
    def test_force_walkdirs_utf8_fs_latin1(self):
1211
1240
        self._save_platform_info()
1229
1258
 
1230
1259
    def test_unicode_walkdirs(self):
1231
1260
        """Walkdirs should always return unicode paths."""
1232
 
        self.requireFeature(tests.UnicodeFilenameFeature)
 
1261
        self.requireFeature(features.UnicodeFilenameFeature)
1233
1262
        name0 = u'0file-\xb6'
1234
1263
        name1 = u'1dir-\u062c\u0648'
1235
1264
        name2 = u'2file-\u0633'
1272
1301
 
1273
1302
        The abspath portion might be in unicode or utf-8
1274
1303
        """
1275
 
        self.requireFeature(tests.UnicodeFilenameFeature)
 
1304
        self.requireFeature(features.UnicodeFilenameFeature)
1276
1305
        name0 = u'0file-\xb6'
1277
1306
        name1 = u'1dir-\u062c\u0648'
1278
1307
        name2 = u'2file-\u0633'
1333
1362
 
1334
1363
        The abspath portion should be in unicode
1335
1364
        """
1336
 
        self.requireFeature(tests.UnicodeFilenameFeature)
 
1365
        self.requireFeature(features.UnicodeFilenameFeature)
1337
1366
        # Use the unicode reader. TODO: split into driver-and-driven unit
1338
1367
        # tests.
1339
1368
        self._save_platform_info()
1380
1409
 
1381
1410
    def test__walkdirs_utf8_win32readdir(self):
1382
1411
        self.requireFeature(test__walkdirs_win32.win32_readdir_feature)
1383
 
        self.requireFeature(tests.UnicodeFilenameFeature)
 
1412
        self.requireFeature(features.UnicodeFilenameFeature)
1384
1413
        from bzrlib._walkdirs_win32 import Win32ReadDir
1385
1414
        self._save_platform_info()
1386
1415
        osutils._selected_dir_reader = Win32ReadDir()
1437
1466
    def test__walkdirs_utf_win32_find_file_stat_file(self):
1438
1467
        """make sure our Stat values are valid"""
1439
1468
        self.requireFeature(test__walkdirs_win32.win32_readdir_feature)
1440
 
        self.requireFeature(tests.UnicodeFilenameFeature)
 
1469
        self.requireFeature(features.UnicodeFilenameFeature)
1441
1470
        from bzrlib._walkdirs_win32 import Win32ReadDir
1442
1471
        name0u = u'0file-\xb6'
1443
1472
        name0 = name0u.encode('utf8')
1461
1490
    def test__walkdirs_utf_win32_find_file_stat_directory(self):
1462
1491
        """make sure our Stat values are valid"""
1463
1492
        self.requireFeature(test__walkdirs_win32.win32_readdir_feature)
1464
 
        self.requireFeature(tests.UnicodeFilenameFeature)
 
1493
        self.requireFeature(features.UnicodeFilenameFeature)
1465
1494
        from bzrlib._walkdirs_win32 import Win32ReadDir
1466
1495
        name0u = u'0dir-\u062c\u0648'
1467
1496
        name0 = name0u.encode('utf8')
1567
1596
        self.assertEqual(['c'], os.listdir('target/b'))
1568
1597
 
1569
1598
    def test_copy_tree_symlinks(self):
1570
 
        self.requireFeature(tests.SymlinkFeature)
 
1599
        self.requireFeature(features.SymlinkFeature)
1571
1600
        self.build_tree(['source/'])
1572
1601
        os.symlink('a/generic/path', 'source/lnk')
1573
1602
        osutils.copy_tree('source', 'target')
1598
1627
                          ('d', 'source/b', 'target/b'),
1599
1628
                          ('f', 'source/b/c', 'target/b/c'),
1600
1629
                         ], processed_files)
1601
 
        self.failIfExists('target')
 
1630
        self.assertPathDoesNotExist('target')
1602
1631
        if osutils.has_symlinks():
1603
1632
            self.assertEqual([('source/lnk', 'target/lnk')], processed_links)
1604
1633
 
1650
1679
        old = osutils.set_or_unset_env('BZR_TEST_ENV_VAR', None)
1651
1680
        self.assertEqual('foo', old)
1652
1681
        self.assertEqual(None, os.environ.get('BZR_TEST_ENV_VAR'))
1653
 
        self.failIf('BZR_TEST_ENV_VAR' in os.environ)
 
1682
        self.assertFalse('BZR_TEST_ENV_VAR' in os.environ)
1654
1683
 
1655
1684
 
1656
1685
class TestSizeShaFile(tests.TestCaseInTempDir):
1733
1762
 
1734
1763
class TestDirReader(tests.TestCaseInTempDir):
1735
1764
 
 
1765
    scenarios = dir_reader_scenarios()
 
1766
 
1736
1767
    # Set by load_tests
1737
1768
    _dir_reader_class = None
1738
1769
    _native_to_unicode = None
1838
1869
        return filtered_dirblocks
1839
1870
 
1840
1871
    def test_walk_unicode_tree(self):
1841
 
        self.requireFeature(tests.UnicodeFilenameFeature)
 
1872
        self.requireFeature(features.UnicodeFilenameFeature)
1842
1873
        tree, expected_dirblocks = self._get_unicode_tree()
1843
1874
        self.build_tree(tree)
1844
1875
        result = list(osutils._walkdirs_utf8('.'))
1845
1876
        self.assertEqual(expected_dirblocks, self._filter_out(result))
1846
1877
 
1847
1878
    def test_symlink(self):
1848
 
        self.requireFeature(tests.SymlinkFeature)
1849
 
        self.requireFeature(tests.UnicodeFilenameFeature)
 
1879
        self.requireFeature(features.SymlinkFeature)
 
1880
        self.requireFeature(features.UnicodeFilenameFeature)
1850
1881
        target = u'target\N{Euro Sign}'
1851
1882
        link_name = u'l\N{Euro Sign}nk'
1852
1883
        os.symlink(target, link_name)
1870
1901
    But prior python versions failed to properly encode the passed unicode
1871
1902
    string.
1872
1903
    """
1873
 
    _test_needs_features = [tests.SymlinkFeature, tests.UnicodeFilenameFeature]
 
1904
    _test_needs_features = [features.SymlinkFeature, features.UnicodeFilenameFeature]
1874
1905
 
1875
1906
    def setUp(self):
1876
1907
        super(tests.TestCaseInTempDir, self).setUp()
1879
1910
        os.symlink(self.target, self.link)
1880
1911
 
1881
1912
    def test_os_readlink_link_encoding(self):
1882
 
        if sys.version_info < (2, 6):
1883
 
            self.assertRaises(UnicodeEncodeError, os.readlink, self.link)
1884
 
        else:
1885
 
            self.assertEquals(self.target,  os.readlink(self.link))
 
1913
        self.assertEquals(self.target,  os.readlink(self.link))
1886
1914
 
1887
1915
    def test_os_readlink_link_decoding(self):
1888
1916
        self.assertEquals(self.target.encode(osutils._fs_enc),
1900
1928
        self.assertIsInstance(concurrency, int)
1901
1929
 
1902
1930
    def test_local_concurrency_environment_variable(self):
1903
 
        os.environ['BZR_CONCURRENCY'] = '2'
 
1931
        self.overrideEnv('BZR_CONCURRENCY', '2')
1904
1932
        self.assertEqual(2, osutils.local_concurrency(use_cache=False))
1905
 
        os.environ['BZR_CONCURRENCY'] = '3'
 
1933
        self.overrideEnv('BZR_CONCURRENCY', '3')
1906
1934
        self.assertEqual(3, osutils.local_concurrency(use_cache=False))
1907
 
        os.environ['BZR_CONCURRENCY'] = 'foo'
 
1935
        self.overrideEnv('BZR_CONCURRENCY', 'foo')
1908
1936
        self.assertEqual(1, osutils.local_concurrency(use_cache=False))
1909
1937
 
1910
1938
    def test_option_concurrency(self):
1911
 
        os.environ['BZR_CONCURRENCY'] = '1'
 
1939
        self.overrideEnv('BZR_CONCURRENCY', '1')
1912
1940
        self.run_bzr('rocks --concurrency 42')
1913
 
        # Command line overrides envrionment variable
 
1941
        # Command line overrides environment variable
1914
1942
        self.assertEquals('42', os.environ['BZR_CONCURRENCY'])
1915
1943
        self.assertEquals(42, osutils.local_concurrency(use_cache=False))
1916
1944
 
1986
2014
    def test_defaults_to_BZR_COLUMNS(self):
1987
2015
        # BZR_COLUMNS is set by the test framework
1988
2016
        self.assertNotEqual('12', os.environ['BZR_COLUMNS'])
1989
 
        os.environ['BZR_COLUMNS'] = '12'
 
2017
        self.overrideEnv('BZR_COLUMNS', '12')
1990
2018
        self.assertEqual(12, osutils.terminal_width())
1991
2019
 
 
2020
    def test_BZR_COLUMNS_0_no_limit(self):
 
2021
        self.overrideEnv('BZR_COLUMNS', '0')
 
2022
        self.assertEqual(None, osutils.terminal_width())
 
2023
 
1992
2024
    def test_falls_back_to_COLUMNS(self):
1993
 
        del os.environ['BZR_COLUMNS']
 
2025
        self.overrideEnv('BZR_COLUMNS', None)
1994
2026
        self.assertNotEqual('42', os.environ['COLUMNS'])
1995
2027
        self.set_fake_tty()
1996
 
        os.environ['COLUMNS'] = '42'
 
2028
        self.overrideEnv('COLUMNS', '42')
1997
2029
        self.assertEqual(42, osutils.terminal_width())
1998
2030
 
1999
2031
    def test_tty_default_without_columns(self):
2000
 
        del os.environ['BZR_COLUMNS']
2001
 
        del os.environ['COLUMNS']
 
2032
        self.overrideEnv('BZR_COLUMNS', None)
 
2033
        self.overrideEnv('COLUMNS', None)
2002
2034
 
2003
2035
        def terminal_size(w, h):
2004
2036
            return 42, 42
2011
2043
        self.assertEqual(42, osutils.terminal_width())
2012
2044
 
2013
2045
    def test_non_tty_default_without_columns(self):
2014
 
        del os.environ['BZR_COLUMNS']
2015
 
        del os.environ['COLUMNS']
 
2046
        self.overrideEnv('BZR_COLUMNS', None)
 
2047
        self.overrideEnv('COLUMNS', None)
2016
2048
        self.replace_stdout(None)
2017
2049
        self.assertEqual(None, osutils.terminal_width())
2018
2050
 
2028
2060
        else:
2029
2061
            self.overrideAttr(termios, 'TIOCGWINSZ')
2030
2062
            del termios.TIOCGWINSZ
2031
 
        del os.environ['BZR_COLUMNS']
2032
 
        del os.environ['COLUMNS']
 
2063
        self.overrideEnv('BZR_COLUMNS', None)
 
2064
        self.overrideEnv('COLUMNS', None)
2033
2065
        # Whatever the result is, if we don't raise an exception, it's ok.
2034
2066
        osutils.terminal_width()
2035
2067
 
 
2068
 
2036
2069
class TestCreationOps(tests.TestCaseInTempDir):
2037
2070
    _test_needs_features = [features.chown_feature]
2038
2071
 
2068
2101
        self.assertEquals(self.uid, s.st_uid)
2069
2102
        self.assertEquals(self.gid, s.st_gid)
2070
2103
 
 
2104
 
2071
2105
class TestGetuserUnicode(tests.TestCase):
2072
2106
 
2073
2107
    def test_ascii_user(self):
2074
 
        osutils.set_or_unset_env('LOGNAME', 'jrandom')
 
2108
        self.overrideEnv('LOGNAME', 'jrandom')
2075
2109
        self.assertEqual(u'jrandom', osutils.getuser_unicode())
2076
2110
 
2077
2111
    def test_unicode_user(self):
2078
2112
        ue = osutils.get_user_encoding()
2079
 
        osutils.set_or_unset_env('LOGNAME', u'jrandom\xb6'.encode(ue))
 
2113
        uni_val, env_val = tests.probe_unicode_in_user_encoding()
 
2114
        if uni_val is None:
 
2115
            raise tests.TestSkipped(
 
2116
                'Cannot find a unicode character that works in encoding %s'
 
2117
                % (osutils.get_user_encoding(),))
 
2118
        uni_username = u'jrandom' + uni_val
 
2119
        encoded_username = uni_username.encode(ue)
 
2120
        self.overrideEnv('LOGNAME', encoded_username)
 
2121
        self.assertEqual(uni_username, osutils.getuser_unicode())
 
2122
        self.overrideEnv('LOGNAME', u'jrandom\xb6'.encode(ue))
2080
2123
        self.assertEqual(u'jrandom\xb6', osutils.getuser_unicode())
 
2124
 
 
2125
    def test_no_username_bug_660174(self):
 
2126
        self.requireFeature(features.win32_feature)
 
2127
        for name in ('LOGNAME', 'USER', 'LNAME', 'USERNAME'):
 
2128
            self.overrideEnv(name, None)
 
2129
        self.assertEqual(u'UNKNOWN', osutils.getuser_unicode())
 
2130
 
 
2131
 
 
2132
class TestBackupNames(tests.TestCase):
 
2133
 
 
2134
    def setUp(self):
 
2135
        super(TestBackupNames, self).setUp()
 
2136
        self.backups = []
 
2137
 
 
2138
    def backup_exists(self, name):
 
2139
        return name in self.backups
 
2140
 
 
2141
    def available_backup_name(self, name):
 
2142
        backup_name = osutils.available_backup_name(name, self.backup_exists)
 
2143
        self.backups.append(backup_name)
 
2144
        return backup_name
 
2145
 
 
2146
    def assertBackupName(self, expected, name):
 
2147
        self.assertEqual(expected, self.available_backup_name(name))
 
2148
 
 
2149
    def test_empty(self):
 
2150
        self.assertBackupName('file.~1~', 'file')
 
2151
 
 
2152
    def test_existing(self):
 
2153
        self.available_backup_name('file')
 
2154
        self.available_backup_name('file')
 
2155
        self.assertBackupName('file.~3~', 'file')
 
2156
        # Empty slots are found, this is not a strict requirement and may be
 
2157
        # revisited if we test against all implementations.
 
2158
        self.backups.remove('file.~2~')
 
2159
        self.assertBackupName('file.~2~', 'file')
 
2160
 
 
2161
 
 
2162
class TestFindExecutableInPath(tests.TestCase):
 
2163
 
 
2164
    def test_windows(self):
 
2165
        if sys.platform != 'win32':
 
2166
            raise tests.TestSkipped('test requires win32')
 
2167
        self.assertTrue(osutils.find_executable_on_path('explorer') is not None)
 
2168
        self.assertTrue(
 
2169
            osutils.find_executable_on_path('explorer.exe') is not None)
 
2170
        self.assertTrue(
 
2171
            osutils.find_executable_on_path('EXPLORER.EXE') is not None)
 
2172
        self.assertTrue(
 
2173
            osutils.find_executable_on_path('THIS SHOULD NOT EXIST') is None)
 
2174
        self.assertTrue(osutils.find_executable_on_path('file.txt') is None)
 
2175
 
 
2176
    def test_other(self):
 
2177
        if sys.platform == 'win32':
 
2178
            raise tests.TestSkipped('test requires non-win32')
 
2179
        self.assertTrue(osutils.find_executable_on_path('sh') is not None)
 
2180
        self.assertTrue(
 
2181
            osutils.find_executable_on_path('THIS SHOULD NOT EXIST') is None)