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
17
"""Tests for win32utils."""
21
20
from bzrlib import (
26
25
from bzrlib.tests import (
30
UnicodeFilenameFeature,
31
from bzrlib.tests.features import backslashdir_feature
32
32
from bzrlib.win32utils import glob_expand, get_app_path
33
from bzrlib.tests import (
38
Win32RegistryFeature = features.ModuleAvailableFeature('_winreg')
39
CtypesFeature = features.ModuleAvailableFeature('ctypes')
40
Win32comShellFeature = features.ModuleAvailableFeature('win32com.shell')
41
Win32ApiFeature = features.ModuleAvailableFeature('win32api')
35
class _BackslashDirSeparatorFeature(tests.Feature):
39
os.lstat(os.getcwd() + '\\')
45
def feature_name(self):
46
return "Filesystem treats '\\' as a directory separator."
48
BackslashDirSeparatorFeature = _BackslashDirSeparatorFeature()
51
class _RequiredModuleFeature(Feature):
53
def __init__(self, mod_name):
54
self.mod_name = mod_name
55
super(_RequiredModuleFeature, self).__init__()
59
__import__(self.mod_name)
64
def feature_name(self):
67
Win32RegistryFeature = _RequiredModuleFeature('_winreg')
68
CtypesFeature = _RequiredModuleFeature('ctypes')
69
Win32comShellFeature = _RequiredModuleFeature('win32com.shell')
63
91
'd/', 'd/d1', 'd/d2', 'd/e/', 'd/e/e1'])
65
93
def build_unicode_tree(self):
66
self.requireFeature(features.UnicodeFilenameFeature)
94
self.requireFeature(UnicodeFilenameFeature)
67
95
self.build_tree([u'\u1234', u'\u1234\u1234', u'\u1235/',
175
202
# typical windows users should have wordpad in the system
176
203
# but there is problem: its path has the format REG_EXPAND_SZ
177
204
# so naive attempt to get the path is not working
178
self.requireFeature(Win32ApiFeature)
179
205
for a in ('wordpad', 'wordpad.exe'):
180
206
p = get_app_path(a)
181
207
d, b = os.path.split(p)
201
227
def test_appdata_not_using_environment(self):
202
228
# Test that we aren't falling back to the environment
203
229
first = win32utils.get_appdata_location()
204
self.overrideEnv("APPDATA", None)
230
self._captureVar("APPDATA", None)
205
231
self.assertPathsEqual(first, win32utils.get_appdata_location())
207
233
def test_appdata_matches_environment(self):
218
244
def test_local_appdata_not_using_environment(self):
219
245
# Test that we aren't falling back to the environment
220
246
first = win32utils.get_local_appdata_location()
221
self.overrideEnv("LOCALAPPDATA", None)
247
self._captureVar("LOCALAPPDATA", None)
222
248
self.assertPathsEqual(first, win32utils.get_local_appdata_location())
224
250
def test_local_appdata_matches_environment(self):
249
275
def test_unicode_dir(self):
250
276
# we should handle unicode paths without errors
251
self.requireFeature(features.UnicodeFilenameFeature)
277
self.requireFeature(UnicodeFilenameFeature)
252
278
os.mkdir(u'\u1234')
253
279
win32utils.set_file_attr_hidden(u'\u1234')
255
281
def test_dot_bzr_in_unicode_dir(self):
256
282
# we should not raise traceback if we try to set hidden attribute
257
283
# on .bzr directory below unicode path
258
self.requireFeature(features.UnicodeFilenameFeature)
284
self.requireFeature(UnicodeFilenameFeature)
259
285
os.makedirs(u'\u1234\\.bzr')
260
286
path = osutils.abspath(u'\u1234\\.bzr')
261
287
win32utils.set_file_attr_hidden(path)
264
292
class Test_CommandLineToArgv(tests.TestCaseInTempDir):
266
def assertCommandLine(self, expected, line, argv=None,
267
single_quotes_allowed=False):
294
def assertCommandLine(self, expected, line, single_quotes_allowed=False):
268
295
# Strictly speaking we should respect parameter order versus glob
269
296
# expansions, but it's not really worth the effort here
272
argv = win32utils._command_line_to_argv(line, argv,
297
argv = win32utils._command_line_to_argv(line,
273
298
single_quotes_allowed=single_quotes_allowed)
274
299
self.assertEqual(expected, sorted(argv))
304
329
def test_single_quote_support(self):
305
330
self.assertCommandLine(["add", "let's-do-it.txt"],
306
"add let's-do-it.txt",
307
["add", "let's-do-it.txt"])
308
self.expectFailure("Using single quotes breaks trimming from argv",
309
self.assertCommandLine, ["add", "lets do it.txt"],
310
"add 'lets do it.txt'", ["add", "'lets", "do", "it.txt'"],
311
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)
313
335
def test_case_insensitive_globs(self):
314
if os.path.normcase("AbC") == "AbC":
315
self.skip("Test requires case insensitive globbing function")
336
self.requireFeature(tests.CaseInsCasePresFilenameFeature)
316
337
self.build_tree(['a/', 'a/b.c', 'a/c.c', 'a/c.h'])
317
338
self.assertCommandLine([u'A/b.c'], 'A/B*')
319
340
def test_backslashes(self):
320
self.requireFeature(backslashdir_feature)
341
self.requireFeature(BackslashDirSeparatorFeature)
321
342
self.build_tree(['a/', 'a/b.c', 'a/c.c', 'a/c.h'])
322
343
self.assertCommandLine([u'a/b.c'], 'a\\b*')
324
def test_with_pdb(self):
325
"""Check stripping Python arguments before bzr script per lp:587868"""
326
self.assertCommandLine([u"rocks"], "-m pdb rocks", ["rocks"])
327
self.build_tree(['d/', 'd/f1', 'd/f2'])
328
self.assertCommandLine([u"rm", u"x*"], "-m pdb rm x*", ["rm", u"x*"])
329
self.assertCommandLine([u"add", u"d/f1", u"d/f2"], "-m pdb add d/*",