~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_smart.py

  • Committer: Launchpad Translations on behalf of bzr-core
  • Date: 2012-05-29 04:30:47 UTC
  • mto: (6581.1.1 trunk)
  • mto: This revision was merged to the branch mainline in revision 6582.
  • Revision ID: launchpad_translations_on_behalf_of_bzr-core-20120529043047-mmj2hkq63ke8kl81
Launchpad automatic translations update.

Show diffs side-by-side

added added

removed removed

Lines of Context:
 
1
# Copyright (C) 2006-2012 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
"""Tests for the smart wire/domain protocol.
 
18
 
 
19
This module contains tests for the domain-level smart requests and responses,
 
20
such as the 'Branch.lock_write' request. Many of these use specific disk
 
21
formats to exercise calls that only make sense for formats with specific
 
22
properties.
 
23
 
 
24
Tests for low-level protocol encoding are found in test_smart_transport.
 
25
"""
 
26
 
 
27
import bz2
 
28
import zlib
 
29
 
 
30
from bzrlib import (
 
31
    bencode,
 
32
    branch as _mod_branch,
 
33
    bzrdir,
 
34
    errors,
 
35
    gpg,
 
36
    inventory_delta,
 
37
    tests,
 
38
    transport,
 
39
    urlutils,
 
40
    versionedfile,
 
41
    )
 
42
from bzrlib.smart import (
 
43
    branch as smart_branch,
 
44
    bzrdir as smart_dir,
 
45
    repository as smart_repo,
 
46
    packrepository as smart_packrepo,
 
47
    request as smart_req,
 
48
    server,
 
49
    vfs,
 
50
    )
 
51
from bzrlib.testament import Testament
 
52
from bzrlib.tests import test_server
 
53
from bzrlib.transport import (
 
54
    chroot,
 
55
    memory,
 
56
    )
 
57
 
 
58
 
 
59
def load_tests(standard_tests, module, loader):
 
60
    """Multiply tests version and protocol consistency."""
 
61
    # FindRepository tests.
 
62
    scenarios = [
 
63
        ("find_repository", {
 
64
            "_request_class": smart_dir.SmartServerRequestFindRepositoryV1}),
 
65
        ("find_repositoryV2", {
 
66
            "_request_class": smart_dir.SmartServerRequestFindRepositoryV2}),
 
67
        ("find_repositoryV3", {
 
68
            "_request_class": smart_dir.SmartServerRequestFindRepositoryV3}),
 
69
        ]
 
70
    to_adapt, result = tests.split_suite_by_re(standard_tests,
 
71
        "TestSmartServerRequestFindRepository")
 
72
    v2_only, v1_and_2 = tests.split_suite_by_re(to_adapt,
 
73
        "_v2")
 
74
    tests.multiply_tests(v1_and_2, scenarios, result)
 
75
    # The first scenario is only applicable to v1 protocols, it is deleted
 
76
    # since.
 
77
    tests.multiply_tests(v2_only, scenarios[1:], result)
 
78
    return result
 
79
 
 
80
 
 
81
class TestCaseWithChrootedTransport(tests.TestCaseWithTransport):
 
82
 
 
83
    def setUp(self):
 
84
        self.vfs_transport_factory = memory.MemoryServer
 
85
        tests.TestCaseWithTransport.setUp(self)
 
86
        self._chroot_server = None
 
87
 
 
88
    def get_transport(self, relpath=None):
 
89
        if self._chroot_server is None:
 
90
            backing_transport = tests.TestCaseWithTransport.get_transport(self)
 
91
            self._chroot_server = chroot.ChrootServer(backing_transport)
 
92
            self.start_server(self._chroot_server)
 
93
        t = transport.get_transport_from_url(self._chroot_server.get_url())
 
94
        if relpath is not None:
 
95
            t = t.clone(relpath)
 
96
        return t
 
97
 
 
98
 
 
99
class TestCaseWithSmartMedium(tests.TestCaseWithMemoryTransport):
 
100
 
 
101
    def setUp(self):
 
102
        super(TestCaseWithSmartMedium, self).setUp()
 
103
        # We're allowed to set  the transport class here, so that we don't use
 
104
        # the default or a parameterized class, but rather use the
 
105
        # TestCaseWithTransport infrastructure to set up a smart server and
 
106
        # transport.
 
107
        self.overrideAttr(self, "transport_server", self.make_transport_server)
 
108
 
 
109
    def make_transport_server(self):
 
110
        return test_server.SmartTCPServer_for_testing('-' + self.id())
 
111
 
 
112
    def get_smart_medium(self):
 
113
        """Get a smart medium to use in tests."""
 
114
        return self.get_transport().get_smart_medium()
 
115
 
 
116
 
 
117
class TestByteStreamToStream(tests.TestCase):
 
118
 
 
119
    def test_repeated_substreams_same_kind_are_one_stream(self):
 
120
        # Make a stream - an iterable of bytestrings.
 
121
        stream = [('text', [versionedfile.FulltextContentFactory(('k1',), None,
 
122
            None, 'foo')]),('text', [
 
123
            versionedfile.FulltextContentFactory(('k2',), None, None, 'bar')])]
 
124
        fmt = bzrdir.format_registry.get('pack-0.92')().repository_format
 
125
        bytes = smart_repo._stream_to_byte_stream(stream, fmt)
 
126
        streams = []
 
127
        # Iterate the resulting iterable; checking that we get only one stream
 
128
        # out.
 
129
        fmt, stream = smart_repo._byte_stream_to_stream(bytes)
 
130
        for kind, substream in stream:
 
131
            streams.append((kind, list(substream)))
 
132
        self.assertLength(1, streams)
 
133
        self.assertLength(2, streams[0][1])
 
134
 
 
135
 
 
136
class TestSmartServerResponse(tests.TestCase):
 
137
 
 
138
    def test__eq__(self):
 
139
        self.assertEqual(smart_req.SmartServerResponse(('ok', )),
 
140
            smart_req.SmartServerResponse(('ok', )))
 
141
        self.assertEqual(smart_req.SmartServerResponse(('ok', ), 'body'),
 
142
            smart_req.SmartServerResponse(('ok', ), 'body'))
 
143
        self.assertNotEqual(smart_req.SmartServerResponse(('ok', )),
 
144
            smart_req.SmartServerResponse(('notok', )))
 
145
        self.assertNotEqual(smart_req.SmartServerResponse(('ok', ), 'body'),
 
146
            smart_req.SmartServerResponse(('ok', )))
 
147
        self.assertNotEqual(None,
 
148
            smart_req.SmartServerResponse(('ok', )))
 
149
 
 
150
    def test__str__(self):
 
151
        """SmartServerResponses can be stringified."""
 
152
        self.assertEqual(
 
153
            "<SuccessfulSmartServerResponse args=('args',) body='body'>",
 
154
            str(smart_req.SuccessfulSmartServerResponse(('args',), 'body')))
 
155
        self.assertEqual(
 
156
            "<FailedSmartServerResponse args=('args',) body='body'>",
 
157
            str(smart_req.FailedSmartServerResponse(('args',), 'body')))
 
158
 
 
159
 
 
160
class TestSmartServerRequest(tests.TestCaseWithMemoryTransport):
 
161
 
 
162
    def test_translate_client_path(self):
 
163
        transport = self.get_transport()
 
164
        request = smart_req.SmartServerRequest(transport, 'foo/')
 
165
        self.assertEqual('./', request.translate_client_path('foo/'))
 
166
        self.assertRaises(
 
167
            errors.InvalidURLJoin, request.translate_client_path, 'foo/..')
 
168
        self.assertRaises(
 
169
            errors.PathNotChild, request.translate_client_path, '/')
 
170
        self.assertRaises(
 
171
            errors.PathNotChild, request.translate_client_path, 'bar/')
 
172
        self.assertEqual('./baz', request.translate_client_path('foo/baz'))
 
173
        e_acute = u'\N{LATIN SMALL LETTER E WITH ACUTE}'.encode('utf-8')
 
174
        self.assertEqual('./' + urlutils.escape(e_acute),
 
175
                         request.translate_client_path('foo/' + e_acute))
 
176
 
 
177
    def test_translate_client_path_vfs(self):
 
178
        """VfsRequests receive escaped paths rather than raw UTF-8."""
 
179
        transport = self.get_transport()
 
180
        request = vfs.VfsRequest(transport, 'foo/')
 
181
        e_acute = u'\N{LATIN SMALL LETTER E WITH ACUTE}'.encode('utf-8')
 
182
        escaped = urlutils.escape('foo/' + e_acute)
 
183
        self.assertEqual('./' + urlutils.escape(e_acute),
 
184
                         request.translate_client_path(escaped))
 
185
 
 
186
    def test_transport_from_client_path(self):
 
187
        transport = self.get_transport()
 
188
        request = smart_req.SmartServerRequest(transport, 'foo/')
 
189
        self.assertEqual(
 
190
            transport.base,
 
191
            request.transport_from_client_path('foo/').base)
 
192
 
 
193
 
 
194
class TestSmartServerBzrDirRequestCloningMetaDir(
 
195
    tests.TestCaseWithMemoryTransport):
 
196
    """Tests for BzrDir.cloning_metadir."""
 
197
 
 
198
    def test_cloning_metadir(self):
 
199
        """When there is a bzrdir present, the call succeeds."""
 
200
        backing = self.get_transport()
 
201
        dir = self.make_bzrdir('.')
 
202
        local_result = dir.cloning_metadir()
 
203
        request_class = smart_dir.SmartServerBzrDirRequestCloningMetaDir
 
204
        request = request_class(backing)
 
205
        expected = smart_req.SuccessfulSmartServerResponse(
 
206
            (local_result.network_name(),
 
207
            local_result.repository_format.network_name(),
 
208
            ('branch', local_result.get_branch_format().network_name())))
 
209
        self.assertEqual(expected, request.execute('', 'False'))
 
210
 
 
211
    def test_cloning_metadir_reference(self):
 
212
        """The request fails when bzrdir contains a branch reference."""
 
213
        backing = self.get_transport()
 
214
        referenced_branch = self.make_branch('referenced')
 
215
        dir = self.make_bzrdir('.')
 
216
        local_result = dir.cloning_metadir()
 
217
        reference = _mod_branch.BranchReferenceFormat().initialize(
 
218
            dir, target_branch=referenced_branch)
 
219
        reference_url = _mod_branch.BranchReferenceFormat().get_reference(dir)
 
220
        # The server shouldn't try to follow the branch reference, so it's fine
 
221
        # if the referenced branch isn't reachable.
 
222
        backing.rename('referenced', 'moved')
 
223
        request_class = smart_dir.SmartServerBzrDirRequestCloningMetaDir
 
224
        request = request_class(backing)
 
225
        expected = smart_req.FailedSmartServerResponse(('BranchReference',))
 
226
        self.assertEqual(expected, request.execute('', 'False'))
 
227
 
 
228
 
 
229
class TestSmartServerBzrDirRequestCloningMetaDir(
 
230
    tests.TestCaseWithMemoryTransport):
 
231
    """Tests for BzrDir.checkout_metadir."""
 
232
 
 
233
    def test_checkout_metadir(self):
 
234
        backing = self.get_transport()
 
235
        request = smart_dir.SmartServerBzrDirRequestCheckoutMetaDir(
 
236
            backing)
 
237
        branch = self.make_branch('.', format='2a')
 
238
        response = request.execute('')
 
239
        self.assertEqual(
 
240
            smart_req.SmartServerResponse(
 
241
                ('Bazaar-NG meta directory, format 1\n',
 
242
                 'Bazaar repository format 2a (needs bzr 1.16 or later)\n',
 
243
                 'Bazaar Branch Format 7 (needs bzr 1.6)\n')),
 
244
            response)
 
245
 
 
246
 
 
247
class TestSmartServerBzrDirRequestDestroyBranch(
 
248
    tests.TestCaseWithMemoryTransport):
 
249
    """Tests for BzrDir.destroy_branch."""
 
250
 
 
251
    def test_destroy_branch_default(self):
 
252
        """The default branch can be removed."""
 
253
        backing = self.get_transport()
 
254
        dir = self.make_branch('.').bzrdir
 
255
        request_class = smart_dir.SmartServerBzrDirRequestDestroyBranch
 
256
        request = request_class(backing)
 
257
        expected = smart_req.SuccessfulSmartServerResponse(('ok',))
 
258
        self.assertEqual(expected, request.execute('', None))
 
259
 
 
260
    def test_destroy_branch_named(self):
 
261
        """A named branch can be removed."""
 
262
        backing = self.get_transport()
 
263
        dir = self.make_repository('.', format="development-colo").bzrdir
 
264
        dir.create_branch(name="branchname")
 
265
        request_class = smart_dir.SmartServerBzrDirRequestDestroyBranch
 
266
        request = request_class(backing)
 
267
        expected = smart_req.SuccessfulSmartServerResponse(('ok',))
 
268
        self.assertEqual(expected, request.execute('', "branchname"))
 
269
 
 
270
    def test_destroy_branch_missing(self):
 
271
        """An error is raised if the branch didn't exist."""
 
272
        backing = self.get_transport()
 
273
        dir = self.make_bzrdir('.', format="development-colo")
 
274
        request_class = smart_dir.SmartServerBzrDirRequestDestroyBranch
 
275
        request = request_class(backing)
 
276
        expected = smart_req.FailedSmartServerResponse(('nobranch',), None)
 
277
        self.assertEqual(expected, request.execute('', "branchname"))
 
278
 
 
279
 
 
280
class TestSmartServerBzrDirRequestHasWorkingTree(
 
281
    tests.TestCaseWithTransport):
 
282
    """Tests for BzrDir.has_workingtree."""
 
283
 
 
284
    def test_has_workingtree_yes(self):
 
285
        """A working tree is present."""
 
286
        backing = self.get_transport()
 
287
        dir = self.make_branch_and_tree('.').bzrdir
 
288
        request_class = smart_dir.SmartServerBzrDirRequestHasWorkingTree
 
289
        request = request_class(backing)
 
290
        expected = smart_req.SuccessfulSmartServerResponse(('yes',))
 
291
        self.assertEqual(expected, request.execute(''))
 
292
 
 
293
    def test_has_workingtree_no(self):
 
294
        """A working tree is missing."""
 
295
        backing = self.get_transport()
 
296
        dir = self.make_bzrdir('.')
 
297
        request_class = smart_dir.SmartServerBzrDirRequestHasWorkingTree
 
298
        request = request_class(backing)
 
299
        expected = smart_req.SuccessfulSmartServerResponse(('no',))
 
300
        self.assertEqual(expected, request.execute(''))
 
301
 
 
302
 
 
303
class TestSmartServerBzrDirRequestDestroyRepository(
 
304
    tests.TestCaseWithMemoryTransport):
 
305
    """Tests for BzrDir.destroy_repository."""
 
306
 
 
307
    def test_destroy_repository_default(self):
 
308
        """The repository can be removed."""
 
309
        backing = self.get_transport()
 
310
        dir = self.make_repository('.').bzrdir
 
311
        request_class = smart_dir.SmartServerBzrDirRequestDestroyRepository
 
312
        request = request_class(backing)
 
313
        expected = smart_req.SuccessfulSmartServerResponse(('ok',))
 
314
        self.assertEqual(expected, request.execute(''))
 
315
 
 
316
    def test_destroy_repository_missing(self):
 
317
        """An error is raised if the repository didn't exist."""
 
318
        backing = self.get_transport()
 
319
        dir = self.make_bzrdir('.')
 
320
        request_class = smart_dir.SmartServerBzrDirRequestDestroyRepository
 
321
        request = request_class(backing)
 
322
        expected = smart_req.FailedSmartServerResponse(
 
323
            ('norepository',), None)
 
324
        self.assertEqual(expected, request.execute(''))
 
325
 
 
326
 
 
327
class TestSmartServerRequestCreateRepository(tests.TestCaseWithMemoryTransport):
 
328
    """Tests for BzrDir.create_repository."""
 
329
 
 
330
    def test_makes_repository(self):
 
331
        """When there is a bzrdir present, the call succeeds."""
 
332
        backing = self.get_transport()
 
333
        self.make_bzrdir('.')
 
334
        request_class = smart_dir.SmartServerRequestCreateRepository
 
335
        request = request_class(backing)
 
336
        reference_bzrdir_format = bzrdir.format_registry.get('pack-0.92')()
 
337
        reference_format = reference_bzrdir_format.repository_format
 
338
        network_name = reference_format.network_name()
 
339
        expected = smart_req.SuccessfulSmartServerResponse(
 
340
            ('ok', 'no', 'no', 'no', network_name))
 
341
        self.assertEqual(expected, request.execute('', network_name, 'True'))
 
342
 
 
343
 
 
344
class TestSmartServerRequestFindRepository(tests.TestCaseWithMemoryTransport):
 
345
    """Tests for BzrDir.find_repository."""
 
346
 
 
347
    def test_no_repository(self):
 
348
        """When there is no repository to be found, ('norepository', ) is returned."""
 
349
        backing = self.get_transport()
 
350
        request = self._request_class(backing)
 
351
        self.make_bzrdir('.')
 
352
        self.assertEqual(smart_req.SmartServerResponse(('norepository', )),
 
353
            request.execute(''))
 
354
 
 
355
    def test_nonshared_repository(self):
 
356
        # nonshared repositorys only allow 'find' to return a handle when the
 
357
        # path the repository is being searched on is the same as that that
 
358
        # the repository is at.
 
359
        backing = self.get_transport()
 
360
        request = self._request_class(backing)
 
361
        result = self._make_repository_and_result()
 
362
        self.assertEqual(result, request.execute(''))
 
363
        self.make_bzrdir('subdir')
 
364
        self.assertEqual(smart_req.SmartServerResponse(('norepository', )),
 
365
            request.execute('subdir'))
 
366
 
 
367
    def _make_repository_and_result(self, shared=False, format=None):
 
368
        """Convenience function to setup a repository.
 
369
 
 
370
        :result: The SmartServerResponse to expect when opening it.
 
371
        """
 
372
        repo = self.make_repository('.', shared=shared, format=format)
 
373
        if repo.supports_rich_root():
 
374
            rich_root = 'yes'
 
375
        else:
 
376
            rich_root = 'no'
 
377
        if repo._format.supports_tree_reference:
 
378
            subtrees = 'yes'
 
379
        else:
 
380
            subtrees = 'no'
 
381
        if repo._format.supports_external_lookups:
 
382
            external = 'yes'
 
383
        else:
 
384
            external = 'no'
 
385
        if (smart_dir.SmartServerRequestFindRepositoryV3 ==
 
386
            self._request_class):
 
387
            return smart_req.SuccessfulSmartServerResponse(
 
388
                ('ok', '', rich_root, subtrees, external,
 
389
                 repo._format.network_name()))
 
390
        elif (smart_dir.SmartServerRequestFindRepositoryV2 ==
 
391
            self._request_class):
 
392
            # All tests so far are on formats, and for non-external
 
393
            # repositories.
 
394
            return smart_req.SuccessfulSmartServerResponse(
 
395
                ('ok', '', rich_root, subtrees, external))
 
396
        else:
 
397
            return smart_req.SuccessfulSmartServerResponse(
 
398
                ('ok', '', rich_root, subtrees))
 
399
 
 
400
    def test_shared_repository(self):
 
401
        """When there is a shared repository, we get 'ok', 'relpath-to-repo'."""
 
402
        backing = self.get_transport()
 
403
        request = self._request_class(backing)
 
404
        result = self._make_repository_and_result(shared=True)
 
405
        self.assertEqual(result, request.execute(''))
 
406
        self.make_bzrdir('subdir')
 
407
        result2 = smart_req.SmartServerResponse(
 
408
            result.args[0:1] + ('..', ) + result.args[2:])
 
409
        self.assertEqual(result2,
 
410
            request.execute('subdir'))
 
411
        self.make_bzrdir('subdir/deeper')
 
412
        result3 = smart_req.SmartServerResponse(
 
413
            result.args[0:1] + ('../..', ) + result.args[2:])
 
414
        self.assertEqual(result3,
 
415
            request.execute('subdir/deeper'))
 
416
 
 
417
    def test_rich_root_and_subtree_encoding(self):
 
418
        """Test for the format attributes for rich root and subtree support."""
 
419
        backing = self.get_transport()
 
420
        request = self._request_class(backing)
 
421
        result = self._make_repository_and_result(
 
422
            format='development-subtree')
 
423
        # check the test will be valid
 
424
        self.assertEqual('yes', result.args[2])
 
425
        self.assertEqual('yes', result.args[3])
 
426
        self.assertEqual(result, request.execute(''))
 
427
 
 
428
    def test_supports_external_lookups_no_v2(self):
 
429
        """Test for the supports_external_lookups attribute."""
 
430
        backing = self.get_transport()
 
431
        request = self._request_class(backing)
 
432
        result = self._make_repository_and_result(
 
433
            format='development-subtree')
 
434
        # check the test will be valid
 
435
        self.assertEqual('yes', result.args[4])
 
436
        self.assertEqual(result, request.execute(''))
 
437
 
 
438
 
 
439
class TestSmartServerBzrDirRequestGetConfigFile(
 
440
    tests.TestCaseWithMemoryTransport):
 
441
    """Tests for BzrDir.get_config_file."""
 
442
 
 
443
    def test_present(self):
 
444
        backing = self.get_transport()
 
445
        dir = self.make_bzrdir('.')
 
446
        dir.get_config().set_default_stack_on("/")
 
447
        local_result = dir._get_config()._get_config_file().read()
 
448
        request_class = smart_dir.SmartServerBzrDirRequestConfigFile
 
449
        request = request_class(backing)
 
450
        expected = smart_req.SuccessfulSmartServerResponse((), local_result)
 
451
        self.assertEqual(expected, request.execute(''))
 
452
 
 
453
    def test_missing(self):
 
454
        backing = self.get_transport()
 
455
        dir = self.make_bzrdir('.')
 
456
        request_class = smart_dir.SmartServerBzrDirRequestConfigFile
 
457
        request = request_class(backing)
 
458
        expected = smart_req.SuccessfulSmartServerResponse((), '')
 
459
        self.assertEqual(expected, request.execute(''))
 
460
 
 
461
 
 
462
class TestSmartServerBzrDirRequestGetBranches(
 
463
    tests.TestCaseWithMemoryTransport):
 
464
    """Tests for BzrDir.get_branches."""
 
465
 
 
466
    def test_simple(self):
 
467
        backing = self.get_transport()
 
468
        branch = self.make_branch('.')
 
469
        request_class = smart_dir.SmartServerBzrDirRequestGetBranches
 
470
        request = request_class(backing)
 
471
        local_result = bencode.bencode(
 
472
            {"": ("branch", branch._format.network_name())})
 
473
        expected = smart_req.SuccessfulSmartServerResponse(
 
474
            ("success", ), local_result)
 
475
        self.assertEqual(expected, request.execute(''))
 
476
 
 
477
    def test_empty(self):
 
478
        backing = self.get_transport()
 
479
        dir = self.make_bzrdir('.')
 
480
        request_class = smart_dir.SmartServerBzrDirRequestGetBranches
 
481
        request = request_class(backing)
 
482
        local_result = bencode.bencode({})
 
483
        expected = smart_req.SuccessfulSmartServerResponse(
 
484
            ('success',), local_result)
 
485
        self.assertEqual(expected, request.execute(''))
 
486
 
 
487
 
 
488
class TestSmartServerRequestInitializeBzrDir(tests.TestCaseWithMemoryTransport):
 
489
 
 
490
    def test_empty_dir(self):
 
491
        """Initializing an empty dir should succeed and do it."""
 
492
        backing = self.get_transport()
 
493
        request = smart_dir.SmartServerRequestInitializeBzrDir(backing)
 
494
        self.assertEqual(smart_req.SmartServerResponse(('ok', )),
 
495
            request.execute(''))
 
496
        made_dir = bzrdir.BzrDir.open_from_transport(backing)
 
497
        # no branch, tree or repository is expected with the current
 
498
        # default formart.
 
499
        self.assertRaises(errors.NoWorkingTree, made_dir.open_workingtree)
 
500
        self.assertRaises(errors.NotBranchError, made_dir.open_branch)
 
501
        self.assertRaises(errors.NoRepositoryPresent, made_dir.open_repository)
 
502
 
 
503
    def test_missing_dir(self):
 
504
        """Initializing a missing directory should fail like the bzrdir api."""
 
505
        backing = self.get_transport()
 
506
        request = smart_dir.SmartServerRequestInitializeBzrDir(backing)
 
507
        self.assertRaises(errors.NoSuchFile,
 
508
            request.execute, 'subdir')
 
509
 
 
510
    def test_initialized_dir(self):
 
511
        """Initializing an extant bzrdir should fail like the bzrdir api."""
 
512
        backing = self.get_transport()
 
513
        request = smart_dir.SmartServerRequestInitializeBzrDir(backing)
 
514
        self.make_bzrdir('subdir')
 
515
        self.assertRaises(errors.AlreadyControlDirError,
 
516
            request.execute, 'subdir')
 
517
 
 
518
 
 
519
class TestSmartServerRequestBzrDirInitializeEx(
 
520
    tests.TestCaseWithMemoryTransport):
 
521
    """Basic tests for BzrDir.initialize_ex_1.16 in the smart server.
 
522
 
 
523
    The main unit tests in test_bzrdir exercise the API comprehensively.
 
524
    """
 
525
 
 
526
    def test_empty_dir(self):
 
527
        """Initializing an empty dir should succeed and do it."""
 
528
        backing = self.get_transport()
 
529
        name = self.make_bzrdir('reference')._format.network_name()
 
530
        request = smart_dir.SmartServerRequestBzrDirInitializeEx(backing)
 
531
        self.assertEqual(
 
532
            smart_req.SmartServerResponse(('', '', '', '', '', '', name,
 
533
                                           'False', '', '', '')),
 
534
            request.execute(name, '', 'True', 'False', 'False', '', '', '', '',
 
535
                            'False'))
 
536
        made_dir = bzrdir.BzrDir.open_from_transport(backing)
 
537
        # no branch, tree or repository is expected with the current
 
538
        # default format.
 
539
        self.assertRaises(errors.NoWorkingTree, made_dir.open_workingtree)
 
540
        self.assertRaises(errors.NotBranchError, made_dir.open_branch)
 
541
        self.assertRaises(errors.NoRepositoryPresent, made_dir.open_repository)
 
542
 
 
543
    def test_missing_dir(self):
 
544
        """Initializing a missing directory should fail like the bzrdir api."""
 
545
        backing = self.get_transport()
 
546
        name = self.make_bzrdir('reference')._format.network_name()
 
547
        request = smart_dir.SmartServerRequestBzrDirInitializeEx(backing)
 
548
        self.assertRaises(errors.NoSuchFile, request.execute, name,
 
549
            'subdir/dir', 'False', 'False', 'False', '', '', '', '', 'False')
 
550
 
 
551
    def test_initialized_dir(self):
 
552
        """Initializing an extant directory should fail like the bzrdir api."""
 
553
        backing = self.get_transport()
 
554
        name = self.make_bzrdir('reference')._format.network_name()
 
555
        request = smart_dir.SmartServerRequestBzrDirInitializeEx(backing)
 
556
        self.make_bzrdir('subdir')
 
557
        self.assertRaises(errors.FileExists, request.execute, name, 'subdir',
 
558
            'False', 'False', 'False', '', '', '', '', 'False')
 
559
 
 
560
 
 
561
class TestSmartServerRequestOpenBzrDir(tests.TestCaseWithMemoryTransport):
 
562
 
 
563
    def test_no_directory(self):
 
564
        backing = self.get_transport()
 
565
        request = smart_dir.SmartServerRequestOpenBzrDir(backing)
 
566
        self.assertEqual(smart_req.SmartServerResponse(('no', )),
 
567
            request.execute('does-not-exist'))
 
568
 
 
569
    def test_empty_directory(self):
 
570
        backing = self.get_transport()
 
571
        backing.mkdir('empty')
 
572
        request = smart_dir.SmartServerRequestOpenBzrDir(backing)
 
573
        self.assertEqual(smart_req.SmartServerResponse(('no', )),
 
574
            request.execute('empty'))
 
575
 
 
576
    def test_outside_root_client_path(self):
 
577
        backing = self.get_transport()
 
578
        request = smart_dir.SmartServerRequestOpenBzrDir(backing,
 
579
            root_client_path='root')
 
580
        self.assertEqual(smart_req.SmartServerResponse(('no', )),
 
581
            request.execute('not-root'))
 
582
 
 
583
 
 
584
class TestSmartServerRequestOpenBzrDir_2_1(tests.TestCaseWithMemoryTransport):
 
585
 
 
586
    def test_no_directory(self):
 
587
        backing = self.get_transport()
 
588
        request = smart_dir.SmartServerRequestOpenBzrDir_2_1(backing)
 
589
        self.assertEqual(smart_req.SmartServerResponse(('no', )),
 
590
            request.execute('does-not-exist'))
 
591
 
 
592
    def test_empty_directory(self):
 
593
        backing = self.get_transport()
 
594
        backing.mkdir('empty')
 
595
        request = smart_dir.SmartServerRequestOpenBzrDir_2_1(backing)
 
596
        self.assertEqual(smart_req.SmartServerResponse(('no', )),
 
597
            request.execute('empty'))
 
598
 
 
599
    def test_present_without_workingtree(self):
 
600
        backing = self.get_transport()
 
601
        request = smart_dir.SmartServerRequestOpenBzrDir_2_1(backing)
 
602
        self.make_bzrdir('.')
 
603
        self.assertEqual(smart_req.SmartServerResponse(('yes', 'no')),
 
604
            request.execute(''))
 
605
 
 
606
    def test_outside_root_client_path(self):
 
607
        backing = self.get_transport()
 
608
        request = smart_dir.SmartServerRequestOpenBzrDir_2_1(backing,
 
609
            root_client_path='root')
 
610
        self.assertEqual(smart_req.SmartServerResponse(('no',)),
 
611
            request.execute('not-root'))
 
612
 
 
613
 
 
614
class TestSmartServerRequestOpenBzrDir_2_1_disk(TestCaseWithChrootedTransport):
 
615
 
 
616
    def test_present_with_workingtree(self):
 
617
        self.vfs_transport_factory = test_server.LocalURLServer
 
618
        backing = self.get_transport()
 
619
        request = smart_dir.SmartServerRequestOpenBzrDir_2_1(backing)
 
620
        bd = self.make_bzrdir('.')
 
621
        bd.create_repository()
 
622
        bd.create_branch()
 
623
        bd.create_workingtree()
 
624
        self.assertEqual(smart_req.SmartServerResponse(('yes', 'yes')),
 
625
            request.execute(''))
 
626
 
 
627
 
 
628
class TestSmartServerRequestOpenBranch(TestCaseWithChrootedTransport):
 
629
 
 
630
    def test_no_branch(self):
 
631
        """When there is no branch, ('nobranch', ) is returned."""
 
632
        backing = self.get_transport()
 
633
        request = smart_dir.SmartServerRequestOpenBranch(backing)
 
634
        self.make_bzrdir('.')
 
635
        self.assertEqual(smart_req.SmartServerResponse(('nobranch', )),
 
636
            request.execute(''))
 
637
 
 
638
    def test_branch(self):
 
639
        """When there is a branch, 'ok' is returned."""
 
640
        backing = self.get_transport()
 
641
        request = smart_dir.SmartServerRequestOpenBranch(backing)
 
642
        self.make_branch('.')
 
643
        self.assertEqual(smart_req.SmartServerResponse(('ok', '')),
 
644
            request.execute(''))
 
645
 
 
646
    def test_branch_reference(self):
 
647
        """When there is a branch reference, the reference URL is returned."""
 
648
        self.vfs_transport_factory = test_server.LocalURLServer
 
649
        backing = self.get_transport()
 
650
        request = smart_dir.SmartServerRequestOpenBranch(backing)
 
651
        branch = self.make_branch('branch')
 
652
        checkout = branch.create_checkout('reference',lightweight=True)
 
653
        reference_url = _mod_branch.BranchReferenceFormat().get_reference(
 
654
            checkout.bzrdir)
 
655
        self.assertFileEqual(reference_url, 'reference/.bzr/branch/location')
 
656
        self.assertEqual(smart_req.SmartServerResponse(('ok', reference_url)),
 
657
            request.execute('reference'))
 
658
 
 
659
    def test_notification_on_branch_from_repository(self):
 
660
        """When there is a repository, the error should return details."""
 
661
        backing = self.get_transport()
 
662
        request = smart_dir.SmartServerRequestOpenBranch(backing)
 
663
        repo = self.make_repository('.')
 
664
        self.assertEqual(smart_req.SmartServerResponse(('nobranch',)),
 
665
            request.execute(''))
 
666
 
 
667
 
 
668
class TestSmartServerRequestOpenBranchV2(TestCaseWithChrootedTransport):
 
669
 
 
670
    def test_no_branch(self):
 
671
        """When there is no branch, ('nobranch', ) is returned."""
 
672
        backing = self.get_transport()
 
673
        self.make_bzrdir('.')
 
674
        request = smart_dir.SmartServerRequestOpenBranchV2(backing)
 
675
        self.assertEqual(smart_req.SmartServerResponse(('nobranch', )),
 
676
            request.execute(''))
 
677
 
 
678
    def test_branch(self):
 
679
        """When there is a branch, 'ok' is returned."""
 
680
        backing = self.get_transport()
 
681
        expected = self.make_branch('.')._format.network_name()
 
682
        request = smart_dir.SmartServerRequestOpenBranchV2(backing)
 
683
        self.assertEqual(smart_req.SuccessfulSmartServerResponse(
 
684
                ('branch', expected)),
 
685
                         request.execute(''))
 
686
 
 
687
    def test_branch_reference(self):
 
688
        """When there is a branch reference, the reference URL is returned."""
 
689
        self.vfs_transport_factory = test_server.LocalURLServer
 
690
        backing = self.get_transport()
 
691
        request = smart_dir.SmartServerRequestOpenBranchV2(backing)
 
692
        branch = self.make_branch('branch')
 
693
        checkout = branch.create_checkout('reference',lightweight=True)
 
694
        reference_url = _mod_branch.BranchReferenceFormat().get_reference(
 
695
            checkout.bzrdir)
 
696
        self.assertFileEqual(reference_url, 'reference/.bzr/branch/location')
 
697
        self.assertEqual(smart_req.SuccessfulSmartServerResponse(
 
698
                ('ref', reference_url)),
 
699
                         request.execute('reference'))
 
700
 
 
701
    def test_stacked_branch(self):
 
702
        """Opening a stacked branch does not open the stacked-on branch."""
 
703
        trunk = self.make_branch('trunk')
 
704
        feature = self.make_branch('feature')
 
705
        feature.set_stacked_on_url(trunk.base)
 
706
        opened_branches = []
 
707
        _mod_branch.Branch.hooks.install_named_hook(
 
708
            'open', opened_branches.append, None)
 
709
        backing = self.get_transport()
 
710
        request = smart_dir.SmartServerRequestOpenBranchV2(backing)
 
711
        request.setup_jail()
 
712
        try:
 
713
            response = request.execute('feature')
 
714
        finally:
 
715
            request.teardown_jail()
 
716
        expected_format = feature._format.network_name()
 
717
        self.assertEqual(smart_req.SuccessfulSmartServerResponse(
 
718
                ('branch', expected_format)),
 
719
                         response)
 
720
        self.assertLength(1, opened_branches)
 
721
 
 
722
    def test_notification_on_branch_from_repository(self):
 
723
        """When there is a repository, the error should return details."""
 
724
        backing = self.get_transport()
 
725
        request = smart_dir.SmartServerRequestOpenBranchV2(backing)
 
726
        repo = self.make_repository('.')
 
727
        self.assertEqual(smart_req.SmartServerResponse(('nobranch',)),
 
728
            request.execute(''))
 
729
 
 
730
 
 
731
class TestSmartServerRequestOpenBranchV3(TestCaseWithChrootedTransport):
 
732
 
 
733
    def test_no_branch(self):
 
734
        """When there is no branch, ('nobranch', ) is returned."""
 
735
        backing = self.get_transport()
 
736
        self.make_bzrdir('.')
 
737
        request = smart_dir.SmartServerRequestOpenBranchV3(backing)
 
738
        self.assertEqual(smart_req.SmartServerResponse(('nobranch',)),
 
739
            request.execute(''))
 
740
 
 
741
    def test_branch(self):
 
742
        """When there is a branch, 'ok' is returned."""
 
743
        backing = self.get_transport()
 
744
        expected = self.make_branch('.')._format.network_name()
 
745
        request = smart_dir.SmartServerRequestOpenBranchV3(backing)
 
746
        self.assertEqual(smart_req.SuccessfulSmartServerResponse(
 
747
                ('branch', expected)),
 
748
                         request.execute(''))
 
749
 
 
750
    def test_branch_reference(self):
 
751
        """When there is a branch reference, the reference URL is returned."""
 
752
        self.vfs_transport_factory = test_server.LocalURLServer
 
753
        backing = self.get_transport()
 
754
        request = smart_dir.SmartServerRequestOpenBranchV3(backing)
 
755
        branch = self.make_branch('branch')
 
756
        checkout = branch.create_checkout('reference',lightweight=True)
 
757
        reference_url = _mod_branch.BranchReferenceFormat().get_reference(
 
758
            checkout.bzrdir)
 
759
        self.assertFileEqual(reference_url, 'reference/.bzr/branch/location')
 
760
        self.assertEqual(smart_req.SuccessfulSmartServerResponse(
 
761
                ('ref', reference_url)),
 
762
                         request.execute('reference'))
 
763
 
 
764
    def test_stacked_branch(self):
 
765
        """Opening a stacked branch does not open the stacked-on branch."""
 
766
        trunk = self.make_branch('trunk')
 
767
        feature = self.make_branch('feature')
 
768
        feature.set_stacked_on_url(trunk.base)
 
769
        opened_branches = []
 
770
        _mod_branch.Branch.hooks.install_named_hook(
 
771
            'open', opened_branches.append, None)
 
772
        backing = self.get_transport()
 
773
        request = smart_dir.SmartServerRequestOpenBranchV3(backing)
 
774
        request.setup_jail()
 
775
        try:
 
776
            response = request.execute('feature')
 
777
        finally:
 
778
            request.teardown_jail()
 
779
        expected_format = feature._format.network_name()
 
780
        self.assertEqual(smart_req.SuccessfulSmartServerResponse(
 
781
                ('branch', expected_format)),
 
782
                         response)
 
783
        self.assertLength(1, opened_branches)
 
784
 
 
785
    def test_notification_on_branch_from_repository(self):
 
786
        """When there is a repository, the error should return details."""
 
787
        backing = self.get_transport()
 
788
        request = smart_dir.SmartServerRequestOpenBranchV3(backing)
 
789
        repo = self.make_repository('.')
 
790
        self.assertEqual(smart_req.SmartServerResponse(
 
791
                ('nobranch', 'location is a repository')),
 
792
                         request.execute(''))
 
793
 
 
794
 
 
795
class TestSmartServerRequestRevisionHistory(tests.TestCaseWithMemoryTransport):
 
796
 
 
797
    def test_empty(self):
 
798
        """For an empty branch, the body is empty."""
 
799
        backing = self.get_transport()
 
800
        request = smart_branch.SmartServerRequestRevisionHistory(backing)
 
801
        self.make_branch('.')
 
802
        self.assertEqual(smart_req.SmartServerResponse(('ok', ), ''),
 
803
            request.execute(''))
 
804
 
 
805
    def test_not_empty(self):
 
806
        """For a non-empty branch, the body is empty."""
 
807
        backing = self.get_transport()
 
808
        request = smart_branch.SmartServerRequestRevisionHistory(backing)
 
809
        tree = self.make_branch_and_memory_tree('.')
 
810
        tree.lock_write()
 
811
        tree.add('')
 
812
        r1 = tree.commit('1st commit')
 
813
        r2 = tree.commit('2nd commit', rev_id=u'\xc8'.encode('utf-8'))
 
814
        tree.unlock()
 
815
        self.assertEqual(
 
816
            smart_req.SmartServerResponse(('ok', ), ('\x00'.join([r1, r2]))),
 
817
            request.execute(''))
 
818
 
 
819
 
 
820
class TestSmartServerBranchRequest(tests.TestCaseWithMemoryTransport):
 
821
 
 
822
    def test_no_branch(self):
 
823
        """When there is a bzrdir and no branch, NotBranchError is raised."""
 
824
        backing = self.get_transport()
 
825
        request = smart_branch.SmartServerBranchRequest(backing)
 
826
        self.make_bzrdir('.')
 
827
        self.assertRaises(errors.NotBranchError,
 
828
            request.execute, '')
 
829
 
 
830
    def test_branch_reference(self):
 
831
        """When there is a branch reference, NotBranchError is raised."""
 
832
        backing = self.get_transport()
 
833
        request = smart_branch.SmartServerBranchRequest(backing)
 
834
        branch = self.make_branch('branch')
 
835
        checkout = branch.create_checkout('reference',lightweight=True)
 
836
        self.assertRaises(errors.NotBranchError,
 
837
            request.execute, 'checkout')
 
838
 
 
839
 
 
840
class TestSmartServerBranchRequestLastRevisionInfo(
 
841
    tests.TestCaseWithMemoryTransport):
 
842
 
 
843
    def test_empty(self):
 
844
        """For an empty branch, the result is ('ok', '0', 'null:')."""
 
845
        backing = self.get_transport()
 
846
        request = smart_branch.SmartServerBranchRequestLastRevisionInfo(backing)
 
847
        self.make_branch('.')
 
848
        self.assertEqual(smart_req.SmartServerResponse(('ok', '0', 'null:')),
 
849
            request.execute(''))
 
850
 
 
851
    def test_not_empty(self):
 
852
        """For a non-empty branch, the result is ('ok', 'revno', 'revid')."""
 
853
        backing = self.get_transport()
 
854
        request = smart_branch.SmartServerBranchRequestLastRevisionInfo(backing)
 
855
        tree = self.make_branch_and_memory_tree('.')
 
856
        tree.lock_write()
 
857
        tree.add('')
 
858
        rev_id_utf8 = u'\xc8'.encode('utf-8')
 
859
        r1 = tree.commit('1st commit')
 
860
        r2 = tree.commit('2nd commit', rev_id=rev_id_utf8)
 
861
        tree.unlock()
 
862
        self.assertEqual(
 
863
            smart_req.SmartServerResponse(('ok', '2', rev_id_utf8)),
 
864
            request.execute(''))
 
865
 
 
866
 
 
867
class TestSmartServerBranchRequestRevisionIdToRevno(
 
868
    tests.TestCaseWithMemoryTransport):
 
869
 
 
870
    def test_null(self):
 
871
        backing = self.get_transport()
 
872
        request = smart_branch.SmartServerBranchRequestRevisionIdToRevno(
 
873
            backing)
 
874
        self.make_branch('.')
 
875
        self.assertEqual(smart_req.SmartServerResponse(('ok', '0')),
 
876
            request.execute('', 'null:'))
 
877
 
 
878
    def test_simple(self):
 
879
        backing = self.get_transport()
 
880
        request = smart_branch.SmartServerBranchRequestRevisionIdToRevno(
 
881
            backing)
 
882
        tree = self.make_branch_and_memory_tree('.')
 
883
        tree.lock_write()
 
884
        tree.add('')
 
885
        r1 = tree.commit('1st commit')
 
886
        tree.unlock()
 
887
        self.assertEqual(
 
888
            smart_req.SmartServerResponse(('ok', '1')),
 
889
            request.execute('', r1))
 
890
 
 
891
    def test_not_found(self):
 
892
        backing = self.get_transport()
 
893
        request = smart_branch.SmartServerBranchRequestRevisionIdToRevno(
 
894
            backing)
 
895
        branch = self.make_branch('.')
 
896
        self.assertEqual(
 
897
            smart_req.FailedSmartServerResponse(
 
898
                ('NoSuchRevision', 'idontexist')),
 
899
            request.execute('', 'idontexist'))
 
900
 
 
901
 
 
902
class TestSmartServerBranchRequestGetConfigFile(
 
903
    tests.TestCaseWithMemoryTransport):
 
904
 
 
905
    def test_default(self):
 
906
        """With no file, we get empty content."""
 
907
        backing = self.get_transport()
 
908
        request = smart_branch.SmartServerBranchGetConfigFile(backing)
 
909
        branch = self.make_branch('.')
 
910
        # there should be no file by default
 
911
        content = ''
 
912
        self.assertEqual(smart_req.SmartServerResponse(('ok', ), content),
 
913
            request.execute(''))
 
914
 
 
915
    def test_with_content(self):
 
916
        # SmartServerBranchGetConfigFile should return the content from
 
917
        # branch.control_files.get('branch.conf') for now - in the future it may
 
918
        # perform more complex processing.
 
919
        backing = self.get_transport()
 
920
        request = smart_branch.SmartServerBranchGetConfigFile(backing)
 
921
        branch = self.make_branch('.')
 
922
        branch._transport.put_bytes('branch.conf', 'foo bar baz')
 
923
        self.assertEqual(smart_req.SmartServerResponse(('ok', ), 'foo bar baz'),
 
924
            request.execute(''))
 
925
 
 
926
 
 
927
class TestLockedBranch(tests.TestCaseWithMemoryTransport):
 
928
 
 
929
    def get_lock_tokens(self, branch):
 
930
        branch_token = branch.lock_write().branch_token
 
931
        repo_token = branch.repository.lock_write().repository_token
 
932
        branch.repository.unlock()
 
933
        return branch_token, repo_token
 
934
 
 
935
 
 
936
class TestSmartServerBranchRequestPutConfigFile(TestLockedBranch):
 
937
 
 
938
    def test_with_content(self):
 
939
        backing = self.get_transport()
 
940
        request = smart_branch.SmartServerBranchPutConfigFile(backing)
 
941
        branch = self.make_branch('.')
 
942
        branch_token, repo_token = self.get_lock_tokens(branch)
 
943
        self.assertIs(None, request.execute('', branch_token, repo_token))
 
944
        self.assertEqual(
 
945
            smart_req.SmartServerResponse(('ok', )),
 
946
            request.do_body('foo bar baz'))
 
947
        self.assertEquals(
 
948
            branch.control_transport.get_bytes('branch.conf'),
 
949
            'foo bar baz')
 
950
        branch.unlock()
 
951
 
 
952
 
 
953
class TestSmartServerBranchRequestSetConfigOption(TestLockedBranch):
 
954
 
 
955
    def test_value_name(self):
 
956
        branch = self.make_branch('.')
 
957
        request = smart_branch.SmartServerBranchRequestSetConfigOption(
 
958
            branch.bzrdir.root_transport)
 
959
        branch_token, repo_token = self.get_lock_tokens(branch)
 
960
        config = branch._get_config()
 
961
        result = request.execute('', branch_token, repo_token, 'bar', 'foo',
 
962
            '')
 
963
        self.assertEqual(smart_req.SuccessfulSmartServerResponse(()), result)
 
964
        self.assertEqual('bar', config.get_option('foo'))
 
965
        # Cleanup
 
966
        branch.unlock()
 
967
 
 
968
    def test_value_name_section(self):
 
969
        branch = self.make_branch('.')
 
970
        request = smart_branch.SmartServerBranchRequestSetConfigOption(
 
971
            branch.bzrdir.root_transport)
 
972
        branch_token, repo_token = self.get_lock_tokens(branch)
 
973
        config = branch._get_config()
 
974
        result = request.execute('', branch_token, repo_token, 'bar', 'foo',
 
975
            'gam')
 
976
        self.assertEqual(smart_req.SuccessfulSmartServerResponse(()), result)
 
977
        self.assertEqual('bar', config.get_option('foo', 'gam'))
 
978
        # Cleanup
 
979
        branch.unlock()
 
980
 
 
981
 
 
982
class TestSmartServerBranchRequestSetConfigOptionDict(TestLockedBranch):
 
983
 
 
984
    def setUp(self):
 
985
        TestLockedBranch.setUp(self)
 
986
        # A dict with non-ascii keys and values to exercise unicode
 
987
        # roundtripping.
 
988
        self.encoded_value_dict = (
 
989
            'd5:ascii1:a11:unicode \xe2\x8c\x9a3:\xe2\x80\xbde')
 
990
        self.value_dict = {
 
991
            'ascii': 'a', u'unicode \N{WATCH}': u'\N{INTERROBANG}'}
 
992
 
 
993
    def test_value_name(self):
 
994
        branch = self.make_branch('.')
 
995
        request = smart_branch.SmartServerBranchRequestSetConfigOptionDict(
 
996
            branch.bzrdir.root_transport)
 
997
        branch_token, repo_token = self.get_lock_tokens(branch)
 
998
        config = branch._get_config()
 
999
        result = request.execute('', branch_token, repo_token,
 
1000
            self.encoded_value_dict, 'foo', '')
 
1001
        self.assertEqual(smart_req.SuccessfulSmartServerResponse(()), result)
 
1002
        self.assertEqual(self.value_dict, config.get_option('foo'))
 
1003
        # Cleanup
 
1004
        branch.unlock()
 
1005
 
 
1006
    def test_value_name_section(self):
 
1007
        branch = self.make_branch('.')
 
1008
        request = smart_branch.SmartServerBranchRequestSetConfigOptionDict(
 
1009
            branch.bzrdir.root_transport)
 
1010
        branch_token, repo_token = self.get_lock_tokens(branch)
 
1011
        config = branch._get_config()
 
1012
        result = request.execute('', branch_token, repo_token,
 
1013
            self.encoded_value_dict, 'foo', 'gam')
 
1014
        self.assertEqual(smart_req.SuccessfulSmartServerResponse(()), result)
 
1015
        self.assertEqual(self.value_dict, config.get_option('foo', 'gam'))
 
1016
        # Cleanup
 
1017
        branch.unlock()
 
1018
 
 
1019
 
 
1020
class TestSmartServerBranchRequestSetTagsBytes(TestLockedBranch):
 
1021
    # Only called when the branch format and tags match [yay factory
 
1022
    # methods] so only need to test straight forward cases.
 
1023
 
 
1024
    def test_set_bytes(self):
 
1025
        base_branch = self.make_branch('base')
 
1026
        tag_bytes = base_branch._get_tags_bytes()
 
1027
        # get_lock_tokens takes out a lock.
 
1028
        branch_token, repo_token = self.get_lock_tokens(base_branch)
 
1029
        request = smart_branch.SmartServerBranchSetTagsBytes(
 
1030
            self.get_transport())
 
1031
        response = request.execute('base', branch_token, repo_token)
 
1032
        self.assertEqual(None, response)
 
1033
        response = request.do_chunk(tag_bytes)
 
1034
        self.assertEqual(None, response)
 
1035
        response = request.do_end()
 
1036
        self.assertEquals(
 
1037
            smart_req.SuccessfulSmartServerResponse(()), response)
 
1038
        base_branch.unlock()
 
1039
 
 
1040
    def test_lock_failed(self):
 
1041
        base_branch = self.make_branch('base')
 
1042
        base_branch.lock_write()
 
1043
        tag_bytes = base_branch._get_tags_bytes()
 
1044
        request = smart_branch.SmartServerBranchSetTagsBytes(
 
1045
            self.get_transport())
 
1046
        self.assertRaises(errors.TokenMismatch, request.execute,
 
1047
            'base', 'wrong token', 'wrong token')
 
1048
        # The request handler will keep processing the message parts, so even
 
1049
        # if the request fails immediately do_chunk and do_end are still
 
1050
        # called.
 
1051
        request.do_chunk(tag_bytes)
 
1052
        request.do_end()
 
1053
        base_branch.unlock()
 
1054
 
 
1055
 
 
1056
 
 
1057
class SetLastRevisionTestBase(TestLockedBranch):
 
1058
    """Base test case for verbs that implement set_last_revision."""
 
1059
 
 
1060
    def setUp(self):
 
1061
        tests.TestCaseWithMemoryTransport.setUp(self)
 
1062
        backing_transport = self.get_transport()
 
1063
        self.request = self.request_class(backing_transport)
 
1064
        self.tree = self.make_branch_and_memory_tree('.')
 
1065
 
 
1066
    def lock_branch(self):
 
1067
        return self.get_lock_tokens(self.tree.branch)
 
1068
 
 
1069
    def unlock_branch(self):
 
1070
        self.tree.branch.unlock()
 
1071
 
 
1072
    def set_last_revision(self, revision_id, revno):
 
1073
        branch_token, repo_token = self.lock_branch()
 
1074
        response = self._set_last_revision(
 
1075
            revision_id, revno, branch_token, repo_token)
 
1076
        self.unlock_branch()
 
1077
        return response
 
1078
 
 
1079
    def assertRequestSucceeds(self, revision_id, revno):
 
1080
        response = self.set_last_revision(revision_id, revno)
 
1081
        self.assertEqual(smart_req.SuccessfulSmartServerResponse(('ok',)),
 
1082
                         response)
 
1083
 
 
1084
 
 
1085
class TestSetLastRevisionVerbMixin(object):
 
1086
    """Mixin test case for verbs that implement set_last_revision."""
 
1087
 
 
1088
    def test_set_null_to_null(self):
 
1089
        """An empty branch can have its last revision set to 'null:'."""
 
1090
        self.assertRequestSucceeds('null:', 0)
 
1091
 
 
1092
    def test_NoSuchRevision(self):
 
1093
        """If the revision_id is not present, the verb returns NoSuchRevision.
 
1094
        """
 
1095
        revision_id = 'non-existent revision'
 
1096
        self.assertEqual(smart_req.FailedSmartServerResponse(('NoSuchRevision',
 
1097
                                                              revision_id)),
 
1098
                         self.set_last_revision(revision_id, 1))
 
1099
 
 
1100
    def make_tree_with_two_commits(self):
 
1101
        self.tree.lock_write()
 
1102
        self.tree.add('')
 
1103
        rev_id_utf8 = u'\xc8'.encode('utf-8')
 
1104
        r1 = self.tree.commit('1st commit', rev_id=rev_id_utf8)
 
1105
        r2 = self.tree.commit('2nd commit', rev_id='rev-2')
 
1106
        self.tree.unlock()
 
1107
 
 
1108
    def test_branch_last_revision_info_is_updated(self):
 
1109
        """A branch's tip can be set to a revision that is present in its
 
1110
        repository.
 
1111
        """
 
1112
        # Make a branch with an empty revision history, but two revisions in
 
1113
        # its repository.
 
1114
        self.make_tree_with_two_commits()
 
1115
        rev_id_utf8 = u'\xc8'.encode('utf-8')
 
1116
        self.tree.branch.set_last_revision_info(0, 'null:')
 
1117
        self.assertEqual(
 
1118
            (0, 'null:'), self.tree.branch.last_revision_info())
 
1119
        # We can update the branch to a revision that is present in the
 
1120
        # repository.
 
1121
        self.assertRequestSucceeds(rev_id_utf8, 1)
 
1122
        self.assertEqual(
 
1123
            (1, rev_id_utf8), self.tree.branch.last_revision_info())
 
1124
 
 
1125
    def test_branch_last_revision_info_rewind(self):
 
1126
        """A branch's tip can be set to a revision that is an ancestor of the
 
1127
        current tip.
 
1128
        """
 
1129
        self.make_tree_with_two_commits()
 
1130
        rev_id_utf8 = u'\xc8'.encode('utf-8')
 
1131
        self.assertEqual(
 
1132
            (2, 'rev-2'), self.tree.branch.last_revision_info())
 
1133
        self.assertRequestSucceeds(rev_id_utf8, 1)
 
1134
        self.assertEqual(
 
1135
            (1, rev_id_utf8), self.tree.branch.last_revision_info())
 
1136
 
 
1137
    def test_TipChangeRejected(self):
 
1138
        """If a pre_change_branch_tip hook raises TipChangeRejected, the verb
 
1139
        returns TipChangeRejected.
 
1140
        """
 
1141
        rejection_message = u'rejection message\N{INTERROBANG}'
 
1142
        def hook_that_rejects(params):
 
1143
            raise errors.TipChangeRejected(rejection_message)
 
1144
        _mod_branch.Branch.hooks.install_named_hook(
 
1145
            'pre_change_branch_tip', hook_that_rejects, None)
 
1146
        self.assertEqual(
 
1147
            smart_req.FailedSmartServerResponse(
 
1148
                ('TipChangeRejected', rejection_message.encode('utf-8'))),
 
1149
            self.set_last_revision('null:', 0))
 
1150
 
 
1151
 
 
1152
class TestSmartServerBranchRequestSetLastRevision(
 
1153
        SetLastRevisionTestBase, TestSetLastRevisionVerbMixin):
 
1154
    """Tests for Branch.set_last_revision verb."""
 
1155
 
 
1156
    request_class = smart_branch.SmartServerBranchRequestSetLastRevision
 
1157
 
 
1158
    def _set_last_revision(self, revision_id, revno, branch_token, repo_token):
 
1159
        return self.request.execute(
 
1160
            '', branch_token, repo_token, revision_id)
 
1161
 
 
1162
 
 
1163
class TestSmartServerBranchRequestSetLastRevisionInfo(
 
1164
        SetLastRevisionTestBase, TestSetLastRevisionVerbMixin):
 
1165
    """Tests for Branch.set_last_revision_info verb."""
 
1166
 
 
1167
    request_class = smart_branch.SmartServerBranchRequestSetLastRevisionInfo
 
1168
 
 
1169
    def _set_last_revision(self, revision_id, revno, branch_token, repo_token):
 
1170
        return self.request.execute(
 
1171
            '', branch_token, repo_token, revno, revision_id)
 
1172
 
 
1173
    def test_NoSuchRevision(self):
 
1174
        """Branch.set_last_revision_info does not have to return
 
1175
        NoSuchRevision if the revision_id is absent.
 
1176
        """
 
1177
        raise tests.TestNotApplicable()
 
1178
 
 
1179
 
 
1180
class TestSmartServerBranchRequestSetLastRevisionEx(
 
1181
        SetLastRevisionTestBase, TestSetLastRevisionVerbMixin):
 
1182
    """Tests for Branch.set_last_revision_ex verb."""
 
1183
 
 
1184
    request_class = smart_branch.SmartServerBranchRequestSetLastRevisionEx
 
1185
 
 
1186
    def _set_last_revision(self, revision_id, revno, branch_token, repo_token):
 
1187
        return self.request.execute(
 
1188
            '', branch_token, repo_token, revision_id, 0, 0)
 
1189
 
 
1190
    def assertRequestSucceeds(self, revision_id, revno):
 
1191
        response = self.set_last_revision(revision_id, revno)
 
1192
        self.assertEqual(
 
1193
            smart_req.SuccessfulSmartServerResponse(('ok', revno, revision_id)),
 
1194
            response)
 
1195
 
 
1196
    def test_branch_last_revision_info_rewind(self):
 
1197
        """A branch's tip can be set to a revision that is an ancestor of the
 
1198
        current tip, but only if allow_overwrite_descendant is passed.
 
1199
        """
 
1200
        self.make_tree_with_two_commits()
 
1201
        rev_id_utf8 = u'\xc8'.encode('utf-8')
 
1202
        self.assertEqual(
 
1203
            (2, 'rev-2'), self.tree.branch.last_revision_info())
 
1204
        # If allow_overwrite_descendant flag is 0, then trying to set the tip
 
1205
        # to an older revision ID has no effect.
 
1206
        branch_token, repo_token = self.lock_branch()
 
1207
        response = self.request.execute(
 
1208
            '', branch_token, repo_token, rev_id_utf8, 0, 0)
 
1209
        self.assertEqual(
 
1210
            smart_req.SuccessfulSmartServerResponse(('ok', 2, 'rev-2')),
 
1211
            response)
 
1212
        self.assertEqual(
 
1213
            (2, 'rev-2'), self.tree.branch.last_revision_info())
 
1214
 
 
1215
        # If allow_overwrite_descendant flag is 1, then setting the tip to an
 
1216
        # ancestor works.
 
1217
        response = self.request.execute(
 
1218
            '', branch_token, repo_token, rev_id_utf8, 0, 1)
 
1219
        self.assertEqual(
 
1220
            smart_req.SuccessfulSmartServerResponse(('ok', 1, rev_id_utf8)),
 
1221
            response)
 
1222
        self.unlock_branch()
 
1223
        self.assertEqual(
 
1224
            (1, rev_id_utf8), self.tree.branch.last_revision_info())
 
1225
 
 
1226
    def make_branch_with_divergent_history(self):
 
1227
        """Make a branch with divergent history in its repo.
 
1228
 
 
1229
        The branch's tip will be 'child-2', and the repo will also contain
 
1230
        'child-1', which diverges from a common base revision.
 
1231
        """
 
1232
        self.tree.lock_write()
 
1233
        self.tree.add('')
 
1234
        r1 = self.tree.commit('1st commit')
 
1235
        revno_1, revid_1 = self.tree.branch.last_revision_info()
 
1236
        r2 = self.tree.commit('2nd commit', rev_id='child-1')
 
1237
        # Undo the second commit
 
1238
        self.tree.branch.set_last_revision_info(revno_1, revid_1)
 
1239
        self.tree.set_parent_ids([revid_1])
 
1240
        # Make a new second commit, child-2.  child-2 has diverged from
 
1241
        # child-1.
 
1242
        new_r2 = self.tree.commit('2nd commit', rev_id='child-2')
 
1243
        self.tree.unlock()
 
1244
 
 
1245
    def test_not_allow_diverged(self):
 
1246
        """If allow_diverged is not passed, then setting a divergent history
 
1247
        returns a Diverged error.
 
1248
        """
 
1249
        self.make_branch_with_divergent_history()
 
1250
        self.assertEqual(
 
1251
            smart_req.FailedSmartServerResponse(('Diverged',)),
 
1252
            self.set_last_revision('child-1', 2))
 
1253
        # The branch tip was not changed.
 
1254
        self.assertEqual('child-2', self.tree.branch.last_revision())
 
1255
 
 
1256
    def test_allow_diverged(self):
 
1257
        """If allow_diverged is passed, then setting a divergent history
 
1258
        succeeds.
 
1259
        """
 
1260
        self.make_branch_with_divergent_history()
 
1261
        branch_token, repo_token = self.lock_branch()
 
1262
        response = self.request.execute(
 
1263
            '', branch_token, repo_token, 'child-1', 1, 0)
 
1264
        self.assertEqual(
 
1265
            smart_req.SuccessfulSmartServerResponse(('ok', 2, 'child-1')),
 
1266
            response)
 
1267
        self.unlock_branch()
 
1268
        # The branch tip was changed.
 
1269
        self.assertEqual('child-1', self.tree.branch.last_revision())
 
1270
 
 
1271
 
 
1272
class TestSmartServerBranchBreakLock(tests.TestCaseWithMemoryTransport):
 
1273
 
 
1274
    def test_lock_to_break(self):
 
1275
        base_branch = self.make_branch('base')
 
1276
        request = smart_branch.SmartServerBranchBreakLock(
 
1277
            self.get_transport())
 
1278
        base_branch.lock_write()
 
1279
        self.assertEqual(
 
1280
            smart_req.SuccessfulSmartServerResponse(('ok', ), None),
 
1281
            request.execute('base'))
 
1282
 
 
1283
    def test_nothing_to_break(self):
 
1284
        base_branch = self.make_branch('base')
 
1285
        request = smart_branch.SmartServerBranchBreakLock(
 
1286
            self.get_transport())
 
1287
        self.assertEqual(
 
1288
            smart_req.SuccessfulSmartServerResponse(('ok', ), None),
 
1289
            request.execute('base'))
 
1290
 
 
1291
 
 
1292
class TestSmartServerBranchRequestGetParent(tests.TestCaseWithMemoryTransport):
 
1293
 
 
1294
    def test_get_parent_none(self):
 
1295
        base_branch = self.make_branch('base')
 
1296
        request = smart_branch.SmartServerBranchGetParent(self.get_transport())
 
1297
        response = request.execute('base')
 
1298
        self.assertEquals(
 
1299
            smart_req.SuccessfulSmartServerResponse(('',)), response)
 
1300
 
 
1301
    def test_get_parent_something(self):
 
1302
        base_branch = self.make_branch('base')
 
1303
        base_branch.set_parent(self.get_url('foo'))
 
1304
        request = smart_branch.SmartServerBranchGetParent(self.get_transport())
 
1305
        response = request.execute('base')
 
1306
        self.assertEquals(
 
1307
            smart_req.SuccessfulSmartServerResponse(("../foo",)),
 
1308
            response)
 
1309
 
 
1310
 
 
1311
class TestSmartServerBranchRequestSetParent(TestLockedBranch):
 
1312
 
 
1313
    def test_set_parent_none(self):
 
1314
        branch = self.make_branch('base', format="1.9")
 
1315
        branch.lock_write()
 
1316
        branch._set_parent_location('foo')
 
1317
        branch.unlock()
 
1318
        request = smart_branch.SmartServerBranchRequestSetParentLocation(
 
1319
            self.get_transport())
 
1320
        branch_token, repo_token = self.get_lock_tokens(branch)
 
1321
        try:
 
1322
            response = request.execute('base', branch_token, repo_token, '')
 
1323
        finally:
 
1324
            branch.unlock()
 
1325
        self.assertEqual(smart_req.SuccessfulSmartServerResponse(()), response)
 
1326
        # Refresh branch as SetParentLocation modified it
 
1327
        branch = branch.bzrdir.open_branch()
 
1328
        self.assertEqual(None, branch.get_parent())
 
1329
 
 
1330
    def test_set_parent_something(self):
 
1331
        branch = self.make_branch('base', format="1.9")
 
1332
        request = smart_branch.SmartServerBranchRequestSetParentLocation(
 
1333
            self.get_transport())
 
1334
        branch_token, repo_token = self.get_lock_tokens(branch)
 
1335
        try:
 
1336
            response = request.execute('base', branch_token, repo_token,
 
1337
                                       'http://bar/')
 
1338
        finally:
 
1339
            branch.unlock()
 
1340
        self.assertEqual(smart_req.SuccessfulSmartServerResponse(()), response)
 
1341
        refreshed = _mod_branch.Branch.open(branch.base)
 
1342
        self.assertEqual('http://bar/', refreshed.get_parent())
 
1343
 
 
1344
 
 
1345
class TestSmartServerBranchRequestGetTagsBytes(
 
1346
    tests.TestCaseWithMemoryTransport):
 
1347
    # Only called when the branch format and tags match [yay factory
 
1348
    # methods] so only need to test straight forward cases.
 
1349
 
 
1350
    def test_get_bytes(self):
 
1351
        base_branch = self.make_branch('base')
 
1352
        request = smart_branch.SmartServerBranchGetTagsBytes(
 
1353
            self.get_transport())
 
1354
        response = request.execute('base')
 
1355
        self.assertEquals(
 
1356
            smart_req.SuccessfulSmartServerResponse(('',)), response)
 
1357
 
 
1358
 
 
1359
class TestSmartServerBranchRequestGetStackedOnURL(tests.TestCaseWithMemoryTransport):
 
1360
 
 
1361
    def test_get_stacked_on_url(self):
 
1362
        base_branch = self.make_branch('base', format='1.6')
 
1363
        stacked_branch = self.make_branch('stacked', format='1.6')
 
1364
        # typically should be relative
 
1365
        stacked_branch.set_stacked_on_url('../base')
 
1366
        request = smart_branch.SmartServerBranchRequestGetStackedOnURL(
 
1367
            self.get_transport())
 
1368
        response = request.execute('stacked')
 
1369
        self.assertEquals(
 
1370
            smart_req.SmartServerResponse(('ok', '../base')),
 
1371
            response)
 
1372
 
 
1373
 
 
1374
class TestSmartServerBranchRequestLockWrite(TestLockedBranch):
 
1375
 
 
1376
    def setUp(self):
 
1377
        tests.TestCaseWithMemoryTransport.setUp(self)
 
1378
 
 
1379
    def test_lock_write_on_unlocked_branch(self):
 
1380
        backing = self.get_transport()
 
1381
        request = smart_branch.SmartServerBranchRequestLockWrite(backing)
 
1382
        branch = self.make_branch('.', format='knit')
 
1383
        repository = branch.repository
 
1384
        response = request.execute('')
 
1385
        branch_nonce = branch.control_files._lock.peek().get('nonce')
 
1386
        repository_nonce = repository.control_files._lock.peek().get('nonce')
 
1387
        self.assertEqual(smart_req.SmartServerResponse(
 
1388
                ('ok', branch_nonce, repository_nonce)),
 
1389
                         response)
 
1390
        # The branch (and associated repository) is now locked.  Verify that
 
1391
        # with a new branch object.
 
1392
        new_branch = repository.bzrdir.open_branch()
 
1393
        self.assertRaises(errors.LockContention, new_branch.lock_write)
 
1394
        # Cleanup
 
1395
        request = smart_branch.SmartServerBranchRequestUnlock(backing)
 
1396
        response = request.execute('', branch_nonce, repository_nonce)
 
1397
 
 
1398
    def test_lock_write_on_locked_branch(self):
 
1399
        backing = self.get_transport()
 
1400
        request = smart_branch.SmartServerBranchRequestLockWrite(backing)
 
1401
        branch = self.make_branch('.')
 
1402
        branch_token = branch.lock_write().branch_token
 
1403
        branch.leave_lock_in_place()
 
1404
        branch.unlock()
 
1405
        response = request.execute('')
 
1406
        self.assertEqual(
 
1407
            smart_req.SmartServerResponse(('LockContention',)), response)
 
1408
        # Cleanup
 
1409
        branch.lock_write(branch_token)
 
1410
        branch.dont_leave_lock_in_place()
 
1411
        branch.unlock()
 
1412
 
 
1413
    def test_lock_write_with_tokens_on_locked_branch(self):
 
1414
        backing = self.get_transport()
 
1415
        request = smart_branch.SmartServerBranchRequestLockWrite(backing)
 
1416
        branch = self.make_branch('.', format='knit')
 
1417
        branch_token, repo_token = self.get_lock_tokens(branch)
 
1418
        branch.leave_lock_in_place()
 
1419
        branch.repository.leave_lock_in_place()
 
1420
        branch.unlock()
 
1421
        response = request.execute('',
 
1422
                                   branch_token, repo_token)
 
1423
        self.assertEqual(
 
1424
            smart_req.SmartServerResponse(('ok', branch_token, repo_token)),
 
1425
            response)
 
1426
        # Cleanup
 
1427
        branch.repository.lock_write(repo_token)
 
1428
        branch.repository.dont_leave_lock_in_place()
 
1429
        branch.repository.unlock()
 
1430
        branch.lock_write(branch_token)
 
1431
        branch.dont_leave_lock_in_place()
 
1432
        branch.unlock()
 
1433
 
 
1434
    def test_lock_write_with_mismatched_tokens_on_locked_branch(self):
 
1435
        backing = self.get_transport()
 
1436
        request = smart_branch.SmartServerBranchRequestLockWrite(backing)
 
1437
        branch = self.make_branch('.', format='knit')
 
1438
        branch_token, repo_token = self.get_lock_tokens(branch)
 
1439
        branch.leave_lock_in_place()
 
1440
        branch.repository.leave_lock_in_place()
 
1441
        branch.unlock()
 
1442
        response = request.execute('',
 
1443
                                   branch_token+'xxx', repo_token)
 
1444
        self.assertEqual(
 
1445
            smart_req.SmartServerResponse(('TokenMismatch',)), response)
 
1446
        # Cleanup
 
1447
        branch.repository.lock_write(repo_token)
 
1448
        branch.repository.dont_leave_lock_in_place()
 
1449
        branch.repository.unlock()
 
1450
        branch.lock_write(branch_token)
 
1451
        branch.dont_leave_lock_in_place()
 
1452
        branch.unlock()
 
1453
 
 
1454
    def test_lock_write_on_locked_repo(self):
 
1455
        backing = self.get_transport()
 
1456
        request = smart_branch.SmartServerBranchRequestLockWrite(backing)
 
1457
        branch = self.make_branch('.', format='knit')
 
1458
        repo = branch.repository
 
1459
        repo_token = repo.lock_write().repository_token
 
1460
        repo.leave_lock_in_place()
 
1461
        repo.unlock()
 
1462
        response = request.execute('')
 
1463
        self.assertEqual(
 
1464
            smart_req.SmartServerResponse(('LockContention',)), response)
 
1465
        # Cleanup
 
1466
        repo.lock_write(repo_token)
 
1467
        repo.dont_leave_lock_in_place()
 
1468
        repo.unlock()
 
1469
 
 
1470
    def test_lock_write_on_readonly_transport(self):
 
1471
        backing = self.get_readonly_transport()
 
1472
        request = smart_branch.SmartServerBranchRequestLockWrite(backing)
 
1473
        branch = self.make_branch('.')
 
1474
        root = self.get_transport().clone('/')
 
1475
        path = urlutils.relative_url(root.base, self.get_transport().base)
 
1476
        response = request.execute(path)
 
1477
        error_name, lock_str, why_str = response.args
 
1478
        self.assertFalse(response.is_successful())
 
1479
        self.assertEqual('LockFailed', error_name)
 
1480
 
 
1481
 
 
1482
class TestSmartServerBranchRequestGetPhysicalLockStatus(TestLockedBranch):
 
1483
 
 
1484
    def setUp(self):
 
1485
        tests.TestCaseWithMemoryTransport.setUp(self)
 
1486
 
 
1487
    def test_true(self):
 
1488
        backing = self.get_transport()
 
1489
        request = smart_branch.SmartServerBranchRequestGetPhysicalLockStatus(
 
1490
            backing)
 
1491
        branch = self.make_branch('.')
 
1492
        branch_token, repo_token = self.get_lock_tokens(branch)
 
1493
        self.assertEquals(True, branch.get_physical_lock_status())
 
1494
        response = request.execute('')
 
1495
        self.assertEqual(
 
1496
            smart_req.SmartServerResponse(('yes',)), response)
 
1497
        branch.unlock()
 
1498
 
 
1499
    def test_false(self):
 
1500
        backing = self.get_transport()
 
1501
        request = smart_branch.SmartServerBranchRequestGetPhysicalLockStatus(
 
1502
            backing)
 
1503
        branch = self.make_branch('.')
 
1504
        self.assertEquals(False, branch.get_physical_lock_status())
 
1505
        response = request.execute('')
 
1506
        self.assertEqual(
 
1507
            smart_req.SmartServerResponse(('no',)), response)
 
1508
 
 
1509
 
 
1510
class TestSmartServerBranchRequestUnlock(TestLockedBranch):
 
1511
 
 
1512
    def setUp(self):
 
1513
        tests.TestCaseWithMemoryTransport.setUp(self)
 
1514
 
 
1515
    def test_unlock_on_locked_branch_and_repo(self):
 
1516
        backing = self.get_transport()
 
1517
        request = smart_branch.SmartServerBranchRequestUnlock(backing)
 
1518
        branch = self.make_branch('.', format='knit')
 
1519
        # Lock the branch
 
1520
        branch_token, repo_token = self.get_lock_tokens(branch)
 
1521
        # Unlock the branch (and repo) object, leaving the physical locks
 
1522
        # in place.
 
1523
        branch.leave_lock_in_place()
 
1524
        branch.repository.leave_lock_in_place()
 
1525
        branch.unlock()
 
1526
        response = request.execute('',
 
1527
                                   branch_token, repo_token)
 
1528
        self.assertEqual(
 
1529
            smart_req.SmartServerResponse(('ok',)), response)
 
1530
        # The branch is now unlocked.  Verify that with a new branch
 
1531
        # object.
 
1532
        new_branch = branch.bzrdir.open_branch()
 
1533
        new_branch.lock_write()
 
1534
        new_branch.unlock()
 
1535
 
 
1536
    def test_unlock_on_unlocked_branch_unlocked_repo(self):
 
1537
        backing = self.get_transport()
 
1538
        request = smart_branch.SmartServerBranchRequestUnlock(backing)
 
1539
        branch = self.make_branch('.', format='knit')
 
1540
        response = request.execute(
 
1541
            '', 'branch token', 'repo token')
 
1542
        self.assertEqual(
 
1543
            smart_req.SmartServerResponse(('TokenMismatch',)), response)
 
1544
 
 
1545
    def test_unlock_on_unlocked_branch_locked_repo(self):
 
1546
        backing = self.get_transport()
 
1547
        request = smart_branch.SmartServerBranchRequestUnlock(backing)
 
1548
        branch = self.make_branch('.', format='knit')
 
1549
        # Lock the repository.
 
1550
        repo_token = branch.repository.lock_write().repository_token
 
1551
        branch.repository.leave_lock_in_place()
 
1552
        branch.repository.unlock()
 
1553
        # Issue branch lock_write request on the unlocked branch (with locked
 
1554
        # repo).
 
1555
        response = request.execute('', 'branch token', repo_token)
 
1556
        self.assertEqual(
 
1557
            smart_req.SmartServerResponse(('TokenMismatch',)), response)
 
1558
        # Cleanup
 
1559
        branch.repository.lock_write(repo_token)
 
1560
        branch.repository.dont_leave_lock_in_place()
 
1561
        branch.repository.unlock()
 
1562
 
 
1563
 
 
1564
class TestSmartServerRepositoryRequest(tests.TestCaseWithMemoryTransport):
 
1565
 
 
1566
    def test_no_repository(self):
 
1567
        """Raise NoRepositoryPresent when there is a bzrdir and no repo."""
 
1568
        # we test this using a shared repository above the named path,
 
1569
        # thus checking the right search logic is used - that is, that
 
1570
        # its the exact path being looked at and the server is not
 
1571
        # searching.
 
1572
        backing = self.get_transport()
 
1573
        request = smart_repo.SmartServerRepositoryRequest(backing)
 
1574
        self.make_repository('.', shared=True)
 
1575
        self.make_bzrdir('subdir')
 
1576
        self.assertRaises(errors.NoRepositoryPresent,
 
1577
            request.execute, 'subdir')
 
1578
 
 
1579
 
 
1580
class TestSmartServerRepositoryAddSignatureText(tests.TestCaseWithMemoryTransport):
 
1581
 
 
1582
    def test_add_text(self):
 
1583
        backing = self.get_transport()
 
1584
        request = smart_repo.SmartServerRepositoryAddSignatureText(backing)
 
1585
        tree = self.make_branch_and_memory_tree('.')
 
1586
        write_token = tree.lock_write()
 
1587
        self.addCleanup(tree.unlock)
 
1588
        tree.add('')
 
1589
        tree.commit("Message", rev_id='rev1')
 
1590
        tree.branch.repository.start_write_group()
 
1591
        write_group_tokens = tree.branch.repository.suspend_write_group()
 
1592
        self.assertEqual(None, request.execute('', write_token,
 
1593
            'rev1', *write_group_tokens))
 
1594
        response = request.do_body('somesignature')
 
1595
        self.assertTrue(response.is_successful())
 
1596
        self.assertEqual(response.args[0], 'ok')
 
1597
        write_group_tokens = response.args[1:]
 
1598
        tree.branch.repository.resume_write_group(write_group_tokens)
 
1599
        tree.branch.repository.commit_write_group()
 
1600
        tree.unlock()
 
1601
        self.assertEqual("somesignature",
 
1602
            tree.branch.repository.get_signature_text("rev1"))
 
1603
 
 
1604
 
 
1605
class TestSmartServerRepositoryAllRevisionIds(
 
1606
    tests.TestCaseWithMemoryTransport):
 
1607
 
 
1608
    def test_empty(self):
 
1609
        """An empty body should be returned for an empty repository."""
 
1610
        backing = self.get_transport()
 
1611
        request = smart_repo.SmartServerRepositoryAllRevisionIds(backing)
 
1612
        self.make_repository('.')
 
1613
        self.assertEquals(
 
1614
            smart_req.SuccessfulSmartServerResponse(("ok", ), ""),
 
1615
            request.execute(''))
 
1616
 
 
1617
    def test_some_revisions(self):
 
1618
        """An empty body should be returned for an empty repository."""
 
1619
        backing = self.get_transport()
 
1620
        request = smart_repo.SmartServerRepositoryAllRevisionIds(backing)
 
1621
        tree = self.make_branch_and_memory_tree('.')
 
1622
        tree.lock_write()
 
1623
        tree.add('')
 
1624
        tree.commit(rev_id='origineel', message="message")
 
1625
        tree.commit(rev_id='nog-een-revisie', message="message")
 
1626
        tree.unlock()
 
1627
        self.assertEquals(
 
1628
            smart_req.SuccessfulSmartServerResponse(("ok", ),
 
1629
                "origineel\nnog-een-revisie"),
 
1630
            request.execute(''))
 
1631
 
 
1632
 
 
1633
class TestSmartServerRepositoryBreakLock(tests.TestCaseWithMemoryTransport):
 
1634
 
 
1635
    def test_lock_to_break(self):
 
1636
        backing = self.get_transport()
 
1637
        request = smart_repo.SmartServerRepositoryBreakLock(backing)
 
1638
        tree = self.make_branch_and_memory_tree('.')
 
1639
        tree.branch.repository.lock_write()
 
1640
        self.assertEqual(
 
1641
            smart_req.SuccessfulSmartServerResponse(('ok', ), None),
 
1642
            request.execute(''))
 
1643
 
 
1644
    def test_nothing_to_break(self):
 
1645
        backing = self.get_transport()
 
1646
        request = smart_repo.SmartServerRepositoryBreakLock(backing)
 
1647
        tree = self.make_branch_and_memory_tree('.')
 
1648
        self.assertEqual(
 
1649
            smart_req.SuccessfulSmartServerResponse(('ok', ), None),
 
1650
            request.execute(''))
 
1651
 
 
1652
 
 
1653
class TestSmartServerRepositoryGetParentMap(tests.TestCaseWithMemoryTransport):
 
1654
 
 
1655
    def test_trivial_bzipped(self):
 
1656
        # This tests that the wire encoding is actually bzipped
 
1657
        backing = self.get_transport()
 
1658
        request = smart_repo.SmartServerRepositoryGetParentMap(backing)
 
1659
        tree = self.make_branch_and_memory_tree('.')
 
1660
 
 
1661
        self.assertEqual(None,
 
1662
            request.execute('', 'missing-id'))
 
1663
        # Note that it returns a body that is bzipped.
 
1664
        self.assertEqual(
 
1665
            smart_req.SuccessfulSmartServerResponse(('ok', ), bz2.compress('')),
 
1666
            request.do_body('\n\n0\n'))
 
1667
 
 
1668
    def test_trivial_include_missing(self):
 
1669
        backing = self.get_transport()
 
1670
        request = smart_repo.SmartServerRepositoryGetParentMap(backing)
 
1671
        tree = self.make_branch_and_memory_tree('.')
 
1672
 
 
1673
        self.assertEqual(None,
 
1674
            request.execute('', 'missing-id', 'include-missing:'))
 
1675
        self.assertEqual(
 
1676
            smart_req.SuccessfulSmartServerResponse(('ok', ),
 
1677
                bz2.compress('missing:missing-id')),
 
1678
            request.do_body('\n\n0\n'))
 
1679
 
 
1680
 
 
1681
class TestSmartServerRepositoryGetRevisionGraph(
 
1682
    tests.TestCaseWithMemoryTransport):
 
1683
 
 
1684
    def test_none_argument(self):
 
1685
        backing = self.get_transport()
 
1686
        request = smart_repo.SmartServerRepositoryGetRevisionGraph(backing)
 
1687
        tree = self.make_branch_and_memory_tree('.')
 
1688
        tree.lock_write()
 
1689
        tree.add('')
 
1690
        r1 = tree.commit('1st commit')
 
1691
        r2 = tree.commit('2nd commit', rev_id=u'\xc8'.encode('utf-8'))
 
1692
        tree.unlock()
 
1693
 
 
1694
        # the lines of revision_id->revision_parent_list has no guaranteed
 
1695
        # order coming out of a dict, so sort both our test and response
 
1696
        lines = sorted([' '.join([r2, r1]), r1])
 
1697
        response = request.execute('', '')
 
1698
        response.body = '\n'.join(sorted(response.body.split('\n')))
 
1699
 
 
1700
        self.assertEqual(
 
1701
            smart_req.SmartServerResponse(('ok', ), '\n'.join(lines)), response)
 
1702
 
 
1703
    def test_specific_revision_argument(self):
 
1704
        backing = self.get_transport()
 
1705
        request = smart_repo.SmartServerRepositoryGetRevisionGraph(backing)
 
1706
        tree = self.make_branch_and_memory_tree('.')
 
1707
        tree.lock_write()
 
1708
        tree.add('')
 
1709
        rev_id_utf8 = u'\xc9'.encode('utf-8')
 
1710
        r1 = tree.commit('1st commit', rev_id=rev_id_utf8)
 
1711
        r2 = tree.commit('2nd commit', rev_id=u'\xc8'.encode('utf-8'))
 
1712
        tree.unlock()
 
1713
 
 
1714
        self.assertEqual(smart_req.SmartServerResponse(('ok', ), rev_id_utf8),
 
1715
            request.execute('', rev_id_utf8))
 
1716
 
 
1717
    def test_no_such_revision(self):
 
1718
        backing = self.get_transport()
 
1719
        request = smart_repo.SmartServerRepositoryGetRevisionGraph(backing)
 
1720
        tree = self.make_branch_and_memory_tree('.')
 
1721
        tree.lock_write()
 
1722
        tree.add('')
 
1723
        r1 = tree.commit('1st commit')
 
1724
        tree.unlock()
 
1725
 
 
1726
        # Note that it still returns body (of zero bytes).
 
1727
        self.assertEqual(smart_req.SmartServerResponse(
 
1728
                ('nosuchrevision', 'missingrevision', ), ''),
 
1729
                         request.execute('', 'missingrevision'))
 
1730
 
 
1731
 
 
1732
class TestSmartServerRepositoryGetRevIdForRevno(
 
1733
    tests.TestCaseWithMemoryTransport):
 
1734
 
 
1735
    def test_revno_found(self):
 
1736
        backing = self.get_transport()
 
1737
        request = smart_repo.SmartServerRepositoryGetRevIdForRevno(backing)
 
1738
        tree = self.make_branch_and_memory_tree('.')
 
1739
        tree.lock_write()
 
1740
        tree.add('')
 
1741
        rev1_id_utf8 = u'\xc8'.encode('utf-8')
 
1742
        rev2_id_utf8 = u'\xc9'.encode('utf-8')
 
1743
        tree.commit('1st commit', rev_id=rev1_id_utf8)
 
1744
        tree.commit('2nd commit', rev_id=rev2_id_utf8)
 
1745
        tree.unlock()
 
1746
 
 
1747
        self.assertEqual(smart_req.SmartServerResponse(('ok', rev1_id_utf8)),
 
1748
            request.execute('', 1, (2, rev2_id_utf8)))
 
1749
 
 
1750
    def test_known_revid_missing(self):
 
1751
        backing = self.get_transport()
 
1752
        request = smart_repo.SmartServerRepositoryGetRevIdForRevno(backing)
 
1753
        repo = self.make_repository('.')
 
1754
        self.assertEqual(
 
1755
            smart_req.FailedSmartServerResponse(('nosuchrevision', 'ghost')),
 
1756
            request.execute('', 1, (2, 'ghost')))
 
1757
 
 
1758
    def test_history_incomplete(self):
 
1759
        backing = self.get_transport()
 
1760
        request = smart_repo.SmartServerRepositoryGetRevIdForRevno(backing)
 
1761
        parent = self.make_branch_and_memory_tree('parent', format='1.9')
 
1762
        parent.lock_write()
 
1763
        parent.add([''], ['TREE_ROOT'])
 
1764
        r1 = parent.commit(message='first commit')
 
1765
        r2 = parent.commit(message='second commit')
 
1766
        parent.unlock()
 
1767
        local = self.make_branch_and_memory_tree('local', format='1.9')
 
1768
        local.branch.pull(parent.branch)
 
1769
        local.set_parent_ids([r2])
 
1770
        r3 = local.commit(message='local commit')
 
1771
        local.branch.create_clone_on_transport(
 
1772
            self.get_transport('stacked'), stacked_on=self.get_url('parent'))
 
1773
        self.assertEqual(
 
1774
            smart_req.SmartServerResponse(('history-incomplete', 2, r2)),
 
1775
            request.execute('stacked', 1, (3, r3)))
 
1776
 
 
1777
 
 
1778
class TestSmartServerRepositoryIterRevisions(
 
1779
    tests.TestCaseWithMemoryTransport):
 
1780
 
 
1781
    def test_basic(self):
 
1782
        backing = self.get_transport()
 
1783
        request = smart_repo.SmartServerRepositoryIterRevisions(backing)
 
1784
        tree = self.make_branch_and_memory_tree('.', format='2a')
 
1785
        tree.lock_write()
 
1786
        tree.add('')
 
1787
        tree.commit('1st commit', rev_id="rev1")
 
1788
        tree.commit('2nd commit', rev_id="rev2")
 
1789
        tree.unlock()
 
1790
 
 
1791
        self.assertIs(None, request.execute(''))
 
1792
        response = request.do_body("rev1\nrev2")
 
1793
        self.assertTrue(response.is_successful())
 
1794
        # Format 2a uses serializer format 10
 
1795
        self.assertEquals(response.args, ("ok", "10"))
 
1796
 
 
1797
        self.addCleanup(tree.branch.lock_read().unlock)
 
1798
        entries = [zlib.compress(record.get_bytes_as("fulltext")) for record in
 
1799
            tree.branch.repository.revisions.get_record_stream(
 
1800
            [("rev1", ), ("rev2", )], "unordered", True)]
 
1801
 
 
1802
        contents = "".join(response.body_stream)
 
1803
        self.assertTrue(contents in (
 
1804
            "".join([entries[0], entries[1]]),
 
1805
            "".join([entries[1], entries[0]])))
 
1806
 
 
1807
    def test_missing(self):
 
1808
        backing = self.get_transport()
 
1809
        request = smart_repo.SmartServerRepositoryIterRevisions(backing)
 
1810
        tree = self.make_branch_and_memory_tree('.', format='2a')
 
1811
 
 
1812
        self.assertIs(None, request.execute(''))
 
1813
        response = request.do_body("rev1\nrev2")
 
1814
        self.assertTrue(response.is_successful())
 
1815
        # Format 2a uses serializer format 10
 
1816
        self.assertEquals(response.args, ("ok", "10"))
 
1817
 
 
1818
        contents = "".join(response.body_stream)
 
1819
        self.assertEquals(contents, "")
 
1820
 
 
1821
 
 
1822
class GetStreamTestBase(tests.TestCaseWithMemoryTransport):
 
1823
 
 
1824
    def make_two_commit_repo(self):
 
1825
        tree = self.make_branch_and_memory_tree('.')
 
1826
        tree.lock_write()
 
1827
        tree.add('')
 
1828
        r1 = tree.commit('1st commit')
 
1829
        r2 = tree.commit('2nd commit', rev_id=u'\xc8'.encode('utf-8'))
 
1830
        tree.unlock()
 
1831
        repo = tree.branch.repository
 
1832
        return repo, r1, r2
 
1833
 
 
1834
 
 
1835
class TestSmartServerRepositoryGetStream(GetStreamTestBase):
 
1836
 
 
1837
    def test_ancestry_of(self):
 
1838
        """The search argument may be a 'ancestry-of' some heads'."""
 
1839
        backing = self.get_transport()
 
1840
        request = smart_repo.SmartServerRepositoryGetStream(backing)
 
1841
        repo, r1, r2 = self.make_two_commit_repo()
 
1842
        fetch_spec = ['ancestry-of', r2]
 
1843
        lines = '\n'.join(fetch_spec)
 
1844
        request.execute('', repo._format.network_name())
 
1845
        response = request.do_body(lines)
 
1846
        self.assertEqual(('ok',), response.args)
 
1847
        stream_bytes = ''.join(response.body_stream)
 
1848
        self.assertStartsWith(stream_bytes, 'Bazaar pack format 1')
 
1849
 
 
1850
    def test_search(self):
 
1851
        """The search argument may be a 'search' of some explicit keys."""
 
1852
        backing = self.get_transport()
 
1853
        request = smart_repo.SmartServerRepositoryGetStream(backing)
 
1854
        repo, r1, r2 = self.make_two_commit_repo()
 
1855
        fetch_spec = ['search', '%s %s' % (r1, r2), 'null:', '2']
 
1856
        lines = '\n'.join(fetch_spec)
 
1857
        request.execute('', repo._format.network_name())
 
1858
        response = request.do_body(lines)
 
1859
        self.assertEqual(('ok',), response.args)
 
1860
        stream_bytes = ''.join(response.body_stream)
 
1861
        self.assertStartsWith(stream_bytes, 'Bazaar pack format 1')
 
1862
 
 
1863
    def test_search_everything(self):
 
1864
        """A search of 'everything' returns a stream."""
 
1865
        backing = self.get_transport()
 
1866
        request = smart_repo.SmartServerRepositoryGetStream_1_19(backing)
 
1867
        repo, r1, r2 = self.make_two_commit_repo()
 
1868
        serialised_fetch_spec = 'everything'
 
1869
        request.execute('', repo._format.network_name())
 
1870
        response = request.do_body(serialised_fetch_spec)
 
1871
        self.assertEqual(('ok',), response.args)
 
1872
        stream_bytes = ''.join(response.body_stream)
 
1873
        self.assertStartsWith(stream_bytes, 'Bazaar pack format 1')
 
1874
 
 
1875
 
 
1876
class TestSmartServerRequestHasRevision(tests.TestCaseWithMemoryTransport):
 
1877
 
 
1878
    def test_missing_revision(self):
 
1879
        """For a missing revision, ('no', ) is returned."""
 
1880
        backing = self.get_transport()
 
1881
        request = smart_repo.SmartServerRequestHasRevision(backing)
 
1882
        self.make_repository('.')
 
1883
        self.assertEqual(smart_req.SmartServerResponse(('no', )),
 
1884
            request.execute('', 'revid'))
 
1885
 
 
1886
    def test_present_revision(self):
 
1887
        """For a present revision, ('yes', ) is returned."""
 
1888
        backing = self.get_transport()
 
1889
        request = smart_repo.SmartServerRequestHasRevision(backing)
 
1890
        tree = self.make_branch_and_memory_tree('.')
 
1891
        tree.lock_write()
 
1892
        tree.add('')
 
1893
        rev_id_utf8 = u'\xc8abc'.encode('utf-8')
 
1894
        r1 = tree.commit('a commit', rev_id=rev_id_utf8)
 
1895
        tree.unlock()
 
1896
        self.assertTrue(tree.branch.repository.has_revision(rev_id_utf8))
 
1897
        self.assertEqual(smart_req.SmartServerResponse(('yes', )),
 
1898
            request.execute('', rev_id_utf8))
 
1899
 
 
1900
 
 
1901
class TestSmartServerRepositoryIterFilesBytes(tests.TestCaseWithTransport):
 
1902
 
 
1903
    def test_single(self):
 
1904
        backing = self.get_transport()
 
1905
        request = smart_repo.SmartServerRepositoryIterFilesBytes(backing)
 
1906
        t = self.make_branch_and_tree('.')
 
1907
        self.addCleanup(t.lock_write().unlock)
 
1908
        self.build_tree_contents([("file", "somecontents")])
 
1909
        t.add(["file"], ["thefileid"])
 
1910
        t.commit(rev_id='somerev', message="add file")
 
1911
        self.assertIs(None, request.execute(''))
 
1912
        response = request.do_body("thefileid\0somerev\n")
 
1913
        self.assertTrue(response.is_successful())
 
1914
        self.assertEquals(response.args, ("ok", ))
 
1915
        self.assertEquals("".join(response.body_stream),
 
1916
            "ok\x000\n" + zlib.compress("somecontents"))
 
1917
 
 
1918
    def test_missing(self):
 
1919
        backing = self.get_transport()
 
1920
        request = smart_repo.SmartServerRepositoryIterFilesBytes(backing)
 
1921
        t = self.make_branch_and_tree('.')
 
1922
        self.addCleanup(t.lock_write().unlock)
 
1923
        self.assertIs(None, request.execute(''))
 
1924
        response = request.do_body("thefileid\0revision\n")
 
1925
        self.assertTrue(response.is_successful())
 
1926
        self.assertEquals(response.args, ("ok", ))
 
1927
        self.assertEquals("".join(response.body_stream),
 
1928
            "absent\x00thefileid\x00revision\x000\n")
 
1929
 
 
1930
 
 
1931
class TestSmartServerRequestHasSignatureForRevisionId(
 
1932
        tests.TestCaseWithMemoryTransport):
 
1933
 
 
1934
    def test_missing_revision(self):
 
1935
        """For a missing revision, NoSuchRevision is returned."""
 
1936
        backing = self.get_transport()
 
1937
        request = smart_repo.SmartServerRequestHasSignatureForRevisionId(
 
1938
            backing)
 
1939
        self.make_repository('.')
 
1940
        self.assertEqual(
 
1941
            smart_req.FailedSmartServerResponse(
 
1942
                ('nosuchrevision', 'revid'), None),
 
1943
            request.execute('', 'revid'))
 
1944
 
 
1945
    def test_missing_signature(self):
 
1946
        """For a missing signature, ('no', ) is returned."""
 
1947
        backing = self.get_transport()
 
1948
        request = smart_repo.SmartServerRequestHasSignatureForRevisionId(
 
1949
            backing)
 
1950
        tree = self.make_branch_and_memory_tree('.')
 
1951
        tree.lock_write()
 
1952
        tree.add('')
 
1953
        r1 = tree.commit('a commit', rev_id='A')
 
1954
        tree.unlock()
 
1955
        self.assertTrue(tree.branch.repository.has_revision('A'))
 
1956
        self.assertEqual(smart_req.SmartServerResponse(('no', )),
 
1957
            request.execute('', 'A'))
 
1958
 
 
1959
    def test_present_signature(self):
 
1960
        """For a present signature, ('yes', ) is returned."""
 
1961
        backing = self.get_transport()
 
1962
        request = smart_repo.SmartServerRequestHasSignatureForRevisionId(
 
1963
            backing)
 
1964
        strategy = gpg.LoopbackGPGStrategy(None)
 
1965
        tree = self.make_branch_and_memory_tree('.')
 
1966
        tree.lock_write()
 
1967
        tree.add('')
 
1968
        r1 = tree.commit('a commit', rev_id='A')
 
1969
        tree.branch.repository.start_write_group()
 
1970
        tree.branch.repository.sign_revision('A', strategy)
 
1971
        tree.branch.repository.commit_write_group()
 
1972
        tree.unlock()
 
1973
        self.assertTrue(tree.branch.repository.has_revision('A'))
 
1974
        self.assertEqual(smart_req.SmartServerResponse(('yes', )),
 
1975
            request.execute('', 'A'))
 
1976
 
 
1977
 
 
1978
class TestSmartServerRepositoryGatherStats(tests.TestCaseWithMemoryTransport):
 
1979
 
 
1980
    def test_empty_revid(self):
 
1981
        """With an empty revid, we get only size an number and revisions"""
 
1982
        backing = self.get_transport()
 
1983
        request = smart_repo.SmartServerRepositoryGatherStats(backing)
 
1984
        repository = self.make_repository('.')
 
1985
        stats = repository.gather_stats()
 
1986
        expected_body = 'revisions: 0\n'
 
1987
        self.assertEqual(smart_req.SmartServerResponse(('ok', ), expected_body),
 
1988
                         request.execute('', '', 'no'))
 
1989
 
 
1990
    def test_revid_with_committers(self):
 
1991
        """For a revid we get more infos."""
 
1992
        backing = self.get_transport()
 
1993
        rev_id_utf8 = u'\xc8abc'.encode('utf-8')
 
1994
        request = smart_repo.SmartServerRepositoryGatherStats(backing)
 
1995
        tree = self.make_branch_and_memory_tree('.')
 
1996
        tree.lock_write()
 
1997
        tree.add('')
 
1998
        # Let's build a predictable result
 
1999
        tree.commit('a commit', timestamp=123456.2, timezone=3600)
 
2000
        tree.commit('a commit', timestamp=654321.4, timezone=0,
 
2001
                    rev_id=rev_id_utf8)
 
2002
        tree.unlock()
 
2003
 
 
2004
        stats = tree.branch.repository.gather_stats()
 
2005
        expected_body = ('firstrev: 123456.200 3600\n'
 
2006
                         'latestrev: 654321.400 0\n'
 
2007
                         'revisions: 2\n')
 
2008
        self.assertEqual(smart_req.SmartServerResponse(('ok', ), expected_body),
 
2009
                         request.execute('',
 
2010
                                         rev_id_utf8, 'no'))
 
2011
 
 
2012
    def test_not_empty_repository_with_committers(self):
 
2013
        """For a revid and requesting committers we get the whole thing."""
 
2014
        backing = self.get_transport()
 
2015
        rev_id_utf8 = u'\xc8abc'.encode('utf-8')
 
2016
        request = smart_repo.SmartServerRepositoryGatherStats(backing)
 
2017
        tree = self.make_branch_and_memory_tree('.')
 
2018
        tree.lock_write()
 
2019
        tree.add('')
 
2020
        # Let's build a predictable result
 
2021
        tree.commit('a commit', timestamp=123456.2, timezone=3600,
 
2022
                    committer='foo')
 
2023
        tree.commit('a commit', timestamp=654321.4, timezone=0,
 
2024
                    committer='bar', rev_id=rev_id_utf8)
 
2025
        tree.unlock()
 
2026
        stats = tree.branch.repository.gather_stats()
 
2027
 
 
2028
        expected_body = ('committers: 2\n'
 
2029
                         'firstrev: 123456.200 3600\n'
 
2030
                         'latestrev: 654321.400 0\n'
 
2031
                         'revisions: 2\n')
 
2032
        self.assertEqual(smart_req.SmartServerResponse(('ok', ), expected_body),
 
2033
                         request.execute('',
 
2034
                                         rev_id_utf8, 'yes'))
 
2035
 
 
2036
    def test_unknown_revid(self):
 
2037
        """An unknown revision id causes a 'nosuchrevision' error."""
 
2038
        backing = self.get_transport()
 
2039
        request = smart_repo.SmartServerRepositoryGatherStats(backing)
 
2040
        repository = self.make_repository('.')
 
2041
        expected_body = 'revisions: 0\n'
 
2042
        self.assertEqual(
 
2043
            smart_req.FailedSmartServerResponse(
 
2044
                ('nosuchrevision', 'mia'), None),
 
2045
            request.execute('', 'mia', 'yes'))
 
2046
 
 
2047
 
 
2048
class TestSmartServerRepositoryIsShared(tests.TestCaseWithMemoryTransport):
 
2049
 
 
2050
    def test_is_shared(self):
 
2051
        """For a shared repository, ('yes', ) is returned."""
 
2052
        backing = self.get_transport()
 
2053
        request = smart_repo.SmartServerRepositoryIsShared(backing)
 
2054
        self.make_repository('.', shared=True)
 
2055
        self.assertEqual(smart_req.SmartServerResponse(('yes', )),
 
2056
            request.execute('', ))
 
2057
 
 
2058
    def test_is_not_shared(self):
 
2059
        """For a shared repository, ('no', ) is returned."""
 
2060
        backing = self.get_transport()
 
2061
        request = smart_repo.SmartServerRepositoryIsShared(backing)
 
2062
        self.make_repository('.', shared=False)
 
2063
        self.assertEqual(smart_req.SmartServerResponse(('no', )),
 
2064
            request.execute('', ))
 
2065
 
 
2066
 
 
2067
class TestSmartServerRepositoryGetRevisionSignatureText(
 
2068
        tests.TestCaseWithMemoryTransport):
 
2069
 
 
2070
    def test_get_signature(self):
 
2071
        backing = self.get_transport()
 
2072
        request = smart_repo.SmartServerRepositoryGetRevisionSignatureText(
 
2073
            backing)
 
2074
        bb = self.make_branch_builder('.')
 
2075
        bb.build_commit(rev_id='A')
 
2076
        repo = bb.get_branch().repository
 
2077
        strategy = gpg.LoopbackGPGStrategy(None)
 
2078
        self.addCleanup(repo.lock_write().unlock)
 
2079
        repo.start_write_group()
 
2080
        repo.sign_revision('A', strategy)
 
2081
        repo.commit_write_group()
 
2082
        expected_body = (
 
2083
            '-----BEGIN PSEUDO-SIGNED CONTENT-----\n' +
 
2084
            Testament.from_revision(repo, 'A').as_short_text() +
 
2085
            '-----END PSEUDO-SIGNED CONTENT-----\n')
 
2086
        self.assertEqual(
 
2087
            smart_req.SmartServerResponse(('ok', ), expected_body),
 
2088
            request.execute('', 'A'))
 
2089
 
 
2090
 
 
2091
class TestSmartServerRepositoryMakeWorkingTrees(
 
2092
        tests.TestCaseWithMemoryTransport):
 
2093
 
 
2094
    def test_make_working_trees(self):
 
2095
        """For a repository with working trees, ('yes', ) is returned."""
 
2096
        backing = self.get_transport()
 
2097
        request = smart_repo.SmartServerRepositoryMakeWorkingTrees(backing)
 
2098
        r = self.make_repository('.')
 
2099
        r.set_make_working_trees(True)
 
2100
        self.assertEqual(smart_req.SmartServerResponse(('yes', )),
 
2101
            request.execute('', ))
 
2102
 
 
2103
    def test_is_not_shared(self):
 
2104
        """For a repository with working trees, ('no', ) is returned."""
 
2105
        backing = self.get_transport()
 
2106
        request = smart_repo.SmartServerRepositoryMakeWorkingTrees(backing)
 
2107
        r = self.make_repository('.')
 
2108
        r.set_make_working_trees(False)
 
2109
        self.assertEqual(smart_req.SmartServerResponse(('no', )),
 
2110
            request.execute('', ))
 
2111
 
 
2112
 
 
2113
class TestSmartServerRepositoryLockWrite(tests.TestCaseWithMemoryTransport):
 
2114
 
 
2115
    def test_lock_write_on_unlocked_repo(self):
 
2116
        backing = self.get_transport()
 
2117
        request = smart_repo.SmartServerRepositoryLockWrite(backing)
 
2118
        repository = self.make_repository('.', format='knit')
 
2119
        response = request.execute('')
 
2120
        nonce = repository.control_files._lock.peek().get('nonce')
 
2121
        self.assertEqual(smart_req.SmartServerResponse(('ok', nonce)), response)
 
2122
        # The repository is now locked.  Verify that with a new repository
 
2123
        # object.
 
2124
        new_repo = repository.bzrdir.open_repository()
 
2125
        self.assertRaises(errors.LockContention, new_repo.lock_write)
 
2126
        # Cleanup
 
2127
        request = smart_repo.SmartServerRepositoryUnlock(backing)
 
2128
        response = request.execute('', nonce)
 
2129
 
 
2130
    def test_lock_write_on_locked_repo(self):
 
2131
        backing = self.get_transport()
 
2132
        request = smart_repo.SmartServerRepositoryLockWrite(backing)
 
2133
        repository = self.make_repository('.', format='knit')
 
2134
        repo_token = repository.lock_write().repository_token
 
2135
        repository.leave_lock_in_place()
 
2136
        repository.unlock()
 
2137
        response = request.execute('')
 
2138
        self.assertEqual(
 
2139
            smart_req.SmartServerResponse(('LockContention',)), response)
 
2140
        # Cleanup
 
2141
        repository.lock_write(repo_token)
 
2142
        repository.dont_leave_lock_in_place()
 
2143
        repository.unlock()
 
2144
 
 
2145
    def test_lock_write_on_readonly_transport(self):
 
2146
        backing = self.get_readonly_transport()
 
2147
        request = smart_repo.SmartServerRepositoryLockWrite(backing)
 
2148
        repository = self.make_repository('.', format='knit')
 
2149
        response = request.execute('')
 
2150
        self.assertFalse(response.is_successful())
 
2151
        self.assertEqual('LockFailed', response.args[0])
 
2152
 
 
2153
 
 
2154
class TestInsertStreamBase(tests.TestCaseWithMemoryTransport):
 
2155
 
 
2156
    def make_empty_byte_stream(self, repo):
 
2157
        byte_stream = smart_repo._stream_to_byte_stream([], repo._format)
 
2158
        return ''.join(byte_stream)
 
2159
 
 
2160
 
 
2161
class TestSmartServerRepositoryInsertStream(TestInsertStreamBase):
 
2162
 
 
2163
    def test_insert_stream_empty(self):
 
2164
        backing = self.get_transport()
 
2165
        request = smart_repo.SmartServerRepositoryInsertStream(backing)
 
2166
        repository = self.make_repository('.')
 
2167
        response = request.execute('', '')
 
2168
        self.assertEqual(None, response)
 
2169
        response = request.do_chunk(self.make_empty_byte_stream(repository))
 
2170
        self.assertEqual(None, response)
 
2171
        response = request.do_end()
 
2172
        self.assertEqual(smart_req.SmartServerResponse(('ok', )), response)
 
2173
 
 
2174
 
 
2175
class TestSmartServerRepositoryInsertStreamLocked(TestInsertStreamBase):
 
2176
 
 
2177
    def test_insert_stream_empty(self):
 
2178
        backing = self.get_transport()
 
2179
        request = smart_repo.SmartServerRepositoryInsertStreamLocked(
 
2180
            backing)
 
2181
        repository = self.make_repository('.', format='knit')
 
2182
        lock_token = repository.lock_write().repository_token
 
2183
        response = request.execute('', '', lock_token)
 
2184
        self.assertEqual(None, response)
 
2185
        response = request.do_chunk(self.make_empty_byte_stream(repository))
 
2186
        self.assertEqual(None, response)
 
2187
        response = request.do_end()
 
2188
        self.assertEqual(smart_req.SmartServerResponse(('ok', )), response)
 
2189
        repository.unlock()
 
2190
 
 
2191
    def test_insert_stream_with_wrong_lock_token(self):
 
2192
        backing = self.get_transport()
 
2193
        request = smart_repo.SmartServerRepositoryInsertStreamLocked(
 
2194
            backing)
 
2195
        repository = self.make_repository('.', format='knit')
 
2196
        lock_token = repository.lock_write().repository_token
 
2197
        self.assertRaises(
 
2198
            errors.TokenMismatch, request.execute, '', '', 'wrong-token')
 
2199
        repository.unlock()
 
2200
 
 
2201
 
 
2202
class TestSmartServerRepositoryUnlock(tests.TestCaseWithMemoryTransport):
 
2203
 
 
2204
    def setUp(self):
 
2205
        tests.TestCaseWithMemoryTransport.setUp(self)
 
2206
 
 
2207
    def test_unlock_on_locked_repo(self):
 
2208
        backing = self.get_transport()
 
2209
        request = smart_repo.SmartServerRepositoryUnlock(backing)
 
2210
        repository = self.make_repository('.', format='knit')
 
2211
        token = repository.lock_write().repository_token
 
2212
        repository.leave_lock_in_place()
 
2213
        repository.unlock()
 
2214
        response = request.execute('', token)
 
2215
        self.assertEqual(
 
2216
            smart_req.SmartServerResponse(('ok',)), response)
 
2217
        # The repository is now unlocked.  Verify that with a new repository
 
2218
        # object.
 
2219
        new_repo = repository.bzrdir.open_repository()
 
2220
        new_repo.lock_write()
 
2221
        new_repo.unlock()
 
2222
 
 
2223
    def test_unlock_on_unlocked_repo(self):
 
2224
        backing = self.get_transport()
 
2225
        request = smart_repo.SmartServerRepositoryUnlock(backing)
 
2226
        repository = self.make_repository('.', format='knit')
 
2227
        response = request.execute('', 'some token')
 
2228
        self.assertEqual(
 
2229
            smart_req.SmartServerResponse(('TokenMismatch',)), response)
 
2230
 
 
2231
 
 
2232
class TestSmartServerRepositoryGetPhysicalLockStatus(
 
2233
    tests.TestCaseWithTransport):
 
2234
 
 
2235
    def test_with_write_lock(self):
 
2236
        backing = self.get_transport()
 
2237
        repo = self.make_repository('.')
 
2238
        self.addCleanup(repo.lock_write().unlock)
 
2239
        # lock_write() doesn't necessarily actually take a physical
 
2240
        # lock out.
 
2241
        if repo.get_physical_lock_status():
 
2242
            expected = 'yes'
 
2243
        else:
 
2244
            expected = 'no'
 
2245
        request_class = smart_repo.SmartServerRepositoryGetPhysicalLockStatus
 
2246
        request = request_class(backing)
 
2247
        self.assertEqual(smart_req.SuccessfulSmartServerResponse((expected,)),
 
2248
            request.execute('', ))
 
2249
 
 
2250
    def test_without_write_lock(self):
 
2251
        backing = self.get_transport()
 
2252
        repo = self.make_repository('.')
 
2253
        self.assertEquals(False, repo.get_physical_lock_status())
 
2254
        request_class = smart_repo.SmartServerRepositoryGetPhysicalLockStatus
 
2255
        request = request_class(backing)
 
2256
        self.assertEqual(smart_req.SuccessfulSmartServerResponse(('no',)),
 
2257
            request.execute('', ))
 
2258
 
 
2259
 
 
2260
class TestSmartServerRepositoryReconcile(tests.TestCaseWithTransport):
 
2261
 
 
2262
    def test_reconcile(self):
 
2263
        backing = self.get_transport()
 
2264
        repo = self.make_repository('.')
 
2265
        token = repo.lock_write().repository_token
 
2266
        self.addCleanup(repo.unlock)
 
2267
        request_class = smart_repo.SmartServerRepositoryReconcile
 
2268
        request = request_class(backing)
 
2269
        self.assertEqual(smart_req.SuccessfulSmartServerResponse(
 
2270
            ('ok', ),
 
2271
             'garbage_inventories: 0\n'
 
2272
             'inconsistent_parents: 0\n'),
 
2273
            request.execute('', token))
 
2274
 
 
2275
 
 
2276
class TestSmartServerIsReadonly(tests.TestCaseWithMemoryTransport):
 
2277
 
 
2278
    def test_is_readonly_no(self):
 
2279
        backing = self.get_transport()
 
2280
        request = smart_req.SmartServerIsReadonly(backing)
 
2281
        response = request.execute()
 
2282
        self.assertEqual(
 
2283
            smart_req.SmartServerResponse(('no',)), response)
 
2284
 
 
2285
    def test_is_readonly_yes(self):
 
2286
        backing = self.get_readonly_transport()
 
2287
        request = smart_req.SmartServerIsReadonly(backing)
 
2288
        response = request.execute()
 
2289
        self.assertEqual(
 
2290
            smart_req.SmartServerResponse(('yes',)), response)
 
2291
 
 
2292
 
 
2293
class TestSmartServerRepositorySetMakeWorkingTrees(
 
2294
    tests.TestCaseWithMemoryTransport):
 
2295
 
 
2296
    def test_set_false(self):
 
2297
        backing = self.get_transport()
 
2298
        repo = self.make_repository('.', shared=True)
 
2299
        repo.set_make_working_trees(True)
 
2300
        request_class = smart_repo.SmartServerRepositorySetMakeWorkingTrees
 
2301
        request = request_class(backing)
 
2302
        self.assertEqual(smart_req.SuccessfulSmartServerResponse(('ok',)),
 
2303
            request.execute('', 'False'))
 
2304
        repo = repo.bzrdir.open_repository()
 
2305
        self.assertFalse(repo.make_working_trees())
 
2306
 
 
2307
    def test_set_true(self):
 
2308
        backing = self.get_transport()
 
2309
        repo = self.make_repository('.', shared=True)
 
2310
        repo.set_make_working_trees(False)
 
2311
        request_class = smart_repo.SmartServerRepositorySetMakeWorkingTrees
 
2312
        request = request_class(backing)
 
2313
        self.assertEqual(smart_req.SuccessfulSmartServerResponse(('ok',)),
 
2314
            request.execute('', 'True'))
 
2315
        repo = repo.bzrdir.open_repository()
 
2316
        self.assertTrue(repo.make_working_trees())
 
2317
 
 
2318
 
 
2319
class TestSmartServerRepositoryGetSerializerFormat(
 
2320
    tests.TestCaseWithMemoryTransport):
 
2321
 
 
2322
    def test_get_serializer_format(self):
 
2323
        backing = self.get_transport()
 
2324
        repo = self.make_repository('.', format='2a')
 
2325
        request_class = smart_repo.SmartServerRepositoryGetSerializerFormat
 
2326
        request = request_class(backing)
 
2327
        self.assertEqual(
 
2328
            smart_req.SuccessfulSmartServerResponse(('ok', '10')),
 
2329
            request.execute(''))
 
2330
 
 
2331
 
 
2332
class TestSmartServerRepositoryWriteGroup(
 
2333
    tests.TestCaseWithMemoryTransport):
 
2334
 
 
2335
    def test_start_write_group(self):
 
2336
        backing = self.get_transport()
 
2337
        repo = self.make_repository('.')
 
2338
        lock_token = repo.lock_write().repository_token
 
2339
        self.addCleanup(repo.unlock)
 
2340
        request_class = smart_repo.SmartServerRepositoryStartWriteGroup
 
2341
        request = request_class(backing)
 
2342
        self.assertEqual(smart_req.SuccessfulSmartServerResponse(('ok', [])),
 
2343
            request.execute('', lock_token))
 
2344
 
 
2345
    def test_start_write_group_unsuspendable(self):
 
2346
        backing = self.get_transport()
 
2347
        repo = self.make_repository('.', format='knit')
 
2348
        lock_token = repo.lock_write().repository_token
 
2349
        self.addCleanup(repo.unlock)
 
2350
        request_class = smart_repo.SmartServerRepositoryStartWriteGroup
 
2351
        request = request_class(backing)
 
2352
        self.assertEqual(
 
2353
            smart_req.FailedSmartServerResponse(('UnsuspendableWriteGroup',)),
 
2354
            request.execute('', lock_token))
 
2355
 
 
2356
    def test_commit_write_group(self):
 
2357
        backing = self.get_transport()
 
2358
        repo = self.make_repository('.')
 
2359
        lock_token = repo.lock_write().repository_token
 
2360
        self.addCleanup(repo.unlock)
 
2361
        repo.start_write_group()
 
2362
        tokens = repo.suspend_write_group()
 
2363
        request_class = smart_repo.SmartServerRepositoryCommitWriteGroup
 
2364
        request = request_class(backing)
 
2365
        self.assertEqual(smart_req.SuccessfulSmartServerResponse(('ok',)),
 
2366
            request.execute('', lock_token, tokens))
 
2367
 
 
2368
    def test_abort_write_group(self):
 
2369
        backing = self.get_transport()
 
2370
        repo = self.make_repository('.')
 
2371
        lock_token = repo.lock_write().repository_token
 
2372
        repo.start_write_group()
 
2373
        tokens = repo.suspend_write_group()
 
2374
        self.addCleanup(repo.unlock)
 
2375
        request_class = smart_repo.SmartServerRepositoryAbortWriteGroup
 
2376
        request = request_class(backing)
 
2377
        self.assertEqual(smart_req.SuccessfulSmartServerResponse(('ok',)),
 
2378
            request.execute('', lock_token, tokens))
 
2379
 
 
2380
    def test_check_write_group(self):
 
2381
        backing = self.get_transport()
 
2382
        repo = self.make_repository('.')
 
2383
        lock_token = repo.lock_write().repository_token
 
2384
        repo.start_write_group()
 
2385
        tokens = repo.suspend_write_group()
 
2386
        self.addCleanup(repo.unlock)
 
2387
        request_class = smart_repo.SmartServerRepositoryCheckWriteGroup
 
2388
        request = request_class(backing)
 
2389
        self.assertEqual(smart_req.SuccessfulSmartServerResponse(('ok',)),
 
2390
            request.execute('', lock_token, tokens))
 
2391
 
 
2392
    def test_check_write_group_invalid(self):
 
2393
        backing = self.get_transport()
 
2394
        repo = self.make_repository('.')
 
2395
        lock_token = repo.lock_write().repository_token
 
2396
        self.addCleanup(repo.unlock)
 
2397
        request_class = smart_repo.SmartServerRepositoryCheckWriteGroup
 
2398
        request = request_class(backing)
 
2399
        self.assertEqual(smart_req.FailedSmartServerResponse(
 
2400
            ('UnresumableWriteGroup', ['random'],
 
2401
                'Malformed write group token')),
 
2402
            request.execute('', lock_token, ["random"]))
 
2403
 
 
2404
 
 
2405
class TestSmartServerPackRepositoryAutopack(tests.TestCaseWithTransport):
 
2406
 
 
2407
    def make_repo_needing_autopacking(self, path='.'):
 
2408
        # Make a repo in need of autopacking.
 
2409
        tree = self.make_branch_and_tree('.', format='pack-0.92')
 
2410
        repo = tree.branch.repository
 
2411
        # monkey-patch the pack collection to disable autopacking
 
2412
        repo._pack_collection._max_pack_count = lambda count: count
 
2413
        for x in range(10):
 
2414
            tree.commit('commit %s' % x)
 
2415
        self.assertEqual(10, len(repo._pack_collection.names()))
 
2416
        del repo._pack_collection._max_pack_count
 
2417
        return repo
 
2418
 
 
2419
    def test_autopack_needed(self):
 
2420
        repo = self.make_repo_needing_autopacking()
 
2421
        repo.lock_write()
 
2422
        self.addCleanup(repo.unlock)
 
2423
        backing = self.get_transport()
 
2424
        request = smart_packrepo.SmartServerPackRepositoryAutopack(
 
2425
            backing)
 
2426
        response = request.execute('')
 
2427
        self.assertEqual(smart_req.SmartServerResponse(('ok',)), response)
 
2428
        repo._pack_collection.reload_pack_names()
 
2429
        self.assertEqual(1, len(repo._pack_collection.names()))
 
2430
 
 
2431
    def test_autopack_not_needed(self):
 
2432
        tree = self.make_branch_and_tree('.', format='pack-0.92')
 
2433
        repo = tree.branch.repository
 
2434
        repo.lock_write()
 
2435
        self.addCleanup(repo.unlock)
 
2436
        for x in range(9):
 
2437
            tree.commit('commit %s' % x)
 
2438
        backing = self.get_transport()
 
2439
        request = smart_packrepo.SmartServerPackRepositoryAutopack(
 
2440
            backing)
 
2441
        response = request.execute('')
 
2442
        self.assertEqual(smart_req.SmartServerResponse(('ok',)), response)
 
2443
        repo._pack_collection.reload_pack_names()
 
2444
        self.assertEqual(9, len(repo._pack_collection.names()))
 
2445
 
 
2446
    def test_autopack_on_nonpack_format(self):
 
2447
        """A request to autopack a non-pack repo is a no-op."""
 
2448
        repo = self.make_repository('.', format='knit')
 
2449
        backing = self.get_transport()
 
2450
        request = smart_packrepo.SmartServerPackRepositoryAutopack(
 
2451
            backing)
 
2452
        response = request.execute('')
 
2453
        self.assertEqual(smart_req.SmartServerResponse(('ok',)), response)
 
2454
 
 
2455
 
 
2456
class TestSmartServerVfsGet(tests.TestCaseWithMemoryTransport):
 
2457
 
 
2458
    def test_unicode_path(self):
 
2459
        """VFS requests expect unicode paths to be escaped."""
 
2460
        filename = u'foo\N{INTERROBANG}'
 
2461
        filename_escaped = urlutils.escape(filename)
 
2462
        backing = self.get_transport()
 
2463
        request = vfs.GetRequest(backing)
 
2464
        backing.put_bytes_non_atomic(filename_escaped, 'contents')
 
2465
        self.assertEqual(smart_req.SmartServerResponse(('ok', ), 'contents'),
 
2466
            request.execute(filename_escaped))
 
2467
 
 
2468
 
 
2469
class TestHandlers(tests.TestCase):
 
2470
    """Tests for the request.request_handlers object."""
 
2471
 
 
2472
    def test_all_registrations_exist(self):
 
2473
        """All registered request_handlers can be found."""
 
2474
        # If there's a typo in a register_lazy call, this loop will fail with
 
2475
        # an AttributeError.
 
2476
        for key in smart_req.request_handlers.keys():
 
2477
            try:
 
2478
                item = smart_req.request_handlers.get(key)
 
2479
            except AttributeError, e:
 
2480
                raise AttributeError('failed to get %s: %s' % (key, e))
 
2481
 
 
2482
    def assertHandlerEqual(self, verb, handler):
 
2483
        self.assertEqual(smart_req.request_handlers.get(verb), handler)
 
2484
 
 
2485
    def test_registered_methods(self):
 
2486
        """Test that known methods are registered to the correct object."""
 
2487
        self.assertHandlerEqual('Branch.break_lock',
 
2488
            smart_branch.SmartServerBranchBreakLock)
 
2489
        self.assertHandlerEqual('Branch.get_config_file',
 
2490
            smart_branch.SmartServerBranchGetConfigFile)
 
2491
        self.assertHandlerEqual('Branch.put_config_file',
 
2492
            smart_branch.SmartServerBranchPutConfigFile)
 
2493
        self.assertHandlerEqual('Branch.get_parent',
 
2494
            smart_branch.SmartServerBranchGetParent)
 
2495
        self.assertHandlerEqual('Branch.get_physical_lock_status',
 
2496
            smart_branch.SmartServerBranchRequestGetPhysicalLockStatus)
 
2497
        self.assertHandlerEqual('Branch.get_tags_bytes',
 
2498
            smart_branch.SmartServerBranchGetTagsBytes)
 
2499
        self.assertHandlerEqual('Branch.lock_write',
 
2500
            smart_branch.SmartServerBranchRequestLockWrite)
 
2501
        self.assertHandlerEqual('Branch.last_revision_info',
 
2502
            smart_branch.SmartServerBranchRequestLastRevisionInfo)
 
2503
        self.assertHandlerEqual('Branch.revision_history',
 
2504
            smart_branch.SmartServerRequestRevisionHistory)
 
2505
        self.assertHandlerEqual('Branch.revision_id_to_revno',
 
2506
            smart_branch.SmartServerBranchRequestRevisionIdToRevno)
 
2507
        self.assertHandlerEqual('Branch.set_config_option',
 
2508
            smart_branch.SmartServerBranchRequestSetConfigOption)
 
2509
        self.assertHandlerEqual('Branch.set_last_revision',
 
2510
            smart_branch.SmartServerBranchRequestSetLastRevision)
 
2511
        self.assertHandlerEqual('Branch.set_last_revision_info',
 
2512
            smart_branch.SmartServerBranchRequestSetLastRevisionInfo)
 
2513
        self.assertHandlerEqual('Branch.set_last_revision_ex',
 
2514
            smart_branch.SmartServerBranchRequestSetLastRevisionEx)
 
2515
        self.assertHandlerEqual('Branch.set_parent_location',
 
2516
            smart_branch.SmartServerBranchRequestSetParentLocation)
 
2517
        self.assertHandlerEqual('Branch.unlock',
 
2518
            smart_branch.SmartServerBranchRequestUnlock)
 
2519
        self.assertHandlerEqual('BzrDir.destroy_branch',
 
2520
            smart_dir.SmartServerBzrDirRequestDestroyBranch)
 
2521
        self.assertHandlerEqual('BzrDir.find_repository',
 
2522
            smart_dir.SmartServerRequestFindRepositoryV1)
 
2523
        self.assertHandlerEqual('BzrDir.find_repositoryV2',
 
2524
            smart_dir.SmartServerRequestFindRepositoryV2)
 
2525
        self.assertHandlerEqual('BzrDirFormat.initialize',
 
2526
            smart_dir.SmartServerRequestInitializeBzrDir)
 
2527
        self.assertHandlerEqual('BzrDirFormat.initialize_ex_1.16',
 
2528
            smart_dir.SmartServerRequestBzrDirInitializeEx)
 
2529
        self.assertHandlerEqual('BzrDir.checkout_metadir',
 
2530
            smart_dir.SmartServerBzrDirRequestCheckoutMetaDir)
 
2531
        self.assertHandlerEqual('BzrDir.cloning_metadir',
 
2532
            smart_dir.SmartServerBzrDirRequestCloningMetaDir)
 
2533
        self.assertHandlerEqual('BzrDir.get_branches',
 
2534
            smart_dir.SmartServerBzrDirRequestGetBranches)
 
2535
        self.assertHandlerEqual('BzrDir.get_config_file',
 
2536
            smart_dir.SmartServerBzrDirRequestConfigFile)
 
2537
        self.assertHandlerEqual('BzrDir.open_branch',
 
2538
            smart_dir.SmartServerRequestOpenBranch)
 
2539
        self.assertHandlerEqual('BzrDir.open_branchV2',
 
2540
            smart_dir.SmartServerRequestOpenBranchV2)
 
2541
        self.assertHandlerEqual('BzrDir.open_branchV3',
 
2542
            smart_dir.SmartServerRequestOpenBranchV3)
 
2543
        self.assertHandlerEqual('PackRepository.autopack',
 
2544
            smart_packrepo.SmartServerPackRepositoryAutopack)
 
2545
        self.assertHandlerEqual('Repository.add_signature_text',
 
2546
            smart_repo.SmartServerRepositoryAddSignatureText)
 
2547
        self.assertHandlerEqual('Repository.all_revision_ids',
 
2548
            smart_repo.SmartServerRepositoryAllRevisionIds)
 
2549
        self.assertHandlerEqual('Repository.break_lock',
 
2550
            smart_repo.SmartServerRepositoryBreakLock)
 
2551
        self.assertHandlerEqual('Repository.gather_stats',
 
2552
            smart_repo.SmartServerRepositoryGatherStats)
 
2553
        self.assertHandlerEqual('Repository.get_parent_map',
 
2554
            smart_repo.SmartServerRepositoryGetParentMap)
 
2555
        self.assertHandlerEqual('Repository.get_physical_lock_status',
 
2556
            smart_repo.SmartServerRepositoryGetPhysicalLockStatus)
 
2557
        self.assertHandlerEqual('Repository.get_rev_id_for_revno',
 
2558
            smart_repo.SmartServerRepositoryGetRevIdForRevno)
 
2559
        self.assertHandlerEqual('Repository.get_revision_graph',
 
2560
            smart_repo.SmartServerRepositoryGetRevisionGraph)
 
2561
        self.assertHandlerEqual('Repository.get_revision_signature_text',
 
2562
            smart_repo.SmartServerRepositoryGetRevisionSignatureText)
 
2563
        self.assertHandlerEqual('Repository.get_stream',
 
2564
            smart_repo.SmartServerRepositoryGetStream)
 
2565
        self.assertHandlerEqual('Repository.get_stream_1.19',
 
2566
            smart_repo.SmartServerRepositoryGetStream_1_19)
 
2567
        self.assertHandlerEqual('Repository.iter_revisions',
 
2568
            smart_repo.SmartServerRepositoryIterRevisions)
 
2569
        self.assertHandlerEqual('Repository.has_revision',
 
2570
            smart_repo.SmartServerRequestHasRevision)
 
2571
        self.assertHandlerEqual('Repository.insert_stream',
 
2572
            smart_repo.SmartServerRepositoryInsertStream)
 
2573
        self.assertHandlerEqual('Repository.insert_stream_locked',
 
2574
            smart_repo.SmartServerRepositoryInsertStreamLocked)
 
2575
        self.assertHandlerEqual('Repository.is_shared',
 
2576
            smart_repo.SmartServerRepositoryIsShared)
 
2577
        self.assertHandlerEqual('Repository.iter_files_bytes',
 
2578
            smart_repo.SmartServerRepositoryIterFilesBytes)
 
2579
        self.assertHandlerEqual('Repository.lock_write',
 
2580
            smart_repo.SmartServerRepositoryLockWrite)
 
2581
        self.assertHandlerEqual('Repository.make_working_trees',
 
2582
            smart_repo.SmartServerRepositoryMakeWorkingTrees)
 
2583
        self.assertHandlerEqual('Repository.pack',
 
2584
            smart_repo.SmartServerRepositoryPack)
 
2585
        self.assertHandlerEqual('Repository.reconcile',
 
2586
            smart_repo.SmartServerRepositoryReconcile)
 
2587
        self.assertHandlerEqual('Repository.tarball',
 
2588
            smart_repo.SmartServerRepositoryTarball)
 
2589
        self.assertHandlerEqual('Repository.unlock',
 
2590
            smart_repo.SmartServerRepositoryUnlock)
 
2591
        self.assertHandlerEqual('Repository.start_write_group',
 
2592
            smart_repo.SmartServerRepositoryStartWriteGroup)
 
2593
        self.assertHandlerEqual('Repository.check_write_group',
 
2594
            smart_repo.SmartServerRepositoryCheckWriteGroup)
 
2595
        self.assertHandlerEqual('Repository.commit_write_group',
 
2596
            smart_repo.SmartServerRepositoryCommitWriteGroup)
 
2597
        self.assertHandlerEqual('Repository.abort_write_group',
 
2598
            smart_repo.SmartServerRepositoryAbortWriteGroup)
 
2599
        self.assertHandlerEqual('VersionedFileRepository.get_serializer_format',
 
2600
            smart_repo.SmartServerRepositoryGetSerializerFormat)
 
2601
        self.assertHandlerEqual('VersionedFileRepository.get_inventories',
 
2602
            smart_repo.SmartServerRepositoryGetInventories)
 
2603
        self.assertHandlerEqual('Transport.is_readonly',
 
2604
            smart_req.SmartServerIsReadonly)
 
2605
 
 
2606
 
 
2607
class SmartTCPServerHookTests(tests.TestCaseWithMemoryTransport):
 
2608
    """Tests for SmartTCPServer hooks."""
 
2609
 
 
2610
    def setUp(self):
 
2611
        super(SmartTCPServerHookTests, self).setUp()
 
2612
        self.server = server.SmartTCPServer(self.get_transport())
 
2613
 
 
2614
    def test_run_server_started_hooks(self):
 
2615
        """Test the server started hooks get fired properly."""
 
2616
        started_calls = []
 
2617
        server.SmartTCPServer.hooks.install_named_hook('server_started',
 
2618
            lambda backing_urls, url: started_calls.append((backing_urls, url)),
 
2619
            None)
 
2620
        started_ex_calls = []
 
2621
        server.SmartTCPServer.hooks.install_named_hook('server_started_ex',
 
2622
            lambda backing_urls, url: started_ex_calls.append((backing_urls, url)),
 
2623
            None)
 
2624
        self.server._sockname = ('example.com', 42)
 
2625
        self.server.run_server_started_hooks()
 
2626
        self.assertEquals(started_calls,
 
2627
            [([self.get_transport().base], 'bzr://example.com:42/')])
 
2628
        self.assertEquals(started_ex_calls,
 
2629
            [([self.get_transport().base], self.server)])
 
2630
 
 
2631
    def test_run_server_started_hooks_ipv6(self):
 
2632
        """Test that socknames can contain 4-tuples."""
 
2633
        self.server._sockname = ('::', 42, 0, 0)
 
2634
        started_calls = []
 
2635
        server.SmartTCPServer.hooks.install_named_hook('server_started',
 
2636
            lambda backing_urls, url: started_calls.append((backing_urls, url)),
 
2637
            None)
 
2638
        self.server.run_server_started_hooks()
 
2639
        self.assertEquals(started_calls,
 
2640
                [([self.get_transport().base], 'bzr://:::42/')])
 
2641
 
 
2642
    def test_run_server_stopped_hooks(self):
 
2643
        """Test the server stopped hooks."""
 
2644
        self.server._sockname = ('example.com', 42)
 
2645
        stopped_calls = []
 
2646
        server.SmartTCPServer.hooks.install_named_hook('server_stopped',
 
2647
            lambda backing_urls, url: stopped_calls.append((backing_urls, url)),
 
2648
            None)
 
2649
        self.server.run_server_stopped_hooks()
 
2650
        self.assertEquals(stopped_calls,
 
2651
            [([self.get_transport().base], 'bzr://example.com:42/')])
 
2652
 
 
2653
 
 
2654
class TestSmartServerRepositoryPack(tests.TestCaseWithMemoryTransport):
 
2655
 
 
2656
    def test_pack(self):
 
2657
        backing = self.get_transport()
 
2658
        request = smart_repo.SmartServerRepositoryPack(backing)
 
2659
        tree = self.make_branch_and_memory_tree('.')
 
2660
        repo_token = tree.branch.repository.lock_write().repository_token
 
2661
 
 
2662
        self.assertIs(None, request.execute('', repo_token, False))
 
2663
 
 
2664
        self.assertEqual(
 
2665
            smart_req.SuccessfulSmartServerResponse(('ok', ), ),
 
2666
            request.do_body(''))
 
2667
 
 
2668
 
 
2669
class TestSmartServerRepositoryGetInventories(tests.TestCaseWithTransport):
 
2670
 
 
2671
    def _get_serialized_inventory_delta(self, repository, base_revid, revid):
 
2672
        base_inv = repository.revision_tree(base_revid).root_inventory
 
2673
        inv = repository.revision_tree(revid).root_inventory
 
2674
        inv_delta = inv._make_delta(base_inv)
 
2675
        serializer = inventory_delta.InventoryDeltaSerializer(True, False)
 
2676
        return "".join(serializer.delta_to_lines(base_revid, revid, inv_delta))
 
2677
 
 
2678
    def test_single(self):
 
2679
        backing = self.get_transport()
 
2680
        request = smart_repo.SmartServerRepositoryGetInventories(backing)
 
2681
        t = self.make_branch_and_tree('.', format='2a')
 
2682
        self.addCleanup(t.lock_write().unlock)
 
2683
        self.build_tree_contents([("file", "somecontents")])
 
2684
        t.add(["file"], ["thefileid"])
 
2685
        t.commit(rev_id='somerev', message="add file")
 
2686
        self.assertIs(None, request.execute('', 'unordered'))
 
2687
        response = request.do_body("somerev\n")
 
2688
        self.assertTrue(response.is_successful())
 
2689
        self.assertEquals(response.args, ("ok", ))
 
2690
        stream = [('inventory-deltas', [
 
2691
            versionedfile.FulltextContentFactory('somerev', None, None,
 
2692
                self._get_serialized_inventory_delta(
 
2693
                    t.branch.repository, 'null:', 'somerev'))])]
 
2694
        fmt = bzrdir.format_registry.get('2a')().repository_format
 
2695
        self.assertEquals(
 
2696
            "".join(response.body_stream),
 
2697
            "".join(smart_repo._stream_to_byte_stream(stream, fmt)))
 
2698
 
 
2699
    def test_empty(self):
 
2700
        backing = self.get_transport()
 
2701
        request = smart_repo.SmartServerRepositoryGetInventories(backing)
 
2702
        t = self.make_branch_and_tree('.', format='2a')
 
2703
        self.addCleanup(t.lock_write().unlock)
 
2704
        self.build_tree_contents([("file", "somecontents")])
 
2705
        t.add(["file"], ["thefileid"])
 
2706
        t.commit(rev_id='somerev', message="add file")
 
2707
        self.assertIs(None, request.execute('', 'unordered'))
 
2708
        response = request.do_body("")
 
2709
        self.assertTrue(response.is_successful())
 
2710
        self.assertEquals(response.args, ("ok", ))
 
2711
        self.assertEquals("".join(response.body_stream),
 
2712
            "Bazaar pack format 1 (introduced in 0.18)\nB54\n\nBazaar repository format 2a (needs bzr 1.16 or later)\nE")