24
23
from bzrlib import (
29
from bzrlib.branch import Branch
30
from bzrlib.config import ensure_config_dir_exists, config_filename
31
31
from bzrlib.msgeditor import (
32
32
make_commit_message_template_encoded,
33
33
edit_commit_message_encoded
35
35
from bzrlib.tests import (
37
38
TestCaseWithTransport,
45
from bzrlib.tests.EncodingAdapter import encoding_scenarios
40
46
from bzrlib.trace import mutter
49
def load_tests(standard_tests, module, loader):
50
"""Parameterize the test for tempfile creation with different encodings."""
51
to_adapt, result = split_suite_by_re(standard_tests,
52
"test__create_temp_file_with_commit_template_in_unicode_dir")
53
return multiply_tests(to_adapt, encoding_scenarios, result)
43
56
class MsgEditorTest(TestCaseWithTransport):
45
58
def make_uncommitted_tree(self):
82
def make_multiple_pending_tree(self):
83
config.GlobalStack().set('email', 'Bilbo Baggins <bb@hobbit.net>')
84
tree = self.make_branch_and_tree('a')
85
tree.commit('Initial checkin.', timestamp=1230912900, timezone=0)
86
tree2 = tree.bzrdir.clone('b').open_workingtree()
87
tree.commit('Minor tweak.', timestamp=1231977840, timezone=0)
88
tree2.commit('Feature X work.', timestamp=1233186240, timezone=0)
89
tree3 = tree2.bzrdir.clone('c').open_workingtree()
90
tree2.commit('Feature X finished.', timestamp=1233187680, timezone=0)
91
tree3.commit('Feature Y, based on initial X work.',
92
timestamp=1233285960, timezone=0)
93
tree.merge_from_branch(tree2.branch)
94
tree.merge_from_branch(tree3.branch, force=True)
97
def test_commit_template_pending_merges(self):
98
"""Test building a commit message template when there are pending
99
merges. The commit message should show all pending merge revisions,
100
as does 'status -v', not only the merge tips.
102
working_tree = self.make_multiple_pending_tree()
103
template = msgeditor.make_commit_message_template(working_tree, None)
104
self.assertEqualDiff(template,
107
Bilbo Baggins 2009-01-29 Feature X finished.
108
Bilbo Baggins 2009-01-28 Feature X work.
109
Bilbo Baggins 2009-01-30 Feature Y, based on initial X work.
69
112
def test_commit_template_encoded(self):
70
113
"""Test building a commit message template"""
71
114
working_tree = self.make_uncommitted_tree()
97
140
""".encode('utf8') in template)
99
def test_run_editor(self):
142
def make_do_nothing_editor(self, basename='fed'):
100
143
if sys.platform == "win32":
101
f = file('fed.bat', 'w')
144
name = basename + '.bat'
102
146
f.write('@rem dummy fed')
104
os.environ['BZR_EDITOR'] = 'fed.bat'
106
f = file('fed.sh', 'wb')
150
name = basename + '.sh'
107
152
f.write('#!/bin/sh\n')
109
os.chmod('fed.sh', 0755)
110
os.environ['BZR_EDITOR'] = './fed.sh'
157
def test_run_editor(self):
158
self.overrideEnv('BZR_EDITOR', self.make_do_nothing_editor())
112
159
self.assertEqual(True, msgeditor._run_editor(''),
113
160
'Unable to run dummy fake editor')
162
def test_parse_editor_name(self):
163
"""Correctly interpret names with spaces.
165
See <https://bugs.launchpad.net/bzr/+bug/220331>
167
self.overrideEnv('BZR_EDITOR',
168
'"%s"' % self.make_do_nothing_editor('name with spaces'))
169
self.assertEqual(True, msgeditor._run_editor('a_filename'))
115
171
def make_fake_editor(self, message='test message from fed\\n'):
116
172
"""Set up environment so that an editor will be a known script.
181
237
working_tree = self.make_uncommitted_tree()
183
239
if sys.platform == 'win32':
184
os.environ['BZR_EDITOR'] = 'cmd.exe /c del'
240
editor = 'cmd.exe /c del'
186
os.environ['BZR_EDITOR'] = 'rm'
243
self.overrideEnv('BZR_EDITOR', editor)
188
245
self.assertRaises((IOError, OSError), msgeditor.edit_commit_message, '')
190
247
def test__get_editor(self):
191
# Test that _get_editor can return a decent list of items
192
bzr_editor = os.environ.get('BZR_EDITOR')
193
visual = os.environ.get('VISUAL')
194
editor = os.environ.get('EDITOR')
248
self.overrideEnv('BZR_EDITOR', 'bzr_editor')
249
self.overrideEnv('VISUAL', 'visual')
250
self.overrideEnv('EDITOR', 'editor')
252
conf = config.GlobalStack()
253
conf.store._load_from_string('[DEFAULT]\neditor = config_editor\n')
255
editors = list(msgeditor._get_editor())
256
editors = [editor for (editor, cfg_src) in editors]
258
self.assertEqual(['bzr_editor', 'config_editor', 'visual', 'editor'],
261
if sys.platform == 'win32':
262
self.assertEqual(['wordpad.exe', 'notepad.exe'], editors[4:])
264
self.assertEqual(['/usr/bin/editor', 'vi', 'pico', 'nano', 'joe'],
268
def test__run_editor_EACCES(self):
269
"""If running a configured editor raises EACESS, the user is warned."""
270
self.overrideEnv('BZR_EDITOR', 'eacces.py')
271
f = file('eacces.py', 'wb')
272
f.write('# Not a real editor')
274
# Make the fake editor unreadable (and unexecutable)
275
os.chmod('eacces.py', 0)
276
# Set $EDITOR so that _run_editor will terminate before trying real
278
self.overrideEnv('EDITOR', self.make_do_nothing_editor())
279
# Call _run_editor, capturing mutter.warning calls.
283
warnings.append(args[0] % args[1:])
285
warnings.append(args[0])
286
_warning = trace.warning
287
trace.warning = warning
196
os.environ['BZR_EDITOR'] = 'bzr_editor'
197
os.environ['VISUAL'] = 'visual'
198
os.environ['EDITOR'] = 'editor'
200
ensure_config_dir_exists()
201
f = open(config_filename(), 'wb')
202
f.write('editor = config_editor\n')
205
editors = list(msgeditor._get_editor())
207
self.assertEqual(['bzr_editor', 'config_editor', 'visual',
208
'editor'], editors[:4])
210
if sys.platform == 'win32':
211
self.assertEqual(['wordpad.exe', 'notepad.exe'], editors[4:])
213
self.assertEqual(['/usr/bin/editor', 'vi', 'pico', 'nano',
289
msgeditor._run_editor('')
217
# Restore the environment
218
if bzr_editor is None:
219
del os.environ['BZR_EDITOR']
221
os.environ['BZR_EDITOR'] = bzr_editor
223
del os.environ['VISUAL']
225
os.environ['VISUAL'] = visual
227
del os.environ['EDITOR']
229
os.environ['EDITOR'] = editor
291
trace.warning = _warning
292
self.assertStartsWith(warnings[0], 'Could not start editor "eacces.py"')
231
294
def test__create_temp_file_with_commit_template(self):
232
295
# check that commit template written properly
244
307
self.assertFileEqual(expected, msgfilename)
309
def test__create_temp_file_with_commit_template_in_unicode_dir(self):
310
self.requireFeature(features.UnicodeFilenameFeature)
311
if hasattr(self, 'info'):
312
tmpdir = self.info['directory']
314
# Force the creation of temp file in a directory whose name
315
# requires some encoding support
316
msgeditor._create_temp_file_with_commit_template('infotext',
319
raise TestNotApplicable('Test run elsewhere with non-ascii data.')
246
321
def test__create_temp_file_with_empty_commit_template(self):
248
323
create_file = msgeditor._create_temp_file_with_commit_template
252
327
self.assertFileEqual('', msgfilename)
254
329
def test_unsupported_encoding_commit_message(self):
255
old_env = osutils.set_or_unset_env('LANG', 'C')
257
# LANG env variable has no effect on Windows
258
# but some characters anyway cannot be represented
259
# in default user encoding
260
char = probe_bad_non_ascii(bzrlib.user_encoding)
262
raise TestSkipped('Cannot find suitable non-ascii character '
263
'for user_encoding (%s)' % bzrlib.user_encoding)
265
self.make_fake_editor(message=char)
267
working_tree = self.make_uncommitted_tree()
268
self.assertRaises(errors.BadCommitMessageEncoding,
269
msgeditor.edit_commit_message, '')
271
osutils.set_or_unset_env('LANG', old_env)
330
self.overrideEnv('LANG', 'C')
331
# LANG env variable has no effect on Windows
332
# but some characters anyway cannot be represented
333
# in default user encoding
334
char = probe_bad_non_ascii(osutils.get_user_encoding())
336
raise TestSkipped('Cannot find suitable non-ascii character '
337
'for user_encoding (%s)' % osutils.get_user_encoding())
339
self.make_fake_editor(message=char)
341
working_tree = self.make_uncommitted_tree()
342
self.assertRaises(errors.BadCommitMessageEncoding,
343
msgeditor.edit_commit_message, '')
345
def test_set_commit_message_no_hooks(self):
346
commit_obj = commit.Commit()
348
msgeditor.set_commit_message(commit_obj))
350
def test_set_commit_message_hook(self):
351
msgeditor.hooks.install_named_hook("set_commit_message",
352
lambda commit_obj, existing_message: "save me some typing\n", None)
353
commit_obj = commit.Commit()
354
self.assertEquals("save me some typing\n",
355
msgeditor.set_commit_message(commit_obj))
357
def test_generate_commit_message_template_no_hooks(self):
358
commit_obj = commit.Commit()
360
msgeditor.generate_commit_message_template(commit_obj))
362
def test_generate_commit_message_template_hook(self):
363
msgeditor.hooks.install_named_hook("commit_message_template",
364
lambda commit_obj, msg: "save me some typing\n", None)
365
commit_obj = commit.Commit()
366
self.assertEquals("save me some typing\n",
367
msgeditor.generate_commit_message_template(commit_obj))
370
# GZ 2009-11-17: This wants moving to osutils when the errno checking code is
371
class TestPlatformErrnoWorkarounds(TestCaseInTempDir):
372
"""Ensuring workarounds enshrined in code actually serve a purpose"""
374
def test_subprocess_call_bad_file(self):
375
if sys.platform != "win32":
376
raise TestNotApplicable("Workarounds for windows only")
377
import subprocess, errno
378
ERROR_BAD_EXE_FORMAT = 193
379
file("textfile.txt", "w").close()
380
e = self.assertRaises(WindowsError, subprocess.call, "textfile.txt")
381
self.assertEqual(e.errno, errno.ENOEXEC)
382
self.assertEqual(e.winerror, ERROR_BAD_EXE_FORMAT)