31
from bzrlib.msgeditor import (
31
from ..msgeditor import (
32
32
make_commit_message_template_encoded,
33
33
edit_commit_message_encoded
35
from bzrlib.tests import (
38
38
TestCaseWithTransport,
42
41
probe_bad_non_ascii,
42
probe_unicode_in_user_encoding,
45
from bzrlib.tests.EncodingAdapter import encoding_scenarios
46
from bzrlib.trace import mutter
49
def load_tests(standard_tests, module, loader):
45
from .EncodingAdapter import encoding_scenarios
46
from ..trace import mutter
49
def load_tests(loader, standard_tests, pattern):
50
50
"""Parameterize the test for tempfile creation with different encodings."""
51
to_adapt, result = split_suite_by_re(standard_tests,
51
to_adapt, result = split_suite_by_re(
52
53
"test__create_temp_file_with_commit_template_in_unicode_dir")
53
54
return multiply_tests(to_adapt, encoding_scenarios, result)
58
59
def make_uncommitted_tree(self):
59
60
"""Build a branch with uncommitted unicode named changes in the cwd."""
60
61
working_tree = self.make_branch_and_tree('.')
61
b = working_tree.branch
62
62
filename = u'hell\u00d8'
64
self.build_tree_contents([(filename, 'contents of hello')])
64
self.build_tree_contents([(filename, b'contents of hello')])
65
65
except UnicodeEncodeError:
66
raise TestSkipped("can't build unicode working tree in "
67
"filesystem encoding %s" % sys.getfilesystemencoding())
66
self.skipTest("can't build unicode working tree in "
67
"filesystem encoding %s" % sys.getfilesystemencoding())
68
68
working_tree.add(filename)
69
69
return working_tree
83
83
config.GlobalStack().set('email', 'Bilbo Baggins <bb@hobbit.net>')
84
84
tree = self.make_branch_and_tree('a')
85
85
tree.commit('Initial checkin.', timestamp=1230912900, timezone=0)
86
tree2 = tree.bzrdir.clone('b').open_workingtree()
86
tree2 = tree.controldir.clone('b').open_workingtree()
87
87
tree.commit('Minor tweak.', timestamp=1231977840, timezone=0)
88
88
tree2.commit('Feature X work.', timestamp=1233186240, timezone=0)
89
tree3 = tree2.bzrdir.clone('c').open_workingtree()
89
tree3 = tree2.controldir.clone('c').open_workingtree()
90
90
tree2.commit('Feature X finished.', timestamp=1233187680, timezone=0)
91
91
tree3.commit('Feature Y, based on initial X work.',
92
92
timestamp=1233285960, timezone=0)
142
141
def make_do_nothing_editor(self, basename='fed'):
143
142
if sys.platform == "win32":
144
143
name = basename + '.bat'
146
f.write('@rem dummy fed')
144
with open(name, 'w') as f:
145
f.write('@rem dummy fed')
150
148
name = basename + '.sh'
152
f.write('#!/bin/sh\n')
149
with open(name, 'wb') as f:
150
f.write(b'#!/bin/sh\n')
151
os.chmod(name, 0o755)
155
152
return './' + name
157
154
def test_run_editor(self):
158
self.overrideEnv('BZR_EDITOR', self.make_do_nothing_editor())
155
self.overrideEnv('BRZ_EDITOR', self.make_do_nothing_editor())
159
156
self.assertEqual(True, msgeditor._run_editor(''),
160
157
'Unable to run dummy fake editor')
165
162
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'))
164
self.overrideEnv('BRZ_EDITOR',
165
'"%s"' % self.make_do_nothing_editor('name with spaces'))
166
self.assertEqual(True, msgeditor._run_editor('a_filename'))
171
def make_fake_editor(self, message='test message from fed\\n'):
168
def make_fake_editor(self, message='test message from fed\n'):
172
169
"""Set up environment so that an editor will be a known script.
174
Sets up BZR_EDITOR so that if an editor is spawned it will run a
171
Sets up BRZ_EDITOR so that if an editor is spawned it will run a
175
172
script that just adds a known message to the start of the file.
177
f = file('fed.py', 'wb')
178
f.write('#!%s\n' % sys.executable)
174
if not isinstance(message, bytes):
175
message = message.encode('utf-8')
176
with open('fed.py', 'w') as f:
177
f.write('#!%s\n' % sys.executable)
182
181
if len(sys.argv) == 2:
183
with open(fn, 'rb') as f:
185
with open(fn, 'wb') as f:
191
188
""" % (message, ))
193
189
if sys.platform == "win32":
194
# [win32] make batch file and set BZR_EDITOR
195
f = file('fed.bat', 'w')
190
# [win32] make batch file and set BRZ_EDITOR
191
with open('fed.bat', 'w') as f:
199
195
""" % sys.executable)
201
self.overrideEnv('BZR_EDITOR', 'fed.bat')
196
self.overrideEnv('BRZ_EDITOR', 'fed.bat')
203
# [non-win32] make python script executable and set BZR_EDITOR
204
os.chmod('fed.py', 0755)
205
self.overrideEnv('BZR_EDITOR', './fed.py')
198
# [non-win32] make python script executable and set BRZ_EDITOR
199
os.chmod('fed.py', 0o755)
200
self.overrideEnv('BRZ_EDITOR', './fed.py')
207
def test_edit_commit_message(self):
208
working_tree = self.make_uncommitted_tree()
202
def test_edit_commit_message_without_infotext(self):
203
self.make_uncommitted_tree()
209
204
self.make_fake_editor()
211
206
mutter('edit_commit_message without infotext')
212
207
self.assertEqual('test message from fed\n',
213
208
msgeditor.edit_commit_message(''))
210
def test_edit_commit_message_with_ascii_infotext(self):
211
self.make_uncommitted_tree()
212
self.make_fake_editor()
215
214
mutter('edit_commit_message with ascii string infotext')
216
215
self.assertEqual('test message from fed\n',
217
216
msgeditor.edit_commit_message('spam'))
218
def test_edit_commit_message_with_unicode_infotext(self):
219
self.make_uncommitted_tree()
220
self.make_fake_editor()
219
222
mutter('edit_commit_message with unicode infotext')
223
uni_val, ue_val = probe_unicode_in_user_encoding()
226
'Cannot find a unicode character that works in encoding %s'
227
% (osutils.get_user_encoding(),))
220
229
self.assertEqual('test message from fed\n',
221
msgeditor.edit_commit_message(u'\u1234'))
230
msgeditor.edit_commit_message(uni_val))
223
232
tmpl = edit_commit_message_encoded(u'\u1234'.encode("utf8"))
224
233
self.assertEqual('test message from fed\n', tmpl)
227
236
self.make_uncommitted_tree()
228
237
self.make_fake_editor()
229
238
self.assertEqual('test message from fed\nstart message\n',
230
msgeditor.edit_commit_message('',
231
start_message='start message\n'))
239
msgeditor.edit_commit_message(
240
'', start_message='start message\n'))
232
241
self.assertEqual('test message from fed\n',
233
msgeditor.edit_commit_message('',
242
msgeditor.edit_commit_message(
243
'', start_message=''))
236
245
def test_deleted_commit_message(self):
237
working_tree = self.make_uncommitted_tree()
246
self.make_uncommitted_tree()
239
248
if sys.platform == 'win32':
240
249
editor = 'cmd.exe /c del'
243
self.overrideEnv('BZR_EDITOR', editor)
252
self.overrideEnv('BRZ_EDITOR', editor)
245
self.assertRaises((IOError, OSError), msgeditor.edit_commit_message, '')
254
self.assertRaises((EnvironmentError, errors.NoSuchFile),
255
msgeditor.edit_commit_message, '')
247
257
def test__get_editor(self):
248
self.overrideEnv('BZR_EDITOR', 'bzr_editor')
258
self.overrideEnv('BRZ_EDITOR', 'bzr_editor')
249
259
self.overrideEnv('VISUAL', 'visual')
250
260
self.overrideEnv('EDITOR', 'editor')
252
262
conf = config.GlobalStack()
253
conf.store._load_from_string('[DEFAULT]\neditor = config_editor\n')
263
conf.store._load_from_string(b'[DEFAULT]\neditor = config_editor\n')
254
264
conf.store.save()
255
265
editors = list(msgeditor._get_editor())
256
266
editors = [editor for (editor, cfg_src) in editors]
264
274
self.assertEqual(['/usr/bin/editor', 'vi', 'pico', 'nano', 'joe'],
268
277
def test__run_editor_EACCES(self):
269
278
"""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')
279
self.overrideEnv('BRZ_EDITOR', 'eacces.py')
280
with open('eacces.py', 'wb') as f:
281
f.write(b'# Not a real editor')
274
282
# Make the fake editor unreadable (and unexecutable)
275
283
os.chmod('eacces.py', 0)
276
284
# Set $EDITOR so that _run_editor will terminate before trying real
289
298
msgeditor._run_editor('')
291
300
trace.warning = _warning
292
self.assertStartsWith(warnings[0], 'Could not start editor "eacces.py"')
301
self.assertStartsWith(
302
warnings[0], 'Could not start editor "eacces.py"')
294
304
def test__create_temp_file_with_commit_template(self):
295
305
# check that commit template written properly
296
306
# and has platform native line-endings (CRLF on win32)
297
307
create_file = msgeditor._create_temp_file_with_commit_template
298
msgfilename, hasinfo = create_file('infotext','----','start message')
308
msgfilename, hasinfo = create_file(
309
b'infotext', '----', b'start message')
299
310
self.assertNotEqual(None, msgfilename)
300
311
self.assertTrue(hasinfo)
301
312
expected = os.linesep.join(['start message',
333
344
# in default user encoding
334
345
char = probe_bad_non_ascii(osutils.get_user_encoding())
336
raise TestSkipped('Cannot find suitable non-ascii character '
348
'Cannot find suitable non-ascii character '
337
349
'for user_encoding (%s)' % osutils.get_user_encoding())
339
351
self.make_fake_editor(message=char)
341
working_tree = self.make_uncommitted_tree()
342
self.assertRaises(errors.BadCommitMessageEncoding,
353
self.make_uncommitted_tree()
354
self.assertRaises(msgeditor.BadCommitMessageEncoding,
343
355
msgeditor.edit_commit_message, '')
345
357
def test_set_commit_message_no_hooks(self):
346
358
commit_obj = commit.Commit()
347
359
self.assertIs(None,
348
msgeditor.set_commit_message(commit_obj))
360
msgeditor.set_commit_message(commit_obj))
350
362
def test_set_commit_message_hook(self):
351
363
msgeditor.hooks.install_named_hook("set_commit_message",
352
lambda commit_obj, existing_message: "save me some typing\n", None)
364
lambda commit_obj, existing_message: "save me some typing\n", None)
353
365
commit_obj = commit.Commit()
354
self.assertEquals("save me some typing\n",
355
msgeditor.set_commit_message(commit_obj))
366
self.assertEqual("save me some typing\n",
367
msgeditor.set_commit_message(commit_obj))
357
369
def test_generate_commit_message_template_no_hooks(self):
358
370
commit_obj = commit.Commit()
359
371
self.assertIs(None,
360
msgeditor.generate_commit_message_template(commit_obj))
372
msgeditor.generate_commit_message_template(commit_obj))
362
374
def test_generate_commit_message_template_hook(self):
363
375
msgeditor.hooks.install_named_hook("commit_message_template",
364
lambda commit_obj, msg: "save me some typing\n", None)
376
lambda commit_obj, msg: "save me some typing\n", None)
365
377
commit_obj = commit.Commit()
366
self.assertEquals("save me some typing\n",
367
msgeditor.generate_commit_message_template(commit_obj))
378
self.assertEqual("save me some typing\n",
379
msgeditor.generate_commit_message_template(commit_obj))
370
382
# GZ 2009-11-17: This wants moving to osutils when the errno checking code is
374
386
def test_subprocess_call_bad_file(self):
375
387
if sys.platform != "win32":
376
388
raise TestNotApplicable("Workarounds for windows only")
377
import subprocess, errno
378
391
ERROR_BAD_EXE_FORMAT = 193
379
file("textfile.txt", "w").close()
392
open("textfile.txt", "w").close()
380
393
e = self.assertRaises(WindowsError, subprocess.call, "textfile.txt")
381
394
self.assertEqual(e.errno, errno.ENOEXEC)
382
395
self.assertEqual(e.winerror, ERROR_BAD_EXE_FORMAT)