1
# Copyright (C) 2006-2011 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.
31
branch as _mod_branch,
40
from bzrlib.smart import (
41
branch as smart_branch,
43
repository as smart_repo,
44
packrepository as smart_packrepo,
49
from bzrlib.testament import Testament
50
from bzrlib.tests import test_server
51
from bzrlib.transport import (
57
def load_tests(standard_tests, module, loader):
58
"""Multiply tests version and protocol consistency."""
59
# FindRepository tests.
62
"_request_class": smart_dir.SmartServerRequestFindRepositoryV1}),
63
("find_repositoryV2", {
64
"_request_class": smart_dir.SmartServerRequestFindRepositoryV2}),
65
("find_repositoryV3", {
66
"_request_class": smart_dir.SmartServerRequestFindRepositoryV3}),
68
to_adapt, result = tests.split_suite_by_re(standard_tests,
69
"TestSmartServerRequestFindRepository")
70
v2_only, v1_and_2 = tests.split_suite_by_re(to_adapt,
72
tests.multiply_tests(v1_and_2, scenarios, result)
73
# The first scenario is only applicable to v1 protocols, it is deleted
75
tests.multiply_tests(v2_only, scenarios[1:], result)
79
class TestCaseWithChrootedTransport(tests.TestCaseWithTransport):
82
self.vfs_transport_factory = memory.MemoryServer
83
tests.TestCaseWithTransport.setUp(self)
84
self._chroot_server = None
86
def get_transport(self, relpath=None):
87
if self._chroot_server is None:
88
backing_transport = tests.TestCaseWithTransport.get_transport(self)
89
self._chroot_server = chroot.ChrootServer(backing_transport)
90
self.start_server(self._chroot_server)
91
t = transport.get_transport_from_url(self._chroot_server.get_url())
92
if relpath is not None:
97
class TestCaseWithSmartMedium(tests.TestCaseWithMemoryTransport):
100
super(TestCaseWithSmartMedium, self).setUp()
101
# We're allowed to set the transport class here, so that we don't use
102
# the default or a parameterized class, but rather use the
103
# TestCaseWithTransport infrastructure to set up a smart server and
105
self.overrideAttr(self, "transport_server", self.make_transport_server)
107
def make_transport_server(self):
108
return test_server.SmartTCPServer_for_testing('-' + self.id())
110
def get_smart_medium(self):
111
"""Get a smart medium to use in tests."""
112
return self.get_transport().get_smart_medium()
115
class TestByteStreamToStream(tests.TestCase):
117
def test_repeated_substreams_same_kind_are_one_stream(self):
118
# Make a stream - an iterable of bytestrings.
119
stream = [('text', [versionedfile.FulltextContentFactory(('k1',), None,
120
None, 'foo')]),('text', [
121
versionedfile.FulltextContentFactory(('k2',), None, None, 'bar')])]
122
fmt = bzrdir.format_registry.get('pack-0.92')().repository_format
123
bytes = smart_repo._stream_to_byte_stream(stream, fmt)
125
# Iterate the resulting iterable; checking that we get only one stream
127
fmt, stream = smart_repo._byte_stream_to_stream(bytes)
128
for kind, substream in stream:
129
streams.append((kind, list(substream)))
130
self.assertLength(1, streams)
131
self.assertLength(2, streams[0][1])
134
class TestSmartServerResponse(tests.TestCase):
136
def test__eq__(self):
137
self.assertEqual(smart_req.SmartServerResponse(('ok', )),
138
smart_req.SmartServerResponse(('ok', )))
139
self.assertEqual(smart_req.SmartServerResponse(('ok', ), 'body'),
140
smart_req.SmartServerResponse(('ok', ), 'body'))
141
self.assertNotEqual(smart_req.SmartServerResponse(('ok', )),
142
smart_req.SmartServerResponse(('notok', )))
143
self.assertNotEqual(smart_req.SmartServerResponse(('ok', ), 'body'),
144
smart_req.SmartServerResponse(('ok', )))
145
self.assertNotEqual(None,
146
smart_req.SmartServerResponse(('ok', )))
148
def test__str__(self):
149
"""SmartServerResponses can be stringified."""
151
"<SuccessfulSmartServerResponse args=('args',) body='body'>",
152
str(smart_req.SuccessfulSmartServerResponse(('args',), 'body')))
154
"<FailedSmartServerResponse args=('args',) body='body'>",
155
str(smart_req.FailedSmartServerResponse(('args',), 'body')))
158
class TestSmartServerRequest(tests.TestCaseWithMemoryTransport):
160
def test_translate_client_path(self):
161
transport = self.get_transport()
162
request = smart_req.SmartServerRequest(transport, 'foo/')
163
self.assertEqual('./', request.translate_client_path('foo/'))
165
errors.InvalidURLJoin, request.translate_client_path, 'foo/..')
167
errors.PathNotChild, request.translate_client_path, '/')
169
errors.PathNotChild, request.translate_client_path, 'bar/')
170
self.assertEqual('./baz', request.translate_client_path('foo/baz'))
171
e_acute = u'\N{LATIN SMALL LETTER E WITH ACUTE}'.encode('utf-8')
172
self.assertEqual('./' + urlutils.escape(e_acute),
173
request.translate_client_path('foo/' + e_acute))
175
def test_translate_client_path_vfs(self):
176
"""VfsRequests receive escaped paths rather than raw UTF-8."""
177
transport = self.get_transport()
178
request = vfs.VfsRequest(transport, 'foo/')
179
e_acute = u'\N{LATIN SMALL LETTER E WITH ACUTE}'.encode('utf-8')
180
escaped = urlutils.escape('foo/' + e_acute)
181
self.assertEqual('./' + urlutils.escape(e_acute),
182
request.translate_client_path(escaped))
184
def test_transport_from_client_path(self):
185
transport = self.get_transport()
186
request = smart_req.SmartServerRequest(transport, 'foo/')
189
request.transport_from_client_path('foo/').base)
192
class TestSmartServerBzrDirRequestCloningMetaDir(
193
tests.TestCaseWithMemoryTransport):
194
"""Tests for BzrDir.cloning_metadir."""
196
def test_cloning_metadir(self):
197
"""When there is a bzrdir present, the call succeeds."""
198
backing = self.get_transport()
199
dir = self.make_bzrdir('.')
200
local_result = dir.cloning_metadir()
201
request_class = smart_dir.SmartServerBzrDirRequestCloningMetaDir
202
request = request_class(backing)
203
expected = smart_req.SuccessfulSmartServerResponse(
204
(local_result.network_name(),
205
local_result.repository_format.network_name(),
206
('branch', local_result.get_branch_format().network_name())))
207
self.assertEqual(expected, request.execute('', 'False'))
209
def test_cloning_metadir_reference(self):
210
"""The request fails when bzrdir contains a branch reference."""
211
backing = self.get_transport()
212
referenced_branch = self.make_branch('referenced')
213
dir = self.make_bzrdir('.')
214
local_result = dir.cloning_metadir()
215
reference = _mod_branch.BranchReferenceFormat().initialize(
216
dir, target_branch=referenced_branch)
217
reference_url = _mod_branch.BranchReferenceFormat().get_reference(dir)
218
# The server shouldn't try to follow the branch reference, so it's fine
219
# if the referenced branch isn't reachable.
220
backing.rename('referenced', 'moved')
221
request_class = smart_dir.SmartServerBzrDirRequestCloningMetaDir
222
request = request_class(backing)
223
expected = smart_req.FailedSmartServerResponse(('BranchReference',))
224
self.assertEqual(expected, request.execute('', 'False'))
227
class TestSmartServerBzrDirRequestDestroyBranch(
228
tests.TestCaseWithMemoryTransport):
229
"""Tests for BzrDir.destroy_branch."""
231
def test_destroy_branch_default(self):
232
"""The default branch can be removed."""
233
backing = self.get_transport()
234
dir = self.make_branch('.').bzrdir
235
request_class = smart_dir.SmartServerBzrDirRequestDestroyBranch
236
request = request_class(backing)
237
expected = smart_req.SuccessfulSmartServerResponse(('ok',))
238
self.assertEqual(expected, request.execute('', None))
240
def test_destroy_branch_named(self):
241
"""A named branch can be removed."""
242
backing = self.get_transport()
243
dir = self.make_repository('.', format="development-colo").bzrdir
244
dir.create_branch(name="branchname")
245
request_class = smart_dir.SmartServerBzrDirRequestDestroyBranch
246
request = request_class(backing)
247
expected = smart_req.SuccessfulSmartServerResponse(('ok',))
248
self.assertEqual(expected, request.execute('', "branchname"))
250
def test_destroy_branch_missing(self):
251
"""An error is raised if the branch didn't exist."""
252
backing = self.get_transport()
253
dir = self.make_bzrdir('.', format="development-colo")
254
request_class = smart_dir.SmartServerBzrDirRequestDestroyBranch
255
request = request_class(backing)
256
expected = smart_req.FailedSmartServerResponse(('nobranch',), None)
257
self.assertEqual(expected, request.execute('', "branchname"))
260
class TestSmartServerBzrDirRequestHasWorkingTree(
261
tests.TestCaseWithTransport):
262
"""Tests for BzrDir.has_workingtree."""
264
def test_has_workingtree_yes(self):
265
"""A working tree is present."""
266
backing = self.get_transport()
267
dir = self.make_branch_and_tree('.').bzrdir
268
request_class = smart_dir.SmartServerBzrDirRequestHasWorkingTree
269
request = request_class(backing)
270
expected = smart_req.SuccessfulSmartServerResponse(('yes',))
271
self.assertEqual(expected, request.execute(''))
273
def test_has_workingtree_no(self):
274
"""A working tree is missing."""
275
backing = self.get_transport()
276
dir = self.make_bzrdir('.')
277
request_class = smart_dir.SmartServerBzrDirRequestHasWorkingTree
278
request = request_class(backing)
279
expected = smart_req.SuccessfulSmartServerResponse(('no',))
280
self.assertEqual(expected, request.execute(''))
283
class TestSmartServerBzrDirRequestDestroyRepository(
284
tests.TestCaseWithMemoryTransport):
285
"""Tests for BzrDir.destroy_repository."""
287
def test_destroy_repository_default(self):
288
"""The repository can be removed."""
289
backing = self.get_transport()
290
dir = self.make_repository('.').bzrdir
291
request_class = smart_dir.SmartServerBzrDirRequestDestroyRepository
292
request = request_class(backing)
293
expected = smart_req.SuccessfulSmartServerResponse(('ok',))
294
self.assertEqual(expected, request.execute(''))
296
def test_destroy_repository_missing(self):
297
"""An error is raised if the repository didn't exist."""
298
backing = self.get_transport()
299
dir = self.make_bzrdir('.')
300
request_class = smart_dir.SmartServerBzrDirRequestDestroyRepository
301
request = request_class(backing)
302
expected = smart_req.FailedSmartServerResponse(
303
('norepository',), None)
304
self.assertEqual(expected, request.execute(''))
307
class TestSmartServerRequestCreateRepository(tests.TestCaseWithMemoryTransport):
308
"""Tests for BzrDir.create_repository."""
310
def test_makes_repository(self):
311
"""When there is a bzrdir present, the call succeeds."""
312
backing = self.get_transport()
313
self.make_bzrdir('.')
314
request_class = smart_dir.SmartServerRequestCreateRepository
315
request = request_class(backing)
316
reference_bzrdir_format = bzrdir.format_registry.get('pack-0.92')()
317
reference_format = reference_bzrdir_format.repository_format
318
network_name = reference_format.network_name()
319
expected = smart_req.SuccessfulSmartServerResponse(
320
('ok', 'no', 'no', 'no', network_name))
321
self.assertEqual(expected, request.execute('', network_name, 'True'))
324
class TestSmartServerRequestFindRepository(tests.TestCaseWithMemoryTransport):
325
"""Tests for BzrDir.find_repository."""
327
def test_no_repository(self):
328
"""When there is no repository to be found, ('norepository', ) is returned."""
329
backing = self.get_transport()
330
request = self._request_class(backing)
331
self.make_bzrdir('.')
332
self.assertEqual(smart_req.SmartServerResponse(('norepository', )),
335
def test_nonshared_repository(self):
336
# nonshared repositorys only allow 'find' to return a handle when the
337
# path the repository is being searched on is the same as that that
338
# the repository is at.
339
backing = self.get_transport()
340
request = self._request_class(backing)
341
result = self._make_repository_and_result()
342
self.assertEqual(result, request.execute(''))
343
self.make_bzrdir('subdir')
344
self.assertEqual(smart_req.SmartServerResponse(('norepository', )),
345
request.execute('subdir'))
347
def _make_repository_and_result(self, shared=False, format=None):
348
"""Convenience function to setup a repository.
350
:result: The SmartServerResponse to expect when opening it.
352
repo = self.make_repository('.', shared=shared, format=format)
353
if repo.supports_rich_root():
357
if repo._format.supports_tree_reference:
361
if repo._format.supports_external_lookups:
365
if (smart_dir.SmartServerRequestFindRepositoryV3 ==
366
self._request_class):
367
return smart_req.SuccessfulSmartServerResponse(
368
('ok', '', rich_root, subtrees, external,
369
repo._format.network_name()))
370
elif (smart_dir.SmartServerRequestFindRepositoryV2 ==
371
self._request_class):
372
# All tests so far are on formats, and for non-external
374
return smart_req.SuccessfulSmartServerResponse(
375
('ok', '', rich_root, subtrees, external))
377
return smart_req.SuccessfulSmartServerResponse(
378
('ok', '', rich_root, subtrees))
380
def test_shared_repository(self):
381
"""When there is a shared repository, we get 'ok', 'relpath-to-repo'."""
382
backing = self.get_transport()
383
request = self._request_class(backing)
384
result = self._make_repository_and_result(shared=True)
385
self.assertEqual(result, request.execute(''))
386
self.make_bzrdir('subdir')
387
result2 = smart_req.SmartServerResponse(
388
result.args[0:1] + ('..', ) + result.args[2:])
389
self.assertEqual(result2,
390
request.execute('subdir'))
391
self.make_bzrdir('subdir/deeper')
392
result3 = smart_req.SmartServerResponse(
393
result.args[0:1] + ('../..', ) + result.args[2:])
394
self.assertEqual(result3,
395
request.execute('subdir/deeper'))
397
def test_rich_root_and_subtree_encoding(self):
398
"""Test for the format attributes for rich root and subtree support."""
399
backing = self.get_transport()
400
request = self._request_class(backing)
401
result = self._make_repository_and_result(
402
format='dirstate-with-subtree')
403
# check the test will be valid
404
self.assertEqual('yes', result.args[2])
405
self.assertEqual('yes', result.args[3])
406
self.assertEqual(result, request.execute(''))
408
def test_supports_external_lookups_no_v2(self):
409
"""Test for the supports_external_lookups attribute."""
410
backing = self.get_transport()
411
request = self._request_class(backing)
412
result = self._make_repository_and_result(
413
format='dirstate-with-subtree')
414
# check the test will be valid
415
self.assertEqual('no', result.args[4])
416
self.assertEqual(result, request.execute(''))
419
class TestSmartServerBzrDirRequestGetConfigFile(
420
tests.TestCaseWithMemoryTransport):
421
"""Tests for BzrDir.get_config_file."""
423
def test_present(self):
424
backing = self.get_transport()
425
dir = self.make_bzrdir('.')
426
dir.get_config().set_default_stack_on("/")
427
local_result = dir._get_config()._get_config_file().read()
428
request_class = smart_dir.SmartServerBzrDirRequestConfigFile
429
request = request_class(backing)
430
expected = smart_req.SuccessfulSmartServerResponse((), local_result)
431
self.assertEqual(expected, request.execute(''))
433
def test_missing(self):
434
backing = self.get_transport()
435
dir = self.make_bzrdir('.')
436
request_class = smart_dir.SmartServerBzrDirRequestConfigFile
437
request = request_class(backing)
438
expected = smart_req.SuccessfulSmartServerResponse((), '')
439
self.assertEqual(expected, request.execute(''))
442
class TestSmartServerRequestInitializeBzrDir(tests.TestCaseWithMemoryTransport):
444
def test_empty_dir(self):
445
"""Initializing an empty dir should succeed and do it."""
446
backing = self.get_transport()
447
request = smart_dir.SmartServerRequestInitializeBzrDir(backing)
448
self.assertEqual(smart_req.SmartServerResponse(('ok', )),
450
made_dir = bzrdir.BzrDir.open_from_transport(backing)
451
# no branch, tree or repository is expected with the current
453
self.assertRaises(errors.NoWorkingTree, made_dir.open_workingtree)
454
self.assertRaises(errors.NotBranchError, made_dir.open_branch)
455
self.assertRaises(errors.NoRepositoryPresent, made_dir.open_repository)
457
def test_missing_dir(self):
458
"""Initializing a missing directory should fail like the bzrdir api."""
459
backing = self.get_transport()
460
request = smart_dir.SmartServerRequestInitializeBzrDir(backing)
461
self.assertRaises(errors.NoSuchFile,
462
request.execute, 'subdir')
464
def test_initialized_dir(self):
465
"""Initializing an extant bzrdir should fail like the bzrdir api."""
466
backing = self.get_transport()
467
request = smart_dir.SmartServerRequestInitializeBzrDir(backing)
468
self.make_bzrdir('subdir')
469
self.assertRaises(errors.FileExists,
470
request.execute, 'subdir')
473
class TestSmartServerRequestBzrDirInitializeEx(
474
tests.TestCaseWithMemoryTransport):
475
"""Basic tests for BzrDir.initialize_ex_1.16 in the smart server.
477
The main unit tests in test_bzrdir exercise the API comprehensively.
480
def test_empty_dir(self):
481
"""Initializing an empty dir should succeed and do it."""
482
backing = self.get_transport()
483
name = self.make_bzrdir('reference')._format.network_name()
484
request = smart_dir.SmartServerRequestBzrDirInitializeEx(backing)
486
smart_req.SmartServerResponse(('', '', '', '', '', '', name,
487
'False', '', '', '')),
488
request.execute(name, '', 'True', 'False', 'False', '', '', '', '',
490
made_dir = bzrdir.BzrDir.open_from_transport(backing)
491
# no branch, tree or repository is expected with the current
493
self.assertRaises(errors.NoWorkingTree, made_dir.open_workingtree)
494
self.assertRaises(errors.NotBranchError, made_dir.open_branch)
495
self.assertRaises(errors.NoRepositoryPresent, made_dir.open_repository)
497
def test_missing_dir(self):
498
"""Initializing a missing directory should fail like the bzrdir api."""
499
backing = self.get_transport()
500
name = self.make_bzrdir('reference')._format.network_name()
501
request = smart_dir.SmartServerRequestBzrDirInitializeEx(backing)
502
self.assertRaises(errors.NoSuchFile, request.execute, name,
503
'subdir/dir', 'False', 'False', 'False', '', '', '', '', 'False')
505
def test_initialized_dir(self):
506
"""Initializing an extant directory should fail like the bzrdir api."""
507
backing = self.get_transport()
508
name = self.make_bzrdir('reference')._format.network_name()
509
request = smart_dir.SmartServerRequestBzrDirInitializeEx(backing)
510
self.make_bzrdir('subdir')
511
self.assertRaises(errors.FileExists, request.execute, name, 'subdir',
512
'False', 'False', 'False', '', '', '', '', 'False')
515
class TestSmartServerRequestOpenBzrDir(tests.TestCaseWithMemoryTransport):
517
def test_no_directory(self):
518
backing = self.get_transport()
519
request = smart_dir.SmartServerRequestOpenBzrDir(backing)
520
self.assertEqual(smart_req.SmartServerResponse(('no', )),
521
request.execute('does-not-exist'))
523
def test_empty_directory(self):
524
backing = self.get_transport()
525
backing.mkdir('empty')
526
request = smart_dir.SmartServerRequestOpenBzrDir(backing)
527
self.assertEqual(smart_req.SmartServerResponse(('no', )),
528
request.execute('empty'))
530
def test_outside_root_client_path(self):
531
backing = self.get_transport()
532
request = smart_dir.SmartServerRequestOpenBzrDir(backing,
533
root_client_path='root')
534
self.assertEqual(smart_req.SmartServerResponse(('no', )),
535
request.execute('not-root'))
538
class TestSmartServerRequestOpenBzrDir_2_1(tests.TestCaseWithMemoryTransport):
540
def test_no_directory(self):
541
backing = self.get_transport()
542
request = smart_dir.SmartServerRequestOpenBzrDir_2_1(backing)
543
self.assertEqual(smart_req.SmartServerResponse(('no', )),
544
request.execute('does-not-exist'))
546
def test_empty_directory(self):
547
backing = self.get_transport()
548
backing.mkdir('empty')
549
request = smart_dir.SmartServerRequestOpenBzrDir_2_1(backing)
550
self.assertEqual(smart_req.SmartServerResponse(('no', )),
551
request.execute('empty'))
553
def test_present_without_workingtree(self):
554
backing = self.get_transport()
555
request = smart_dir.SmartServerRequestOpenBzrDir_2_1(backing)
556
self.make_bzrdir('.')
557
self.assertEqual(smart_req.SmartServerResponse(('yes', 'no')),
560
def test_outside_root_client_path(self):
561
backing = self.get_transport()
562
request = smart_dir.SmartServerRequestOpenBzrDir_2_1(backing,
563
root_client_path='root')
564
self.assertEqual(smart_req.SmartServerResponse(('no',)),
565
request.execute('not-root'))
568
class TestSmartServerRequestOpenBzrDir_2_1_disk(TestCaseWithChrootedTransport):
570
def test_present_with_workingtree(self):
571
self.vfs_transport_factory = test_server.LocalURLServer
572
backing = self.get_transport()
573
request = smart_dir.SmartServerRequestOpenBzrDir_2_1(backing)
574
bd = self.make_bzrdir('.')
575
bd.create_repository()
577
bd.create_workingtree()
578
self.assertEqual(smart_req.SmartServerResponse(('yes', 'yes')),
582
class TestSmartServerRequestOpenBranch(TestCaseWithChrootedTransport):
584
def test_no_branch(self):
585
"""When there is no branch, ('nobranch', ) is returned."""
586
backing = self.get_transport()
587
request = smart_dir.SmartServerRequestOpenBranch(backing)
588
self.make_bzrdir('.')
589
self.assertEqual(smart_req.SmartServerResponse(('nobranch', )),
592
def test_branch(self):
593
"""When there is a branch, 'ok' is returned."""
594
backing = self.get_transport()
595
request = smart_dir.SmartServerRequestOpenBranch(backing)
596
self.make_branch('.')
597
self.assertEqual(smart_req.SmartServerResponse(('ok', '')),
600
def test_branch_reference(self):
601
"""When there is a branch reference, the reference URL is returned."""
602
self.vfs_transport_factory = test_server.LocalURLServer
603
backing = self.get_transport()
604
request = smart_dir.SmartServerRequestOpenBranch(backing)
605
branch = self.make_branch('branch')
606
checkout = branch.create_checkout('reference',lightweight=True)
607
reference_url = _mod_branch.BranchReferenceFormat().get_reference(
609
self.assertFileEqual(reference_url, 'reference/.bzr/branch/location')
610
self.assertEqual(smart_req.SmartServerResponse(('ok', reference_url)),
611
request.execute('reference'))
613
def test_notification_on_branch_from_repository(self):
614
"""When there is a repository, the error should return details."""
615
backing = self.get_transport()
616
request = smart_dir.SmartServerRequestOpenBranch(backing)
617
repo = self.make_repository('.')
618
self.assertEqual(smart_req.SmartServerResponse(('nobranch',)),
622
class TestSmartServerRequestOpenBranchV2(TestCaseWithChrootedTransport):
624
def test_no_branch(self):
625
"""When there is no branch, ('nobranch', ) is returned."""
626
backing = self.get_transport()
627
self.make_bzrdir('.')
628
request = smart_dir.SmartServerRequestOpenBranchV2(backing)
629
self.assertEqual(smart_req.SmartServerResponse(('nobranch', )),
632
def test_branch(self):
633
"""When there is a branch, 'ok' is returned."""
634
backing = self.get_transport()
635
expected = self.make_branch('.')._format.network_name()
636
request = smart_dir.SmartServerRequestOpenBranchV2(backing)
637
self.assertEqual(smart_req.SuccessfulSmartServerResponse(
638
('branch', expected)),
641
def test_branch_reference(self):
642
"""When there is a branch reference, the reference URL is returned."""
643
self.vfs_transport_factory = test_server.LocalURLServer
644
backing = self.get_transport()
645
request = smart_dir.SmartServerRequestOpenBranchV2(backing)
646
branch = self.make_branch('branch')
647
checkout = branch.create_checkout('reference',lightweight=True)
648
reference_url = _mod_branch.BranchReferenceFormat().get_reference(
650
self.assertFileEqual(reference_url, 'reference/.bzr/branch/location')
651
self.assertEqual(smart_req.SuccessfulSmartServerResponse(
652
('ref', reference_url)),
653
request.execute('reference'))
655
def test_stacked_branch(self):
656
"""Opening a stacked branch does not open the stacked-on branch."""
657
trunk = self.make_branch('trunk')
658
feature = self.make_branch('feature')
659
feature.set_stacked_on_url(trunk.base)
661
_mod_branch.Branch.hooks.install_named_hook(
662
'open', opened_branches.append, None)
663
backing = self.get_transport()
664
request = smart_dir.SmartServerRequestOpenBranchV2(backing)
667
response = request.execute('feature')
669
request.teardown_jail()
670
expected_format = feature._format.network_name()
671
self.assertEqual(smart_req.SuccessfulSmartServerResponse(
672
('branch', expected_format)),
674
self.assertLength(1, opened_branches)
676
def test_notification_on_branch_from_repository(self):
677
"""When there is a repository, the error should return details."""
678
backing = self.get_transport()
679
request = smart_dir.SmartServerRequestOpenBranchV2(backing)
680
repo = self.make_repository('.')
681
self.assertEqual(smart_req.SmartServerResponse(('nobranch',)),
685
class TestSmartServerRequestOpenBranchV3(TestCaseWithChrootedTransport):
687
def test_no_branch(self):
688
"""When there is no branch, ('nobranch', ) is returned."""
689
backing = self.get_transport()
690
self.make_bzrdir('.')
691
request = smart_dir.SmartServerRequestOpenBranchV3(backing)
692
self.assertEqual(smart_req.SmartServerResponse(('nobranch',)),
695
def test_branch(self):
696
"""When there is a branch, 'ok' is returned."""
697
backing = self.get_transport()
698
expected = self.make_branch('.')._format.network_name()
699
request = smart_dir.SmartServerRequestOpenBranchV3(backing)
700
self.assertEqual(smart_req.SuccessfulSmartServerResponse(
701
('branch', expected)),
704
def test_branch_reference(self):
705
"""When there is a branch reference, the reference URL is returned."""
706
self.vfs_transport_factory = test_server.LocalURLServer
707
backing = self.get_transport()
708
request = smart_dir.SmartServerRequestOpenBranchV3(backing)
709
branch = self.make_branch('branch')
710
checkout = branch.create_checkout('reference',lightweight=True)
711
reference_url = _mod_branch.BranchReferenceFormat().get_reference(
713
self.assertFileEqual(reference_url, 'reference/.bzr/branch/location')
714
self.assertEqual(smart_req.SuccessfulSmartServerResponse(
715
('ref', reference_url)),
716
request.execute('reference'))
718
def test_stacked_branch(self):
719
"""Opening a stacked branch does not open the stacked-on branch."""
720
trunk = self.make_branch('trunk')
721
feature = self.make_branch('feature')
722
feature.set_stacked_on_url(trunk.base)
724
_mod_branch.Branch.hooks.install_named_hook(
725
'open', opened_branches.append, None)
726
backing = self.get_transport()
727
request = smart_dir.SmartServerRequestOpenBranchV3(backing)
730
response = request.execute('feature')
732
request.teardown_jail()
733
expected_format = feature._format.network_name()
734
self.assertEqual(smart_req.SuccessfulSmartServerResponse(
735
('branch', expected_format)),
737
self.assertLength(1, opened_branches)
739
def test_notification_on_branch_from_repository(self):
740
"""When there is a repository, the error should return details."""
741
backing = self.get_transport()
742
request = smart_dir.SmartServerRequestOpenBranchV3(backing)
743
repo = self.make_repository('.')
744
self.assertEqual(smart_req.SmartServerResponse(
745
('nobranch', 'location is a repository')),
749
class TestSmartServerRequestRevisionHistory(tests.TestCaseWithMemoryTransport):
751
def test_empty(self):
752
"""For an empty branch, the body is empty."""
753
backing = self.get_transport()
754
request = smart_branch.SmartServerRequestRevisionHistory(backing)
755
self.make_branch('.')
756
self.assertEqual(smart_req.SmartServerResponse(('ok', ), ''),
759
def test_not_empty(self):
760
"""For a non-empty branch, the body is empty."""
761
backing = self.get_transport()
762
request = smart_branch.SmartServerRequestRevisionHistory(backing)
763
tree = self.make_branch_and_memory_tree('.')
766
r1 = tree.commit('1st commit')
767
r2 = tree.commit('2nd commit', rev_id=u'\xc8'.encode('utf-8'))
770
smart_req.SmartServerResponse(('ok', ), ('\x00'.join([r1, r2]))),
774
class TestSmartServerBranchRequest(tests.TestCaseWithMemoryTransport):
776
def test_no_branch(self):
777
"""When there is a bzrdir and no branch, NotBranchError is raised."""
778
backing = self.get_transport()
779
request = smart_branch.SmartServerBranchRequest(backing)
780
self.make_bzrdir('.')
781
self.assertRaises(errors.NotBranchError,
784
def test_branch_reference(self):
785
"""When there is a branch reference, NotBranchError is raised."""
786
backing = self.get_transport()
787
request = smart_branch.SmartServerBranchRequest(backing)
788
branch = self.make_branch('branch')
789
checkout = branch.create_checkout('reference',lightweight=True)
790
self.assertRaises(errors.NotBranchError,
791
request.execute, 'checkout')
794
class TestSmartServerBranchRequestLastRevisionInfo(
795
tests.TestCaseWithMemoryTransport):
797
def test_empty(self):
798
"""For an empty branch, the result is ('ok', '0', 'null:')."""
799
backing = self.get_transport()
800
request = smart_branch.SmartServerBranchRequestLastRevisionInfo(backing)
801
self.make_branch('.')
802
self.assertEqual(smart_req.SmartServerResponse(('ok', '0', 'null:')),
805
def test_not_empty(self):
806
"""For a non-empty branch, the result is ('ok', 'revno', 'revid')."""
807
backing = self.get_transport()
808
request = smart_branch.SmartServerBranchRequestLastRevisionInfo(backing)
809
tree = self.make_branch_and_memory_tree('.')
812
rev_id_utf8 = u'\xc8'.encode('utf-8')
813
r1 = tree.commit('1st commit')
814
r2 = tree.commit('2nd commit', rev_id=rev_id_utf8)
817
smart_req.SmartServerResponse(('ok', '2', rev_id_utf8)),
821
class TestSmartServerBranchRequestRevisionIdToRevno(
822
tests.TestCaseWithMemoryTransport):
825
backing = self.get_transport()
826
request = smart_branch.SmartServerBranchRequestRevisionIdToRevno(
828
self.make_branch('.')
829
self.assertEqual(smart_req.SmartServerResponse(('ok', '0')),
830
request.execute('', 'null:'))
832
def test_simple(self):
833
backing = self.get_transport()
834
request = smart_branch.SmartServerBranchRequestRevisionIdToRevno(
836
tree = self.make_branch_and_memory_tree('.')
839
r1 = tree.commit('1st commit')
842
smart_req.SmartServerResponse(('ok', '1')),
843
request.execute('', r1))
845
def test_not_found(self):
846
backing = self.get_transport()
847
request = smart_branch.SmartServerBranchRequestRevisionIdToRevno(
849
branch = self.make_branch('.')
851
smart_req.FailedSmartServerResponse(
852
('NoSuchRevision', 'idontexist')),
853
request.execute('', 'idontexist'))
856
class TestSmartServerBranchRequestGetConfigFile(
857
tests.TestCaseWithMemoryTransport):
859
def test_default(self):
860
"""With no file, we get empty content."""
861
backing = self.get_transport()
862
request = smart_branch.SmartServerBranchGetConfigFile(backing)
863
branch = self.make_branch('.')
864
# there should be no file by default
866
self.assertEqual(smart_req.SmartServerResponse(('ok', ), content),
869
def test_with_content(self):
870
# SmartServerBranchGetConfigFile should return the content from
871
# branch.control_files.get('branch.conf') for now - in the future it may
872
# perform more complex processing.
873
backing = self.get_transport()
874
request = smart_branch.SmartServerBranchGetConfigFile(backing)
875
branch = self.make_branch('.')
876
branch._transport.put_bytes('branch.conf', 'foo bar baz')
877
self.assertEqual(smart_req.SmartServerResponse(('ok', ), 'foo bar baz'),
881
class TestLockedBranch(tests.TestCaseWithMemoryTransport):
883
def get_lock_tokens(self, branch):
884
branch_token = branch.lock_write().branch_token
885
repo_token = branch.repository.lock_write().repository_token
886
branch.repository.unlock()
887
return branch_token, repo_token
890
class TestSmartServerBranchRequestPutConfigFile(TestLockedBranch):
892
def test_with_content(self):
893
backing = self.get_transport()
894
request = smart_branch.SmartServerBranchPutConfigFile(backing)
895
branch = self.make_branch('.')
896
branch_token, repo_token = self.get_lock_tokens(branch)
897
self.assertIs(None, request.execute('', branch_token, repo_token))
899
smart_req.SmartServerResponse(('ok', )),
900
request.do_body('foo bar baz'))
902
branch.control_transport.get_bytes('branch.conf'),
907
class TestSmartServerBranchRequestSetConfigOption(TestLockedBranch):
909
def test_value_name(self):
910
branch = self.make_branch('.')
911
request = smart_branch.SmartServerBranchRequestSetConfigOption(
912
branch.bzrdir.root_transport)
913
branch_token, repo_token = self.get_lock_tokens(branch)
914
config = branch._get_config()
915
result = request.execute('', branch_token, repo_token, 'bar', 'foo',
917
self.assertEqual(smart_req.SuccessfulSmartServerResponse(()), result)
918
self.assertEqual('bar', config.get_option('foo'))
922
def test_value_name_section(self):
923
branch = self.make_branch('.')
924
request = smart_branch.SmartServerBranchRequestSetConfigOption(
925
branch.bzrdir.root_transport)
926
branch_token, repo_token = self.get_lock_tokens(branch)
927
config = branch._get_config()
928
result = request.execute('', branch_token, repo_token, 'bar', 'foo',
930
self.assertEqual(smart_req.SuccessfulSmartServerResponse(()), result)
931
self.assertEqual('bar', config.get_option('foo', 'gam'))
936
class TestSmartServerBranchRequestSetConfigOptionDict(TestLockedBranch):
939
TestLockedBranch.setUp(self)
940
# A dict with non-ascii keys and values to exercise unicode
942
self.encoded_value_dict = (
943
'd5:ascii1:a11:unicode \xe2\x8c\x9a3:\xe2\x80\xbde')
945
'ascii': 'a', u'unicode \N{WATCH}': u'\N{INTERROBANG}'}
947
def test_value_name(self):
948
branch = self.make_branch('.')
949
request = smart_branch.SmartServerBranchRequestSetConfigOptionDict(
950
branch.bzrdir.root_transport)
951
branch_token, repo_token = self.get_lock_tokens(branch)
952
config = branch._get_config()
953
result = request.execute('', branch_token, repo_token,
954
self.encoded_value_dict, 'foo', '')
955
self.assertEqual(smart_req.SuccessfulSmartServerResponse(()), result)
956
self.assertEqual(self.value_dict, config.get_option('foo'))
960
def test_value_name_section(self):
961
branch = self.make_branch('.')
962
request = smart_branch.SmartServerBranchRequestSetConfigOptionDict(
963
branch.bzrdir.root_transport)
964
branch_token, repo_token = self.get_lock_tokens(branch)
965
config = branch._get_config()
966
result = request.execute('', branch_token, repo_token,
967
self.encoded_value_dict, 'foo', 'gam')
968
self.assertEqual(smart_req.SuccessfulSmartServerResponse(()), result)
969
self.assertEqual(self.value_dict, config.get_option('foo', 'gam'))
974
class TestSmartServerBranchRequestSetTagsBytes(TestLockedBranch):
975
# Only called when the branch format and tags match [yay factory
976
# methods] so only need to test straight forward cases.
978
def test_set_bytes(self):
979
base_branch = self.make_branch('base')
980
tag_bytes = base_branch._get_tags_bytes()
981
# get_lock_tokens takes out a lock.
982
branch_token, repo_token = self.get_lock_tokens(base_branch)
983
request = smart_branch.SmartServerBranchSetTagsBytes(
984
self.get_transport())
985
response = request.execute('base', branch_token, repo_token)
986
self.assertEqual(None, response)
987
response = request.do_chunk(tag_bytes)
988
self.assertEqual(None, response)
989
response = request.do_end()
991
smart_req.SuccessfulSmartServerResponse(()), response)
994
def test_lock_failed(self):
995
base_branch = self.make_branch('base')
996
base_branch.lock_write()
997
tag_bytes = base_branch._get_tags_bytes()
998
request = smart_branch.SmartServerBranchSetTagsBytes(
999
self.get_transport())
1000
self.assertRaises(errors.TokenMismatch, request.execute,
1001
'base', 'wrong token', 'wrong token')
1002
# The request handler will keep processing the message parts, so even
1003
# if the request fails immediately do_chunk and do_end are still
1005
request.do_chunk(tag_bytes)
1007
base_branch.unlock()
1011
class SetLastRevisionTestBase(TestLockedBranch):
1012
"""Base test case for verbs that implement set_last_revision."""
1015
tests.TestCaseWithMemoryTransport.setUp(self)
1016
backing_transport = self.get_transport()
1017
self.request = self.request_class(backing_transport)
1018
self.tree = self.make_branch_and_memory_tree('.')
1020
def lock_branch(self):
1021
return self.get_lock_tokens(self.tree.branch)
1023
def unlock_branch(self):
1024
self.tree.branch.unlock()
1026
def set_last_revision(self, revision_id, revno):
1027
branch_token, repo_token = self.lock_branch()
1028
response = self._set_last_revision(
1029
revision_id, revno, branch_token, repo_token)
1030
self.unlock_branch()
1033
def assertRequestSucceeds(self, revision_id, revno):
1034
response = self.set_last_revision(revision_id, revno)
1035
self.assertEqual(smart_req.SuccessfulSmartServerResponse(('ok',)),
1039
class TestSetLastRevisionVerbMixin(object):
1040
"""Mixin test case for verbs that implement set_last_revision."""
1042
def test_set_null_to_null(self):
1043
"""An empty branch can have its last revision set to 'null:'."""
1044
self.assertRequestSucceeds('null:', 0)
1046
def test_NoSuchRevision(self):
1047
"""If the revision_id is not present, the verb returns NoSuchRevision.
1049
revision_id = 'non-existent revision'
1050
self.assertEqual(smart_req.FailedSmartServerResponse(('NoSuchRevision',
1052
self.set_last_revision(revision_id, 1))
1054
def make_tree_with_two_commits(self):
1055
self.tree.lock_write()
1057
rev_id_utf8 = u'\xc8'.encode('utf-8')
1058
r1 = self.tree.commit('1st commit', rev_id=rev_id_utf8)
1059
r2 = self.tree.commit('2nd commit', rev_id='rev-2')
1062
def test_branch_last_revision_info_is_updated(self):
1063
"""A branch's tip can be set to a revision that is present in its
1066
# Make a branch with an empty revision history, but two revisions in
1068
self.make_tree_with_two_commits()
1069
rev_id_utf8 = u'\xc8'.encode('utf-8')
1070
self.tree.branch.set_last_revision_info(0, 'null:')
1072
(0, 'null:'), self.tree.branch.last_revision_info())
1073
# We can update the branch to a revision that is present in the
1075
self.assertRequestSucceeds(rev_id_utf8, 1)
1077
(1, rev_id_utf8), self.tree.branch.last_revision_info())
1079
def test_branch_last_revision_info_rewind(self):
1080
"""A branch's tip can be set to a revision that is an ancestor of the
1083
self.make_tree_with_two_commits()
1084
rev_id_utf8 = u'\xc8'.encode('utf-8')
1086
(2, 'rev-2'), self.tree.branch.last_revision_info())
1087
self.assertRequestSucceeds(rev_id_utf8, 1)
1089
(1, rev_id_utf8), self.tree.branch.last_revision_info())
1091
def test_TipChangeRejected(self):
1092
"""If a pre_change_branch_tip hook raises TipChangeRejected, the verb
1093
returns TipChangeRejected.
1095
rejection_message = u'rejection message\N{INTERROBANG}'
1096
def hook_that_rejects(params):
1097
raise errors.TipChangeRejected(rejection_message)
1098
_mod_branch.Branch.hooks.install_named_hook(
1099
'pre_change_branch_tip', hook_that_rejects, None)
1101
smart_req.FailedSmartServerResponse(
1102
('TipChangeRejected', rejection_message.encode('utf-8'))),
1103
self.set_last_revision('null:', 0))
1106
class TestSmartServerBranchRequestSetLastRevision(
1107
SetLastRevisionTestBase, TestSetLastRevisionVerbMixin):
1108
"""Tests for Branch.set_last_revision verb."""
1110
request_class = smart_branch.SmartServerBranchRequestSetLastRevision
1112
def _set_last_revision(self, revision_id, revno, branch_token, repo_token):
1113
return self.request.execute(
1114
'', branch_token, repo_token, revision_id)
1117
class TestSmartServerBranchRequestSetLastRevisionInfo(
1118
SetLastRevisionTestBase, TestSetLastRevisionVerbMixin):
1119
"""Tests for Branch.set_last_revision_info verb."""
1121
request_class = smart_branch.SmartServerBranchRequestSetLastRevisionInfo
1123
def _set_last_revision(self, revision_id, revno, branch_token, repo_token):
1124
return self.request.execute(
1125
'', branch_token, repo_token, revno, revision_id)
1127
def test_NoSuchRevision(self):
1128
"""Branch.set_last_revision_info does not have to return
1129
NoSuchRevision if the revision_id is absent.
1131
raise tests.TestNotApplicable()
1134
class TestSmartServerBranchRequestSetLastRevisionEx(
1135
SetLastRevisionTestBase, TestSetLastRevisionVerbMixin):
1136
"""Tests for Branch.set_last_revision_ex verb."""
1138
request_class = smart_branch.SmartServerBranchRequestSetLastRevisionEx
1140
def _set_last_revision(self, revision_id, revno, branch_token, repo_token):
1141
return self.request.execute(
1142
'', branch_token, repo_token, revision_id, 0, 0)
1144
def assertRequestSucceeds(self, revision_id, revno):
1145
response = self.set_last_revision(revision_id, revno)
1147
smart_req.SuccessfulSmartServerResponse(('ok', revno, revision_id)),
1150
def test_branch_last_revision_info_rewind(self):
1151
"""A branch's tip can be set to a revision that is an ancestor of the
1152
current tip, but only if allow_overwrite_descendant is passed.
1154
self.make_tree_with_two_commits()
1155
rev_id_utf8 = u'\xc8'.encode('utf-8')
1157
(2, 'rev-2'), self.tree.branch.last_revision_info())
1158
# If allow_overwrite_descendant flag is 0, then trying to set the tip
1159
# to an older revision ID has no effect.
1160
branch_token, repo_token = self.lock_branch()
1161
response = self.request.execute(
1162
'', branch_token, repo_token, rev_id_utf8, 0, 0)
1164
smart_req.SuccessfulSmartServerResponse(('ok', 2, 'rev-2')),
1167
(2, 'rev-2'), self.tree.branch.last_revision_info())
1169
# If allow_overwrite_descendant flag is 1, then setting the tip to an
1171
response = self.request.execute(
1172
'', branch_token, repo_token, rev_id_utf8, 0, 1)
1174
smart_req.SuccessfulSmartServerResponse(('ok', 1, rev_id_utf8)),
1176
self.unlock_branch()
1178
(1, rev_id_utf8), self.tree.branch.last_revision_info())
1180
def make_branch_with_divergent_history(self):
1181
"""Make a branch with divergent history in its repo.
1183
The branch's tip will be 'child-2', and the repo will also contain
1184
'child-1', which diverges from a common base revision.
1186
self.tree.lock_write()
1188
r1 = self.tree.commit('1st commit')
1189
revno_1, revid_1 = self.tree.branch.last_revision_info()
1190
r2 = self.tree.commit('2nd commit', rev_id='child-1')
1191
# Undo the second commit
1192
self.tree.branch.set_last_revision_info(revno_1, revid_1)
1193
self.tree.set_parent_ids([revid_1])
1194
# Make a new second commit, child-2. child-2 has diverged from
1196
new_r2 = self.tree.commit('2nd commit', rev_id='child-2')
1199
def test_not_allow_diverged(self):
1200
"""If allow_diverged is not passed, then setting a divergent history
1201
returns a Diverged error.
1203
self.make_branch_with_divergent_history()
1205
smart_req.FailedSmartServerResponse(('Diverged',)),
1206
self.set_last_revision('child-1', 2))
1207
# The branch tip was not changed.
1208
self.assertEqual('child-2', self.tree.branch.last_revision())
1210
def test_allow_diverged(self):
1211
"""If allow_diverged is passed, then setting a divergent history
1214
self.make_branch_with_divergent_history()
1215
branch_token, repo_token = self.lock_branch()
1216
response = self.request.execute(
1217
'', branch_token, repo_token, 'child-1', 1, 0)
1219
smart_req.SuccessfulSmartServerResponse(('ok', 2, 'child-1')),
1221
self.unlock_branch()
1222
# The branch tip was changed.
1223
self.assertEqual('child-1', self.tree.branch.last_revision())
1226
class TestSmartServerBranchBreakLock(tests.TestCaseWithMemoryTransport):
1228
def test_lock_to_break(self):
1229
base_branch = self.make_branch('base')
1230
request = smart_branch.SmartServerBranchBreakLock(
1231
self.get_transport())
1232
base_branch.lock_write()
1234
smart_req.SuccessfulSmartServerResponse(('ok', ), None),
1235
request.execute('base'))
1237
def test_nothing_to_break(self):
1238
base_branch = self.make_branch('base')
1239
request = smart_branch.SmartServerBranchBreakLock(
1240
self.get_transport())
1242
smart_req.SuccessfulSmartServerResponse(('ok', ), None),
1243
request.execute('base'))
1246
class TestSmartServerBranchRequestGetParent(tests.TestCaseWithMemoryTransport):
1248
def test_get_parent_none(self):
1249
base_branch = self.make_branch('base')
1250
request = smart_branch.SmartServerBranchGetParent(self.get_transport())
1251
response = request.execute('base')
1253
smart_req.SuccessfulSmartServerResponse(('',)), response)
1255
def test_get_parent_something(self):
1256
base_branch = self.make_branch('base')
1257
base_branch.set_parent(self.get_url('foo'))
1258
request = smart_branch.SmartServerBranchGetParent(self.get_transport())
1259
response = request.execute('base')
1261
smart_req.SuccessfulSmartServerResponse(("../foo",)),
1265
class TestSmartServerBranchRequestSetParent(TestLockedBranch):
1267
def test_set_parent_none(self):
1268
branch = self.make_branch('base', format="1.9")
1270
branch._set_parent_location('foo')
1272
request = smart_branch.SmartServerBranchRequestSetParentLocation(
1273
self.get_transport())
1274
branch_token, repo_token = self.get_lock_tokens(branch)
1276
response = request.execute('base', branch_token, repo_token, '')
1279
self.assertEqual(smart_req.SuccessfulSmartServerResponse(()), response)
1280
self.assertEqual(None, branch.get_parent())
1282
def test_set_parent_something(self):
1283
branch = self.make_branch('base', format="1.9")
1284
request = smart_branch.SmartServerBranchRequestSetParentLocation(
1285
self.get_transport())
1286
branch_token, repo_token = self.get_lock_tokens(branch)
1288
response = request.execute('base', branch_token, repo_token,
1292
self.assertEqual(smart_req.SuccessfulSmartServerResponse(()), response)
1293
self.assertEqual('http://bar/', branch.get_parent())
1296
class TestSmartServerBranchRequestGetTagsBytes(
1297
tests.TestCaseWithMemoryTransport):
1298
# Only called when the branch format and tags match [yay factory
1299
# methods] so only need to test straight forward cases.
1301
def test_get_bytes(self):
1302
base_branch = self.make_branch('base')
1303
request = smart_branch.SmartServerBranchGetTagsBytes(
1304
self.get_transport())
1305
response = request.execute('base')
1307
smart_req.SuccessfulSmartServerResponse(('',)), response)
1310
class TestSmartServerBranchRequestGetStackedOnURL(tests.TestCaseWithMemoryTransport):
1312
def test_get_stacked_on_url(self):
1313
base_branch = self.make_branch('base', format='1.6')
1314
stacked_branch = self.make_branch('stacked', format='1.6')
1315
# typically should be relative
1316
stacked_branch.set_stacked_on_url('../base')
1317
request = smart_branch.SmartServerBranchRequestGetStackedOnURL(
1318
self.get_transport())
1319
response = request.execute('stacked')
1321
smart_req.SmartServerResponse(('ok', '../base')),
1325
class TestSmartServerBranchRequestLockWrite(TestLockedBranch):
1328
tests.TestCaseWithMemoryTransport.setUp(self)
1330
def test_lock_write_on_unlocked_branch(self):
1331
backing = self.get_transport()
1332
request = smart_branch.SmartServerBranchRequestLockWrite(backing)
1333
branch = self.make_branch('.', format='knit')
1334
repository = branch.repository
1335
response = request.execute('')
1336
branch_nonce = branch.control_files._lock.peek().get('nonce')
1337
repository_nonce = repository.control_files._lock.peek().get('nonce')
1338
self.assertEqual(smart_req.SmartServerResponse(
1339
('ok', branch_nonce, repository_nonce)),
1341
# The branch (and associated repository) is now locked. Verify that
1342
# with a new branch object.
1343
new_branch = repository.bzrdir.open_branch()
1344
self.assertRaises(errors.LockContention, new_branch.lock_write)
1346
request = smart_branch.SmartServerBranchRequestUnlock(backing)
1347
response = request.execute('', branch_nonce, repository_nonce)
1349
def test_lock_write_on_locked_branch(self):
1350
backing = self.get_transport()
1351
request = smart_branch.SmartServerBranchRequestLockWrite(backing)
1352
branch = self.make_branch('.')
1353
branch_token = branch.lock_write().branch_token
1354
branch.leave_lock_in_place()
1356
response = request.execute('')
1358
smart_req.SmartServerResponse(('LockContention',)), response)
1360
branch.lock_write(branch_token)
1361
branch.dont_leave_lock_in_place()
1364
def test_lock_write_with_tokens_on_locked_branch(self):
1365
backing = self.get_transport()
1366
request = smart_branch.SmartServerBranchRequestLockWrite(backing)
1367
branch = self.make_branch('.', format='knit')
1368
branch_token, repo_token = self.get_lock_tokens(branch)
1369
branch.leave_lock_in_place()
1370
branch.repository.leave_lock_in_place()
1372
response = request.execute('',
1373
branch_token, repo_token)
1375
smart_req.SmartServerResponse(('ok', branch_token, repo_token)),
1378
branch.repository.lock_write(repo_token)
1379
branch.repository.dont_leave_lock_in_place()
1380
branch.repository.unlock()
1381
branch.lock_write(branch_token)
1382
branch.dont_leave_lock_in_place()
1385
def test_lock_write_with_mismatched_tokens_on_locked_branch(self):
1386
backing = self.get_transport()
1387
request = smart_branch.SmartServerBranchRequestLockWrite(backing)
1388
branch = self.make_branch('.', format='knit')
1389
branch_token, repo_token = self.get_lock_tokens(branch)
1390
branch.leave_lock_in_place()
1391
branch.repository.leave_lock_in_place()
1393
response = request.execute('',
1394
branch_token+'xxx', repo_token)
1396
smart_req.SmartServerResponse(('TokenMismatch',)), response)
1398
branch.repository.lock_write(repo_token)
1399
branch.repository.dont_leave_lock_in_place()
1400
branch.repository.unlock()
1401
branch.lock_write(branch_token)
1402
branch.dont_leave_lock_in_place()
1405
def test_lock_write_on_locked_repo(self):
1406
backing = self.get_transport()
1407
request = smart_branch.SmartServerBranchRequestLockWrite(backing)
1408
branch = self.make_branch('.', format='knit')
1409
repo = branch.repository
1410
repo_token = repo.lock_write().repository_token
1411
repo.leave_lock_in_place()
1413
response = request.execute('')
1415
smart_req.SmartServerResponse(('LockContention',)), response)
1417
repo.lock_write(repo_token)
1418
repo.dont_leave_lock_in_place()
1421
def test_lock_write_on_readonly_transport(self):
1422
backing = self.get_readonly_transport()
1423
request = smart_branch.SmartServerBranchRequestLockWrite(backing)
1424
branch = self.make_branch('.')
1425
root = self.get_transport().clone('/')
1426
path = urlutils.relative_url(root.base, self.get_transport().base)
1427
response = request.execute(path)
1428
error_name, lock_str, why_str = response.args
1429
self.assertFalse(response.is_successful())
1430
self.assertEqual('LockFailed', error_name)
1433
class TestSmartServerBranchRequestGetPhysicalLockStatus(TestLockedBranch):
1436
tests.TestCaseWithMemoryTransport.setUp(self)
1438
def test_true(self):
1439
backing = self.get_transport()
1440
request = smart_branch.SmartServerBranchRequestGetPhysicalLockStatus(
1442
branch = self.make_branch('.')
1443
branch_token, repo_token = self.get_lock_tokens(branch)
1444
self.assertEquals(True, branch.get_physical_lock_status())
1445
response = request.execute('')
1447
smart_req.SmartServerResponse(('yes',)), response)
1450
def test_false(self):
1451
backing = self.get_transport()
1452
request = smart_branch.SmartServerBranchRequestGetPhysicalLockStatus(
1454
branch = self.make_branch('.')
1455
self.assertEquals(False, branch.get_physical_lock_status())
1456
response = request.execute('')
1458
smart_req.SmartServerResponse(('no',)), response)
1461
class TestSmartServerBranchRequestUnlock(TestLockedBranch):
1464
tests.TestCaseWithMemoryTransport.setUp(self)
1466
def test_unlock_on_locked_branch_and_repo(self):
1467
backing = self.get_transport()
1468
request = smart_branch.SmartServerBranchRequestUnlock(backing)
1469
branch = self.make_branch('.', format='knit')
1471
branch_token, repo_token = self.get_lock_tokens(branch)
1472
# Unlock the branch (and repo) object, leaving the physical locks
1474
branch.leave_lock_in_place()
1475
branch.repository.leave_lock_in_place()
1477
response = request.execute('',
1478
branch_token, repo_token)
1480
smart_req.SmartServerResponse(('ok',)), response)
1481
# The branch is now unlocked. Verify that with a new branch
1483
new_branch = branch.bzrdir.open_branch()
1484
new_branch.lock_write()
1487
def test_unlock_on_unlocked_branch_unlocked_repo(self):
1488
backing = self.get_transport()
1489
request = smart_branch.SmartServerBranchRequestUnlock(backing)
1490
branch = self.make_branch('.', format='knit')
1491
response = request.execute(
1492
'', 'branch token', 'repo token')
1494
smart_req.SmartServerResponse(('TokenMismatch',)), response)
1496
def test_unlock_on_unlocked_branch_locked_repo(self):
1497
backing = self.get_transport()
1498
request = smart_branch.SmartServerBranchRequestUnlock(backing)
1499
branch = self.make_branch('.', format='knit')
1500
# Lock the repository.
1501
repo_token = branch.repository.lock_write().repository_token
1502
branch.repository.leave_lock_in_place()
1503
branch.repository.unlock()
1504
# Issue branch lock_write request on the unlocked branch (with locked
1506
response = request.execute('', 'branch token', repo_token)
1508
smart_req.SmartServerResponse(('TokenMismatch',)), response)
1510
branch.repository.lock_write(repo_token)
1511
branch.repository.dont_leave_lock_in_place()
1512
branch.repository.unlock()
1515
class TestSmartServerRepositoryRequest(tests.TestCaseWithMemoryTransport):
1517
def test_no_repository(self):
1518
"""Raise NoRepositoryPresent when there is a bzrdir and no repo."""
1519
# we test this using a shared repository above the named path,
1520
# thus checking the right search logic is used - that is, that
1521
# its the exact path being looked at and the server is not
1523
backing = self.get_transport()
1524
request = smart_repo.SmartServerRepositoryRequest(backing)
1525
self.make_repository('.', shared=True)
1526
self.make_bzrdir('subdir')
1527
self.assertRaises(errors.NoRepositoryPresent,
1528
request.execute, 'subdir')
1531
class TestSmartServerRepositoryAddSignatureText(tests.TestCaseWithMemoryTransport):
1533
def test_add_text(self):
1534
backing = self.get_transport()
1535
request = smart_repo.SmartServerRepositoryAddSignatureText(backing)
1536
tree = self.make_branch_and_memory_tree('.')
1537
write_token = tree.lock_write()
1538
self.addCleanup(tree.unlock)
1540
tree.commit("Message", rev_id='rev1')
1541
tree.branch.repository.start_write_group()
1542
write_group_tokens = tree.branch.repository.suspend_write_group()
1543
self.assertEqual(None, request.execute('', write_token,
1544
'rev1', *write_group_tokens))
1545
response = request.do_body('somesignature')
1546
self.assertTrue(response.is_successful())
1547
self.assertEqual(response.args[0], 'ok')
1548
write_group_tokens = response.args[1:]
1549
tree.branch.repository.resume_write_group(write_group_tokens)
1550
tree.branch.repository.commit_write_group()
1552
self.assertEqual("somesignature",
1553
tree.branch.repository.get_signature_text("rev1"))
1556
class TestSmartServerRepositoryAllRevisionIds(
1557
tests.TestCaseWithMemoryTransport):
1559
def test_empty(self):
1560
"""An empty body should be returned for an empty repository."""
1561
backing = self.get_transport()
1562
request = smart_repo.SmartServerRepositoryAllRevisionIds(backing)
1563
self.make_repository('.')
1565
smart_req.SuccessfulSmartServerResponse(("ok", ), ""),
1566
request.execute(''))
1568
def test_some_revisions(self):
1569
"""An empty body should be returned for an empty repository."""
1570
backing = self.get_transport()
1571
request = smart_repo.SmartServerRepositoryAllRevisionIds(backing)
1572
tree = self.make_branch_and_memory_tree('.')
1575
tree.commit(rev_id='origineel', message="message")
1576
tree.commit(rev_id='nog-een-revisie', message="message")
1579
smart_req.SuccessfulSmartServerResponse(("ok", ),
1580
"origineel\nnog-een-revisie"),
1581
request.execute(''))
1584
class TestSmartServerRepositoryBreakLock(tests.TestCaseWithMemoryTransport):
1586
def test_lock_to_break(self):
1587
backing = self.get_transport()
1588
request = smart_repo.SmartServerRepositoryBreakLock(backing)
1589
tree = self.make_branch_and_memory_tree('.')
1590
tree.branch.repository.lock_write()
1592
smart_req.SuccessfulSmartServerResponse(('ok', ), None),
1593
request.execute(''))
1595
def test_nothing_to_break(self):
1596
backing = self.get_transport()
1597
request = smart_repo.SmartServerRepositoryBreakLock(backing)
1598
tree = self.make_branch_and_memory_tree('.')
1600
smart_req.SuccessfulSmartServerResponse(('ok', ), None),
1601
request.execute(''))
1604
class TestSmartServerRepositoryGetParentMap(tests.TestCaseWithMemoryTransport):
1606
def test_trivial_bzipped(self):
1607
# This tests that the wire encoding is actually bzipped
1608
backing = self.get_transport()
1609
request = smart_repo.SmartServerRepositoryGetParentMap(backing)
1610
tree = self.make_branch_and_memory_tree('.')
1612
self.assertEqual(None,
1613
request.execute('', 'missing-id'))
1614
# Note that it returns a body that is bzipped.
1616
smart_req.SuccessfulSmartServerResponse(('ok', ), bz2.compress('')),
1617
request.do_body('\n\n0\n'))
1619
def test_trivial_include_missing(self):
1620
backing = self.get_transport()
1621
request = smart_repo.SmartServerRepositoryGetParentMap(backing)
1622
tree = self.make_branch_and_memory_tree('.')
1624
self.assertEqual(None,
1625
request.execute('', 'missing-id', 'include-missing:'))
1627
smart_req.SuccessfulSmartServerResponse(('ok', ),
1628
bz2.compress('missing:missing-id')),
1629
request.do_body('\n\n0\n'))
1632
class TestSmartServerRepositoryGetRevisionGraph(
1633
tests.TestCaseWithMemoryTransport):
1635
def test_none_argument(self):
1636
backing = self.get_transport()
1637
request = smart_repo.SmartServerRepositoryGetRevisionGraph(backing)
1638
tree = self.make_branch_and_memory_tree('.')
1641
r1 = tree.commit('1st commit')
1642
r2 = tree.commit('2nd commit', rev_id=u'\xc8'.encode('utf-8'))
1645
# the lines of revision_id->revision_parent_list has no guaranteed
1646
# order coming out of a dict, so sort both our test and response
1647
lines = sorted([' '.join([r2, r1]), r1])
1648
response = request.execute('', '')
1649
response.body = '\n'.join(sorted(response.body.split('\n')))
1652
smart_req.SmartServerResponse(('ok', ), '\n'.join(lines)), response)
1654
def test_specific_revision_argument(self):
1655
backing = self.get_transport()
1656
request = smart_repo.SmartServerRepositoryGetRevisionGraph(backing)
1657
tree = self.make_branch_and_memory_tree('.')
1660
rev_id_utf8 = u'\xc9'.encode('utf-8')
1661
r1 = tree.commit('1st commit', rev_id=rev_id_utf8)
1662
r2 = tree.commit('2nd commit', rev_id=u'\xc8'.encode('utf-8'))
1665
self.assertEqual(smart_req.SmartServerResponse(('ok', ), rev_id_utf8),
1666
request.execute('', rev_id_utf8))
1668
def test_no_such_revision(self):
1669
backing = self.get_transport()
1670
request = smart_repo.SmartServerRepositoryGetRevisionGraph(backing)
1671
tree = self.make_branch_and_memory_tree('.')
1674
r1 = tree.commit('1st commit')
1677
# Note that it still returns body (of zero bytes).
1678
self.assertEqual(smart_req.SmartServerResponse(
1679
('nosuchrevision', 'missingrevision', ), ''),
1680
request.execute('', 'missingrevision'))
1683
class TestSmartServerRepositoryGetRevIdForRevno(
1684
tests.TestCaseWithMemoryTransport):
1686
def test_revno_found(self):
1687
backing = self.get_transport()
1688
request = smart_repo.SmartServerRepositoryGetRevIdForRevno(backing)
1689
tree = self.make_branch_and_memory_tree('.')
1692
rev1_id_utf8 = u'\xc8'.encode('utf-8')
1693
rev2_id_utf8 = u'\xc9'.encode('utf-8')
1694
tree.commit('1st commit', rev_id=rev1_id_utf8)
1695
tree.commit('2nd commit', rev_id=rev2_id_utf8)
1698
self.assertEqual(smart_req.SmartServerResponse(('ok', rev1_id_utf8)),
1699
request.execute('', 1, (2, rev2_id_utf8)))
1701
def test_known_revid_missing(self):
1702
backing = self.get_transport()
1703
request = smart_repo.SmartServerRepositoryGetRevIdForRevno(backing)
1704
repo = self.make_repository('.')
1706
smart_req.FailedSmartServerResponse(('nosuchrevision', 'ghost')),
1707
request.execute('', 1, (2, 'ghost')))
1709
def test_history_incomplete(self):
1710
backing = self.get_transport()
1711
request = smart_repo.SmartServerRepositoryGetRevIdForRevno(backing)
1712
parent = self.make_branch_and_memory_tree('parent', format='1.9')
1714
parent.add([''], ['TREE_ROOT'])
1715
r1 = parent.commit(message='first commit')
1716
r2 = parent.commit(message='second commit')
1718
local = self.make_branch_and_memory_tree('local', format='1.9')
1719
local.branch.pull(parent.branch)
1720
local.set_parent_ids([r2])
1721
r3 = local.commit(message='local commit')
1722
local.branch.create_clone_on_transport(
1723
self.get_transport('stacked'), stacked_on=self.get_url('parent'))
1725
smart_req.SmartServerResponse(('history-incomplete', 2, r2)),
1726
request.execute('stacked', 1, (3, r3)))
1729
class TestSmartServerRepositoryIterRevisions(
1730
tests.TestCaseWithMemoryTransport):
1732
def test_basic(self):
1733
backing = self.get_transport()
1734
request = smart_repo.SmartServerRepositoryIterRevisions(backing)
1735
tree = self.make_branch_and_memory_tree('.', format='2a')
1738
tree.commit('1st commit', rev_id="rev1")
1739
tree.commit('2nd commit', rev_id="rev2")
1742
self.assertIs(None, request.execute(''))
1743
response = request.do_body("rev1\nrev2")
1744
self.assertTrue(response.is_successful())
1745
# Format 2a uses serializer format 10
1746
self.assertEquals(response.args, ("ok", "10"))
1748
self.addCleanup(tree.branch.lock_read().unlock)
1749
entries = [zlib.compress(record.get_bytes_as("fulltext")) for record in
1750
tree.branch.repository.revisions.get_record_stream(
1751
[("rev1", ), ("rev2", )], "unordered", True)]
1753
contents = "".join(response.body_stream)
1754
self.assertTrue(contents in (
1755
"".join([entries[0], entries[1]]),
1756
"".join([entries[1], entries[0]])))
1758
def test_missing(self):
1759
backing = self.get_transport()
1760
request = smart_repo.SmartServerRepositoryIterRevisions(backing)
1761
tree = self.make_branch_and_memory_tree('.', format='2a')
1763
self.assertIs(None, request.execute(''))
1764
response = request.do_body("rev1\nrev2")
1765
self.assertTrue(response.is_successful())
1766
# Format 2a uses serializer format 10
1767
self.assertEquals(response.args, ("ok", "10"))
1769
contents = "".join(response.body_stream)
1770
self.assertEquals(contents, "")
1773
class GetStreamTestBase(tests.TestCaseWithMemoryTransport):
1775
def make_two_commit_repo(self):
1776
tree = self.make_branch_and_memory_tree('.')
1779
r1 = tree.commit('1st commit')
1780
r2 = tree.commit('2nd commit', rev_id=u'\xc8'.encode('utf-8'))
1782
repo = tree.branch.repository
1786
class TestSmartServerRepositoryGetStream(GetStreamTestBase):
1788
def test_ancestry_of(self):
1789
"""The search argument may be a 'ancestry-of' some heads'."""
1790
backing = self.get_transport()
1791
request = smart_repo.SmartServerRepositoryGetStream(backing)
1792
repo, r1, r2 = self.make_two_commit_repo()
1793
fetch_spec = ['ancestry-of', r2]
1794
lines = '\n'.join(fetch_spec)
1795
request.execute('', repo._format.network_name())
1796
response = request.do_body(lines)
1797
self.assertEqual(('ok',), response.args)
1798
stream_bytes = ''.join(response.body_stream)
1799
self.assertStartsWith(stream_bytes, 'Bazaar pack format 1')
1801
def test_search(self):
1802
"""The search argument may be a 'search' of some explicit keys."""
1803
backing = self.get_transport()
1804
request = smart_repo.SmartServerRepositoryGetStream(backing)
1805
repo, r1, r2 = self.make_two_commit_repo()
1806
fetch_spec = ['search', '%s %s' % (r1, r2), 'null:', '2']
1807
lines = '\n'.join(fetch_spec)
1808
request.execute('', repo._format.network_name())
1809
response = request.do_body(lines)
1810
self.assertEqual(('ok',), response.args)
1811
stream_bytes = ''.join(response.body_stream)
1812
self.assertStartsWith(stream_bytes, 'Bazaar pack format 1')
1814
def test_search_everything(self):
1815
"""A search of 'everything' returns a stream."""
1816
backing = self.get_transport()
1817
request = smart_repo.SmartServerRepositoryGetStream_1_19(backing)
1818
repo, r1, r2 = self.make_two_commit_repo()
1819
serialised_fetch_spec = 'everything'
1820
request.execute('', repo._format.network_name())
1821
response = request.do_body(serialised_fetch_spec)
1822
self.assertEqual(('ok',), response.args)
1823
stream_bytes = ''.join(response.body_stream)
1824
self.assertStartsWith(stream_bytes, 'Bazaar pack format 1')
1827
class TestSmartServerRequestHasRevision(tests.TestCaseWithMemoryTransport):
1829
def test_missing_revision(self):
1830
"""For a missing revision, ('no', ) is returned."""
1831
backing = self.get_transport()
1832
request = smart_repo.SmartServerRequestHasRevision(backing)
1833
self.make_repository('.')
1834
self.assertEqual(smart_req.SmartServerResponse(('no', )),
1835
request.execute('', 'revid'))
1837
def test_present_revision(self):
1838
"""For a present revision, ('yes', ) is returned."""
1839
backing = self.get_transport()
1840
request = smart_repo.SmartServerRequestHasRevision(backing)
1841
tree = self.make_branch_and_memory_tree('.')
1844
rev_id_utf8 = u'\xc8abc'.encode('utf-8')
1845
r1 = tree.commit('a commit', rev_id=rev_id_utf8)
1847
self.assertTrue(tree.branch.repository.has_revision(rev_id_utf8))
1848
self.assertEqual(smart_req.SmartServerResponse(('yes', )),
1849
request.execute('', rev_id_utf8))
1852
class TestSmartServerRepositoryIterFilesBytes(tests.TestCaseWithTransport):
1854
def test_single(self):
1855
backing = self.get_transport()
1856
request = smart_repo.SmartServerRepositoryIterFilesBytes(backing)
1857
t = self.make_branch_and_tree('.')
1858
self.addCleanup(t.lock_write().unlock)
1859
self.build_tree_contents([("file", "somecontents")])
1860
t.add(["file"], ["thefileid"])
1861
t.commit(rev_id='somerev', message="add file")
1862
self.assertIs(None, request.execute(''))
1863
response = request.do_body("thefileid\0somerev\n")
1864
self.assertTrue(response.is_successful())
1865
self.assertEquals(response.args, ("ok", ))
1866
self.assertEquals("".join(response.body_stream),
1867
"ok\x000\n" + zlib.compress("somecontents"))
1869
def test_missing(self):
1870
backing = self.get_transport()
1871
request = smart_repo.SmartServerRepositoryIterFilesBytes(backing)
1872
t = self.make_branch_and_tree('.')
1873
self.addCleanup(t.lock_write().unlock)
1874
self.assertIs(None, request.execute(''))
1875
response = request.do_body("thefileid\0revision\n")
1876
self.assertTrue(response.is_successful())
1877
self.assertEquals(response.args, ("ok", ))
1878
self.assertEquals("".join(response.body_stream),
1879
"absent\x00thefileid\x00revision\x000\n")
1882
class TestSmartServerRequestHasSignatureForRevisionId(
1883
tests.TestCaseWithMemoryTransport):
1885
def test_missing_revision(self):
1886
"""For a missing revision, NoSuchRevision is returned."""
1887
backing = self.get_transport()
1888
request = smart_repo.SmartServerRequestHasSignatureForRevisionId(
1890
self.make_repository('.')
1892
smart_req.FailedSmartServerResponse(
1893
('nosuchrevision', 'revid'), None),
1894
request.execute('', 'revid'))
1896
def test_missing_signature(self):
1897
"""For a missing signature, ('no', ) is returned."""
1898
backing = self.get_transport()
1899
request = smart_repo.SmartServerRequestHasSignatureForRevisionId(
1901
tree = self.make_branch_and_memory_tree('.')
1904
r1 = tree.commit('a commit', rev_id='A')
1906
self.assertTrue(tree.branch.repository.has_revision('A'))
1907
self.assertEqual(smart_req.SmartServerResponse(('no', )),
1908
request.execute('', 'A'))
1910
def test_present_signature(self):
1911
"""For a present signature, ('yes', ) is returned."""
1912
backing = self.get_transport()
1913
request = smart_repo.SmartServerRequestHasSignatureForRevisionId(
1915
strategy = gpg.LoopbackGPGStrategy(None)
1916
tree = self.make_branch_and_memory_tree('.')
1919
r1 = tree.commit('a commit', rev_id='A')
1920
tree.branch.repository.start_write_group()
1921
tree.branch.repository.sign_revision('A', strategy)
1922
tree.branch.repository.commit_write_group()
1924
self.assertTrue(tree.branch.repository.has_revision('A'))
1925
self.assertEqual(smart_req.SmartServerResponse(('yes', )),
1926
request.execute('', 'A'))
1929
class TestSmartServerRepositoryGatherStats(tests.TestCaseWithMemoryTransport):
1931
def test_empty_revid(self):
1932
"""With an empty revid, we get only size an number and revisions"""
1933
backing = self.get_transport()
1934
request = smart_repo.SmartServerRepositoryGatherStats(backing)
1935
repository = self.make_repository('.')
1936
stats = repository.gather_stats()
1937
expected_body = 'revisions: 0\n'
1938
self.assertEqual(smart_req.SmartServerResponse(('ok', ), expected_body),
1939
request.execute('', '', 'no'))
1941
def test_revid_with_committers(self):
1942
"""For a revid we get more infos."""
1943
backing = self.get_transport()
1944
rev_id_utf8 = u'\xc8abc'.encode('utf-8')
1945
request = smart_repo.SmartServerRepositoryGatherStats(backing)
1946
tree = self.make_branch_and_memory_tree('.')
1949
# Let's build a predictable result
1950
tree.commit('a commit', timestamp=123456.2, timezone=3600)
1951
tree.commit('a commit', timestamp=654321.4, timezone=0,
1955
stats = tree.branch.repository.gather_stats()
1956
expected_body = ('firstrev: 123456.200 3600\n'
1957
'latestrev: 654321.400 0\n'
1959
self.assertEqual(smart_req.SmartServerResponse(('ok', ), expected_body),
1963
def test_not_empty_repository_with_committers(self):
1964
"""For a revid and requesting committers we get the whole thing."""
1965
backing = self.get_transport()
1966
rev_id_utf8 = u'\xc8abc'.encode('utf-8')
1967
request = smart_repo.SmartServerRepositoryGatherStats(backing)
1968
tree = self.make_branch_and_memory_tree('.')
1971
# Let's build a predictable result
1972
tree.commit('a commit', timestamp=123456.2, timezone=3600,
1974
tree.commit('a commit', timestamp=654321.4, timezone=0,
1975
committer='bar', rev_id=rev_id_utf8)
1977
stats = tree.branch.repository.gather_stats()
1979
expected_body = ('committers: 2\n'
1980
'firstrev: 123456.200 3600\n'
1981
'latestrev: 654321.400 0\n'
1983
self.assertEqual(smart_req.SmartServerResponse(('ok', ), expected_body),
1985
rev_id_utf8, 'yes'))
1987
def test_unknown_revid(self):
1988
"""An unknown revision id causes a 'nosuchrevision' error."""
1989
backing = self.get_transport()
1990
request = smart_repo.SmartServerRepositoryGatherStats(backing)
1991
repository = self.make_repository('.')
1992
expected_body = 'revisions: 0\n'
1994
smart_req.FailedSmartServerResponse(
1995
('nosuchrevision', 'mia'), None),
1996
request.execute('', 'mia', 'yes'))
1999
class TestSmartServerRepositoryIsShared(tests.TestCaseWithMemoryTransport):
2001
def test_is_shared(self):
2002
"""For a shared repository, ('yes', ) is returned."""
2003
backing = self.get_transport()
2004
request = smart_repo.SmartServerRepositoryIsShared(backing)
2005
self.make_repository('.', shared=True)
2006
self.assertEqual(smart_req.SmartServerResponse(('yes', )),
2007
request.execute('', ))
2009
def test_is_not_shared(self):
2010
"""For a shared repository, ('no', ) is returned."""
2011
backing = self.get_transport()
2012
request = smart_repo.SmartServerRepositoryIsShared(backing)
2013
self.make_repository('.', shared=False)
2014
self.assertEqual(smart_req.SmartServerResponse(('no', )),
2015
request.execute('', ))
2018
class TestSmartServerRepositoryGetRevisionSignatureText(
2019
tests.TestCaseWithMemoryTransport):
2021
def test_get_signature(self):
2022
backing = self.get_transport()
2023
request = smart_repo.SmartServerRepositoryGetRevisionSignatureText(
2025
bb = self.make_branch_builder('.')
2026
bb.build_commit(rev_id='A')
2027
repo = bb.get_branch().repository
2028
strategy = gpg.LoopbackGPGStrategy(None)
2029
self.addCleanup(repo.lock_write().unlock)
2030
repo.start_write_group()
2031
repo.sign_revision('A', strategy)
2032
repo.commit_write_group()
2034
'-----BEGIN PSEUDO-SIGNED CONTENT-----\n' +
2035
Testament.from_revision(repo, 'A').as_short_text() +
2036
'-----END PSEUDO-SIGNED CONTENT-----\n')
2038
smart_req.SmartServerResponse(('ok', ), expected_body),
2039
request.execute('', 'A'))
2042
class TestSmartServerRepositoryMakeWorkingTrees(
2043
tests.TestCaseWithMemoryTransport):
2045
def test_make_working_trees(self):
2046
"""For a repository with working trees, ('yes', ) is returned."""
2047
backing = self.get_transport()
2048
request = smart_repo.SmartServerRepositoryMakeWorkingTrees(backing)
2049
r = self.make_repository('.')
2050
r.set_make_working_trees(True)
2051
self.assertEqual(smart_req.SmartServerResponse(('yes', )),
2052
request.execute('', ))
2054
def test_is_not_shared(self):
2055
"""For a repository with working trees, ('no', ) is returned."""
2056
backing = self.get_transport()
2057
request = smart_repo.SmartServerRepositoryMakeWorkingTrees(backing)
2058
r = self.make_repository('.')
2059
r.set_make_working_trees(False)
2060
self.assertEqual(smart_req.SmartServerResponse(('no', )),
2061
request.execute('', ))
2064
class TestSmartServerRepositoryLockWrite(tests.TestCaseWithMemoryTransport):
2066
def test_lock_write_on_unlocked_repo(self):
2067
backing = self.get_transport()
2068
request = smart_repo.SmartServerRepositoryLockWrite(backing)
2069
repository = self.make_repository('.', format='knit')
2070
response = request.execute('')
2071
nonce = repository.control_files._lock.peek().get('nonce')
2072
self.assertEqual(smart_req.SmartServerResponse(('ok', nonce)), response)
2073
# The repository is now locked. Verify that with a new repository
2075
new_repo = repository.bzrdir.open_repository()
2076
self.assertRaises(errors.LockContention, new_repo.lock_write)
2078
request = smart_repo.SmartServerRepositoryUnlock(backing)
2079
response = request.execute('', nonce)
2081
def test_lock_write_on_locked_repo(self):
2082
backing = self.get_transport()
2083
request = smart_repo.SmartServerRepositoryLockWrite(backing)
2084
repository = self.make_repository('.', format='knit')
2085
repo_token = repository.lock_write().repository_token
2086
repository.leave_lock_in_place()
2088
response = request.execute('')
2090
smart_req.SmartServerResponse(('LockContention',)), response)
2092
repository.lock_write(repo_token)
2093
repository.dont_leave_lock_in_place()
2096
def test_lock_write_on_readonly_transport(self):
2097
backing = self.get_readonly_transport()
2098
request = smart_repo.SmartServerRepositoryLockWrite(backing)
2099
repository = self.make_repository('.', format='knit')
2100
response = request.execute('')
2101
self.assertFalse(response.is_successful())
2102
self.assertEqual('LockFailed', response.args[0])
2105
class TestInsertStreamBase(tests.TestCaseWithMemoryTransport):
2107
def make_empty_byte_stream(self, repo):
2108
byte_stream = smart_repo._stream_to_byte_stream([], repo._format)
2109
return ''.join(byte_stream)
2112
class TestSmartServerRepositoryInsertStream(TestInsertStreamBase):
2114
def test_insert_stream_empty(self):
2115
backing = self.get_transport()
2116
request = smart_repo.SmartServerRepositoryInsertStream(backing)
2117
repository = self.make_repository('.')
2118
response = request.execute('', '')
2119
self.assertEqual(None, response)
2120
response = request.do_chunk(self.make_empty_byte_stream(repository))
2121
self.assertEqual(None, response)
2122
response = request.do_end()
2123
self.assertEqual(smart_req.SmartServerResponse(('ok', )), response)
2126
class TestSmartServerRepositoryInsertStreamLocked(TestInsertStreamBase):
2128
def test_insert_stream_empty(self):
2129
backing = self.get_transport()
2130
request = smart_repo.SmartServerRepositoryInsertStreamLocked(
2132
repository = self.make_repository('.', format='knit')
2133
lock_token = repository.lock_write().repository_token
2134
response = request.execute('', '', lock_token)
2135
self.assertEqual(None, response)
2136
response = request.do_chunk(self.make_empty_byte_stream(repository))
2137
self.assertEqual(None, response)
2138
response = request.do_end()
2139
self.assertEqual(smart_req.SmartServerResponse(('ok', )), response)
2142
def test_insert_stream_with_wrong_lock_token(self):
2143
backing = self.get_transport()
2144
request = smart_repo.SmartServerRepositoryInsertStreamLocked(
2146
repository = self.make_repository('.', format='knit')
2147
lock_token = repository.lock_write().repository_token
2149
errors.TokenMismatch, request.execute, '', '', 'wrong-token')
2153
class TestSmartServerRepositoryUnlock(tests.TestCaseWithMemoryTransport):
2156
tests.TestCaseWithMemoryTransport.setUp(self)
2158
def test_unlock_on_locked_repo(self):
2159
backing = self.get_transport()
2160
request = smart_repo.SmartServerRepositoryUnlock(backing)
2161
repository = self.make_repository('.', format='knit')
2162
token = repository.lock_write().repository_token
2163
repository.leave_lock_in_place()
2165
response = request.execute('', token)
2167
smart_req.SmartServerResponse(('ok',)), response)
2168
# The repository is now unlocked. Verify that with a new repository
2170
new_repo = repository.bzrdir.open_repository()
2171
new_repo.lock_write()
2174
def test_unlock_on_unlocked_repo(self):
2175
backing = self.get_transport()
2176
request = smart_repo.SmartServerRepositoryUnlock(backing)
2177
repository = self.make_repository('.', format='knit')
2178
response = request.execute('', 'some token')
2180
smart_req.SmartServerResponse(('TokenMismatch',)), response)
2183
class TestSmartServerRepositoryGetPhysicalLockStatus(
2184
tests.TestCaseWithTransport):
2186
def test_with_write_lock(self):
2187
backing = self.get_transport()
2188
repo = self.make_repository('.')
2189
self.addCleanup(repo.lock_write().unlock)
2190
# lock_write() doesn't necessarily actually take a physical
2192
if repo.get_physical_lock_status():
2196
request_class = smart_repo.SmartServerRepositoryGetPhysicalLockStatus
2197
request = request_class(backing)
2198
self.assertEqual(smart_req.SuccessfulSmartServerResponse((expected,)),
2199
request.execute('', ))
2201
def test_without_write_lock(self):
2202
backing = self.get_transport()
2203
repo = self.make_repository('.')
2204
self.assertEquals(False, repo.get_physical_lock_status())
2205
request_class = smart_repo.SmartServerRepositoryGetPhysicalLockStatus
2206
request = request_class(backing)
2207
self.assertEqual(smart_req.SuccessfulSmartServerResponse(('no',)),
2208
request.execute('', ))
2211
class TestSmartServerIsReadonly(tests.TestCaseWithMemoryTransport):
2213
def test_is_readonly_no(self):
2214
backing = self.get_transport()
2215
request = smart_req.SmartServerIsReadonly(backing)
2216
response = request.execute()
2218
smart_req.SmartServerResponse(('no',)), response)
2220
def test_is_readonly_yes(self):
2221
backing = self.get_readonly_transport()
2222
request = smart_req.SmartServerIsReadonly(backing)
2223
response = request.execute()
2225
smart_req.SmartServerResponse(('yes',)), response)
2228
class TestSmartServerRepositorySetMakeWorkingTrees(
2229
tests.TestCaseWithMemoryTransport):
2231
def test_set_false(self):
2232
backing = self.get_transport()
2233
repo = self.make_repository('.', shared=True)
2234
repo.set_make_working_trees(True)
2235
request_class = smart_repo.SmartServerRepositorySetMakeWorkingTrees
2236
request = request_class(backing)
2237
self.assertEqual(smart_req.SuccessfulSmartServerResponse(('ok',)),
2238
request.execute('', 'False'))
2239
repo = repo.bzrdir.open_repository()
2240
self.assertFalse(repo.make_working_trees())
2242
def test_set_true(self):
2243
backing = self.get_transport()
2244
repo = self.make_repository('.', shared=True)
2245
repo.set_make_working_trees(False)
2246
request_class = smart_repo.SmartServerRepositorySetMakeWorkingTrees
2247
request = request_class(backing)
2248
self.assertEqual(smart_req.SuccessfulSmartServerResponse(('ok',)),
2249
request.execute('', 'True'))
2250
repo = repo.bzrdir.open_repository()
2251
self.assertTrue(repo.make_working_trees())
2254
class TestSmartServerRepositoryGetSerializerFormat(
2255
tests.TestCaseWithMemoryTransport):
2257
def test_get_serializer_format(self):
2258
backing = self.get_transport()
2259
repo = self.make_repository('.', format='2a')
2260
request_class = smart_repo.SmartServerRepositoryGetSerializerFormat
2261
request = request_class(backing)
2263
smart_req.SuccessfulSmartServerResponse(('ok', '10')),
2264
request.execute(''))
2267
class TestSmartServerRepositoryWriteGroup(
2268
tests.TestCaseWithMemoryTransport):
2270
def test_start_write_group(self):
2271
backing = self.get_transport()
2272
repo = self.make_repository('.')
2273
lock_token = repo.lock_write().repository_token
2274
self.addCleanup(repo.unlock)
2275
request_class = smart_repo.SmartServerRepositoryStartWriteGroup
2276
request = request_class(backing)
2277
self.assertEqual(smart_req.SuccessfulSmartServerResponse(('ok', [])),
2278
request.execute('', lock_token))
2280
def test_start_write_group_unsuspendable(self):
2281
backing = self.get_transport()
2282
repo = self.make_repository('.', format='knit')
2283
lock_token = repo.lock_write().repository_token
2284
self.addCleanup(repo.unlock)
2285
request_class = smart_repo.SmartServerRepositoryStartWriteGroup
2286
request = request_class(backing)
2288
smart_req.FailedSmartServerResponse(('UnsuspendableWriteGroup',)),
2289
request.execute('', lock_token))
2291
def test_commit_write_group(self):
2292
backing = self.get_transport()
2293
repo = self.make_repository('.')
2294
lock_token = repo.lock_write().repository_token
2295
self.addCleanup(repo.unlock)
2296
repo.start_write_group()
2297
tokens = repo.suspend_write_group()
2298
request_class = smart_repo.SmartServerRepositoryCommitWriteGroup
2299
request = request_class(backing)
2300
self.assertEqual(smart_req.SuccessfulSmartServerResponse(('ok',)),
2301
request.execute('', lock_token, tokens))
2303
def test_abort_write_group(self):
2304
backing = self.get_transport()
2305
repo = self.make_repository('.')
2306
lock_token = repo.lock_write().repository_token
2307
repo.start_write_group()
2308
tokens = repo.suspend_write_group()
2309
self.addCleanup(repo.unlock)
2310
request_class = smart_repo.SmartServerRepositoryAbortWriteGroup
2311
request = request_class(backing)
2312
self.assertEqual(smart_req.SuccessfulSmartServerResponse(('ok',)),
2313
request.execute('', lock_token, tokens))
2315
def test_check_write_group(self):
2316
backing = self.get_transport()
2317
repo = self.make_repository('.')
2318
lock_token = repo.lock_write().repository_token
2319
repo.start_write_group()
2320
tokens = repo.suspend_write_group()
2321
self.addCleanup(repo.unlock)
2322
request_class = smart_repo.SmartServerRepositoryCheckWriteGroup
2323
request = request_class(backing)
2324
self.assertEqual(smart_req.SuccessfulSmartServerResponse(('ok',)),
2325
request.execute('', lock_token, tokens))
2327
def test_check_write_group_invalid(self):
2328
backing = self.get_transport()
2329
repo = self.make_repository('.')
2330
lock_token = repo.lock_write().repository_token
2331
self.addCleanup(repo.unlock)
2332
request_class = smart_repo.SmartServerRepositoryCheckWriteGroup
2333
request = request_class(backing)
2334
self.assertEqual(smart_req.FailedSmartServerResponse(
2335
('UnresumableWriteGroup', ['random'],
2336
'Malformed write group token')),
2337
request.execute('', lock_token, ["random"]))
2340
class TestSmartServerPackRepositoryAutopack(tests.TestCaseWithTransport):
2342
def make_repo_needing_autopacking(self, path='.'):
2343
# Make a repo in need of autopacking.
2344
tree = self.make_branch_and_tree('.', format='pack-0.92')
2345
repo = tree.branch.repository
2346
# monkey-patch the pack collection to disable autopacking
2347
repo._pack_collection._max_pack_count = lambda count: count
2349
tree.commit('commit %s' % x)
2350
self.assertEqual(10, len(repo._pack_collection.names()))
2351
del repo._pack_collection._max_pack_count
2354
def test_autopack_needed(self):
2355
repo = self.make_repo_needing_autopacking()
2357
self.addCleanup(repo.unlock)
2358
backing = self.get_transport()
2359
request = smart_packrepo.SmartServerPackRepositoryAutopack(
2361
response = request.execute('')
2362
self.assertEqual(smart_req.SmartServerResponse(('ok',)), response)
2363
repo._pack_collection.reload_pack_names()
2364
self.assertEqual(1, len(repo._pack_collection.names()))
2366
def test_autopack_not_needed(self):
2367
tree = self.make_branch_and_tree('.', format='pack-0.92')
2368
repo = tree.branch.repository
2370
self.addCleanup(repo.unlock)
2372
tree.commit('commit %s' % x)
2373
backing = self.get_transport()
2374
request = smart_packrepo.SmartServerPackRepositoryAutopack(
2376
response = request.execute('')
2377
self.assertEqual(smart_req.SmartServerResponse(('ok',)), response)
2378
repo._pack_collection.reload_pack_names()
2379
self.assertEqual(9, len(repo._pack_collection.names()))
2381
def test_autopack_on_nonpack_format(self):
2382
"""A request to autopack a non-pack repo is a no-op."""
2383
repo = self.make_repository('.', format='knit')
2384
backing = self.get_transport()
2385
request = smart_packrepo.SmartServerPackRepositoryAutopack(
2387
response = request.execute('')
2388
self.assertEqual(smart_req.SmartServerResponse(('ok',)), response)
2391
class TestSmartServerVfsGet(tests.TestCaseWithMemoryTransport):
2393
def test_unicode_path(self):
2394
"""VFS requests expect unicode paths to be escaped."""
2395
filename = u'foo\N{INTERROBANG}'
2396
filename_escaped = urlutils.escape(filename)
2397
backing = self.get_transport()
2398
request = vfs.GetRequest(backing)
2399
backing.put_bytes_non_atomic(filename_escaped, 'contents')
2400
self.assertEqual(smart_req.SmartServerResponse(('ok', ), 'contents'),
2401
request.execute(filename_escaped))
2404
class TestHandlers(tests.TestCase):
2405
"""Tests for the request.request_handlers object."""
2407
def test_all_registrations_exist(self):
2408
"""All registered request_handlers can be found."""
2409
# If there's a typo in a register_lazy call, this loop will fail with
2410
# an AttributeError.
2411
for key in smart_req.request_handlers.keys():
2413
item = smart_req.request_handlers.get(key)
2414
except AttributeError, e:
2415
raise AttributeError('failed to get %s: %s' % (key, e))
2417
def assertHandlerEqual(self, verb, handler):
2418
self.assertEqual(smart_req.request_handlers.get(verb), handler)
2420
def test_registered_methods(self):
2421
"""Test that known methods are registered to the correct object."""
2422
self.assertHandlerEqual('Branch.break_lock',
2423
smart_branch.SmartServerBranchBreakLock)
2424
self.assertHandlerEqual('Branch.get_config_file',
2425
smart_branch.SmartServerBranchGetConfigFile)
2426
self.assertHandlerEqual('Branch.put_config_file',
2427
smart_branch.SmartServerBranchPutConfigFile)
2428
self.assertHandlerEqual('Branch.get_parent',
2429
smart_branch.SmartServerBranchGetParent)
2430
self.assertHandlerEqual('Branch.get_physical_lock_status',
2431
smart_branch.SmartServerBranchRequestGetPhysicalLockStatus)
2432
self.assertHandlerEqual('Branch.get_tags_bytes',
2433
smart_branch.SmartServerBranchGetTagsBytes)
2434
self.assertHandlerEqual('Branch.lock_write',
2435
smart_branch.SmartServerBranchRequestLockWrite)
2436
self.assertHandlerEqual('Branch.last_revision_info',
2437
smart_branch.SmartServerBranchRequestLastRevisionInfo)
2438
self.assertHandlerEqual('Branch.revision_history',
2439
smart_branch.SmartServerRequestRevisionHistory)
2440
self.assertHandlerEqual('Branch.revision_id_to_revno',
2441
smart_branch.SmartServerBranchRequestRevisionIdToRevno)
2442
self.assertHandlerEqual('Branch.set_config_option',
2443
smart_branch.SmartServerBranchRequestSetConfigOption)
2444
self.assertHandlerEqual('Branch.set_last_revision',
2445
smart_branch.SmartServerBranchRequestSetLastRevision)
2446
self.assertHandlerEqual('Branch.set_last_revision_info',
2447
smart_branch.SmartServerBranchRequestSetLastRevisionInfo)
2448
self.assertHandlerEqual('Branch.set_last_revision_ex',
2449
smart_branch.SmartServerBranchRequestSetLastRevisionEx)
2450
self.assertHandlerEqual('Branch.set_parent_location',
2451
smart_branch.SmartServerBranchRequestSetParentLocation)
2452
self.assertHandlerEqual('Branch.unlock',
2453
smart_branch.SmartServerBranchRequestUnlock)
2454
self.assertHandlerEqual('BzrDir.destroy_branch',
2455
smart_dir.SmartServerBzrDirRequestDestroyBranch)
2456
self.assertHandlerEqual('BzrDir.find_repository',
2457
smart_dir.SmartServerRequestFindRepositoryV1)
2458
self.assertHandlerEqual('BzrDir.find_repositoryV2',
2459
smart_dir.SmartServerRequestFindRepositoryV2)
2460
self.assertHandlerEqual('BzrDirFormat.initialize',
2461
smart_dir.SmartServerRequestInitializeBzrDir)
2462
self.assertHandlerEqual('BzrDirFormat.initialize_ex_1.16',
2463
smart_dir.SmartServerRequestBzrDirInitializeEx)
2464
self.assertHandlerEqual('BzrDir.cloning_metadir',
2465
smart_dir.SmartServerBzrDirRequestCloningMetaDir)
2466
self.assertHandlerEqual('BzrDir.get_config_file',
2467
smart_dir.SmartServerBzrDirRequestConfigFile)
2468
self.assertHandlerEqual('BzrDir.open_branch',
2469
smart_dir.SmartServerRequestOpenBranch)
2470
self.assertHandlerEqual('BzrDir.open_branchV2',
2471
smart_dir.SmartServerRequestOpenBranchV2)
2472
self.assertHandlerEqual('BzrDir.open_branchV3',
2473
smart_dir.SmartServerRequestOpenBranchV3)
2474
self.assertHandlerEqual('PackRepository.autopack',
2475
smart_packrepo.SmartServerPackRepositoryAutopack)
2476
self.assertHandlerEqual('Repository.add_signature_text',
2477
smart_repo.SmartServerRepositoryAddSignatureText)
2478
self.assertHandlerEqual('Repository.all_revision_ids',
2479
smart_repo.SmartServerRepositoryAllRevisionIds)
2480
self.assertHandlerEqual('Repository.break_lock',
2481
smart_repo.SmartServerRepositoryBreakLock)
2482
self.assertHandlerEqual('Repository.gather_stats',
2483
smart_repo.SmartServerRepositoryGatherStats)
2484
self.assertHandlerEqual('Repository.get_parent_map',
2485
smart_repo.SmartServerRepositoryGetParentMap)
2486
self.assertHandlerEqual('Repository.get_physical_lock_status',
2487
smart_repo.SmartServerRepositoryGetPhysicalLockStatus)
2488
self.assertHandlerEqual('Repository.get_rev_id_for_revno',
2489
smart_repo.SmartServerRepositoryGetRevIdForRevno)
2490
self.assertHandlerEqual('Repository.get_revision_graph',
2491
smart_repo.SmartServerRepositoryGetRevisionGraph)
2492
self.assertHandlerEqual('Repository.get_revision_signature_text',
2493
smart_repo.SmartServerRepositoryGetRevisionSignatureText)
2494
self.assertHandlerEqual('Repository.get_stream',
2495
smart_repo.SmartServerRepositoryGetStream)
2496
self.assertHandlerEqual('Repository.get_stream_1.19',
2497
smart_repo.SmartServerRepositoryGetStream_1_19)
2498
self.assertHandlerEqual('Repository.iter_revisions',
2499
smart_repo.SmartServerRepositoryIterRevisions)
2500
self.assertHandlerEqual('Repository.has_revision',
2501
smart_repo.SmartServerRequestHasRevision)
2502
self.assertHandlerEqual('Repository.insert_stream',
2503
smart_repo.SmartServerRepositoryInsertStream)
2504
self.assertHandlerEqual('Repository.insert_stream_locked',
2505
smart_repo.SmartServerRepositoryInsertStreamLocked)
2506
self.assertHandlerEqual('Repository.is_shared',
2507
smart_repo.SmartServerRepositoryIsShared)
2508
self.assertHandlerEqual('Repository.iter_files_bytes',
2509
smart_repo.SmartServerRepositoryIterFilesBytes)
2510
self.assertHandlerEqual('Repository.lock_write',
2511
smart_repo.SmartServerRepositoryLockWrite)
2512
self.assertHandlerEqual('Repository.make_working_trees',
2513
smart_repo.SmartServerRepositoryMakeWorkingTrees)
2514
self.assertHandlerEqual('Repository.pack',
2515
smart_repo.SmartServerRepositoryPack)
2516
self.assertHandlerEqual('Repository.tarball',
2517
smart_repo.SmartServerRepositoryTarball)
2518
self.assertHandlerEqual('Repository.unlock',
2519
smart_repo.SmartServerRepositoryUnlock)
2520
self.assertHandlerEqual('Repository.start_write_group',
2521
smart_repo.SmartServerRepositoryStartWriteGroup)
2522
self.assertHandlerEqual('Repository.check_write_group',
2523
smart_repo.SmartServerRepositoryCheckWriteGroup)
2524
self.assertHandlerEqual('Repository.commit_write_group',
2525
smart_repo.SmartServerRepositoryCommitWriteGroup)
2526
self.assertHandlerEqual('Repository.abort_write_group',
2527
smart_repo.SmartServerRepositoryAbortWriteGroup)
2528
self.assertHandlerEqual('VersionedFileRepository.get_serializer_format',
2529
smart_repo.SmartServerRepositoryGetSerializerFormat)
2530
self.assertHandlerEqual('Transport.is_readonly',
2531
smart_req.SmartServerIsReadonly)
2534
class SmartTCPServerHookTests(tests.TestCaseWithMemoryTransport):
2535
"""Tests for SmartTCPServer hooks."""
2538
super(SmartTCPServerHookTests, self).setUp()
2539
self.server = server.SmartTCPServer(self.get_transport())
2541
def test_run_server_started_hooks(self):
2542
"""Test the server started hooks get fired properly."""
2544
server.SmartTCPServer.hooks.install_named_hook('server_started',
2545
lambda backing_urls, url: started_calls.append((backing_urls, url)),
2547
started_ex_calls = []
2548
server.SmartTCPServer.hooks.install_named_hook('server_started_ex',
2549
lambda backing_urls, url: started_ex_calls.append((backing_urls, url)),
2551
self.server._sockname = ('example.com', 42)
2552
self.server.run_server_started_hooks()
2553
self.assertEquals(started_calls,
2554
[([self.get_transport().base], 'bzr://example.com:42/')])
2555
self.assertEquals(started_ex_calls,
2556
[([self.get_transport().base], self.server)])
2558
def test_run_server_started_hooks_ipv6(self):
2559
"""Test that socknames can contain 4-tuples."""
2560
self.server._sockname = ('::', 42, 0, 0)
2562
server.SmartTCPServer.hooks.install_named_hook('server_started',
2563
lambda backing_urls, url: started_calls.append((backing_urls, url)),
2565
self.server.run_server_started_hooks()
2566
self.assertEquals(started_calls,
2567
[([self.get_transport().base], 'bzr://:::42/')])
2569
def test_run_server_stopped_hooks(self):
2570
"""Test the server stopped hooks."""
2571
self.server._sockname = ('example.com', 42)
2573
server.SmartTCPServer.hooks.install_named_hook('server_stopped',
2574
lambda backing_urls, url: stopped_calls.append((backing_urls, url)),
2576
self.server.run_server_stopped_hooks()
2577
self.assertEquals(stopped_calls,
2578
[([self.get_transport().base], 'bzr://example.com:42/')])
2581
class TestSmartServerRepositoryPack(tests.TestCaseWithMemoryTransport):
2583
def test_pack(self):
2584
backing = self.get_transport()
2585
request = smart_repo.SmartServerRepositoryPack(backing)
2586
tree = self.make_branch_and_memory_tree('.')
2587
repo_token = tree.branch.repository.lock_write().repository_token
2589
self.assertIs(None, request.execute('', repo_token, False))
2592
smart_req.SuccessfulSmartServerResponse(('ok', ), ),
2593
request.do_body(''))
added
removed
bzrlib/tests/test_smart.py
