/b-gtk/fix-viz

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/b-gtk/fix-viz

« back to all changes in this revision

Viewing changes to tests/test_commit.py

  • Committer: Vincent Ladeuil
  • Date: 2009-05-06 11:51:38 UTC
  • mfrom: (628.1.2 gtk)
  • Revision ID: v.ladeuil+lp@free.fr-20090506115138-7abxvg2qz4gzraup
Upgrade COMPATIBLE_BZR_VERSIONS

Show diffs side-by-side

added added

removed removed

Lines of Context:
tests/test_commit.py
21
21
import gtk
22
22
 
23
23
from bzrlib import (
24
 
    branch,
 
24
    tests,
25
25
    revision,
26
 
    tests,
27
 
    uncommit,
28
26
    )
29
 
try:
30
 
    from bzrlib import bencode
31
 
except ImportError:
32
 
    from bzrlib.util import bencode
 
27
from bzrlib.util import bencode
33
28
 
34
29
from bzrlib.plugins.gtk import commit
35
30
 
91
86
        rev_id4 = tree3.commit('four')
92
87
        rev_id5 = tree3.commit('five')
93
88
        tree.merge_from_branch(tree2.branch)
94
 
        tree.merge_from_branch(tree3.branch, force=True)
 
89
        tree.merge_from_branch(tree3.branch)
95
90
        self.assertEqual([rev_id1, rev_id3, rev_id5], tree.get_parent_ids())
96
91
 
97
92
        pending_revisions = commit.pending_revisions(tree)
271
266
                               committer='Jerry Foo <jerry@foo.com>',
272
267
                               timestamp=1191372278.05,
273
268
                               timezone=+7200)
274
 
        tree.merge_from_branch(tree3.branch, force=True)
 
269
        tree.merge_from_branch(tree3.branch)
275
270
 
276
271
        dlg = commit.CommitDialog(tree)
277
272
        # TODO: assert that the pending box is set to show
672
667
                          ]), dlg._get_specific_files())
673
668
 
674
669
 
675
 
class QuestionHelpers(object):
 
670
class TestCommitDialog_Commit(tests.TestCaseWithTransport):
 
671
    """Tests on the actual 'commit' button being pushed."""
676
672
 
677
673
    def _set_question_yes(self, dlg):
678
674
        """Set the dialog to answer YES to any questions."""
692
688
            return gtk.RESPONSE_NO
693
689
        dlg._question_dialog = _question_no
694
690
 
695
 
 
696
 
class TestCommitDialog_Commit(tests.TestCaseWithTransport, QuestionHelpers):
697
 
    """Tests on the actual 'commit' button being pushed."""
698
 
 
699
691
    def test_bound_commit_local(self):
700
692
        tree = self.make_branch_and_tree('tree')
701
693
        self.build_tree(['tree/a'])
988
980
        rev = tree.branch.repository.get_revision(rev_id2)
989
981
        self.assertEqual('Commit just "a"', rev.message)
990
982
        file_info = rev.properties['file-info']
991
 
        self.assertEqual(u'ld7:file_id4:a-id'
992
 
                         '7:message14:Message for A\n'
993
 
                         '4:path1:a'
994
 
                         'ee',
995
 
                         file_info)
 
983
        self.assertEqual('ld7:file_id4:a-id'
 
984
                           '7:message14:Message for A\n'
 
985
                           '4:path1:a'
 
986
                         'ee', file_info)
996
987
        self.assertEqual([{'path':'a', 'file_id':'a-id',
997
 
                           'message':'Message for A\n'},],
998
 
                         bencode.bdecode(file_info.encode('UTF-8')))
 
988
                           'message':'Message for A\n'},
 
989
                         ], bencode.bdecode(file_info))
999
990
 
1000
991
    def test_commit_messages_after_merge(self):
1001
992
        tree = self.make_branch_and_tree('tree')
1022
1013
        self.assertEqual('Merging from "tree2"\n', rev.message)
1023
1014
        self.assertEqual([rev_id1, rev_id2], rev.parent_ids)
1024
1015
        file_info = rev.properties['file-info']
1025
 
        self.assertEqual(u'ld7:file_id4:a-id'
1026
 
                         '7:message14:Message for A\n'
1027
 
                         '4:path1:a'
1028
 
                         'ee',
1029
 
                         file_info)
 
1016
        self.assertEqual('ld7:file_id4:a-id'
 
1017
                           '7:message14:Message for A\n'
 
1018
                           '4:path1:a'
 
1019
                         'ee', file_info)
1030
1020
        self.assertEqual([{'path':'a', 'file_id':'a-id',
1031
 
                           'message':'Message for A\n'},],
1032
 
                         bencode.bdecode(file_info.encode('UTF-8')))
 
1021
                           'message':'Message for A\n'},
 
1022
                         ], bencode.bdecode(file_info))
1033
1023
 
1034
1024
    def test_commit_unicode_messages(self):
1035
1025
        self.requireFeature(tests.UnicodeFilenameFeature)
1096
1086
 
1097
1087
    def test_converts_mixed_to_lf(self):
1098
1088
        self.assertSanitize('foo\nbar\nbaz\n', 'foo\r\nbar\rbaz\n')
1099
 
 
1100
 
 
1101
 
class TestSavedCommitMessages(tests.TestCaseWithTransport):
1102
 
 
1103
 
    def setUp(self):
1104
 
        super(TestSavedCommitMessages, self).setUp()
1105
 
        # Install our hook
1106
 
        branch.Branch.hooks.install_named_hook(
1107
 
            'post_uncommit', commit.save_commit_messages, None)
1108
 
 
1109
 
    def _get_file_info_dict(self, rank):
1110
 
        file_info = [dict(path='a', file_id='a-id', message='a msg %d' % rank),
1111
 
                     dict(path='b', file_id='b-id', message='b msg %d' % rank)]
1112
 
        return file_info
1113
 
 
1114
 
    def _get_file_info_revprops(self, rank):
1115
 
        file_info_prop = self._get_file_info_dict(rank)
1116
 
        return {'file-info': bencode.bencode(file_info_prop).decode('UTF-8')}
1117
 
 
1118
 
    def _get_commit_message(self):
1119
 
        return self.config.get_user_option('gtk_global_commit_message')
1120
 
 
1121
 
    def _get_file_commit_messages(self):
1122
 
        return self.config.get_user_option('gtk_file_commit_messages')
1123
 
 
1124
 
 
1125
 
class TestUncommitHook(TestSavedCommitMessages):
1126
 
 
1127
 
    def setUp(self):
1128
 
        super(TestUncommitHook, self).setUp()
1129
 
        self.tree = self.make_branch_and_tree('tree')
1130
 
        self.config = self.tree.branch.get_config()
1131
 
        self.build_tree(['tree/a', 'tree/b'])
1132
 
        self.tree.add(['a'], ['a-id'])
1133
 
        self.tree.add(['b'], ['b-id'])
1134
 
        rev1 = self.tree.commit('one', rev_id='one-id',
1135
 
                                revprops=self._get_file_info_revprops(1))
1136
 
        rev2 = self.tree.commit('two', rev_id='two-id',
1137
 
                                revprops=self._get_file_info_revprops(2))
1138
 
        rev3 = self.tree.commit('three', rev_id='three-id',
1139
 
                                revprops=self._get_file_info_revprops(3))
1140
 
 
1141
 
    def test_uncommit_one_by_one(self):
1142
 
        uncommit.uncommit(self.tree.branch, tree=self.tree)
1143
 
        self.assertEquals(u'three', self._get_commit_message())
1144
 
        self.assertEquals(u'd4:a-id7:a msg 34:b-id7:b msg 3e',
1145
 
                          self._get_file_commit_messages())
1146
 
 
1147
 
        uncommit.uncommit(self.tree.branch, tree=self.tree)
1148
 
        self.assertEquals(u'two\n******\nthree', self._get_commit_message())
1149
 
        self.assertEquals(u'd4:a-id22:a msg 2\n******\na msg 3'
1150
 
                          '4:b-id22:b msg 2\n******\nb msg 3e',
1151
 
                          self._get_file_commit_messages())
1152
 
 
1153
 
        uncommit.uncommit(self.tree.branch, tree=self.tree)
1154
 
        self.assertEquals(u'one\n******\ntwo\n******\nthree',
1155
 
                          self._get_commit_message())
1156
 
        self.assertEquals(u'd4:a-id37:a msg 1\n******\na msg 2\n******\na msg 3'
1157
 
                          '4:b-id37:b msg 1\n******\nb msg 2\n******\nb msg 3e',
1158
 
                          self._get_file_commit_messages())
1159
 
 
1160
 
    def test_uncommit_all_at_once(self):
1161
 
        uncommit.uncommit(self.tree.branch, tree=self.tree, revno=1)
1162
 
        self.assertEquals(u'one\n******\ntwo\n******\nthree',
1163
 
                          self._get_commit_message())
1164
 
        self.assertEquals(u'd4:a-id37:a msg 1\n******\na msg 2\n******\na msg 3'
1165
 
                          '4:b-id37:b msg 1\n******\nb msg 2\n******\nb msg 3e',
1166
 
                          self._get_file_commit_messages())
1167
 
 
1168
 
 
1169
 
class TestReusingSavedCommitMessages(TestSavedCommitMessages, QuestionHelpers):
1170
 
 
1171
 
    def setUp(self):
1172
 
        super(TestReusingSavedCommitMessages, self).setUp()
1173
 
        self.tree = self.make_branch_and_tree('tree')
1174
 
        self.config = self.tree.branch.get_config()
1175
 
        self.config.set_user_option('per_file_commits', 'true')
1176
 
        self.build_tree(['tree/a', 'tree/b'])
1177
 
        self.tree.add(['a'], ['a-id'])
1178
 
        self.tree.add(['b'], ['b-id'])
1179
 
        rev1 = self.tree.commit('one', revprops=self._get_file_info_revprops(1))
1180
 
        rev2 = self.tree.commit('two', revprops=self._get_file_info_revprops(2))
1181
 
        uncommit.uncommit(self.tree.branch, tree=self.tree)
1182
 
        self.build_tree_contents([('tree/a', 'new a content\n'),
1183
 
                                  ('tree/b', 'new b content'),])
1184
 
 
1185
 
    def _get_commit_dialog(self, tree):
1186
 
        # Ensure we will never use a dialog that can actually prompt the user
1187
 
        # during the test suite. Test *can* and *should* override with the
1188
 
        # correct question dialog type.
1189
 
        dlg = commit.CommitDialog(tree)
1190
 
        self._set_question_no(dlg)
1191
 
        return dlg
1192
 
 
1193
 
    def test_setup_saved_messages(self):
1194
 
        # Check the initial setup
1195
 
        self.assertEquals(u'two', self._get_commit_message())
1196
 
        self.assertEquals(u'd4:a-id7:a msg 24:b-id7:b msg 2e',
1197
 
                          self._get_file_commit_messages())
1198
 
 
1199
 
    def test_messages_are_reloaded(self):
1200
 
        dlg = self._get_commit_dialog(self.tree)
1201
 
        self.assertEquals(u'two', dlg._get_global_commit_message())
1202
 
        self.assertEquals(([u'a', u'b'],
1203
 
                           [{ 'path': 'a',
1204
 
                             'file_id': 'a-id', 'message': 'a msg 2',},
1205
 
                           {'path': 'b',
1206
 
                            'file_id': 'b-id', 'message': 'b msg 2',}],),
1207
 
                          dlg._get_specific_files())
1208
 
 
1209
 
    def test_messages_are_consumed(self):
1210
 
        dlg = self._get_commit_dialog(self.tree)
1211
 
        dlg._do_commit()
1212
 
        self.assertEquals(u'', self._get_commit_message())
1213
 
        self.assertEquals(u'de', self._get_file_commit_messages())
1214
 
 
1215
 
    def test_messages_are_saved_on_cancel_if_required(self):
1216
 
        dlg = self._get_commit_dialog(self.tree)
1217
 
        self._set_question_yes(dlg) # Save messages
1218
 
        dlg._do_cancel()
1219
 
        self.assertEquals(u'two', self._get_commit_message())
1220
 
        self.assertEquals(u'd4:a-id7:a msg 24:b-id7:b msg 2e',
1221
 
                          self._get_file_commit_messages())
1222
 
 
1223
 
    def test_messages_are_cleared_on_cancel_if_required(self):
1224
 
        dlg = self._get_commit_dialog(self.tree)
1225
 
        self._set_question_no(dlg) # Don't save messages
1226
 
        dlg._do_cancel()
1227
 
        self.assertEquals(u'', self._get_commit_message())
1228
 
        self.assertEquals(u'de',
1229
 
                          self._get_file_commit_messages())