466
409
dlg = commit.CommitDialog(tree)
467
410
diff_buffer = dlg._diff_view.buffer
468
411
text = diff_buffer.get_text(diff_buffer.get_start_iter(),
469
diff_buffer.get_end_iter(),
470
True).splitlines(True)
412
diff_buffer.get_end_iter()).splitlines(True)
472
self.assertEqual("=== modified file 'a'\n", text[0])
414
self.assertEqual("=== removed file 'b'\n", text[0])
473
415
self.assertContainsRe(text[1],
416
r"--- b\t\d\d\d\d-\d\d-\d\d \d\d:\d\d:\d\d [+-]\d\d\d\d")
417
self.assertEqual('+++ b\t1970-01-01 00:00:00 +0000\n', text[2])
418
self.assertEqual('@@ -1,1 +0,0 @@\n', text[3])
419
self.assertEqual('-contents of tree/b\n', text[4])
420
self.assertEqual('\n', text[5])
422
self.assertEqual("=== modified file 'a'\n", text[6])
423
self.assertContainsRe(text[7],
474
424
r"--- a\t\d\d\d\d-\d\d-\d\d \d\d:\d\d:\d\d [+-]\d\d\d\d")
475
self.assertContainsRe(text[2],
476
r"\+\+\+ a\t\d\d\d\d-\d\d-\d\d \d\d:\d\d:\d\d [+-]\d\d\d\d")
477
self.assertEqual('@@ -1,1 +1,1 @@\n', text[3])
478
self.assertEqual('-contents of tree/a\n', text[4])
479
self.assertEqual('+new contents for a\n', text[5])
480
self.assertEqual('\n', text[6])
482
self.assertEqual("=== removed file 'b'\n", text[7])
483
425
self.assertContainsRe(text[8],
484
r"--- b\t\d\d\d\d-\d\d-\d\d \d\d:\d\d:\d\d [+-]\d\d\d\d")
485
self.assertEqual('+++ b\t1970-01-01 00:00:00 +0000\n', text[9])
486
self.assertEqual('@@ -1,1 +0,0 @@\n', text[10])
487
self.assertEqual('-contents of tree/b\n', text[11])
426
r"\+\+\+ a\t\d\d\d\d-\d\d-\d\d \d\d:\d\d:\d\d [+-]\d\d\d\d")
427
self.assertEqual('@@ -1,1 +1,1 @@\n', text[9])
428
self.assertEqual('-contents of tree/a\n', text[10])
429
self.assertEqual('+new contents for a\n', text[11])
488
430
self.assertEqual('\n', text[12])
490
432
self.assertEqual('Diff for All Files', dlg._diff_label.get_text())
640
573
# do with. So instead, we just call toggle directly, and assume
641
574
# that toggle is hooked in correctly
642
575
# column = dlg._treeview_files.get_column(0)
643
# renderer = column.get_cells()[0]
576
# renderer = column.get_cell_renderers()[0]
645
578
# Toggle a single entry should set just that entry to False
646
579
dlg._toggle_commit(None, 1, dlg._files_store)
647
self.assertEqual([("", "", True),
580
self.assertEqual([(None, None, True),
648
581
('a-id', 'a', False),
649
582
('b-id', 'b', True),
650
583
], [(r[0], r[1], r[2]) for r in dlg._files_store])
652
585
# Toggling the main entry should set all entries
653
586
dlg._toggle_commit(None, 0, dlg._files_store)
654
self.assertEqual([("", "", False),
587
self.assertEqual([(None, None, False),
655
588
('a-id', 'a', False),
656
589
('b-id', 'b', False),
657
590
], [(r[0], r[1], r[2]) for r in dlg._files_store])
659
592
dlg._toggle_commit(None, 2, dlg._files_store)
660
self.assertEqual([("", "", False),
593
self.assertEqual([(None, None, False),
661
594
('a-id', 'a', False),
662
595
('b-id', 'b', True),
663
596
], [(r[0], r[1], r[2]) for r in dlg._files_store])
665
598
dlg._toggle_commit(None, 0, dlg._files_store)
666
self.assertEqual([("", "", True),
599
self.assertEqual([(None, None, True),
667
600
('a-id', 'a', True),
668
601
('b-id', 'b', True),
669
602
], [(r[0], r[1], r[2]) for r in dlg._files_store])
713
644
'message':'message\nfor b_dir\n'},
714
645
]), dlg._get_specific_files())
716
def test_specific_files_sanitizes_messages(self):
717
tree = self.make_branch_and_tree('tree')
718
tree.branch.get_config().set_user_option('per_file_commits', 'true')
719
self.build_tree(['tree/a_file', 'tree/b_dir/'])
720
tree.add(['a_file', 'b_dir'], ['1a-id', '0b-id'])
722
dlg = commit.CommitDialog(tree)
723
dlg._commit_selected_radio.set_active(True)
724
self.assertEqual((['a_file', 'b_dir'], []), dlg._get_specific_files())
726
dlg._treeview_files.set_cursor(
727
Gtk.TreePath(path=1), None, False)
728
dlg._set_file_commit_message('Test\r\nmessage\rfor a_file\n')
729
dlg._treeview_files.set_cursor(
730
Gtk.TreePath(path=2), None, False)
731
dlg._set_file_commit_message('message\r\nfor\nb_dir\r')
733
self.assertEqual((['a_file', 'b_dir'],
734
[{'path':'a_file', 'file_id':'1a-id',
735
'message':'Test\nmessage\nfor a_file\n'},
736
{'path':'b_dir', 'file_id':'0b-id',
737
'message':'message\nfor\nb_dir\n'},
738
]), dlg._get_specific_files())
741
class QuestionHelpers(object):
648
class TestCommitDialog_Commit(tests.TestCaseWithTransport):
649
"""Tests on the actual 'commit' button being pushed."""
743
651
def _set_question_yes(self, dlg):
744
652
"""Set the dialog to answer YES to any questions."""
745
653
self.questions = []
746
def _question_yes(*args, **kwargs):
654
def _question_yes(*args):
747
655
self.questions.append(args)
748
656
self.questions.append('YES')
749
return Gtk.ResponseType.YES
657
return gtk.RESPONSE_YES
750
658
dlg._question_dialog = _question_yes
752
660
def _set_question_no(self, dlg):
753
661
"""Set the dialog to answer NO to any questions."""
754
662
self.questions = []
755
def _question_no(*args, **kwargs):
663
def _question_no(*args):
756
664
self.questions.append(args)
757
665
self.questions.append('NO')
758
return Gtk.ResponseType.NO
666
return gtk.RESPONSE_NO
759
667
dlg._question_dialog = _question_no
762
class TestCommitDialog_Commit(tests.TestCaseWithTransport, QuestionHelpers):
763
"""Tests on the actual 'commit' button being pushed."""
765
669
def test_bound_commit_local(self):
766
670
tree = self.make_branch_and_tree('tree')
767
671
self.build_tree(['tree/a'])
1148
1031
{'path':u'\u03a9', 'file_id':'omega-id',
1149
1032
'message':u'\u03a9 is the end of all things.\n'},
1150
1033
], file_info_decoded)
1153
class TestSanitizeMessage(tests.TestCase):
1155
def assertSanitize(self, expected, original):
1156
self.assertEqual(expected,
1157
commit._sanitize_and_decode_message(original))
1159
def test_untouched(self):
1160
self.assertSanitize('foo\nbar\nbaz\n', 'foo\nbar\nbaz\n')
1162
def test_converts_cr_to_lf(self):
1163
self.assertSanitize('foo\nbar\nbaz\n', 'foo\rbar\rbaz\r')
1165
def test_converts_crlf_to_lf(self):
1166
self.assertSanitize('foo\nbar\nbaz\n', 'foo\r\nbar\r\nbaz\r\n')
1168
def test_converts_mixed_to_lf(self):
1169
self.assertSanitize('foo\nbar\nbaz\n', 'foo\r\nbar\rbaz\n')
1172
class TestSavedCommitMessages(tests.TestCaseWithTransport):
1175
super(TestSavedCommitMessages, self).setUp()
1177
branch.Branch.hooks.install_named_hook(
1178
'post_uncommit', commitmsgs.save_commit_messages, None)
1180
def _get_file_info_dict(self, rank):
1181
file_info = [dict(path='a', file_id='a-id', message='a msg %d' % rank),
1182
dict(path='b', file_id='b-id', message='b msg %d' % rank)]
1185
def _get_file_info_revprops(self, rank):
1186
file_info_prop = self._get_file_info_dict(rank)
1187
return {'file-info': bencode.bencode(file_info_prop).decode('UTF-8')}
1189
def _get_commit_message(self):
1190
return self.config.get_user_option('gtk_global_commit_message')
1192
def _get_file_commit_messages(self):
1193
return self.config.get_user_option('gtk_file_commit_messages')
1196
class TestUncommitHook(TestSavedCommitMessages):
1199
super(TestUncommitHook, self).setUp()
1200
self.tree = self.make_branch_and_tree('tree')
1201
self.config = self.tree.branch.get_config()
1202
self.build_tree(['tree/a', 'tree/b'])
1203
self.tree.add(['a'], ['a-id'])
1204
self.tree.add(['b'], ['b-id'])
1205
rev1 = self.tree.commit('one', rev_id='one-id',
1206
revprops=self._get_file_info_revprops(1))
1207
rev2 = self.tree.commit('two', rev_id='two-id',
1208
revprops=self._get_file_info_revprops(2))
1209
rev3 = self.tree.commit('three', rev_id='three-id',
1210
revprops=self._get_file_info_revprops(3))
1212
def test_uncommit_one_by_one(self):
1213
uncommit.uncommit(self.tree.branch, tree=self.tree)
1214
self.assertEquals(u'three', self._get_commit_message())
1215
self.assertEquals(u'd4:a-id7:a msg 34:b-id7:b msg 3e',
1216
self._get_file_commit_messages())
1218
uncommit.uncommit(self.tree.branch, tree=self.tree)
1219
self.assertEquals(u'two\n******\nthree', self._get_commit_message())
1220
self.assertEquals(u'd4:a-id22:a msg 2\n******\na msg 3'
1221
'4:b-id22:b msg 2\n******\nb msg 3e',
1222
self._get_file_commit_messages())
1224
uncommit.uncommit(self.tree.branch, tree=self.tree)
1225
self.assertEquals(u'one\n******\ntwo\n******\nthree',
1226
self._get_commit_message())
1227
self.assertEquals(u'd4:a-id37:a msg 1\n******\na msg 2\n******\na msg 3'
1228
'4:b-id37:b msg 1\n******\nb msg 2\n******\nb msg 3e',
1229
self._get_file_commit_messages())
1231
def test_uncommit_all_at_once(self):
1232
uncommit.uncommit(self.tree.branch, tree=self.tree, revno=1)
1233
self.assertEquals(u'one\n******\ntwo\n******\nthree',
1234
self._get_commit_message())
1235
self.assertEquals(u'd4:a-id37:a msg 1\n******\na msg 2\n******\na msg 3'
1236
'4:b-id37:b msg 1\n******\nb msg 2\n******\nb msg 3e',
1237
self._get_file_commit_messages())
1240
class TestReusingSavedCommitMessages(TestSavedCommitMessages, QuestionHelpers):
1243
super(TestReusingSavedCommitMessages, self).setUp()
1244
self.tree = self.make_branch_and_tree('tree')
1245
self.config = self.tree.branch.get_config()
1246
self.config.set_user_option('per_file_commits', 'true')
1247
self.build_tree(['tree/a', 'tree/b'])
1248
self.tree.add(['a'], ['a-id'])
1249
self.tree.add(['b'], ['b-id'])
1250
rev1 = self.tree.commit('one', revprops=self._get_file_info_revprops(1))
1251
rev2 = self.tree.commit('two', revprops=self._get_file_info_revprops(2))
1252
uncommit.uncommit(self.tree.branch, tree=self.tree)
1253
self.build_tree_contents([('tree/a', 'new a content\n'),
1254
('tree/b', 'new b content'),])
1256
def _get_commit_dialog(self, tree):
1257
# Ensure we will never use a dialog that can actually prompt the user
1258
# during the test suite. Test *can* and *should* override with the
1259
# correct question dialog type.
1260
dlg = commit.CommitDialog(tree)
1261
self._set_question_no(dlg)
1264
def test_setup_saved_messages(self):
1265
# Check the initial setup
1266
self.assertEquals(u'two', self._get_commit_message())
1267
self.assertEquals(u'd4:a-id7:a msg 24:b-id7:b msg 2e',
1268
self._get_file_commit_messages())
1270
def test_messages_are_reloaded(self):
1271
dlg = self._get_commit_dialog(self.tree)
1272
self.assertEquals(u'two', dlg._get_global_commit_message())
1273
self.assertEquals(([u'a', u'b'],
1275
'file_id': 'a-id', 'message': 'a msg 2',},
1277
'file_id': 'b-id', 'message': 'b msg 2',}],),
1278
dlg._get_specific_files())
1280
def test_messages_are_consumed(self):
1281
dlg = self._get_commit_dialog(self.tree)
1283
self.assertEquals(u'', self._get_commit_message())
1284
self.assertEquals(u'de', self._get_file_commit_messages())
1286
def test_messages_are_saved_on_cancel_if_required(self):
1287
dlg = self._get_commit_dialog(self.tree)
1288
self._set_question_yes(dlg) # Save messages
1290
self.assertEquals(u'two', self._get_commit_message())
1291
self.assertEquals(u'd4:a-id7:a msg 24:b-id7:b msg 2e',
1292
self._get_file_commit_messages())
1294
def test_messages_are_cleared_on_cancel_if_required(self):
1295
dlg = self._get_commit_dialog(self.tree)
1296
self._set_question_no(dlg) # Don't save messages
1298
self.assertEquals(u'', self._get_commit_message())
1299
self.assertEquals(u'de',
1300
self._get_file_commit_messages())
1303
class BzrHandlePatchTestCase(tests.TestCase):
1306
top = os.path.abspath(os.path.join(
1307
os.path.dirname(__file__), os.pardir))
1308
self.script = os.path.join(top, 'bzr-handle-patch')
1309
self.env = dict(os.environ)
1310
self.env['BZR_PLUGINS_AT'] = 'gtk@%s' % top
1311
self.patch = NamedTemporaryFile()
1312
self.patch.write('\n'.join([
1313
"=== added file '_test.txt'",
1314
"--- _test.txt 1970-01-01 00:00:00 +0000",
1315
"+++ _test.txt 2012-02-03 20:00:34 +0000",
1320
super(BzrHandlePatchTestCase, self).setUp()
1322
def test_smoketest(self):
1323
# This is a smoke test to verify the process starts.
1324
bzr_notify = subprocess.Popen(
1325
[self.script, self.patch.name, 'test'],
1326
stdout=subprocess.PIPE, stderr=subprocess.PIPE, env=self.env)
1327
stdout, stderr = bzr_notify.communicate()
1328
self.assertEqual('', stdout)
1329
self.assertEqual('', stderr)