1
# Copyright (C) 2007 Canonical Ltd
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.
8
# This program is distributed in the hope that it will be useful,
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11
# GNU General Public License for more details.
13
# You should have received a copy of the GNU General Public License
14
# along with this program; if not, write to the Free Software
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
26
OUTPUT1 = """# Bazaar merge directive format 1
27
# revision_id: example:
28
# target_branch: http://example.com
30
# timestamp: 1970-01-01 00:09:33 +0002
35
OUTPUT2 = """# Bazaar merge directive format 1
36
# revision_id: example:
37
# target_branch: http://example.com
39
# timestamp: 1970-01-01 00:09:33 +0002
40
# source_branch: http://example.org
47
I was thinking today about creating a merge directive.
53
(I've pasted it in the body of this message)
57
# Bazaar merge directive format 1\r
58
# revision_id: example:
59
# target_branch: http://example.com
61
# timestamp: 1970-01-01 00:09:33 +0002
62
# source_branch: http://example.org
65
booga""".splitlines(True)
68
class TestMergeDirective(tests.TestCase):
70
def test_merge_source(self):
73
self.assertRaises(errors.NoMergeSource, merge_directive.MergeDirective,
74
'example:', 'sha', time, timezone, 'http://example.com')
75
self.assertRaises(errors.NoMergeSource, merge_directive.MergeDirective,
76
'example:', 'sha', time, timezone, 'http://example.com',
78
merge_directive.MergeDirective('example:', 'sha', time, timezone,
79
'http://example.com', source_branch='http://example.org')
80
md = merge_directive.MergeDirective('null:', 'sha', time, timezone,
81
'http://example.com', patch='blah', patch_type='bundle')
82
self.assertIs(None, md.source_branch)
83
md2 = merge_directive.MergeDirective('null:', 'sha', time, timezone,
84
'http://example.com', patch='blah', patch_type='bundle',
86
self.assertEqual('bar', md2.source_branch)
88
def test_require_patch(self):
91
self.assertRaises(errors.PatchMissing, merge_directive.MergeDirective,
92
'example:', 'sha', time, timezone, 'http://example.com',
94
md = merge_directive.MergeDirective('example:', 'sha1', time, timezone,
95
'http://example.com', source_branch="http://example.org",
96
patch='', patch_type='diff')
97
self.assertEqual(md.patch, '')
99
def test_serialization(self):
102
md = merge_directive.MergeDirective('example:', 'sha', time, timezone,
103
'http://example.com', patch='booga', patch_type='bundle')
104
self.assertEqualDiff(OUTPUT1, ''.join(md.to_lines()))
105
md = merge_directive.MergeDirective('example:', 'sha', time, timezone,
106
'http://example.com', source_branch="http://example.org",
107
patch='booga', patch_type='diff', message="Hi mom!")
108
self.assertEqualDiff(OUTPUT2, ''.join(md.to_lines()))
110
def test_deserialize_junk(self):
112
self.assertRaises(errors.NotAMergeDirective,
113
merge_directive.MergeDirective.from_lines, 'lala')
115
def test_deserialize_empty(self):
116
self.assertRaises(errors.NotAMergeDirective,
117
merge_directive.MergeDirective.from_lines, [])
119
def test_deserialize_leading_junk(self):
120
md = merge_directive.MergeDirective.from_lines(INPUT1)
121
self.assertEqual('example:', md.revision_id)
122
self.assertEqual('sha', md.testament_sha1)
123
self.assertEqual('http://example.com', md.target_branch)
124
self.assertEqual('http://example.org', md.source_branch)
125
self.assertEqual(453, md.time)
126
self.assertEqual(120, md.timezone)
127
self.assertEqual('booga', md.patch)
128
self.assertEqual('diff', md.patch_type)
129
self.assertEqual('Hi mom!', md.message)
131
def test_roundtrip(self):
133
timezone = 7.5 * 3600
134
md = merge_directive.MergeDirective('example:', 'sha', time, timezone,
135
'http://example.com', source_branch="http://example.org",
136
patch='booga', patch_type='diff')
137
md2 = merge_directive.MergeDirective.from_lines(md.to_lines())
138
self.assertEqual('example:', md2.revision_id)
139
self.assertIsInstance(md2.revision_id, str)
140
self.assertEqual('sha', md2.testament_sha1)
141
self.assertEqual('http://example.com', md2.target_branch)
142
self.assertEqual('http://example.org', md2.source_branch)
143
self.assertEqual(time, md2.time)
144
self.assertEqual(timezone, md2.timezone)
145
self.assertEqual('diff', md2.patch_type)
146
self.assertEqual('booga', md2.patch)
147
self.assertEqual(None, md2.message)
148
md.patch = "# Bazaar revision bundle v0.9\n#\n"
149
md.message = "Hi mom!"
150
md3 = merge_directive.MergeDirective.from_lines(md.to_lines())
151
self.assertEqual("# Bazaar revision bundle v0.9\n#\n", md3.patch)
152
self.assertEqual("bundle", md3.patch_type)
153
self.assertContainsRe(md3.to_lines()[0],
154
'^# Bazaar merge directive format ')
155
self.assertEqual("Hi mom!", md3.message)
156
md3.patch_type = None
158
md4 = merge_directive.MergeDirective.from_lines(md3.to_lines())
159
self.assertIs(None, md4.patch_type)
162
EMAIL1 = """To: pqm@example.com
163
From: J. Random Hacker <jrandom@example.com>
164
Subject: Commit of rev2a
166
# Bazaar merge directive format 1
168
# target_branch: (.|\n)*
170
# timestamp: 1970-01-01 00:08:56 \\+0001
171
# source_branch: (.|\n)*
175
EMAIL2 = """To: pqm@example.com
176
From: J. Random Hacker <jrandom@example.com>
177
Subject: Commit of rev2a with special message
179
# Bazaar merge directive format 1
181
# target_branch: (.|\n)*
183
# timestamp: 1970-01-01 00:08:56 \\+0001
184
# source_branch: (.|\n)*
185
# message: Commit of rev2a with special message
189
class TestMergeDirectiveBranch(tests.TestCaseWithTransport):
191
def make_trees(self):
192
tree_a = self.make_branch_and_tree('tree_a')
193
tree_a.branch.get_config().set_user_option('email',
194
'J. Random Hacker <jrandom@example.com>')
195
self.build_tree_contents([('tree_a/file', 'content_a\ncontent_b\n')])
197
tree_a.commit('message', rev_id='rev1')
198
tree_b = tree_a.bzrdir.sprout('tree_b').open_workingtree()
199
branch_c = tree_a.bzrdir.sprout('branch_c').open_branch()
200
tree_b.commit('message', rev_id='rev2b')
201
self.build_tree_contents([('tree_a/file', 'content_a\ncontent_c\n')])
202
tree_a.commit('Commit of rev2a', rev_id='rev2a')
203
return tree_a, tree_b, branch_c
205
def test_empty_target(self):
206
tree_a, tree_b, branch_c = self.make_trees()
207
tree_d = self.make_branch_and_tree('tree_d')
208
md2 = merge_directive.MergeDirective.from_objects(
209
tree_a.branch.repository, 'rev2a', 500, 120, tree_d.branch.base,
210
patch_type='diff', public_branch=tree_a.branch.base)
212
def test_generate_patch(self):
213
tree_a, tree_b, branch_c = self.make_trees()
214
md2 = merge_directive.MergeDirective.from_objects(
215
tree_a.branch.repository, 'rev2a', 500, 120, tree_b.branch.base,
216
patch_type='diff', public_branch=tree_a.branch.base)
217
self.assertNotContainsRe(md2.patch, 'Bazaar revision bundle')
218
self.assertContainsRe(md2.patch, '\\+content_c')
219
self.assertNotContainsRe(md2.patch, '\\+\\+\\+ b/')
220
self.assertContainsRe(md2.patch, '\\+\\+\\+ file')
222
def test_public_branch(self):
223
tree_a, tree_b, branch_c = self.make_trees()
224
self.assertRaises(errors.PublicBranchOutOfDate,
225
merge_directive.MergeDirective.from_objects,
226
tree_a.branch.repository, 'rev2a', 500, 144, tree_b.branch.base,
227
public_branch=branch_c.base, patch_type='diff')
228
# public branch is not checked if patch format is bundle.
229
md1 = merge_directive.MergeDirective.from_objects(
230
tree_a.branch.repository, 'rev2a', 500, 144, tree_b.branch.base,
231
public_branch=branch_c.base)
232
# public branch is provided with a bundle, despite possibly being out
233
# of date, because it's not required if a bundle is present.
234
self.assertEqual(md1.source_branch, branch_c.base)
235
# Once we update the public branch, we can generate a diff.
236
branch_c.pull(tree_a.branch)
237
md3 = merge_directive.MergeDirective.from_objects(
238
tree_a.branch.repository, 'rev2a', 500, 144, tree_b.branch.base,
239
patch_type=None, public_branch=branch_c.base)
241
def test_use_public_submit_branch(self):
242
tree_a, tree_b, branch_c = self.make_trees()
243
branch_c.pull(tree_a.branch)
244
md = merge_directive.MergeDirective.from_objects(
245
tree_a.branch.repository, 'rev2a', 500, 144, tree_b.branch.base,
246
patch_type=None, public_branch=branch_c.base)
247
self.assertEqual(md.target_branch, tree_b.branch.base)
248
tree_b.branch.set_public_branch('http://example.com')
249
md2 = merge_directive.MergeDirective.from_objects(
250
tree_a.branch.repository, 'rev2a', 500, 144, tree_b.branch.base,
251
patch_type=None, public_branch=branch_c.base)
252
self.assertEqual(md2.target_branch, 'http://example.com')
254
def test_message(self):
255
tree_a, tree_b, branch_c = self.make_trees()
256
md3 = merge_directive.MergeDirective.from_objects(
257
tree_a.branch.repository, 'rev2a', 500, 120, tree_b.branch.base,
258
patch_type=None, public_branch=branch_c.base,
259
message='Merge message')
261
self.assertIs(None, md3.patch)
262
self.assertEqual('Merge message', md3.message)
264
def test_generate_bundle(self):
265
tree_a, tree_b, branch_c = self.make_trees()
266
md1 = merge_directive.MergeDirective.from_objects(
267
tree_a.branch.repository, 'rev2a', 500, 120, tree_b.branch.base,
268
public_branch=branch_c.base)
269
self.assertContainsRe(md1.patch, 'Bazaar revision bundle')
270
self.assertContainsRe(md1.patch, '\\+content_c')
271
self.assertNotContainsRe(md1.patch, '\\+content_a')
272
self.assertContainsRe(md1.patch, '\\+content_c')
273
self.assertNotContainsRe(md1.patch, '\\+content_a')
275
def test_broken_bundle(self):
276
tree_a, tree_b, branch_c = self.make_trees()
277
md1 = merge_directive.MergeDirective.from_objects(
278
tree_a.branch.repository, 'rev2a', 500, 120, tree_b.branch.base,
279
public_branch=branch_c.base)
280
lines = md1.to_lines()
281
lines = [l.replace('\n', '\r\n') for l in lines]
282
md2 = merge_directive.MergeDirective.from_lines(lines)
283
self.assertEqual('rev2a', md2.revision_id)
285
def test_signing(self):
288
class FakeBranch(object):
289
def get_config(self):
291
def gpg_signing_command(self):
293
md = merge_directive.MergeDirective('example:', 'sha', time, timezone,
294
'http://example.com', source_branch="http://example.org",
295
patch='booga', patch_type='diff')
296
old_strategy = gpg.GPGStrategy
297
gpg.GPGStrategy = gpg.LoopbackGPGStrategy
299
signed = md.to_signed(FakeBranch())
301
gpg.GPGStrategy = old_strategy
302
self.assertContainsRe(signed, '^-----BEGIN PSEUDO-SIGNED CONTENT')
303
self.assertContainsRe(signed, 'example.org')
304
self.assertContainsRe(signed, 'booga')
306
def test_email(self):
307
tree_a, tree_b, branch_c = self.make_trees()
308
md = merge_directive.MergeDirective.from_objects(
309
tree_a.branch.repository, 'rev2a', 476, 60, tree_b.branch.base,
310
patch_type=None, public_branch=tree_a.branch.base)
311
message = md.to_email('pqm@example.com', tree_a.branch)
312
self.assertContainsRe(message.as_string(), EMAIL1)
313
md.message = 'Commit of rev2a with special message'
314
message = md.to_email('pqm@example.com', tree_a.branch)
315
self.assertContainsRe(message.as_string(), EMAIL2)
317
def test_install_revisions_branch(self):
318
tree_a, tree_b, branch_c = self.make_trees()
319
md = merge_directive.MergeDirective.from_objects(
320
tree_a.branch.repository, 'rev2a', 500, 36, tree_b.branch.base,
321
patch_type=None, public_branch=tree_a.branch.base)
322
self.assertFalse(tree_b.branch.repository.has_revision('rev2a'))
323
revision = md.install_revisions(tree_b.branch.repository)
324
self.assertEqual('rev2a', revision)
325
self.assertTrue(tree_b.branch.repository.has_revision('rev2a'))
327
def test_install_revisions_bundle(self):
328
tree_a, tree_b, branch_c = self.make_trees()
329
md = merge_directive.MergeDirective.from_objects(
330
tree_a.branch.repository, 'rev2a', 500, 36, tree_b.branch.base,
331
patch_type='bundle', public_branch=tree_a.branch.base)
332
self.assertFalse(tree_b.branch.repository.has_revision('rev2a'))
333
revision = md.install_revisions(tree_b.branch.repository)
334
self.assertEqual('rev2a', revision)
335
self.assertTrue(tree_b.branch.repository.has_revision('rev2a'))
337
def test_get_target_revision_nofetch(self):
338
tree_a, tree_b, branch_c = self.make_trees()
339
tree_b.branch.fetch(tree_a.branch)
340
md = merge_directive.MergeDirective.from_objects(
341
tree_a.branch.repository, 'rev2a', 500, 36, tree_b.branch.base,
342
patch_type=None, public_branch=tree_a.branch.base)
343
md.source_branch = '/dev/null'
344
revision = md.install_revisions(tree_b.branch.repository)
345
self.assertEqual('rev2a', revision)
added
removed
bzrlib/tests/test_merge_directive.py
