1
# Copyright (C) 2005 Canonical Ltd
1
# Copyright (C) 2005 by Canonical Ltd
3
3
# This program is free software; you can redistribute it and/or modify
4
# it under the terms of the GNU General Public License as published by
5
# the Free Software Foundation; either version 2 of the License, or
6
# (at your option) any later version.
4
# it under the terms of the GNU General Public License version 2 as published by
5
# the Free Software Foundation.
8
7
# This program is distributed in the hope that it will be useful,
9
8
# but WITHOUT ANY WARRANTY; without even the implied warranty of
29
22
from bzrlib.branch import Branch
30
23
from bzrlib.config import ensure_config_dir_exists, config_filename
31
from bzrlib.msgeditor import (
32
make_commit_message_template_encoded,
33
edit_commit_message_encoded
35
from bzrlib.tests import (
39
TestCaseWithTransport,
43
from bzrlib.tests.EncodingAdapter import EncodingTestAdapter
44
from bzrlib.trace import mutter
47
def load_tests(standard_tests, module, loader):
48
"""Parameterize the test for tempfile creation with different encodings."""
49
to_adapt, result = split_suite_by_re(standard_tests,
50
"test__create_temp_file_with_commit_template_in_unicode_dir")
51
for test in iter_suite_tests(to_adapt):
52
result.addTests(EncodingTestAdapter().adapt(test))
24
from bzrlib.msgeditor import make_commit_message_template, _get_editor
25
from bzrlib.tests import TestCaseWithTransport, TestSkipped
56
28
class MsgEditorTest(TestCaseWithTransport):
71
43
def test_commit_template(self):
72
44
"""Test building a commit message template"""
73
45
working_tree = self.make_uncommitted_tree()
74
template = msgeditor.make_commit_message_template(working_tree,
46
template = make_commit_message_template(working_tree, None)
76
47
self.assertEqualDiff(template,
82
def test_commit_template_encoded(self):
83
"""Test building a commit message template"""
84
working_tree = self.make_uncommitted_tree()
85
template = make_commit_message_template_encoded(working_tree,
87
output_encoding='utf8')
88
self.assertEqualDiff(template,
95
def test_commit_template_and_diff(self):
96
"""Test building a commit message template"""
97
working_tree = self.make_uncommitted_tree()
98
template = make_commit_message_template_encoded(working_tree,
101
output_encoding='utf8')
107
self.assertTrue(u"""\
110
""".encode('utf8') in template)
112
def test_run_editor(self):
113
if sys.platform == "win32":
114
f = file('fed.bat', 'w')
115
f.write('@rem dummy fed')
117
os.environ['BZR_EDITOR'] = 'fed.bat'
119
f = file('fed.sh', 'wb')
120
f.write('#!/bin/sh\n')
122
os.chmod('fed.sh', 0755)
123
os.environ['BZR_EDITOR'] = './fed.sh'
125
self.assertEqual(True, msgeditor._run_editor(''),
126
'Unable to run dummy fake editor')
128
def make_fake_editor(self, message='test message from fed\\n'):
129
"""Set up environment so that an editor will be a known script.
131
Sets up BZR_EDITOR so that if an editor is spawned it will run a
132
script that just adds a known message to the start of the file.
134
f = file('fed.py', 'wb')
135
f.write('#!%s\n' % sys.executable)
139
if len(sys.argv) == 2:
150
if sys.platform == "win32":
151
# [win32] make batch file and set BZR_EDITOR
152
f = file('fed.bat', 'w')
156
""" % sys.executable)
158
os.environ['BZR_EDITOR'] = 'fed.bat'
160
# [non-win32] make python script executable and set BZR_EDITOR
161
os.chmod('fed.py', 0755)
162
os.environ['BZR_EDITOR'] = './fed.py'
164
def test_edit_commit_message(self):
165
working_tree = self.make_uncommitted_tree()
166
self.make_fake_editor()
168
mutter('edit_commit_message without infotext')
169
self.assertEqual('test message from fed\n',
170
msgeditor.edit_commit_message(''))
172
mutter('edit_commit_message with ascii string infotext')
173
self.assertEqual('test message from fed\n',
174
msgeditor.edit_commit_message('spam'))
176
mutter('edit_commit_message with unicode infotext')
177
self.assertEqual('test message from fed\n',
178
msgeditor.edit_commit_message(u'\u1234'))
180
tmpl = edit_commit_message_encoded(u'\u1234'.encode("utf8"))
181
self.assertEqual('test message from fed\n', tmpl)
183
def test_start_message(self):
184
self.make_uncommitted_tree()
185
self.make_fake_editor()
186
self.assertEqual('test message from fed\nstart message\n',
187
msgeditor.edit_commit_message('',
188
start_message='start message\n'))
189
self.assertEqual('test message from fed\n',
190
msgeditor.edit_commit_message('',
193
def test_deleted_commit_message(self):
194
working_tree = self.make_uncommitted_tree()
196
if sys.platform == 'win32':
197
os.environ['BZR_EDITOR'] = 'cmd.exe /c del'
199
os.environ['BZR_EDITOR'] = 'rm'
201
self.assertRaises((IOError, OSError), msgeditor.edit_commit_message, '')
203
53
def test__get_editor(self):
204
54
# Test that _get_editor can return a decent list of items
205
55
bzr_editor = os.environ.get('BZR_EDITOR')
240
90
del os.environ['EDITOR']
242
92
os.environ['EDITOR'] = editor
244
def test__create_temp_file_with_commit_template(self):
245
# check that commit template written properly
246
# and has platform native line-endings (CRLF on win32)
247
create_file = msgeditor._create_temp_file_with_commit_template
248
msgfilename, hasinfo = create_file('infotext','----','start message')
249
self.assertNotEqual(None, msgfilename)
250
self.assertTrue(hasinfo)
251
expected = os.linesep.join(['start message',
257
self.assertFileEqual(expected, msgfilename)
259
def test__create_temp_file_with_commit_template_in_unicode_dir(self):
260
self.requireFeature(tests.UnicodeFilenameFeature)
261
if hasattr(self, 'info'):
262
os.mkdir(self.info['directory'])
263
os.chdir(self.info['directory'])
264
msgeditor._create_temp_file_with_commit_template('infotext')
266
raise TestNotApplicable('Test run elsewhere with non-ascii data.')
268
def test__create_temp_file_with_empty_commit_template(self):
270
create_file = msgeditor._create_temp_file_with_commit_template
271
msgfilename, hasinfo = create_file('')
272
self.assertNotEqual(None, msgfilename)
273
self.assertFalse(hasinfo)
274
self.assertFileEqual('', msgfilename)
276
def test_unsupported_encoding_commit_message(self):
277
old_env = osutils.set_or_unset_env('LANG', 'C')
279
# LANG env variable has no effect on Windows
280
# but some characters anyway cannot be represented
281
# in default user encoding
282
char = probe_bad_non_ascii(osutils.get_user_encoding())
284
raise TestSkipped('Cannot find suitable non-ascii character '
285
'for user_encoding (%s)' % osutils.get_user_encoding())
287
self.make_fake_editor(message=char)
289
working_tree = self.make_uncommitted_tree()
290
self.assertRaises(errors.BadCommitMessageEncoding,
291
msgeditor.edit_commit_message, '')
293
osutils.set_or_unset_env('LANG', old_env)
added
removed
bzrlib/tests/test_msgeditor.py
