/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to breezy/git/tests/test_builder.py

  • Committer: Jelmer Vernooij
  • Date: 2017-07-23 22:06:41 UTC
  • mfrom: (6738 trunk)
  • mto: This revision was merged to the branch mainline in revision 6739.
  • Revision ID: jelmer@jelmer.uk-20170723220641-69eczax9bmv8d6kk
Merge trunk, address review comments.

Show diffs side-by-side

added added

removed removed

Lines of Context:
breezy/git/tests/test_builder.py
1
 
# Copyright (C) 2007 Canonical Ltd
2
 
#
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.
7
 
#
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.
12
 
#
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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
 
 
17
 
"""Test our ability to build up test repositories"""
18
 
 
19
 
from io import BytesIO
20
 
 
21
 
from dulwich.repo import Repo as GitRepo
22
 
 
23
 
from .. import tests
24
 
 
25
 
 
26
 
class TestGitBranchBuilder(tests.TestCase):
27
 
 
28
 
    def test__create_blob(self):
29
 
        stream = BytesIO()
30
 
        builder = tests.GitBranchBuilder(stream)
31
 
        self.assertEqual(1, builder._create_blob(b'foo\nbar\n'))
32
 
        self.assertEqualDiff(b'blob\nmark :1\ndata 8\nfoo\nbar\n\n',
33
 
                             stream.getvalue())
34
 
 
35
 
    def test_set_file(self):
36
 
        stream = BytesIO()
37
 
        builder = tests.GitBranchBuilder(stream)
38
 
        builder.set_file('foobar', b'foo\nbar\n', False)
39
 
        self.assertEqualDiff(b'blob\nmark :1\ndata 8\nfoo\nbar\n\n',
40
 
                             stream.getvalue())
41
 
        self.assertEqual([b'M 100644 :1 foobar\n'], builder.commit_info)
42
 
 
43
 
    def test_set_file_unicode(self):
44
 
        stream = BytesIO()
45
 
        builder = tests.GitBranchBuilder(stream)
46
 
        builder.set_file(u'f\xb5/bar', b'contents\nbar\n', False)
47
 
        self.assertEqualDiff(b'blob\nmark :1\ndata 13\ncontents\nbar\n\n',
48
 
                             stream.getvalue())
49
 
        self.assertEqual([b'M 100644 :1 f\xc2\xb5/bar\n'], builder.commit_info)
50
 
 
51
 
    def test_set_file_newline(self):
52
 
        stream = BytesIO()
53
 
        builder = tests.GitBranchBuilder(stream)
54
 
        builder.set_file(u'foo\nbar', b'contents\nbar\n', False)
55
 
        self.assertEqualDiff(b'blob\nmark :1\ndata 13\ncontents\nbar\n\n',
56
 
                             stream.getvalue())
57
 
        self.assertEqual([b'M 100644 :1 "foo\\nbar"\n'], builder.commit_info)
58
 
 
59
 
    def test_set_file_executable(self):
60
 
        stream = BytesIO()
61
 
        builder = tests.GitBranchBuilder(stream)
62
 
        builder.set_file(u'f\xb5/bar', b'contents\nbar\n', True)
63
 
        self.assertEqualDiff(b'blob\nmark :1\ndata 13\ncontents\nbar\n\n',
64
 
                             stream.getvalue())
65
 
        self.assertEqual([b'M 100755 :1 f\xc2\xb5/bar\n'], builder.commit_info)
66
 
 
67
 
    def test_set_symlink(self):
68
 
        stream = BytesIO()
69
 
        builder = tests.GitBranchBuilder(stream)
70
 
        builder.set_symlink(u'f\xb5/bar', b'link/contents')
71
 
        self.assertEqualDiff(b'blob\nmark :1\ndata 13\nlink/contents\n',
72
 
                             stream.getvalue())
73
 
        self.assertEqual([b'M 120000 :1 f\xc2\xb5/bar\n'], builder.commit_info)
74
 
 
75
 
    def test_set_symlink_newline(self):
76
 
        stream = BytesIO()
77
 
        builder = tests.GitBranchBuilder(stream)
78
 
        builder.set_symlink(u'foo\nbar', 'link/contents')
79
 
        self.assertEqualDiff(b'blob\nmark :1\ndata 13\nlink/contents\n',
80
 
                             stream.getvalue())
81
 
        self.assertEqual([b'M 120000 :1 "foo\\nbar"\n'], builder.commit_info)
82
 
 
83
 
    def test_delete_entry(self):
84
 
        stream = BytesIO()
85
 
        builder = tests.GitBranchBuilder(stream)
86
 
        builder.delete_entry(u'path/to/f\xb5')
87
 
        self.assertEqual([b'D path/to/f\xc2\xb5\n'], builder.commit_info)
88
 
 
89
 
    def test_delete_entry_newline(self):
90
 
        stream = BytesIO()
91
 
        builder = tests.GitBranchBuilder(stream)
92
 
        builder.delete_entry(u'path/to/foo\nbar')
93
 
        self.assertEqual([b'D "path/to/foo\\nbar"\n'], builder.commit_info)
94
 
 
95
 
    def test_encode_path(self):
96
 
        encode = tests.GitBranchBuilder._encode_path
97
 
        # Unicode is encoded to utf-8
98
 
        self.assertEqual(encode(u'f\xb5'), b'f\xc2\xb5')
99
 
        # The name must be quoted if it starts by a double quote or contains a
100
 
        # newline.
101
 
        self.assertEqual(encode(u'"foo'), b'"\\"foo"')
102
 
        self.assertEqual(encode(u'fo\no'), b'"fo\\no"')
103
 
        # When the name is quoted, all backslash and quote chars must be
104
 
        # escaped.
105
 
        self.assertEqual(encode(u'fo\\o\nbar'), b'"fo\\\\o\\nbar"')
106
 
        self.assertEqual(encode(u'fo"o"\nbar'), b'"fo\\"o\\"\\nbar"')
107
 
        # Other control chars, such as \r, need not be escaped.
108
 
        self.assertEqual(encode(u'foo\r\nbar'), b'"foo\r\\nbar"')
109
 
 
110
 
    def test_add_and_commit(self):
111
 
        stream = BytesIO()
112
 
        builder = tests.GitBranchBuilder(stream)
113
 
 
114
 
        builder.set_file(u'f\xb5/bar', b'contents\nbar\n', False)
115
 
        self.assertEqual(b'2', builder.commit(b'Joe Foo <joe@foo.com>',
116
 
                                              u'committing f\xb5/bar',
117
 
                                              timestamp=1194586400,
118
 
                                              timezone=b'+0100'))
119
 
        self.assertEqualDiff(b'blob\nmark :1\ndata 13\ncontents\nbar\n\n'
120
 
                             b'commit refs/heads/master\n'
121
 
                             b'mark :2\n'
122
 
                             b'committer Joe Foo <joe@foo.com> 1194586400 +0100\n'
123
 
                             b'data 18\n'
124
 
                             b'committing f\xc2\xb5/bar'
125
 
                             b'\n'
126
 
                             b'M 100644 :1 f\xc2\xb5/bar\n'
127
 
                             b'\n',
128
 
                             stream.getvalue())
129
 
 
130
 
    def test_commit_base(self):
131
 
        stream = BytesIO()
132
 
        builder = tests.GitBranchBuilder(stream)
133
 
 
134
 
        builder.set_file(u'foo', b'contents\nfoo\n', False)
135
 
        r1 = builder.commit(b'Joe Foo <joe@foo.com>', u'first',
136
 
                            timestamp=1194586400)
137
 
        r2 = builder.commit(b'Joe Foo <joe@foo.com>', u'second',
138
 
                            timestamp=1194586405)
139
 
        r3 = builder.commit(b'Joe Foo <joe@foo.com>', u'third',
140
 
                            timestamp=1194586410,
141
 
                            base=r1)
142
 
 
143
 
        self.assertEqualDiff(b'blob\nmark :1\ndata 13\ncontents\nfoo\n\n'
144
 
                             b'commit refs/heads/master\n'
145
 
                             b'mark :2\n'
146
 
                             b'committer Joe Foo <joe@foo.com> 1194586400 +0000\n'
147
 
                             b'data 5\n'
148
 
                             b'first'
149
 
                             b'\n'
150
 
                             b'M 100644 :1 foo\n'
151
 
                             b'\n'
152
 
                             b'commit refs/heads/master\n'
153
 
                             b'mark :3\n'
154
 
                             b'committer Joe Foo <joe@foo.com> 1194586405 +0000\n'
155
 
                             b'data 6\n'
156
 
                             b'second'
157
 
                             b'\n'
158
 
                             b'\n'
159
 
                             b'commit refs/heads/master\n'
160
 
                             b'mark :4\n'
161
 
                             b'committer Joe Foo <joe@foo.com> 1194586410 +0000\n'
162
 
                             b'data 5\n'
163
 
                             b'third'
164
 
                             b'\n'
165
 
                             b'from :2\n'
166
 
                             b'\n', stream.getvalue())
167
 
 
168
 
    def test_commit_merge(self):
169
 
        stream = BytesIO()
170
 
        builder = tests.GitBranchBuilder(stream)
171
 
 
172
 
        builder.set_file(u'foo', b'contents\nfoo\n', False)
173
 
        r1 = builder.commit(b'Joe Foo <joe@foo.com>', u'first',
174
 
                            timestamp=1194586400)
175
 
        r2 = builder.commit(b'Joe Foo <joe@foo.com>', u'second',
176
 
                            timestamp=1194586405)
177
 
        r3 = builder.commit(b'Joe Foo <joe@foo.com>', u'third',
178
 
                            timestamp=1194586410,
179
 
                            base=r1)
180
 
        r4 = builder.commit(b'Joe Foo <joe@foo.com>', u'Merge',
181
 
                            timestamp=1194586415,
182
 
                            merge=[r2])
183
 
 
184
 
        self.assertEqualDiff(b'blob\nmark :1\ndata 13\ncontents\nfoo\n\n'
185
 
                             b'commit refs/heads/master\n'
186
 
                             b'mark :2\n'
187
 
                             b'committer Joe Foo <joe@foo.com> 1194586400 +0000\n'
188
 
                             b'data 5\n'
189
 
                             b'first'
190
 
                             b'\n'
191
 
                             b'M 100644 :1 foo\n'
192
 
                             b'\n'
193
 
                             b'commit refs/heads/master\n'
194
 
                             b'mark :3\n'
195
 
                             b'committer Joe Foo <joe@foo.com> 1194586405 +0000\n'
196
 
                             b'data 6\n'
197
 
                             b'second'
198
 
                             b'\n'
199
 
                             b'\n'
200
 
                             b'commit refs/heads/master\n'
201
 
                             b'mark :4\n'
202
 
                             b'committer Joe Foo <joe@foo.com> 1194586410 +0000\n'
203
 
                             b'data 5\n'
204
 
                             b'third'
205
 
                             b'\n'
206
 
                             b'from :2\n'
207
 
                             b'\n'
208
 
                             b'commit refs/heads/master\n'
209
 
                             b'mark :5\n'
210
 
                             b'committer Joe Foo <joe@foo.com> 1194586415 +0000\n'
211
 
                             b'data 5\n'
212
 
                             b'Merge'
213
 
                             b'\n'
214
 
                             b'merge :3\n'
215
 
                             b'\n', stream.getvalue())
216
 
 
217
 
    def test_auto_timestamp(self):
218
 
        stream = BytesIO()
219
 
        builder = tests.GitBranchBuilder(stream)
220
 
        builder.commit(b'Joe Foo <joe@foo.com>', u'message')
221
 
        self.assertContainsRe(stream.getvalue(),
222
 
                              br'committer Joe Foo <joe@foo\.com> \d+ \+0000')
223
 
 
224
 
    def test_reset(self):
225
 
        stream = BytesIO()
226
 
        builder = tests.GitBranchBuilder(stream)
227
 
        builder.reset()
228
 
        self.assertEqualDiff(b'reset refs/heads/master\n\n', stream.getvalue())
229
 
 
230
 
    def test_reset_named_ref(self):
231
 
        stream = BytesIO()
232
 
        builder = tests.GitBranchBuilder(stream)
233
 
        builder.reset(b'refs/heads/branch')
234
 
        self.assertEqualDiff(b'reset refs/heads/branch\n\n', stream.getvalue())
235
 
 
236
 
    def test_reset_revision(self):
237
 
        stream = BytesIO()
238
 
        builder = tests.GitBranchBuilder(stream)
239
 
        builder.reset(mark=b'123')
240
 
        self.assertEqualDiff(
241
 
            b'reset refs/heads/master\n'
242
 
            b'from :123\n'
243
 
            b'\n', stream.getvalue())
244
 
 
245
 
 
246
 
class TestGitBranchBuilderReal(tests.TestCaseInTempDir):
247
 
 
248
 
    def test_create_real_branch(self):
249
 
        GitRepo.init(".")
250
 
 
251
 
        builder = tests.GitBranchBuilder()
252
 
        builder.set_file(u'foo', b'contents\nfoo\n', False)
253
 
        r1 = builder.commit(b'Joe Foo <joe@foo.com>', u'first',
254
 
                            timestamp=1194586400)
255
 
        mapping = builder.finish()
256
 
        self.assertEqual({b'1': b'44411e8e9202177dd19b6599d7a7991059fa3cb4',
257
 
                          b'2': b'b0b62e674f67306fddcf72fa888c3b56df100d64',
258
 
                          }, mapping)