1
# Copyright (C) 2006-2010 Canonical Ltd
3
# This program is free software; you can redistribute it and/or modify
4
# it under the terms of the GNU General Public License as published by
5
# the Free Software Foundation; either version 2 of the License, or
6
# (at your option) any later version.
8
# This program is distributed in the hope that it will be useful,
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11
# GNU General Public License for more details.
13
# You should have received a copy of the GNU General Public License
14
# along with this program; if not, write to the Free Software
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
17
"""Tests for the smart wire/domain protocol.
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
24
Tests for low-level protocol encoding are found in test_smart_transport.
28
from cStringIO import StringIO
33
branch as _mod_branch,
42
from bzrlib.smart import (
43
branch as smart_branch,
45
repository as smart_repo,
46
packrepository as smart_packrepo,
51
from bzrlib.tests import test_server
52
from bzrlib.transport import (
58
def load_tests(standard_tests, module, loader):
59
"""Multiply tests version and protocol consistency."""
60
# FindRepository tests.
63
"_request_class": smart_dir.SmartServerRequestFindRepositoryV1}),
64
("find_repositoryV2", {
65
"_request_class": smart_dir.SmartServerRequestFindRepositoryV2}),
66
("find_repositoryV3", {
67
"_request_class": smart_dir.SmartServerRequestFindRepositoryV3}),
69
to_adapt, result = tests.split_suite_by_re(standard_tests,
70
"TestSmartServerRequestFindRepository")
71
v2_only, v1_and_2 = tests.split_suite_by_re(to_adapt,
73
tests.multiply_tests(v1_and_2, scenarios, result)
74
# The first scenario is only applicable to v1 protocols, it is deleted
76
tests.multiply_tests(v2_only, scenarios[1:], result)
80
class TestCaseWithChrootedTransport(tests.TestCaseWithTransport):
83
self.vfs_transport_factory = memory.MemoryServer
84
tests.TestCaseWithTransport.setUp(self)
85
self._chroot_server = None
87
def get_transport(self, relpath=None):
88
if self._chroot_server is None:
89
backing_transport = tests.TestCaseWithTransport.get_transport(self)
90
self._chroot_server = chroot.ChrootServer(backing_transport)
91
self.start_server(self._chroot_server)
92
t = transport.get_transport(self._chroot_server.get_url())
93
if relpath is not None:
98
class TestCaseWithSmartMedium(tests.TestCaseWithMemoryTransport):
101
super(TestCaseWithSmartMedium, self).setUp()
102
# We're allowed to set the transport class here, so that we don't use
103
# the default or a parameterized class, but rather use the
104
# TestCaseWithTransport infrastructure to set up a smart server and
106
self.transport_server = self.make_transport_server
108
def make_transport_server(self):
109
return test_server.SmartTCPServer_for_testing('-' + self.id())
111
def get_smart_medium(self):
112
"""Get a smart medium to use in tests."""
113
return self.get_transport().get_smart_medium()
116
class TestByteStreamToStream(tests.TestCase):
118
def test_repeated_substreams_same_kind_are_one_stream(self):
119
# Make a stream - an iterable of bytestrings.
120
stream = [('text', [versionedfile.FulltextContentFactory(('k1',), None,
121
None, 'foo')]),('text', [
122
versionedfile.FulltextContentFactory(('k2',), None, None, 'bar')])]
123
fmt = bzrdir.format_registry.get('pack-0.92')().repository_format
124
bytes = smart_repo._stream_to_byte_stream(stream, fmt)
126
# Iterate the resulting iterable; checking that we get only one stream
128
fmt, stream = smart_repo._byte_stream_to_stream(bytes)
129
for kind, substream in stream:
130
streams.append((kind, list(substream)))
131
self.assertLength(1, streams)
132
self.assertLength(2, streams[0][1])
135
class TestSmartServerResponse(tests.TestCase):
137
def test__eq__(self):
138
self.assertEqual(smart_req.SmartServerResponse(('ok', )),
139
smart_req.SmartServerResponse(('ok', )))
140
self.assertEqual(smart_req.SmartServerResponse(('ok', ), 'body'),
141
smart_req.SmartServerResponse(('ok', ), 'body'))
142
self.assertNotEqual(smart_req.SmartServerResponse(('ok', )),
143
smart_req.SmartServerResponse(('notok', )))
144
self.assertNotEqual(smart_req.SmartServerResponse(('ok', ), 'body'),
145
smart_req.SmartServerResponse(('ok', )))
146
self.assertNotEqual(None,
147
smart_req.SmartServerResponse(('ok', )))
149
def test__str__(self):
150
"""SmartServerResponses can be stringified."""
152
"<SuccessfulSmartServerResponse args=('args',) body='body'>",
153
str(smart_req.SuccessfulSmartServerResponse(('args',), 'body')))
155
"<FailedSmartServerResponse args=('args',) body='body'>",
156
str(smart_req.FailedSmartServerResponse(('args',), 'body')))
159
class TestSmartServerRequest(tests.TestCaseWithMemoryTransport):
161
def test_translate_client_path(self):
162
transport = self.get_transport()
163
request = smart_req.SmartServerRequest(transport, 'foo/')
164
self.assertEqual('./', request.translate_client_path('foo/'))
166
errors.InvalidURLJoin, request.translate_client_path, 'foo/..')
168
errors.PathNotChild, request.translate_client_path, '/')
170
errors.PathNotChild, request.translate_client_path, 'bar/')
171
self.assertEqual('./baz', request.translate_client_path('foo/baz'))
172
e_acute = u'\N{LATIN SMALL LETTER E WITH ACUTE}'.encode('utf-8')
173
self.assertEqual('./' + urlutils.escape(e_acute),
174
request.translate_client_path('foo/' + e_acute))
176
def test_translate_client_path_vfs(self):
177
"""VfsRequests receive escaped paths rather than raw UTF-8."""
178
transport = self.get_transport()
179
request = vfs.VfsRequest(transport, 'foo/')
180
e_acute = u'\N{LATIN SMALL LETTER E WITH ACUTE}'.encode('utf-8')
181
escaped = urlutils.escape('foo/' + e_acute)
182
self.assertEqual('./' + urlutils.escape(e_acute),
183
request.translate_client_path(escaped))
185
def test_transport_from_client_path(self):
186
transport = self.get_transport()
187
request = smart_req.SmartServerRequest(transport, 'foo/')
190
request.transport_from_client_path('foo/').base)
193
class TestSmartServerBzrDirRequestCloningMetaDir(
194
tests.TestCaseWithMemoryTransport):
195
"""Tests for BzrDir.cloning_metadir."""
197
def test_cloning_metadir(self):
198
"""When there is a bzrdir present, the call succeeds."""
199
backing = self.get_transport()
200
dir = self.make_bzrdir('.')
201
local_result = dir.cloning_metadir()
202
request_class = smart_dir.SmartServerBzrDirRequestCloningMetaDir
203
request = request_class(backing)
204
expected = smart_req.SuccessfulSmartServerResponse(
205
(local_result.network_name(),
206
local_result.repository_format.network_name(),
207
('branch', local_result.get_branch_format().network_name())))
208
self.assertEqual(expected, request.execute('', 'False'))
210
def test_cloning_metadir_reference(self):
211
"""The request fails when bzrdir contains a branch reference."""
212
backing = self.get_transport()
213
referenced_branch = self.make_branch('referenced')
214
dir = self.make_bzrdir('.')
215
local_result = dir.cloning_metadir()
216
reference = _mod_branch.BranchReferenceFormat().initialize(
217
dir, target_branch=referenced_branch)
218
reference_url = _mod_branch.BranchReferenceFormat().get_reference(dir)
219
# The server shouldn't try to follow the branch reference, so it's fine
220
# if the referenced branch isn't reachable.
221
backing.rename('referenced', 'moved')
222
request_class = smart_dir.SmartServerBzrDirRequestCloningMetaDir
223
request = request_class(backing)
224
expected = smart_req.FailedSmartServerResponse(('BranchReference',))
225
self.assertEqual(expected, request.execute('', 'False'))
228
class TestSmartServerRequestCreateRepository(tests.TestCaseWithMemoryTransport):
229
"""Tests for BzrDir.create_repository."""
231
def test_makes_repository(self):
232
"""When there is a bzrdir present, the call succeeds."""
233
backing = self.get_transport()
234
self.make_bzrdir('.')
235
request_class = smart_dir.SmartServerRequestCreateRepository
236
request = request_class(backing)
237
reference_bzrdir_format = bzrdir.format_registry.get('pack-0.92')()
238
reference_format = reference_bzrdir_format.repository_format
239
network_name = reference_format.network_name()
240
expected = smart_req.SuccessfulSmartServerResponse(
241
('ok', 'no', 'no', 'no', network_name))
242
self.assertEqual(expected, request.execute('', network_name, 'True'))
245
class TestSmartServerRequestFindRepository(tests.TestCaseWithMemoryTransport):
246
"""Tests for BzrDir.find_repository."""
248
def test_no_repository(self):
249
"""When there is no repository to be found, ('norepository', ) is returned."""
250
backing = self.get_transport()
251
request = self._request_class(backing)
252
self.make_bzrdir('.')
253
self.assertEqual(smart_req.SmartServerResponse(('norepository', )),
256
def test_nonshared_repository(self):
257
# nonshared repositorys only allow 'find' to return a handle when the
258
# path the repository is being searched on is the same as that that
259
# the repository is at.
260
backing = self.get_transport()
261
request = self._request_class(backing)
262
result = self._make_repository_and_result()
263
self.assertEqual(result, request.execute(''))
264
self.make_bzrdir('subdir')
265
self.assertEqual(smart_req.SmartServerResponse(('norepository', )),
266
request.execute('subdir'))
268
def _make_repository_and_result(self, shared=False, format=None):
269
"""Convenience function to setup a repository.
271
:result: The SmartServerResponse to expect when opening it.
273
repo = self.make_repository('.', shared=shared, format=format)
274
if repo.supports_rich_root():
278
if repo._format.supports_tree_reference:
282
if repo._format.supports_external_lookups:
286
if (smart_dir.SmartServerRequestFindRepositoryV3 ==
287
self._request_class):
288
return smart_req.SuccessfulSmartServerResponse(
289
('ok', '', rich_root, subtrees, external,
290
repo._format.network_name()))
291
elif (smart_dir.SmartServerRequestFindRepositoryV2 ==
292
self._request_class):
293
# All tests so far are on formats, and for non-external
295
return smart_req.SuccessfulSmartServerResponse(
296
('ok', '', rich_root, subtrees, external))
298
return smart_req.SuccessfulSmartServerResponse(
299
('ok', '', rich_root, subtrees))
301
def test_shared_repository(self):
302
"""When there is a shared repository, we get 'ok', 'relpath-to-repo'."""
303
backing = self.get_transport()
304
request = self._request_class(backing)
305
result = self._make_repository_and_result(shared=True)
306
self.assertEqual(result, request.execute(''))
307
self.make_bzrdir('subdir')
308
result2 = smart_req.SmartServerResponse(
309
result.args[0:1] + ('..', ) + result.args[2:])
310
self.assertEqual(result2,
311
request.execute('subdir'))
312
self.make_bzrdir('subdir/deeper')
313
result3 = smart_req.SmartServerResponse(
314
result.args[0:1] + ('../..', ) + result.args[2:])
315
self.assertEqual(result3,
316
request.execute('subdir/deeper'))
318
def test_rich_root_and_subtree_encoding(self):
319
"""Test for the format attributes for rich root and subtree support."""
320
backing = self.get_transport()
321
request = self._request_class(backing)
322
result = self._make_repository_and_result(
323
format='dirstate-with-subtree')
324
# check the test will be valid
325
self.assertEqual('yes', result.args[2])
326
self.assertEqual('yes', result.args[3])
327
self.assertEqual(result, request.execute(''))
329
def test_supports_external_lookups_no_v2(self):
330
"""Test for the supports_external_lookups attribute."""
331
backing = self.get_transport()
332
request = self._request_class(backing)
333
result = self._make_repository_and_result(
334
format='dirstate-with-subtree')
335
# check the test will be valid
336
self.assertEqual('no', result.args[4])
337
self.assertEqual(result, request.execute(''))
340
class TestSmartServerBzrDirRequestGetConfigFile(
341
tests.TestCaseWithMemoryTransport):
342
"""Tests for BzrDir.get_config_file."""
344
def test_present(self):
345
backing = self.get_transport()
346
dir = self.make_bzrdir('.')
347
dir.get_config().set_default_stack_on("/")
348
local_result = dir._get_config()._get_config_file().read()
349
request_class = smart_dir.SmartServerBzrDirRequestConfigFile
350
request = request_class(backing)
351
expected = smart_req.SuccessfulSmartServerResponse((), local_result)
352
self.assertEqual(expected, request.execute(''))
354
def test_missing(self):
355
backing = self.get_transport()
356
dir = self.make_bzrdir('.')
357
request_class = smart_dir.SmartServerBzrDirRequestConfigFile
358
request = request_class(backing)
359
expected = smart_req.SuccessfulSmartServerResponse((), '')
360
self.assertEqual(expected, request.execute(''))
363
class TestSmartServerRequestInitializeBzrDir(tests.TestCaseWithMemoryTransport):
365
def test_empty_dir(self):
366
"""Initializing an empty dir should succeed and do it."""
367
backing = self.get_transport()
368
request = smart_dir.SmartServerRequestInitializeBzrDir(backing)
369
self.assertEqual(smart_req.SmartServerResponse(('ok', )),
371
made_dir = bzrdir.BzrDir.open_from_transport(backing)
372
# no branch, tree or repository is expected with the current
374
self.assertRaises(errors.NoWorkingTree, made_dir.open_workingtree)
375
self.assertRaises(errors.NotBranchError, made_dir.open_branch)
376
self.assertRaises(errors.NoRepositoryPresent, made_dir.open_repository)
378
def test_missing_dir(self):
379
"""Initializing a missing directory should fail like the bzrdir api."""
380
backing = self.get_transport()
381
request = smart_dir.SmartServerRequestInitializeBzrDir(backing)
382
self.assertRaises(errors.NoSuchFile,
383
request.execute, 'subdir')
385
def test_initialized_dir(self):
386
"""Initializing an extant bzrdir should fail like the bzrdir api."""
387
backing = self.get_transport()
388
request = smart_dir.SmartServerRequestInitializeBzrDir(backing)
389
self.make_bzrdir('subdir')
390
self.assertRaises(errors.FileExists,
391
request.execute, 'subdir')
394
class TestSmartServerRequestBzrDirInitializeEx(
395
tests.TestCaseWithMemoryTransport):
396
"""Basic tests for BzrDir.initialize_ex_1.16 in the smart server.
398
The main unit tests in test_bzrdir exercise the API comprehensively.
401
def test_empty_dir(self):
402
"""Initializing an empty dir should succeed and do it."""
403
backing = self.get_transport()
404
name = self.make_bzrdir('reference')._format.network_name()
405
request = smart_dir.SmartServerRequestBzrDirInitializeEx(backing)
407
smart_req.SmartServerResponse(('', '', '', '', '', '', name,
408
'False', '', '', '')),
409
request.execute(name, '', 'True', 'False', 'False', '', '', '', '',
411
made_dir = bzrdir.BzrDir.open_from_transport(backing)
412
# no branch, tree or repository is expected with the current
414
self.assertRaises(errors.NoWorkingTree, made_dir.open_workingtree)
415
self.assertRaises(errors.NotBranchError, made_dir.open_branch)
416
self.assertRaises(errors.NoRepositoryPresent, made_dir.open_repository)
418
def test_missing_dir(self):
419
"""Initializing a missing directory should fail like the bzrdir api."""
420
backing = self.get_transport()
421
name = self.make_bzrdir('reference')._format.network_name()
422
request = smart_dir.SmartServerRequestBzrDirInitializeEx(backing)
423
self.assertRaises(errors.NoSuchFile, request.execute, name,
424
'subdir/dir', 'False', 'False', 'False', '', '', '', '', 'False')
426
def test_initialized_dir(self):
427
"""Initializing an extant directory should fail like the bzrdir api."""
428
backing = self.get_transport()
429
name = self.make_bzrdir('reference')._format.network_name()
430
request = smart_dir.SmartServerRequestBzrDirInitializeEx(backing)
431
self.make_bzrdir('subdir')
432
self.assertRaises(errors.FileExists, request.execute, name, 'subdir',
433
'False', 'False', 'False', '', '', '', '', 'False')
436
class TestSmartServerRequestOpenBzrDir(tests.TestCaseWithMemoryTransport):
438
def test_no_directory(self):
439
backing = self.get_transport()
440
request = smart_dir.SmartServerRequestOpenBzrDir(backing)
441
self.assertEqual(smart_req.SmartServerResponse(('no', )),
442
request.execute('does-not-exist'))
444
def test_empty_directory(self):
445
backing = self.get_transport()
446
backing.mkdir('empty')
447
request = smart_dir.SmartServerRequestOpenBzrDir(backing)
448
self.assertEqual(smart_req.SmartServerResponse(('no', )),
449
request.execute('empty'))
451
def test_outside_root_client_path(self):
452
backing = self.get_transport()
453
request = smart_dir.SmartServerRequestOpenBzrDir(backing,
454
root_client_path='root')
455
self.assertEqual(smart_req.SmartServerResponse(('no', )),
456
request.execute('not-root'))
459
class TestSmartServerRequestOpenBzrDir_2_1(tests.TestCaseWithMemoryTransport):
461
def test_no_directory(self):
462
backing = self.get_transport()
463
request = smart_dir.SmartServerRequestOpenBzrDir_2_1(backing)
464
self.assertEqual(smart_req.SmartServerResponse(('no', )),
465
request.execute('does-not-exist'))
467
def test_empty_directory(self):
468
backing = self.get_transport()
469
backing.mkdir('empty')
470
request = smart_dir.SmartServerRequestOpenBzrDir_2_1(backing)
471
self.assertEqual(smart_req.SmartServerResponse(('no', )),
472
request.execute('empty'))
474
def test_present_without_workingtree(self):
475
backing = self.get_transport()
476
request = smart_dir.SmartServerRequestOpenBzrDir_2_1(backing)
477
self.make_bzrdir('.')
478
self.assertEqual(smart_req.SmartServerResponse(('yes', 'no')),
481
def test_outside_root_client_path(self):
482
backing = self.get_transport()
483
request = smart_dir.SmartServerRequestOpenBzrDir_2_1(backing,
484
root_client_path='root')
485
self.assertEqual(smart_req.SmartServerResponse(('no',)),
486
request.execute('not-root'))
489
class TestSmartServerRequestOpenBzrDir_2_1_disk(TestCaseWithChrootedTransport):
491
def test_present_with_workingtree(self):
492
self.vfs_transport_factory = test_server.LocalURLServer
493
backing = self.get_transport()
494
request = smart_dir.SmartServerRequestOpenBzrDir_2_1(backing)
495
bd = self.make_bzrdir('.')
496
bd.create_repository()
498
bd.create_workingtree()
499
self.assertEqual(smart_req.SmartServerResponse(('yes', 'yes')),
503
class TestSmartServerRequestOpenBranch(TestCaseWithChrootedTransport):
505
def test_no_branch(self):
506
"""When there is no branch, ('nobranch', ) is returned."""
507
backing = self.get_transport()
508
request = smart_dir.SmartServerRequestOpenBranch(backing)
509
self.make_bzrdir('.')
510
self.assertEqual(smart_req.SmartServerResponse(('nobranch', )),
513
def test_branch(self):
514
"""When there is a branch, 'ok' is returned."""
515
backing = self.get_transport()
516
request = smart_dir.SmartServerRequestOpenBranch(backing)
517
self.make_branch('.')
518
self.assertEqual(smart_req.SmartServerResponse(('ok', '')),
521
def test_branch_reference(self):
522
"""When there is a branch reference, the reference URL is returned."""
523
self.vfs_transport_factory = test_server.LocalURLServer
524
backing = self.get_transport()
525
request = smart_dir.SmartServerRequestOpenBranch(backing)
526
branch = self.make_branch('branch')
527
checkout = branch.create_checkout('reference',lightweight=True)
528
reference_url = _mod_branch.BranchReferenceFormat().get_reference(
530
self.assertFileEqual(reference_url, 'reference/.bzr/branch/location')
531
self.assertEqual(smart_req.SmartServerResponse(('ok', reference_url)),
532
request.execute('reference'))
534
def test_notification_on_branch_from_repository(self):
535
"""When there is a repository, the error should return details."""
536
backing = self.get_transport()
537
request = smart_dir.SmartServerRequestOpenBranch(backing)
538
repo = self.make_repository('.')
539
self.assertEqual(smart_req.SmartServerResponse(('nobranch',)),
543
class TestSmartServerRequestOpenBranchV2(TestCaseWithChrootedTransport):
545
def test_no_branch(self):
546
"""When there is no branch, ('nobranch', ) is returned."""
547
backing = self.get_transport()
548
self.make_bzrdir('.')
549
request = smart_dir.SmartServerRequestOpenBranchV2(backing)
550
self.assertEqual(smart_req.SmartServerResponse(('nobranch', )),
553
def test_branch(self):
554
"""When there is a branch, 'ok' is returned."""
555
backing = self.get_transport()
556
expected = self.make_branch('.')._format.network_name()
557
request = smart_dir.SmartServerRequestOpenBranchV2(backing)
558
self.assertEqual(smart_req.SuccessfulSmartServerResponse(
559
('branch', expected)),
562
def test_branch_reference(self):
563
"""When there is a branch reference, the reference URL is returned."""
564
self.vfs_transport_factory = test_server.LocalURLServer
565
backing = self.get_transport()
566
request = smart_dir.SmartServerRequestOpenBranchV2(backing)
567
branch = self.make_branch('branch')
568
checkout = branch.create_checkout('reference',lightweight=True)
569
reference_url = _mod_branch.BranchReferenceFormat().get_reference(
571
self.assertFileEqual(reference_url, 'reference/.bzr/branch/location')
572
self.assertEqual(smart_req.SuccessfulSmartServerResponse(
573
('ref', reference_url)),
574
request.execute('reference'))
576
def test_stacked_branch(self):
577
"""Opening a stacked branch does not open the stacked-on branch."""
578
trunk = self.make_branch('trunk')
579
feature = self.make_branch('feature')
580
feature.set_stacked_on_url(trunk.base)
582
_mod_branch.Branch.hooks.install_named_hook(
583
'open', opened_branches.append, None)
584
backing = self.get_transport()
585
request = smart_dir.SmartServerRequestOpenBranchV2(backing)
588
response = request.execute('feature')
590
request.teardown_jail()
591
expected_format = feature._format.network_name()
592
self.assertEqual(smart_req.SuccessfulSmartServerResponse(
593
('branch', expected_format)),
595
self.assertLength(1, opened_branches)
597
def test_notification_on_branch_from_repository(self):
598
"""When there is a repository, the error should return details."""
599
backing = self.get_transport()
600
request = smart_dir.SmartServerRequestOpenBranchV2(backing)
601
repo = self.make_repository('.')
602
self.assertEqual(smart_req.SmartServerResponse(('nobranch',)),
606
class TestSmartServerRequestOpenBranchV3(TestCaseWithChrootedTransport):
608
def test_no_branch(self):
609
"""When there is no branch, ('nobranch', ) is returned."""
610
backing = self.get_transport()
611
self.make_bzrdir('.')
612
request = smart_dir.SmartServerRequestOpenBranchV3(backing)
613
self.assertEqual(smart_req.SmartServerResponse(('nobranch',)),
616
def test_branch(self):
617
"""When there is a branch, 'ok' is returned."""
618
backing = self.get_transport()
619
expected = self.make_branch('.')._format.network_name()
620
request = smart_dir.SmartServerRequestOpenBranchV3(backing)
621
self.assertEqual(smart_req.SuccessfulSmartServerResponse(
622
('branch', expected)),
625
def test_branch_reference(self):
626
"""When there is a branch reference, the reference URL is returned."""
627
self.vfs_transport_factory = test_server.LocalURLServer
628
backing = self.get_transport()
629
request = smart_dir.SmartServerRequestOpenBranchV3(backing)
630
branch = self.make_branch('branch')
631
checkout = branch.create_checkout('reference',lightweight=True)
632
reference_url = _mod_branch.BranchReferenceFormat().get_reference(
634
self.assertFileEqual(reference_url, 'reference/.bzr/branch/location')
635
self.assertEqual(smart_req.SuccessfulSmartServerResponse(
636
('ref', reference_url)),
637
request.execute('reference'))
639
def test_stacked_branch(self):
640
"""Opening a stacked branch does not open the stacked-on branch."""
641
trunk = self.make_branch('trunk')
642
feature = self.make_branch('feature')
643
feature.set_stacked_on_url(trunk.base)
645
_mod_branch.Branch.hooks.install_named_hook(
646
'open', opened_branches.append, None)
647
backing = self.get_transport()
648
request = smart_dir.SmartServerRequestOpenBranchV3(backing)
651
response = request.execute('feature')
653
request.teardown_jail()
654
expected_format = feature._format.network_name()
655
self.assertEqual(smart_req.SuccessfulSmartServerResponse(
656
('branch', expected_format)),
658
self.assertLength(1, opened_branches)
660
def test_notification_on_branch_from_repository(self):
661
"""When there is a repository, the error should return details."""
662
backing = self.get_transport()
663
request = smart_dir.SmartServerRequestOpenBranchV3(backing)
664
repo = self.make_repository('.')
665
self.assertEqual(smart_req.SmartServerResponse(
666
('nobranch', 'location is a repository')),
670
class TestSmartServerRequestRevisionHistory(tests.TestCaseWithMemoryTransport):
672
def test_empty(self):
673
"""For an empty branch, the body is empty."""
674
backing = self.get_transport()
675
request = smart_branch.SmartServerRequestRevisionHistory(backing)
676
self.make_branch('.')
677
self.assertEqual(smart_req.SmartServerResponse(('ok', ), ''),
680
def test_not_empty(self):
681
"""For a non-empty branch, the body is empty."""
682
backing = self.get_transport()
683
request = smart_branch.SmartServerRequestRevisionHistory(backing)
684
tree = self.make_branch_and_memory_tree('.')
687
r1 = tree.commit('1st commit')
688
r2 = tree.commit('2nd commit', rev_id=u'\xc8'.encode('utf-8'))
691
smart_req.SmartServerResponse(('ok', ), ('\x00'.join([r1, r2]))),
695
class TestSmartServerBranchRequest(tests.TestCaseWithMemoryTransport):
697
def test_no_branch(self):
698
"""When there is a bzrdir and no branch, NotBranchError is raised."""
699
backing = self.get_transport()
700
request = smart_branch.SmartServerBranchRequest(backing)
701
self.make_bzrdir('.')
702
self.assertRaises(errors.NotBranchError,
705
def test_branch_reference(self):
706
"""When there is a branch reference, NotBranchError is raised."""
707
backing = self.get_transport()
708
request = smart_branch.SmartServerBranchRequest(backing)
709
branch = self.make_branch('branch')
710
checkout = branch.create_checkout('reference',lightweight=True)
711
self.assertRaises(errors.NotBranchError,
712
request.execute, 'checkout')
715
class TestSmartServerBranchRequestLastRevisionInfo(
716
tests.TestCaseWithMemoryTransport):
718
def test_empty(self):
719
"""For an empty branch, the result is ('ok', '0', 'null:')."""
720
backing = self.get_transport()
721
request = smart_branch.SmartServerBranchRequestLastRevisionInfo(backing)
722
self.make_branch('.')
723
self.assertEqual(smart_req.SmartServerResponse(('ok', '0', 'null:')),
726
def test_not_empty(self):
727
"""For a non-empty branch, the result is ('ok', 'revno', 'revid')."""
728
backing = self.get_transport()
729
request = smart_branch.SmartServerBranchRequestLastRevisionInfo(backing)
730
tree = self.make_branch_and_memory_tree('.')
733
rev_id_utf8 = u'\xc8'.encode('utf-8')
734
r1 = tree.commit('1st commit')
735
r2 = tree.commit('2nd commit', rev_id=rev_id_utf8)
738
smart_req.SmartServerResponse(('ok', '2', rev_id_utf8)),
742
class TestSmartServerBranchRequestGetConfigFile(
743
tests.TestCaseWithMemoryTransport):
745
def test_default(self):
746
"""With no file, we get empty content."""
747
backing = self.get_transport()
748
request = smart_branch.SmartServerBranchGetConfigFile(backing)
749
branch = self.make_branch('.')
750
# there should be no file by default
752
self.assertEqual(smart_req.SmartServerResponse(('ok', ), content),
755
def test_with_content(self):
756
# SmartServerBranchGetConfigFile should return the content from
757
# branch.control_files.get('branch.conf') for now - in the future it may
758
# perform more complex processing.
759
backing = self.get_transport()
760
request = smart_branch.SmartServerBranchGetConfigFile(backing)
761
branch = self.make_branch('.')
762
branch._transport.put_bytes('branch.conf', 'foo bar baz')
763
self.assertEqual(smart_req.SmartServerResponse(('ok', ), 'foo bar baz'),
767
class TestLockedBranch(tests.TestCaseWithMemoryTransport):
769
def get_lock_tokens(self, branch):
770
branch_token = branch.lock_write().branch_token
771
repo_token = branch.repository.lock_write().repository_token
772
branch.repository.unlock()
773
return branch_token, repo_token
776
class TestSmartServerBranchRequestSetConfigOption(TestLockedBranch):
778
def test_value_name(self):
779
branch = self.make_branch('.')
780
request = smart_branch.SmartServerBranchRequestSetConfigOption(
781
branch.bzrdir.root_transport)
782
branch_token, repo_token = self.get_lock_tokens(branch)
783
config = branch._get_config()
784
result = request.execute('', branch_token, repo_token, 'bar', 'foo',
786
self.assertEqual(smart_req.SuccessfulSmartServerResponse(()), result)
787
self.assertEqual('bar', config.get_option('foo'))
791
def test_value_name_section(self):
792
branch = self.make_branch('.')
793
request = smart_branch.SmartServerBranchRequestSetConfigOption(
794
branch.bzrdir.root_transport)
795
branch_token, repo_token = self.get_lock_tokens(branch)
796
config = branch._get_config()
797
result = request.execute('', branch_token, repo_token, 'bar', 'foo',
799
self.assertEqual(smart_req.SuccessfulSmartServerResponse(()), result)
800
self.assertEqual('bar', config.get_option('foo', 'gam'))
805
class TestSmartServerBranchRequestSetConfigOptionDict(TestLockedBranch):
808
TestLockedBranch.setUp(self)
809
# A dict with non-ascii keys and values to exercise unicode
811
self.encoded_value_dict = (
812
'd5:ascii1:a11:unicode \xe2\x8c\x9a3:\xe2\x80\xbde')
814
'ascii': 'a', u'unicode \N{WATCH}': u'\N{INTERROBANG}'}
816
def test_value_name(self):
817
branch = self.make_branch('.')
818
request = smart_branch.SmartServerBranchRequestSetConfigOptionDict(
819
branch.bzrdir.root_transport)
820
branch_token, repo_token = self.get_lock_tokens(branch)
821
config = branch._get_config()
822
result = request.execute('', branch_token, repo_token,
823
self.encoded_value_dict, 'foo', '')
824
self.assertEqual(smart_req.SuccessfulSmartServerResponse(()), result)
825
self.assertEqual(self.value_dict, config.get_option('foo'))
829
def test_value_name_section(self):
830
branch = self.make_branch('.')
831
request = smart_branch.SmartServerBranchRequestSetConfigOptionDict(
832
branch.bzrdir.root_transport)
833
branch_token, repo_token = self.get_lock_tokens(branch)
834
config = branch._get_config()
835
result = request.execute('', branch_token, repo_token,
836
self.encoded_value_dict, 'foo', 'gam')
837
self.assertEqual(smart_req.SuccessfulSmartServerResponse(()), result)
838
self.assertEqual(self.value_dict, config.get_option('foo', 'gam'))
843
class TestSmartServerBranchRequestSetTagsBytes(TestLockedBranch):
844
# Only called when the branch format and tags match [yay factory
845
# methods] so only need to test straight forward cases.
847
def test_set_bytes(self):
848
base_branch = self.make_branch('base')
849
tag_bytes = base_branch._get_tags_bytes()
850
# get_lock_tokens takes out a lock.
851
branch_token, repo_token = self.get_lock_tokens(base_branch)
852
request = smart_branch.SmartServerBranchSetTagsBytes(
853
self.get_transport())
854
response = request.execute('base', branch_token, repo_token)
855
self.assertEqual(None, response)
856
response = request.do_chunk(tag_bytes)
857
self.assertEqual(None, response)
858
response = request.do_end()
860
smart_req.SuccessfulSmartServerResponse(()), response)
863
def test_lock_failed(self):
864
base_branch = self.make_branch('base')
865
base_branch.lock_write()
866
tag_bytes = base_branch._get_tags_bytes()
867
request = smart_branch.SmartServerBranchSetTagsBytes(
868
self.get_transport())
869
self.assertRaises(errors.TokenMismatch, request.execute,
870
'base', 'wrong token', 'wrong token')
871
# The request handler will keep processing the message parts, so even
872
# if the request fails immediately do_chunk and do_end are still
874
request.do_chunk(tag_bytes)
880
class SetLastRevisionTestBase(TestLockedBranch):
881
"""Base test case for verbs that implement set_last_revision."""
884
tests.TestCaseWithMemoryTransport.setUp(self)
885
backing_transport = self.get_transport()
886
self.request = self.request_class(backing_transport)
887
self.tree = self.make_branch_and_memory_tree('.')
889
def lock_branch(self):
890
return self.get_lock_tokens(self.tree.branch)
892
def unlock_branch(self):
893
self.tree.branch.unlock()
895
def set_last_revision(self, revision_id, revno):
896
branch_token, repo_token = self.lock_branch()
897
response = self._set_last_revision(
898
revision_id, revno, branch_token, repo_token)
902
def assertRequestSucceeds(self, revision_id, revno):
903
response = self.set_last_revision(revision_id, revno)
904
self.assertEqual(smart_req.SuccessfulSmartServerResponse(('ok',)),
908
class TestSetLastRevisionVerbMixin(object):
909
"""Mixin test case for verbs that implement set_last_revision."""
911
def test_set_null_to_null(self):
912
"""An empty branch can have its last revision set to 'null:'."""
913
self.assertRequestSucceeds('null:', 0)
915
def test_NoSuchRevision(self):
916
"""If the revision_id is not present, the verb returns NoSuchRevision.
918
revision_id = 'non-existent revision'
919
self.assertEqual(smart_req.FailedSmartServerResponse(('NoSuchRevision',
921
self.set_last_revision(revision_id, 1))
923
def make_tree_with_two_commits(self):
924
self.tree.lock_write()
926
rev_id_utf8 = u'\xc8'.encode('utf-8')
927
r1 = self.tree.commit('1st commit', rev_id=rev_id_utf8)
928
r2 = self.tree.commit('2nd commit', rev_id='rev-2')
931
def test_branch_last_revision_info_is_updated(self):
932
"""A branch's tip can be set to a revision that is present in its
935
# Make a branch with an empty revision history, but two revisions in
937
self.make_tree_with_two_commits()
938
rev_id_utf8 = u'\xc8'.encode('utf-8')
939
self.tree.branch.set_revision_history([])
941
(0, 'null:'), self.tree.branch.last_revision_info())
942
# We can update the branch to a revision that is present in the
944
self.assertRequestSucceeds(rev_id_utf8, 1)
946
(1, rev_id_utf8), self.tree.branch.last_revision_info())
948
def test_branch_last_revision_info_rewind(self):
949
"""A branch's tip can be set to a revision that is an ancestor of the
952
self.make_tree_with_two_commits()
953
rev_id_utf8 = u'\xc8'.encode('utf-8')
955
(2, 'rev-2'), self.tree.branch.last_revision_info())
956
self.assertRequestSucceeds(rev_id_utf8, 1)
958
(1, rev_id_utf8), self.tree.branch.last_revision_info())
960
def test_TipChangeRejected(self):
961
"""If a pre_change_branch_tip hook raises TipChangeRejected, the verb
962
returns TipChangeRejected.
964
rejection_message = u'rejection message\N{INTERROBANG}'
965
def hook_that_rejects(params):
966
raise errors.TipChangeRejected(rejection_message)
967
_mod_branch.Branch.hooks.install_named_hook(
968
'pre_change_branch_tip', hook_that_rejects, None)
970
smart_req.FailedSmartServerResponse(
971
('TipChangeRejected', rejection_message.encode('utf-8'))),
972
self.set_last_revision('null:', 0))
975
class TestSmartServerBranchRequestSetLastRevision(
976
SetLastRevisionTestBase, TestSetLastRevisionVerbMixin):
977
"""Tests for Branch.set_last_revision verb."""
979
request_class = smart_branch.SmartServerBranchRequestSetLastRevision
981
def _set_last_revision(self, revision_id, revno, branch_token, repo_token):
982
return self.request.execute(
983
'', branch_token, repo_token, revision_id)
986
class TestSmartServerBranchRequestSetLastRevisionInfo(
987
SetLastRevisionTestBase, TestSetLastRevisionVerbMixin):
988
"""Tests for Branch.set_last_revision_info verb."""
990
request_class = smart_branch.SmartServerBranchRequestSetLastRevisionInfo
992
def _set_last_revision(self, revision_id, revno, branch_token, repo_token):
993
return self.request.execute(
994
'', branch_token, repo_token, revno, revision_id)
996
def test_NoSuchRevision(self):
997
"""Branch.set_last_revision_info does not have to return
998
NoSuchRevision if the revision_id is absent.
1000
raise tests.TestNotApplicable()
1003
class TestSmartServerBranchRequestSetLastRevisionEx(
1004
SetLastRevisionTestBase, TestSetLastRevisionVerbMixin):
1005
"""Tests for Branch.set_last_revision_ex verb."""
1007
request_class = smart_branch.SmartServerBranchRequestSetLastRevisionEx
1009
def _set_last_revision(self, revision_id, revno, branch_token, repo_token):
1010
return self.request.execute(
1011
'', branch_token, repo_token, revision_id, 0, 0)
1013
def assertRequestSucceeds(self, revision_id, revno):
1014
response = self.set_last_revision(revision_id, revno)
1016
smart_req.SuccessfulSmartServerResponse(('ok', revno, revision_id)),
1019
def test_branch_last_revision_info_rewind(self):
1020
"""A branch's tip can be set to a revision that is an ancestor of the
1021
current tip, but only if allow_overwrite_descendant is passed.
1023
self.make_tree_with_two_commits()
1024
rev_id_utf8 = u'\xc8'.encode('utf-8')
1026
(2, 'rev-2'), self.tree.branch.last_revision_info())
1027
# If allow_overwrite_descendant flag is 0, then trying to set the tip
1028
# to an older revision ID has no effect.
1029
branch_token, repo_token = self.lock_branch()
1030
response = self.request.execute(
1031
'', branch_token, repo_token, rev_id_utf8, 0, 0)
1033
smart_req.SuccessfulSmartServerResponse(('ok', 2, 'rev-2')),
1036
(2, 'rev-2'), self.tree.branch.last_revision_info())
1038
# If allow_overwrite_descendant flag is 1, then setting the tip to an
1040
response = self.request.execute(
1041
'', branch_token, repo_token, rev_id_utf8, 0, 1)
1043
smart_req.SuccessfulSmartServerResponse(('ok', 1, rev_id_utf8)),
1045
self.unlock_branch()
1047
(1, rev_id_utf8), self.tree.branch.last_revision_info())
1049
def make_branch_with_divergent_history(self):
1050
"""Make a branch with divergent history in its repo.
1052
The branch's tip will be 'child-2', and the repo will also contain
1053
'child-1', which diverges from a common base revision.
1055
self.tree.lock_write()
1057
r1 = self.tree.commit('1st commit')
1058
revno_1, revid_1 = self.tree.branch.last_revision_info()
1059
r2 = self.tree.commit('2nd commit', rev_id='child-1')
1060
# Undo the second commit
1061
self.tree.branch.set_last_revision_info(revno_1, revid_1)
1062
self.tree.set_parent_ids([revid_1])
1063
# Make a new second commit, child-2. child-2 has diverged from
1065
new_r2 = self.tree.commit('2nd commit', rev_id='child-2')
1068
def test_not_allow_diverged(self):
1069
"""If allow_diverged is not passed, then setting a divergent history
1070
returns a Diverged error.
1072
self.make_branch_with_divergent_history()
1074
smart_req.FailedSmartServerResponse(('Diverged',)),
1075
self.set_last_revision('child-1', 2))
1076
# The branch tip was not changed.
1077
self.assertEqual('child-2', self.tree.branch.last_revision())
1079
def test_allow_diverged(self):
1080
"""If allow_diverged is passed, then setting a divergent history
1083
self.make_branch_with_divergent_history()
1084
branch_token, repo_token = self.lock_branch()
1085
response = self.request.execute(
1086
'', branch_token, repo_token, 'child-1', 1, 0)
1088
smart_req.SuccessfulSmartServerResponse(('ok', 2, 'child-1')),
1090
self.unlock_branch()
1091
# The branch tip was changed.
1092
self.assertEqual('child-1', self.tree.branch.last_revision())
1095
class TestSmartServerBranchRequestGetParent(tests.TestCaseWithMemoryTransport):
1097
def test_get_parent_none(self):
1098
base_branch = self.make_branch('base')
1099
request = smart_branch.SmartServerBranchGetParent(self.get_transport())
1100
response = request.execute('base')
1102
smart_req.SuccessfulSmartServerResponse(('',)), response)
1104
def test_get_parent_something(self):
1105
base_branch = self.make_branch('base')
1106
base_branch.set_parent(self.get_url('foo'))
1107
request = smart_branch.SmartServerBranchGetParent(self.get_transport())
1108
response = request.execute('base')
1110
smart_req.SuccessfulSmartServerResponse(("../foo",)),
1114
class TestSmartServerBranchRequestSetParent(TestLockedBranch):
1116
def test_set_parent_none(self):
1117
branch = self.make_branch('base', format="1.9")
1119
branch._set_parent_location('foo')
1121
request = smart_branch.SmartServerBranchRequestSetParentLocation(
1122
self.get_transport())
1123
branch_token, repo_token = self.get_lock_tokens(branch)
1125
response = request.execute('base', branch_token, repo_token, '')
1128
self.assertEqual(smart_req.SuccessfulSmartServerResponse(()), response)
1129
self.assertEqual(None, branch.get_parent())
1131
def test_set_parent_something(self):
1132
branch = self.make_branch('base', format="1.9")
1133
request = smart_branch.SmartServerBranchRequestSetParentLocation(
1134
self.get_transport())
1135
branch_token, repo_token = self.get_lock_tokens(branch)
1137
response = request.execute('base', branch_token, repo_token,
1141
self.assertEqual(smart_req.SuccessfulSmartServerResponse(()), response)
1142
self.assertEqual('http://bar/', branch.get_parent())
1145
class TestSmartServerBranchRequestGetTagsBytes(
1146
tests.TestCaseWithMemoryTransport):
1147
# Only called when the branch format and tags match [yay factory
1148
# methods] so only need to test straight forward cases.
1150
def test_get_bytes(self):
1151
base_branch = self.make_branch('base')
1152
request = smart_branch.SmartServerBranchGetTagsBytes(
1153
self.get_transport())
1154
response = request.execute('base')
1156
smart_req.SuccessfulSmartServerResponse(('',)), response)
1159
class TestSmartServerBranchRequestGetStackedOnURL(tests.TestCaseWithMemoryTransport):
1161
def test_get_stacked_on_url(self):
1162
base_branch = self.make_branch('base', format='1.6')
1163
stacked_branch = self.make_branch('stacked', format='1.6')
1164
# typically should be relative
1165
stacked_branch.set_stacked_on_url('../base')
1166
request = smart_branch.SmartServerBranchRequestGetStackedOnURL(
1167
self.get_transport())
1168
response = request.execute('stacked')
1170
smart_req.SmartServerResponse(('ok', '../base')),
1174
class TestSmartServerBranchRequestLockWrite(TestLockedBranch):
1177
tests.TestCaseWithMemoryTransport.setUp(self)
1179
def test_lock_write_on_unlocked_branch(self):
1180
backing = self.get_transport()
1181
request = smart_branch.SmartServerBranchRequestLockWrite(backing)
1182
branch = self.make_branch('.', format='knit')
1183
repository = branch.repository
1184
response = request.execute('')
1185
branch_nonce = branch.control_files._lock.peek().get('nonce')
1186
repository_nonce = repository.control_files._lock.peek().get('nonce')
1187
self.assertEqual(smart_req.SmartServerResponse(
1188
('ok', branch_nonce, repository_nonce)),
1190
# The branch (and associated repository) is now locked. Verify that
1191
# with a new branch object.
1192
new_branch = repository.bzrdir.open_branch()
1193
self.assertRaises(errors.LockContention, new_branch.lock_write)
1195
request = smart_branch.SmartServerBranchRequestUnlock(backing)
1196
response = request.execute('', branch_nonce, repository_nonce)
1198
def test_lock_write_on_locked_branch(self):
1199
backing = self.get_transport()
1200
request = smart_branch.SmartServerBranchRequestLockWrite(backing)
1201
branch = self.make_branch('.')
1202
branch_token = branch.lock_write().branch_token
1203
branch.leave_lock_in_place()
1205
response = request.execute('')
1207
smart_req.SmartServerResponse(('LockContention',)), response)
1209
branch.lock_write(branch_token)
1210
branch.dont_leave_lock_in_place()
1213
def test_lock_write_with_tokens_on_locked_branch(self):
1214
backing = self.get_transport()
1215
request = smart_branch.SmartServerBranchRequestLockWrite(backing)
1216
branch = self.make_branch('.', format='knit')
1217
branch_token, repo_token = self.get_lock_tokens(branch)
1218
branch.leave_lock_in_place()
1219
branch.repository.leave_lock_in_place()
1221
response = request.execute('',
1222
branch_token, repo_token)
1224
smart_req.SmartServerResponse(('ok', branch_token, repo_token)),
1227
branch.repository.lock_write(repo_token)
1228
branch.repository.dont_leave_lock_in_place()
1229
branch.repository.unlock()
1230
branch.lock_write(branch_token)
1231
branch.dont_leave_lock_in_place()
1234
def test_lock_write_with_mismatched_tokens_on_locked_branch(self):
1235
backing = self.get_transport()
1236
request = smart_branch.SmartServerBranchRequestLockWrite(backing)
1237
branch = self.make_branch('.', format='knit')
1238
branch_token, repo_token = self.get_lock_tokens(branch)
1239
branch.leave_lock_in_place()
1240
branch.repository.leave_lock_in_place()
1242
response = request.execute('',
1243
branch_token+'xxx', repo_token)
1245
smart_req.SmartServerResponse(('TokenMismatch',)), response)
1247
branch.repository.lock_write(repo_token)
1248
branch.repository.dont_leave_lock_in_place()
1249
branch.repository.unlock()
1250
branch.lock_write(branch_token)
1251
branch.dont_leave_lock_in_place()
1254
def test_lock_write_on_locked_repo(self):
1255
backing = self.get_transport()
1256
request = smart_branch.SmartServerBranchRequestLockWrite(backing)
1257
branch = self.make_branch('.', format='knit')
1258
repo = branch.repository
1259
repo_token = repo.lock_write().repository_token
1260
repo.leave_lock_in_place()
1262
response = request.execute('')
1264
smart_req.SmartServerResponse(('LockContention',)), response)
1266
repo.lock_write(repo_token)
1267
repo.dont_leave_lock_in_place()
1270
def test_lock_write_on_readonly_transport(self):
1271
backing = self.get_readonly_transport()
1272
request = smart_branch.SmartServerBranchRequestLockWrite(backing)
1273
branch = self.make_branch('.')
1274
root = self.get_transport().clone('/')
1275
path = urlutils.relative_url(root.base, self.get_transport().base)
1276
response = request.execute(path)
1277
error_name, lock_str, why_str = response.args
1278
self.assertFalse(response.is_successful())
1279
self.assertEqual('LockFailed', error_name)
1282
class TestSmartServerBranchRequestUnlock(TestLockedBranch):
1285
tests.TestCaseWithMemoryTransport.setUp(self)
1287
def test_unlock_on_locked_branch_and_repo(self):
1288
backing = self.get_transport()
1289
request = smart_branch.SmartServerBranchRequestUnlock(backing)
1290
branch = self.make_branch('.', format='knit')
1292
branch_token, repo_token = self.get_lock_tokens(branch)
1293
# Unlock the branch (and repo) object, leaving the physical locks
1295
branch.leave_lock_in_place()
1296
branch.repository.leave_lock_in_place()
1298
response = request.execute('',
1299
branch_token, repo_token)
1301
smart_req.SmartServerResponse(('ok',)), response)
1302
# The branch is now unlocked. Verify that with a new branch
1304
new_branch = branch.bzrdir.open_branch()
1305
new_branch.lock_write()
1308
def test_unlock_on_unlocked_branch_unlocked_repo(self):
1309
backing = self.get_transport()
1310
request = smart_branch.SmartServerBranchRequestUnlock(backing)
1311
branch = self.make_branch('.', format='knit')
1312
response = request.execute(
1313
'', 'branch token', 'repo token')
1315
smart_req.SmartServerResponse(('TokenMismatch',)), response)
1317
def test_unlock_on_unlocked_branch_locked_repo(self):
1318
backing = self.get_transport()
1319
request = smart_branch.SmartServerBranchRequestUnlock(backing)
1320
branch = self.make_branch('.', format='knit')
1321
# Lock the repository.
1322
repo_token = branch.repository.lock_write().repository_token
1323
branch.repository.leave_lock_in_place()
1324
branch.repository.unlock()
1325
# Issue branch lock_write request on the unlocked branch (with locked
1327
response = request.execute('', 'branch token', repo_token)
1329
smart_req.SmartServerResponse(('TokenMismatch',)), response)
1331
branch.repository.lock_write(repo_token)
1332
branch.repository.dont_leave_lock_in_place()
1333
branch.repository.unlock()
1336
class TestSmartServerRepositoryRequest(tests.TestCaseWithMemoryTransport):
1338
def test_no_repository(self):
1339
"""Raise NoRepositoryPresent when there is a bzrdir and no repo."""
1340
# we test this using a shared repository above the named path,
1341
# thus checking the right search logic is used - that is, that
1342
# its the exact path being looked at and the server is not
1344
backing = self.get_transport()
1345
request = smart_repo.SmartServerRepositoryRequest(backing)
1346
self.make_repository('.', shared=True)
1347
self.make_bzrdir('subdir')
1348
self.assertRaises(errors.NoRepositoryPresent,
1349
request.execute, 'subdir')
1352
class TestSmartServerRepositoryGetParentMap(tests.TestCaseWithMemoryTransport):
1354
def test_trivial_bzipped(self):
1355
# This tests that the wire encoding is actually bzipped
1356
backing = self.get_transport()
1357
request = smart_repo.SmartServerRepositoryGetParentMap(backing)
1358
tree = self.make_branch_and_memory_tree('.')
1360
self.assertEqual(None,
1361
request.execute('', 'missing-id'))
1362
# Note that it returns a body that is bzipped.
1364
smart_req.SuccessfulSmartServerResponse(('ok', ), bz2.compress('')),
1365
request.do_body('\n\n0\n'))
1367
def test_trivial_include_missing(self):
1368
backing = self.get_transport()
1369
request = smart_repo.SmartServerRepositoryGetParentMap(backing)
1370
tree = self.make_branch_and_memory_tree('.')
1372
self.assertEqual(None,
1373
request.execute('', 'missing-id', 'include-missing:'))
1375
smart_req.SuccessfulSmartServerResponse(('ok', ),
1376
bz2.compress('missing:missing-id')),
1377
request.do_body('\n\n0\n'))
1380
class TestSmartServerRepositoryGetRevisionGraph(
1381
tests.TestCaseWithMemoryTransport):
1383
def test_none_argument(self):
1384
backing = self.get_transport()
1385
request = smart_repo.SmartServerRepositoryGetRevisionGraph(backing)
1386
tree = self.make_branch_and_memory_tree('.')
1389
r1 = tree.commit('1st commit')
1390
r2 = tree.commit('2nd commit', rev_id=u'\xc8'.encode('utf-8'))
1393
# the lines of revision_id->revision_parent_list has no guaranteed
1394
# order coming out of a dict, so sort both our test and response
1395
lines = sorted([' '.join([r2, r1]), r1])
1396
response = request.execute('', '')
1397
response.body = '\n'.join(sorted(response.body.split('\n')))
1400
smart_req.SmartServerResponse(('ok', ), '\n'.join(lines)), response)
1402
def test_specific_revision_argument(self):
1403
backing = self.get_transport()
1404
request = smart_repo.SmartServerRepositoryGetRevisionGraph(backing)
1405
tree = self.make_branch_and_memory_tree('.')
1408
rev_id_utf8 = u'\xc9'.encode('utf-8')
1409
r1 = tree.commit('1st commit', rev_id=rev_id_utf8)
1410
r2 = tree.commit('2nd commit', rev_id=u'\xc8'.encode('utf-8'))
1413
self.assertEqual(smart_req.SmartServerResponse(('ok', ), rev_id_utf8),
1414
request.execute('', rev_id_utf8))
1416
def test_no_such_revision(self):
1417
backing = self.get_transport()
1418
request = smart_repo.SmartServerRepositoryGetRevisionGraph(backing)
1419
tree = self.make_branch_and_memory_tree('.')
1422
r1 = tree.commit('1st commit')
1425
# Note that it still returns body (of zero bytes).
1426
self.assertEqual(smart_req.SmartServerResponse(
1427
('nosuchrevision', 'missingrevision', ), ''),
1428
request.execute('', 'missingrevision'))
1431
class TestSmartServerRepositoryGetRevIdForRevno(
1432
tests.TestCaseWithMemoryTransport):
1434
def test_revno_found(self):
1435
backing = self.get_transport()
1436
request = smart_repo.SmartServerRepositoryGetRevIdForRevno(backing)
1437
tree = self.make_branch_and_memory_tree('.')
1440
rev1_id_utf8 = u'\xc8'.encode('utf-8')
1441
rev2_id_utf8 = u'\xc9'.encode('utf-8')
1442
tree.commit('1st commit', rev_id=rev1_id_utf8)
1443
tree.commit('2nd commit', rev_id=rev2_id_utf8)
1446
self.assertEqual(smart_req.SmartServerResponse(('ok', rev1_id_utf8)),
1447
request.execute('', 1, (2, rev2_id_utf8)))
1449
def test_known_revid_missing(self):
1450
backing = self.get_transport()
1451
request = smart_repo.SmartServerRepositoryGetRevIdForRevno(backing)
1452
repo = self.make_repository('.')
1454
smart_req.FailedSmartServerResponse(('nosuchrevision', 'ghost')),
1455
request.execute('', 1, (2, 'ghost')))
1457
def test_history_incomplete(self):
1458
backing = self.get_transport()
1459
request = smart_repo.SmartServerRepositoryGetRevIdForRevno(backing)
1460
parent = self.make_branch_and_memory_tree('parent', format='1.9')
1462
parent.add([''], ['TREE_ROOT'])
1463
r1 = parent.commit(message='first commit')
1464
r2 = parent.commit(message='second commit')
1466
local = self.make_branch_and_memory_tree('local', format='1.9')
1467
local.branch.pull(parent.branch)
1468
local.set_parent_ids([r2])
1469
r3 = local.commit(message='local commit')
1470
local.branch.create_clone_on_transport(
1471
self.get_transport('stacked'), stacked_on=self.get_url('parent'))
1473
smart_req.SmartServerResponse(('history-incomplete', 2, r2)),
1474
request.execute('stacked', 1, (3, r3)))
1477
class TestSmartServerRepositoryGetStream(tests.TestCaseWithMemoryTransport):
1479
def make_two_commit_repo(self):
1480
tree = self.make_branch_and_memory_tree('.')
1483
r1 = tree.commit('1st commit')
1484
r2 = tree.commit('2nd commit', rev_id=u'\xc8'.encode('utf-8'))
1486
repo = tree.branch.repository
1489
def test_ancestry_of(self):
1490
"""The search argument may be a 'ancestry-of' some heads'."""
1491
backing = self.get_transport()
1492
request = smart_repo.SmartServerRepositoryGetStream(backing)
1493
repo, r1, r2 = self.make_two_commit_repo()
1494
fetch_spec = ['ancestry-of', r2]
1495
lines = '\n'.join(fetch_spec)
1496
request.execute('', repo._format.network_name())
1497
response = request.do_body(lines)
1498
self.assertEqual(('ok',), response.args)
1499
stream_bytes = ''.join(response.body_stream)
1500
self.assertStartsWith(stream_bytes, 'Bazaar pack format 1')
1502
def test_search(self):
1503
"""The search argument may be a 'search' of some explicit keys."""
1504
backing = self.get_transport()
1505
request = smart_repo.SmartServerRepositoryGetStream(backing)
1506
repo, r1, r2 = self.make_two_commit_repo()
1507
fetch_spec = ['search', '%s %s' % (r1, r2), 'null:', '2']
1508
lines = '\n'.join(fetch_spec)
1509
request.execute('', repo._format.network_name())
1510
response = request.do_body(lines)
1511
self.assertEqual(('ok',), response.args)
1512
stream_bytes = ''.join(response.body_stream)
1513
self.assertStartsWith(stream_bytes, 'Bazaar pack format 1')
1516
class TestSmartServerRequestHasRevision(tests.TestCaseWithMemoryTransport):
1518
def test_missing_revision(self):
1519
"""For a missing revision, ('no', ) is returned."""
1520
backing = self.get_transport()
1521
request = smart_repo.SmartServerRequestHasRevision(backing)
1522
self.make_repository('.')
1523
self.assertEqual(smart_req.SmartServerResponse(('no', )),
1524
request.execute('', 'revid'))
1526
def test_present_revision(self):
1527
"""For a present revision, ('yes', ) is returned."""
1528
backing = self.get_transport()
1529
request = smart_repo.SmartServerRequestHasRevision(backing)
1530
tree = self.make_branch_and_memory_tree('.')
1533
rev_id_utf8 = u'\xc8abc'.encode('utf-8')
1534
r1 = tree.commit('a commit', rev_id=rev_id_utf8)
1536
self.assertTrue(tree.branch.repository.has_revision(rev_id_utf8))
1537
self.assertEqual(smart_req.SmartServerResponse(('yes', )),
1538
request.execute('', rev_id_utf8))
1541
class TestSmartServerRepositoryGatherStats(tests.TestCaseWithMemoryTransport):
1543
def test_empty_revid(self):
1544
"""With an empty revid, we get only size an number and revisions"""
1545
backing = self.get_transport()
1546
request = smart_repo.SmartServerRepositoryGatherStats(backing)
1547
repository = self.make_repository('.')
1548
stats = repository.gather_stats()
1549
expected_body = 'revisions: 0\n'
1550
self.assertEqual(smart_req.SmartServerResponse(('ok', ), expected_body),
1551
request.execute('', '', 'no'))
1553
def test_revid_with_committers(self):
1554
"""For a revid we get more infos."""
1555
backing = self.get_transport()
1556
rev_id_utf8 = u'\xc8abc'.encode('utf-8')
1557
request = smart_repo.SmartServerRepositoryGatherStats(backing)
1558
tree = self.make_branch_and_memory_tree('.')
1561
# Let's build a predictable result
1562
tree.commit('a commit', timestamp=123456.2, timezone=3600)
1563
tree.commit('a commit', timestamp=654321.4, timezone=0,
1567
stats = tree.branch.repository.gather_stats()
1568
expected_body = ('firstrev: 123456.200 3600\n'
1569
'latestrev: 654321.400 0\n'
1571
self.assertEqual(smart_req.SmartServerResponse(('ok', ), expected_body),
1575
def test_not_empty_repository_with_committers(self):
1576
"""For a revid and requesting committers we get the whole thing."""
1577
backing = self.get_transport()
1578
rev_id_utf8 = u'\xc8abc'.encode('utf-8')
1579
request = smart_repo.SmartServerRepositoryGatherStats(backing)
1580
tree = self.make_branch_and_memory_tree('.')
1583
# Let's build a predictable result
1584
tree.commit('a commit', timestamp=123456.2, timezone=3600,
1586
tree.commit('a commit', timestamp=654321.4, timezone=0,
1587
committer='bar', rev_id=rev_id_utf8)
1589
stats = tree.branch.repository.gather_stats()
1591
expected_body = ('committers: 2\n'
1592
'firstrev: 123456.200 3600\n'
1593
'latestrev: 654321.400 0\n'
1595
self.assertEqual(smart_req.SmartServerResponse(('ok', ), expected_body),
1597
rev_id_utf8, 'yes'))
1600
class TestSmartServerRepositoryIsShared(tests.TestCaseWithMemoryTransport):
1602
def test_is_shared(self):
1603
"""For a shared repository, ('yes', ) is returned."""
1604
backing = self.get_transport()
1605
request = smart_repo.SmartServerRepositoryIsShared(backing)
1606
self.make_repository('.', shared=True)
1607
self.assertEqual(smart_req.SmartServerResponse(('yes', )),
1608
request.execute('', ))
1610
def test_is_not_shared(self):
1611
"""For a shared repository, ('no', ) is returned."""
1612
backing = self.get_transport()
1613
request = smart_repo.SmartServerRepositoryIsShared(backing)
1614
self.make_repository('.', shared=False)
1615
self.assertEqual(smart_req.SmartServerResponse(('no', )),
1616
request.execute('', ))
1619
class TestSmartServerRepositoryLockWrite(tests.TestCaseWithMemoryTransport):
1621
def test_lock_write_on_unlocked_repo(self):
1622
backing = self.get_transport()
1623
request = smart_repo.SmartServerRepositoryLockWrite(backing)
1624
repository = self.make_repository('.', format='knit')
1625
response = request.execute('')
1626
nonce = repository.control_files._lock.peek().get('nonce')
1627
self.assertEqual(smart_req.SmartServerResponse(('ok', nonce)), response)
1628
# The repository is now locked. Verify that with a new repository
1630
new_repo = repository.bzrdir.open_repository()
1631
self.assertRaises(errors.LockContention, new_repo.lock_write)
1633
request = smart_repo.SmartServerRepositoryUnlock(backing)
1634
response = request.execute('', nonce)
1636
def test_lock_write_on_locked_repo(self):
1637
backing = self.get_transport()
1638
request = smart_repo.SmartServerRepositoryLockWrite(backing)
1639
repository = self.make_repository('.', format='knit')
1640
repo_token = repository.lock_write().repository_token
1641
repository.leave_lock_in_place()
1643
response = request.execute('')
1645
smart_req.SmartServerResponse(('LockContention',)), response)
1647
repository.lock_write(repo_token)
1648
repository.dont_leave_lock_in_place()
1651
def test_lock_write_on_readonly_transport(self):
1652
backing = self.get_readonly_transport()
1653
request = smart_repo.SmartServerRepositoryLockWrite(backing)
1654
repository = self.make_repository('.', format='knit')
1655
response = request.execute('')
1656
self.assertFalse(response.is_successful())
1657
self.assertEqual('LockFailed', response.args[0])
1660
class TestInsertStreamBase(tests.TestCaseWithMemoryTransport):
1662
def make_empty_byte_stream(self, repo):
1663
byte_stream = smart_repo._stream_to_byte_stream([], repo._format)
1664
return ''.join(byte_stream)
1667
class TestSmartServerRepositoryInsertStream(TestInsertStreamBase):
1669
def test_insert_stream_empty(self):
1670
backing = self.get_transport()
1671
request = smart_repo.SmartServerRepositoryInsertStream(backing)
1672
repository = self.make_repository('.')
1673
response = request.execute('', '')
1674
self.assertEqual(None, response)
1675
response = request.do_chunk(self.make_empty_byte_stream(repository))
1676
self.assertEqual(None, response)
1677
response = request.do_end()
1678
self.assertEqual(smart_req.SmartServerResponse(('ok', )), response)
1681
class TestSmartServerRepositoryInsertStreamLocked(TestInsertStreamBase):
1683
def test_insert_stream_empty(self):
1684
backing = self.get_transport()
1685
request = smart_repo.SmartServerRepositoryInsertStreamLocked(
1687
repository = self.make_repository('.', format='knit')
1688
lock_token = repository.lock_write().repository_token
1689
response = request.execute('', '', lock_token)
1690
self.assertEqual(None, response)
1691
response = request.do_chunk(self.make_empty_byte_stream(repository))
1692
self.assertEqual(None, response)
1693
response = request.do_end()
1694
self.assertEqual(smart_req.SmartServerResponse(('ok', )), response)
1697
def test_insert_stream_with_wrong_lock_token(self):
1698
backing = self.get_transport()
1699
request = smart_repo.SmartServerRepositoryInsertStreamLocked(
1701
repository = self.make_repository('.', format='knit')
1702
lock_token = repository.lock_write().repository_token
1704
errors.TokenMismatch, request.execute, '', '', 'wrong-token')
1708
class TestSmartServerRepositoryUnlock(tests.TestCaseWithMemoryTransport):
1711
tests.TestCaseWithMemoryTransport.setUp(self)
1713
def test_unlock_on_locked_repo(self):
1714
backing = self.get_transport()
1715
request = smart_repo.SmartServerRepositoryUnlock(backing)
1716
repository = self.make_repository('.', format='knit')
1717
token = repository.lock_write().repository_token
1718
repository.leave_lock_in_place()
1720
response = request.execute('', token)
1722
smart_req.SmartServerResponse(('ok',)), response)
1723
# The repository is now unlocked. Verify that with a new repository
1725
new_repo = repository.bzrdir.open_repository()
1726
new_repo.lock_write()
1729
def test_unlock_on_unlocked_repo(self):
1730
backing = self.get_transport()
1731
request = smart_repo.SmartServerRepositoryUnlock(backing)
1732
repository = self.make_repository('.', format='knit')
1733
response = request.execute('', 'some token')
1735
smart_req.SmartServerResponse(('TokenMismatch',)), response)
1738
class TestSmartServerIsReadonly(tests.TestCaseWithMemoryTransport):
1740
def test_is_readonly_no(self):
1741
backing = self.get_transport()
1742
request = smart_req.SmartServerIsReadonly(backing)
1743
response = request.execute()
1745
smart_req.SmartServerResponse(('no',)), response)
1747
def test_is_readonly_yes(self):
1748
backing = self.get_readonly_transport()
1749
request = smart_req.SmartServerIsReadonly(backing)
1750
response = request.execute()
1752
smart_req.SmartServerResponse(('yes',)), response)
1755
class TestSmartServerRepositorySetMakeWorkingTrees(
1756
tests.TestCaseWithMemoryTransport):
1758
def test_set_false(self):
1759
backing = self.get_transport()
1760
repo = self.make_repository('.', shared=True)
1761
repo.set_make_working_trees(True)
1762
request_class = smart_repo.SmartServerRepositorySetMakeWorkingTrees
1763
request = request_class(backing)
1764
self.assertEqual(smart_req.SuccessfulSmartServerResponse(('ok',)),
1765
request.execute('', 'False'))
1766
repo = repo.bzrdir.open_repository()
1767
self.assertFalse(repo.make_working_trees())
1769
def test_set_true(self):
1770
backing = self.get_transport()
1771
repo = self.make_repository('.', shared=True)
1772
repo.set_make_working_trees(False)
1773
request_class = smart_repo.SmartServerRepositorySetMakeWorkingTrees
1774
request = request_class(backing)
1775
self.assertEqual(smart_req.SuccessfulSmartServerResponse(('ok',)),
1776
request.execute('', 'True'))
1777
repo = repo.bzrdir.open_repository()
1778
self.assertTrue(repo.make_working_trees())
1781
class TestSmartServerPackRepositoryAutopack(tests.TestCaseWithTransport):
1783
def make_repo_needing_autopacking(self, path='.'):
1784
# Make a repo in need of autopacking.
1785
tree = self.make_branch_and_tree('.', format='pack-0.92')
1786
repo = tree.branch.repository
1787
# monkey-patch the pack collection to disable autopacking
1788
repo._pack_collection._max_pack_count = lambda count: count
1790
tree.commit('commit %s' % x)
1791
self.assertEqual(10, len(repo._pack_collection.names()))
1792
del repo._pack_collection._max_pack_count
1795
def test_autopack_needed(self):
1796
repo = self.make_repo_needing_autopacking()
1798
self.addCleanup(repo.unlock)
1799
backing = self.get_transport()
1800
request = smart_packrepo.SmartServerPackRepositoryAutopack(
1802
response = request.execute('')
1803
self.assertEqual(smart_req.SmartServerResponse(('ok',)), response)
1804
repo._pack_collection.reload_pack_names()
1805
self.assertEqual(1, len(repo._pack_collection.names()))
1807
def test_autopack_not_needed(self):
1808
tree = self.make_branch_and_tree('.', format='pack-0.92')
1809
repo = tree.branch.repository
1811
self.addCleanup(repo.unlock)
1813
tree.commit('commit %s' % x)
1814
backing = self.get_transport()
1815
request = smart_packrepo.SmartServerPackRepositoryAutopack(
1817
response = request.execute('')
1818
self.assertEqual(smart_req.SmartServerResponse(('ok',)), response)
1819
repo._pack_collection.reload_pack_names()
1820
self.assertEqual(9, len(repo._pack_collection.names()))
1822
def test_autopack_on_nonpack_format(self):
1823
"""A request to autopack a non-pack repo is a no-op."""
1824
repo = self.make_repository('.', format='knit')
1825
backing = self.get_transport()
1826
request = smart_packrepo.SmartServerPackRepositoryAutopack(
1828
response = request.execute('')
1829
self.assertEqual(smart_req.SmartServerResponse(('ok',)), response)
1832
class TestSmartServerVfsGet(tests.TestCaseWithMemoryTransport):
1834
def test_unicode_path(self):
1835
"""VFS requests expect unicode paths to be escaped."""
1836
filename = u'foo\N{INTERROBANG}'
1837
filename_escaped = urlutils.escape(filename)
1838
backing = self.get_transport()
1839
request = vfs.GetRequest(backing)
1840
backing.put_bytes_non_atomic(filename_escaped, 'contents')
1841
self.assertEqual(smart_req.SmartServerResponse(('ok', ), 'contents'),
1842
request.execute(filename_escaped))
1845
class TestHandlers(tests.TestCase):
1846
"""Tests for the request.request_handlers object."""
1848
def test_all_registrations_exist(self):
1849
"""All registered request_handlers can be found."""
1850
# If there's a typo in a register_lazy call, this loop will fail with
1851
# an AttributeError.
1852
for key, item in smart_req.request_handlers.iteritems():
1855
def assertHandlerEqual(self, verb, handler):
1856
self.assertEqual(smart_req.request_handlers.get(verb), handler)
1858
def test_registered_methods(self):
1859
"""Test that known methods are registered to the correct object."""
1860
self.assertHandlerEqual('Branch.get_config_file',
1861
smart_branch.SmartServerBranchGetConfigFile)
1862
self.assertHandlerEqual('Branch.get_parent',
1863
smart_branch.SmartServerBranchGetParent)
1864
self.assertHandlerEqual('Branch.get_tags_bytes',
1865
smart_branch.SmartServerBranchGetTagsBytes)
1866
self.assertHandlerEqual('Branch.lock_write',
1867
smart_branch.SmartServerBranchRequestLockWrite)
1868
self.assertHandlerEqual('Branch.last_revision_info',
1869
smart_branch.SmartServerBranchRequestLastRevisionInfo)
1870
self.assertHandlerEqual('Branch.revision_history',
1871
smart_branch.SmartServerRequestRevisionHistory)
1872
self.assertHandlerEqual('Branch.set_config_option',
1873
smart_branch.SmartServerBranchRequestSetConfigOption)
1874
self.assertHandlerEqual('Branch.set_last_revision',
1875
smart_branch.SmartServerBranchRequestSetLastRevision)
1876
self.assertHandlerEqual('Branch.set_last_revision_info',
1877
smart_branch.SmartServerBranchRequestSetLastRevisionInfo)
1878
self.assertHandlerEqual('Branch.set_last_revision_ex',
1879
smart_branch.SmartServerBranchRequestSetLastRevisionEx)
1880
self.assertHandlerEqual('Branch.set_parent_location',
1881
smart_branch.SmartServerBranchRequestSetParentLocation)
1882
self.assertHandlerEqual('Branch.unlock',
1883
smart_branch.SmartServerBranchRequestUnlock)
1884
self.assertHandlerEqual('BzrDir.find_repository',
1885
smart_dir.SmartServerRequestFindRepositoryV1)
1886
self.assertHandlerEqual('BzrDir.find_repositoryV2',
1887
smart_dir.SmartServerRequestFindRepositoryV2)
1888
self.assertHandlerEqual('BzrDirFormat.initialize',
1889
smart_dir.SmartServerRequestInitializeBzrDir)
1890
self.assertHandlerEqual('BzrDirFormat.initialize_ex_1.16',
1891
smart_dir.SmartServerRequestBzrDirInitializeEx)
1892
self.assertHandlerEqual('BzrDir.cloning_metadir',
1893
smart_dir.SmartServerBzrDirRequestCloningMetaDir)
1894
self.assertHandlerEqual('BzrDir.get_config_file',
1895
smart_dir.SmartServerBzrDirRequestConfigFile)
1896
self.assertHandlerEqual('BzrDir.open_branch',
1897
smart_dir.SmartServerRequestOpenBranch)
1898
self.assertHandlerEqual('BzrDir.open_branchV2',
1899
smart_dir.SmartServerRequestOpenBranchV2)
1900
self.assertHandlerEqual('BzrDir.open_branchV3',
1901
smart_dir.SmartServerRequestOpenBranchV3)
1902
self.assertHandlerEqual('PackRepository.autopack',
1903
smart_packrepo.SmartServerPackRepositoryAutopack)
1904
self.assertHandlerEqual('Repository.gather_stats',
1905
smart_repo.SmartServerRepositoryGatherStats)
1906
self.assertHandlerEqual('Repository.get_parent_map',
1907
smart_repo.SmartServerRepositoryGetParentMap)
1908
self.assertHandlerEqual('Repository.get_rev_id_for_revno',
1909
smart_repo.SmartServerRepositoryGetRevIdForRevno)
1910
self.assertHandlerEqual('Repository.get_revision_graph',
1911
smart_repo.SmartServerRepositoryGetRevisionGraph)
1912
self.assertHandlerEqual('Repository.get_stream',
1913
smart_repo.SmartServerRepositoryGetStream)
1914
self.assertHandlerEqual('Repository.has_revision',
1915
smart_repo.SmartServerRequestHasRevision)
1916
self.assertHandlerEqual('Repository.insert_stream',
1917
smart_repo.SmartServerRepositoryInsertStream)
1918
self.assertHandlerEqual('Repository.insert_stream_locked',
1919
smart_repo.SmartServerRepositoryInsertStreamLocked)
1920
self.assertHandlerEqual('Repository.is_shared',
1921
smart_repo.SmartServerRepositoryIsShared)
1922
self.assertHandlerEqual('Repository.lock_write',
1923
smart_repo.SmartServerRepositoryLockWrite)
1924
self.assertHandlerEqual('Repository.tarball',
1925
smart_repo.SmartServerRepositoryTarball)
1926
self.assertHandlerEqual('Repository.unlock',
1927
smart_repo.SmartServerRepositoryUnlock)
1928
self.assertHandlerEqual('Transport.is_readonly',
1929
smart_req.SmartServerIsReadonly)
added
removed
bzrlib/tests/test_smart.py
