~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_win32utils.py

  • Committer: Martin Pool
  • Date: 2010-04-01 04:41:18 UTC
  • mto: This revision was merged to the branch mainline in revision 5128.
  • Revision ID: mbp@sourcefrog.net-20100401044118-shyctqc02ob08ngz
ignore .testrepository

Show diffs side-by-side

added added

removed removed

Lines of Context:
14
14
# along with this program; if not, write to the Free Software
15
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
16
 
17
 
"""Tests for win32utils."""
18
 
 
19
17
import os
20
18
import sys
21
19
 
31
29
    TestSkipped,
32
30
    UnicodeFilenameFeature,
33
31
    )
34
 
from bzrlib.tests.features import backslashdir_feature
35
32
from bzrlib.win32utils import glob_expand, get_app_path
36
33
 
37
34
 
 
35
class _BackslashDirSeparatorFeature(tests.Feature):
 
36
 
 
37
    def _probe(self):
 
38
        try:
 
39
            os.lstat(os.getcwd() + '\\')
 
40
        except OSError:
 
41
            return False
 
42
        else:
 
43
            return True
 
44
 
 
45
    def feature_name(self):
 
46
        return "Filesystem treats '\\' as a directory separator."
 
47
 
 
48
BackslashDirSeparatorFeature = _BackslashDirSeparatorFeature()
 
49
 
 
50
 
38
51
class _RequiredModuleFeature(Feature):
39
52
 
40
53
    def __init__(self, mod_name):
54
67
Win32RegistryFeature = _RequiredModuleFeature('_winreg')
55
68
CtypesFeature = _RequiredModuleFeature('ctypes')
56
69
Win32comShellFeature = _RequiredModuleFeature('win32com.shell')
57
 
Win32ApiFeature = _RequiredModuleFeature('win32api') 
58
70
 
59
71
 
60
72
# Tests
109
121
            ])
110
122
 
111
123
    def test_backslash_globbing(self):
112
 
        self.requireFeature(backslashdir_feature)
 
124
        self.requireFeature(BackslashDirSeparatorFeature)
113
125
        self.build_ascii_tree()
114
126
        self._run_testset([
115
127
            [[u'd\\'], [u'd/']],
152
164
            ])
153
165
 
154
166
    def test_unicode_backslashes(self):
155
 
        self.requireFeature(backslashdir_feature)
 
167
        self.requireFeature(BackslashDirSeparatorFeature)
156
168
        self.build_unicode_tree()
157
169
        self._run_testset([
158
170
            # no wildcards
190
202
        # typical windows users should have wordpad in the system
191
203
        # but there is problem: its path has the format REG_EXPAND_SZ
192
204
        # so naive attempt to get the path is not working
193
 
        self.requireFeature(Win32ApiFeature)
194
205
        for a in ('wordpad', 'wordpad.exe'):
195
206
            p = get_app_path(a)
196
207
            d, b = os.path.split(p)
276
287
        win32utils.set_file_attr_hidden(path)
277
288
 
278
289
 
 
290
 
 
291
 
279
292
class Test_CommandLineToArgv(tests.TestCaseInTempDir):
280
293
 
281
 
    def assertCommandLine(self, expected, line, argv=None,
282
 
            single_quotes_allowed=False):
 
294
    def assertCommandLine(self, expected, line, single_quotes_allowed=False):
283
295
        # Strictly speaking we should respect parameter order versus glob
284
296
        # expansions, but it's not really worth the effort here
285
 
        if argv is None:
286
 
            argv = [line]
287
 
        argv = win32utils._command_line_to_argv(line, argv,
 
297
        argv = win32utils._command_line_to_argv(line,
288
298
                single_quotes_allowed=single_quotes_allowed)
289
299
        self.assertEqual(expected, sorted(argv))
290
300
 
318
328
 
319
329
    def test_single_quote_support(self):
320
330
        self.assertCommandLine(["add", "let's-do-it.txt"],
321
 
            "add let's-do-it.txt",
322
 
            ["add", "let's-do-it.txt"])
323
 
        self.expectFailure("Using single quotes breaks trimming from argv",
324
 
            self.assertCommandLine, ["add", "lets do it.txt"],
325
 
            "add 'lets do it.txt'", ["add", "'lets", "do", "it.txt'"],
326
 
            single_quotes_allowed=True)
 
331
            "add let's-do-it.txt")
 
332
        self.assertCommandLine(["add", "lets do it.txt"],
 
333
            "add 'lets do it.txt'", single_quotes_allowed=True)
327
334
 
328
335
    def test_case_insensitive_globs(self):
329
336
        self.requireFeature(tests.CaseInsCasePresFilenameFeature)
331
338
        self.assertCommandLine([u'A/b.c'], 'A/B*')
332
339
 
333
340
    def test_backslashes(self):
334
 
        self.requireFeature(backslashdir_feature)
 
341
        self.requireFeature(BackslashDirSeparatorFeature)
335
342
        self.build_tree(['a/', 'a/b.c', 'a/c.c', 'a/c.h'])
336
343
        self.assertCommandLine([u'a/b.c'], 'a\\b*')
337
 
 
338
 
    def test_with_pdb(self):
339
 
        """Check stripping Python arguments before bzr script per lp:587868"""
340
 
        self.assertCommandLine([u"rocks"], "-m pdb rocks", ["rocks"])
341
 
        self.build_tree(['d/', 'd/f1', 'd/f2'])
342
 
        self.assertCommandLine([u"rm", u"x*"], "-m pdb rm x*", ["rm", u"x*"])
343
 
        self.assertCommandLine([u"add", u"d/f1", u"d/f2"], "-m pdb add d/*",
344
 
            ["add", u"d/*"])