103
49
# the default or a parameterized class, but rather use the
104
50
# TestCaseWithTransport infrastructure to set up a smart server and
106
self.transport_server = self.make_transport_server
108
def make_transport_server(self):
109
return test_server.SmartTCPServer_for_testing('-' + self.id())
52
self.transport_server = smart.server.SmartTCPServer_for_testing
111
54
def get_smart_medium(self):
112
55
"""Get a smart medium to use in tests."""
113
56
return self.get_transport().get_smart_medium()
116
class TestByteStreamToStream(tests.TestCase):
118
def test_repeated_substreams_same_kind_are_one_stream(self):
119
# Make a stream - an iterable of bytestrings.
120
stream = [('text', [versionedfile.FulltextContentFactory(('k1',), None,
121
None, 'foo')]),('text', [
122
versionedfile.FulltextContentFactory(('k2',), None, None, 'bar')])]
123
fmt = bzrdir.format_registry.get('pack-0.92')().repository_format
124
bytes = smart_repo._stream_to_byte_stream(stream, fmt)
126
# Iterate the resulting iterable; checking that we get only one stream
128
fmt, stream = smart_repo._byte_stream_to_stream(bytes)
129
for kind, substream in stream:
130
streams.append((kind, list(substream)))
131
self.assertLength(1, streams)
132
self.assertLength(2, streams[0][1])
135
59
class TestSmartServerResponse(tests.TestCase):
137
61
def test__eq__(self):
138
self.assertEqual(smart_req.SmartServerResponse(('ok', )),
139
smart_req.SmartServerResponse(('ok', )))
140
self.assertEqual(smart_req.SmartServerResponse(('ok', ), 'body'),
141
smart_req.SmartServerResponse(('ok', ), 'body'))
142
self.assertNotEqual(smart_req.SmartServerResponse(('ok', )),
143
smart_req.SmartServerResponse(('notok', )))
144
self.assertNotEqual(smart_req.SmartServerResponse(('ok', ), 'body'),
145
smart_req.SmartServerResponse(('ok', )))
62
self.assertEqual(SmartServerResponse(('ok', )),
63
SmartServerResponse(('ok', )))
64
self.assertEqual(SmartServerResponse(('ok', ), 'body'),
65
SmartServerResponse(('ok', ), 'body'))
66
self.assertNotEqual(SmartServerResponse(('ok', )),
67
SmartServerResponse(('notok', )))
68
self.assertNotEqual(SmartServerResponse(('ok', ), 'body'),
69
SmartServerResponse(('ok', )))
146
70
self.assertNotEqual(None,
147
smart_req.SmartServerResponse(('ok', )))
149
def test__str__(self):
150
"""SmartServerResponses can be stringified."""
152
"<SuccessfulSmartServerResponse args=('args',) body='body'>",
153
str(smart_req.SuccessfulSmartServerResponse(('args',), 'body')))
155
"<FailedSmartServerResponse args=('args',) body='body'>",
156
str(smart_req.FailedSmartServerResponse(('args',), 'body')))
159
class TestSmartServerRequest(tests.TestCaseWithMemoryTransport):
161
def test_translate_client_path(self):
162
transport = self.get_transport()
163
request = smart_req.SmartServerRequest(transport, 'foo/')
164
self.assertEqual('./', request.translate_client_path('foo/'))
166
errors.InvalidURLJoin, request.translate_client_path, 'foo/..')
168
errors.PathNotChild, request.translate_client_path, '/')
170
errors.PathNotChild, request.translate_client_path, 'bar/')
171
self.assertEqual('./baz', request.translate_client_path('foo/baz'))
172
e_acute = u'\N{LATIN SMALL LETTER E WITH ACUTE}'.encode('utf-8')
173
self.assertEqual('./' + urlutils.escape(e_acute),
174
request.translate_client_path('foo/' + e_acute))
176
def test_translate_client_path_vfs(self):
177
"""VfsRequests receive escaped paths rather than raw UTF-8."""
178
transport = self.get_transport()
179
request = vfs.VfsRequest(transport, 'foo/')
180
e_acute = u'\N{LATIN SMALL LETTER E WITH ACUTE}'.encode('utf-8')
181
escaped = urlutils.escape('foo/' + e_acute)
182
self.assertEqual('./' + urlutils.escape(e_acute),
183
request.translate_client_path(escaped))
185
def test_transport_from_client_path(self):
186
transport = self.get_transport()
187
request = smart_req.SmartServerRequest(transport, 'foo/')
190
request.transport_from_client_path('foo/').base)
193
class TestSmartServerBzrDirRequestCloningMetaDir(
194
tests.TestCaseWithMemoryTransport):
195
"""Tests for BzrDir.cloning_metadir."""
197
def test_cloning_metadir(self):
198
"""When there is a bzrdir present, the call succeeds."""
199
backing = self.get_transport()
200
dir = self.make_bzrdir('.')
201
local_result = dir.cloning_metadir()
202
request_class = smart_dir.SmartServerBzrDirRequestCloningMetaDir
203
request = request_class(backing)
204
expected = smart_req.SuccessfulSmartServerResponse(
205
(local_result.network_name(),
206
local_result.repository_format.network_name(),
207
('branch', local_result.get_branch_format().network_name())))
208
self.assertEqual(expected, request.execute('', 'False'))
210
def test_cloning_metadir_reference(self):
211
"""The request fails when bzrdir contains a branch reference."""
212
backing = self.get_transport()
213
referenced_branch = self.make_branch('referenced')
214
dir = self.make_bzrdir('.')
215
local_result = dir.cloning_metadir()
216
reference = _mod_branch.BranchReferenceFormat().initialize(
217
dir, target_branch=referenced_branch)
218
reference_url = _mod_branch.BranchReferenceFormat().get_reference(dir)
219
# The server shouldn't try to follow the branch reference, so it's fine
220
# if the referenced branch isn't reachable.
221
backing.rename('referenced', 'moved')
222
request_class = smart_dir.SmartServerBzrDirRequestCloningMetaDir
223
request = request_class(backing)
224
expected = smart_req.FailedSmartServerResponse(('BranchReference',))
225
self.assertEqual(expected, request.execute('', 'False'))
228
class TestSmartServerRequestCreateRepository(tests.TestCaseWithMemoryTransport):
229
"""Tests for BzrDir.create_repository."""
231
def test_makes_repository(self):
232
"""When there is a bzrdir present, the call succeeds."""
233
backing = self.get_transport()
234
self.make_bzrdir('.')
235
request_class = smart_dir.SmartServerRequestCreateRepository
236
request = request_class(backing)
237
reference_bzrdir_format = bzrdir.format_registry.get('pack-0.92')()
238
reference_format = reference_bzrdir_format.repository_format
239
network_name = reference_format.network_name()
240
expected = smart_req.SuccessfulSmartServerResponse(
241
('ok', 'no', 'no', 'no', network_name))
242
self.assertEqual(expected, request.execute('', network_name, 'True'))
245
class TestSmartServerRequestFindRepository(tests.TestCaseWithMemoryTransport):
71
SmartServerResponse(('ok', )))
74
class TestSmartServerRequestFindRepository(tests.TestCaseWithTransport):
246
75
"""Tests for BzrDir.find_repository."""
248
77
def test_no_repository(self):
249
78
"""When there is no repository to be found, ('norepository', ) is returned."""
250
79
backing = self.get_transport()
251
request = self._request_class(backing)
80
request = smart.bzrdir.SmartServerRequestFindRepository(backing)
252
81
self.make_bzrdir('.')
253
self.assertEqual(smart_req.SmartServerResponse(('norepository', )),
82
self.assertEqual(SmartServerResponse(('norepository', )),
83
request.execute(backing.local_abspath('')))
256
85
def test_nonshared_repository(self):
257
# nonshared repositorys only allow 'find' to return a handle when the
258
# path the repository is being searched on is the same as that that
86
# nonshared repositorys only allow 'find' to return a handle when the
87
# path the repository is being searched on is the same as that that
259
88
# the repository is at.
260
89
backing = self.get_transport()
261
request = self._request_class(backing)
90
request = smart.bzrdir.SmartServerRequestFindRepository(backing)
262
91
result = self._make_repository_and_result()
263
self.assertEqual(result, request.execute(''))
92
self.assertEqual(result, request.execute(backing.local_abspath('')))
264
93
self.make_bzrdir('subdir')
265
self.assertEqual(smart_req.SmartServerResponse(('norepository', )),
266
request.execute('subdir'))
94
self.assertEqual(SmartServerResponse(('norepository', )),
95
request.execute(backing.local_abspath('subdir')))
268
97
def _make_repository_and_result(self, shared=False, format=None):
269
98
"""Convenience function to setup a repository.
282
if repo._format.supports_external_lookups:
286
if (smart_dir.SmartServerRequestFindRepositoryV3 ==
287
self._request_class):
288
return smart_req.SuccessfulSmartServerResponse(
289
('ok', '', rich_root, subtrees, external,
290
repo._format.network_name()))
291
elif (smart_dir.SmartServerRequestFindRepositoryV2 ==
292
self._request_class):
293
# All tests so far are on formats, and for non-external
295
return smart_req.SuccessfulSmartServerResponse(
296
('ok', '', rich_root, subtrees, external))
298
return smart_req.SuccessfulSmartServerResponse(
299
('ok', '', rich_root, subtrees))
111
return SmartServerResponse(('ok', '', rich_root, subtrees))
301
113
def test_shared_repository(self):
302
114
"""When there is a shared repository, we get 'ok', 'relpath-to-repo'."""
303
115
backing = self.get_transport()
304
request = self._request_class(backing)
116
request = smart.bzrdir.SmartServerRequestFindRepository(backing)
305
117
result = self._make_repository_and_result(shared=True)
306
self.assertEqual(result, request.execute(''))
118
self.assertEqual(result, request.execute(backing.local_abspath('')))
307
119
self.make_bzrdir('subdir')
308
result2 = smart_req.SmartServerResponse(
309
result.args[0:1] + ('..', ) + result.args[2:])
120
result2 = SmartServerResponse(result.args[0:1] + ('..', ) + result.args[2:])
310
121
self.assertEqual(result2,
311
request.execute('subdir'))
122
request.execute(backing.local_abspath('subdir')))
312
123
self.make_bzrdir('subdir/deeper')
313
result3 = smart_req.SmartServerResponse(
314
result.args[0:1] + ('../..', ) + result.args[2:])
124
result3 = SmartServerResponse(result.args[0:1] + ('../..', ) + result.args[2:])
315
125
self.assertEqual(result3,
316
request.execute('subdir/deeper'))
126
request.execute(backing.local_abspath('subdir/deeper')))
318
128
def test_rich_root_and_subtree_encoding(self):
319
129
"""Test for the format attributes for rich root and subtree support."""
320
130
backing = self.get_transport()
321
request = self._request_class(backing)
322
result = self._make_repository_and_result(
323
format='dirstate-with-subtree')
131
request = smart.bzrdir.SmartServerRequestFindRepository(backing)
132
result = self._make_repository_and_result(format='dirstate-with-subtree')
324
133
# check the test will be valid
325
134
self.assertEqual('yes', result.args[2])
326
135
self.assertEqual('yes', result.args[3])
327
self.assertEqual(result, request.execute(''))
329
def test_supports_external_lookups_no_v2(self):
330
"""Test for the supports_external_lookups attribute."""
331
backing = self.get_transport()
332
request = self._request_class(backing)
333
result = self._make_repository_and_result(
334
format='dirstate-with-subtree')
335
# check the test will be valid
336
self.assertEqual('no', result.args[4])
337
self.assertEqual(result, request.execute(''))
340
class TestSmartServerBzrDirRequestGetConfigFile(
341
tests.TestCaseWithMemoryTransport):
342
"""Tests for BzrDir.get_config_file."""
344
def test_present(self):
345
backing = self.get_transport()
346
dir = self.make_bzrdir('.')
347
dir.get_config().set_default_stack_on("/")
348
local_result = dir._get_config()._get_config_file().read()
349
request_class = smart_dir.SmartServerBzrDirRequestConfigFile
350
request = request_class(backing)
351
expected = smart_req.SuccessfulSmartServerResponse((), local_result)
352
self.assertEqual(expected, request.execute(''))
354
def test_missing(self):
355
backing = self.get_transport()
356
dir = self.make_bzrdir('.')
357
request_class = smart_dir.SmartServerBzrDirRequestConfigFile
358
request = request_class(backing)
359
expected = smart_req.SuccessfulSmartServerResponse((), '')
360
self.assertEqual(expected, request.execute(''))
363
class TestSmartServerRequestInitializeBzrDir(tests.TestCaseWithMemoryTransport):
136
self.assertEqual(result, request.execute(backing.local_abspath('')))
139
class TestSmartServerRequestInitializeBzrDir(tests.TestCaseWithTransport):
365
141
def test_empty_dir(self):
366
142
"""Initializing an empty dir should succeed and do it."""
367
143
backing = self.get_transport()
368
request = smart_dir.SmartServerRequestInitializeBzrDir(backing)
369
self.assertEqual(smart_req.SmartServerResponse(('ok', )),
144
request = smart.bzrdir.SmartServerRequestInitializeBzrDir(backing)
145
self.assertEqual(SmartServerResponse(('ok', )),
146
request.execute(backing.local_abspath('.')))
371
147
made_dir = bzrdir.BzrDir.open_from_transport(backing)
372
# no branch, tree or repository is expected with the current
148
# no branch, tree or repository is expected with the current
373
149
# default formart.
374
150
self.assertRaises(errors.NoWorkingTree, made_dir.open_workingtree)
375
151
self.assertRaises(errors.NotBranchError, made_dir.open_branch)
378
154
def test_missing_dir(self):
379
155
"""Initializing a missing directory should fail like the bzrdir api."""
380
156
backing = self.get_transport()
381
request = smart_dir.SmartServerRequestInitializeBzrDir(backing)
157
request = smart.bzrdir.SmartServerRequestInitializeBzrDir(backing)
382
158
self.assertRaises(errors.NoSuchFile,
383
request.execute, 'subdir')
159
request.execute, backing.local_abspath('subdir'))
385
161
def test_initialized_dir(self):
386
162
"""Initializing an extant bzrdir should fail like the bzrdir api."""
387
163
backing = self.get_transport()
388
request = smart_dir.SmartServerRequestInitializeBzrDir(backing)
164
request = smart.bzrdir.SmartServerRequestInitializeBzrDir(backing)
389
165
self.make_bzrdir('subdir')
390
166
self.assertRaises(errors.FileExists,
391
request.execute, 'subdir')
394
class TestSmartServerRequestBzrDirInitializeEx(
395
tests.TestCaseWithMemoryTransport):
396
"""Basic tests for BzrDir.initialize_ex_1.16 in the smart server.
398
The main unit tests in test_bzrdir exercise the API comprehensively.
401
def test_empty_dir(self):
402
"""Initializing an empty dir should succeed and do it."""
403
backing = self.get_transport()
404
name = self.make_bzrdir('reference')._format.network_name()
405
request = smart_dir.SmartServerRequestBzrDirInitializeEx(backing)
407
smart_req.SmartServerResponse(('', '', '', '', '', '', name,
408
'False', '', '', '')),
409
request.execute(name, '', 'True', 'False', 'False', '', '', '', '',
411
made_dir = bzrdir.BzrDir.open_from_transport(backing)
412
# no branch, tree or repository is expected with the current
414
self.assertRaises(errors.NoWorkingTree, made_dir.open_workingtree)
415
self.assertRaises(errors.NotBranchError, made_dir.open_branch)
416
self.assertRaises(errors.NoRepositoryPresent, made_dir.open_repository)
418
def test_missing_dir(self):
419
"""Initializing a missing directory should fail like the bzrdir api."""
420
backing = self.get_transport()
421
name = self.make_bzrdir('reference')._format.network_name()
422
request = smart_dir.SmartServerRequestBzrDirInitializeEx(backing)
423
self.assertRaises(errors.NoSuchFile, request.execute, name,
424
'subdir/dir', 'False', 'False', 'False', '', '', '', '', 'False')
426
def test_initialized_dir(self):
427
"""Initializing an extant directory should fail like the bzrdir api."""
428
backing = self.get_transport()
429
name = self.make_bzrdir('reference')._format.network_name()
430
request = smart_dir.SmartServerRequestBzrDirInitializeEx(backing)
431
self.make_bzrdir('subdir')
432
self.assertRaises(errors.FileExists, request.execute, name, 'subdir',
433
'False', 'False', 'False', '', '', '', '', 'False')
436
class TestSmartServerRequestOpenBzrDir(tests.TestCaseWithMemoryTransport):
438
def test_no_directory(self):
439
backing = self.get_transport()
440
request = smart_dir.SmartServerRequestOpenBzrDir(backing)
441
self.assertEqual(smart_req.SmartServerResponse(('no', )),
442
request.execute('does-not-exist'))
444
def test_empty_directory(self):
445
backing = self.get_transport()
446
backing.mkdir('empty')
447
request = smart_dir.SmartServerRequestOpenBzrDir(backing)
448
self.assertEqual(smart_req.SmartServerResponse(('no', )),
449
request.execute('empty'))
451
def test_outside_root_client_path(self):
452
backing = self.get_transport()
453
request = smart_dir.SmartServerRequestOpenBzrDir(backing,
454
root_client_path='root')
455
self.assertEqual(smart_req.SmartServerResponse(('no', )),
456
request.execute('not-root'))
459
class TestSmartServerRequestOpenBzrDir_2_1(tests.TestCaseWithMemoryTransport):
461
def test_no_directory(self):
462
backing = self.get_transport()
463
request = smart_dir.SmartServerRequestOpenBzrDir_2_1(backing)
464
self.assertEqual(smart_req.SmartServerResponse(('no', )),
465
request.execute('does-not-exist'))
467
def test_empty_directory(self):
468
backing = self.get_transport()
469
backing.mkdir('empty')
470
request = smart_dir.SmartServerRequestOpenBzrDir_2_1(backing)
471
self.assertEqual(smart_req.SmartServerResponse(('no', )),
472
request.execute('empty'))
474
def test_present_without_workingtree(self):
475
backing = self.get_transport()
476
request = smart_dir.SmartServerRequestOpenBzrDir_2_1(backing)
477
self.make_bzrdir('.')
478
self.assertEqual(smart_req.SmartServerResponse(('yes', 'no')),
481
def test_outside_root_client_path(self):
482
backing = self.get_transport()
483
request = smart_dir.SmartServerRequestOpenBzrDir_2_1(backing,
484
root_client_path='root')
485
self.assertEqual(smart_req.SmartServerResponse(('no',)),
486
request.execute('not-root'))
489
class TestSmartServerRequestOpenBzrDir_2_1_disk(TestCaseWithChrootedTransport):
491
def test_present_with_workingtree(self):
492
self.vfs_transport_factory = test_server.LocalURLServer
493
backing = self.get_transport()
494
request = smart_dir.SmartServerRequestOpenBzrDir_2_1(backing)
495
bd = self.make_bzrdir('.')
496
bd.create_repository()
498
bd.create_workingtree()
499
self.assertEqual(smart_req.SmartServerResponse(('yes', 'yes')),
503
class TestSmartServerRequestOpenBranch(TestCaseWithChrootedTransport):
167
request.execute, backing.local_abspath('subdir'))
170
class TestSmartServerRequestOpenBranch(tests.TestCaseWithTransport):
505
172
def test_no_branch(self):
506
173
"""When there is no branch, ('nobranch', ) is returned."""
507
174
backing = self.get_transport()
508
request = smart_dir.SmartServerRequestOpenBranch(backing)
175
request = smart.bzrdir.SmartServerRequestOpenBranch(backing)
509
176
self.make_bzrdir('.')
510
self.assertEqual(smart_req.SmartServerResponse(('nobranch', )),
177
self.assertEqual(SmartServerResponse(('nobranch', )),
178
request.execute(backing.local_abspath('')))
513
180
def test_branch(self):
514
181
"""When there is a branch, 'ok' is returned."""
515
182
backing = self.get_transport()
516
request = smart_dir.SmartServerRequestOpenBranch(backing)
183
request = smart.bzrdir.SmartServerRequestOpenBranch(backing)
517
184
self.make_branch('.')
518
self.assertEqual(smart_req.SmartServerResponse(('ok', '')),
521
def test_branch_reference(self):
522
"""When there is a branch reference, the reference URL is returned."""
523
self.vfs_transport_factory = test_server.LocalURLServer
524
backing = self.get_transport()
525
request = smart_dir.SmartServerRequestOpenBranch(backing)
526
branch = self.make_branch('branch')
527
checkout = branch.create_checkout('reference',lightweight=True)
528
reference_url = _mod_branch.BranchReferenceFormat().get_reference(
530
self.assertFileEqual(reference_url, 'reference/.bzr/branch/location')
531
self.assertEqual(smart_req.SmartServerResponse(('ok', reference_url)),
532
request.execute('reference'))
534
def test_notification_on_branch_from_repository(self):
535
"""When there is a repository, the error should return details."""
536
backing = self.get_transport()
537
request = smart_dir.SmartServerRequestOpenBranch(backing)
538
repo = self.make_repository('.')
539
self.assertEqual(smart_req.SmartServerResponse(('nobranch',)),
543
class TestSmartServerRequestOpenBranchV2(TestCaseWithChrootedTransport):
545
def test_no_branch(self):
546
"""When there is no branch, ('nobranch', ) is returned."""
547
backing = self.get_transport()
548
self.make_bzrdir('.')
549
request = smart_dir.SmartServerRequestOpenBranchV2(backing)
550
self.assertEqual(smart_req.SmartServerResponse(('nobranch', )),
553
def test_branch(self):
554
"""When there is a branch, 'ok' is returned."""
555
backing = self.get_transport()
556
expected = self.make_branch('.')._format.network_name()
557
request = smart_dir.SmartServerRequestOpenBranchV2(backing)
558
self.assertEqual(smart_req.SuccessfulSmartServerResponse(
559
('branch', expected)),
562
def test_branch_reference(self):
563
"""When there is a branch reference, the reference URL is returned."""
564
self.vfs_transport_factory = test_server.LocalURLServer
565
backing = self.get_transport()
566
request = smart_dir.SmartServerRequestOpenBranchV2(backing)
567
branch = self.make_branch('branch')
568
checkout = branch.create_checkout('reference',lightweight=True)
569
reference_url = _mod_branch.BranchReferenceFormat().get_reference(
571
self.assertFileEqual(reference_url, 'reference/.bzr/branch/location')
572
self.assertEqual(smart_req.SuccessfulSmartServerResponse(
573
('ref', reference_url)),
574
request.execute('reference'))
576
def test_stacked_branch(self):
577
"""Opening a stacked branch does not open the stacked-on branch."""
578
trunk = self.make_branch('trunk')
579
feature = self.make_branch('feature')
580
feature.set_stacked_on_url(trunk.base)
582
_mod_branch.Branch.hooks.install_named_hook(
583
'open', opened_branches.append, None)
584
backing = self.get_transport()
585
request = smart_dir.SmartServerRequestOpenBranchV2(backing)
588
response = request.execute('feature')
590
request.teardown_jail()
591
expected_format = feature._format.network_name()
592
self.assertEqual(smart_req.SuccessfulSmartServerResponse(
593
('branch', expected_format)),
595
self.assertLength(1, opened_branches)
597
def test_notification_on_branch_from_repository(self):
598
"""When there is a repository, the error should return details."""
599
backing = self.get_transport()
600
request = smart_dir.SmartServerRequestOpenBranchV2(backing)
601
repo = self.make_repository('.')
602
self.assertEqual(smart_req.SmartServerResponse(('nobranch',)),
606
class TestSmartServerRequestOpenBranchV3(TestCaseWithChrootedTransport):
608
def test_no_branch(self):
609
"""When there is no branch, ('nobranch', ) is returned."""
610
backing = self.get_transport()
611
self.make_bzrdir('.')
612
request = smart_dir.SmartServerRequestOpenBranchV3(backing)
613
self.assertEqual(smart_req.SmartServerResponse(('nobranch',)),
616
def test_branch(self):
617
"""When there is a branch, 'ok' is returned."""
618
backing = self.get_transport()
619
expected = self.make_branch('.')._format.network_name()
620
request = smart_dir.SmartServerRequestOpenBranchV3(backing)
621
self.assertEqual(smart_req.SuccessfulSmartServerResponse(
622
('branch', expected)),
625
def test_branch_reference(self):
626
"""When there is a branch reference, the reference URL is returned."""
627
self.vfs_transport_factory = test_server.LocalURLServer
628
backing = self.get_transport()
629
request = smart_dir.SmartServerRequestOpenBranchV3(backing)
630
branch = self.make_branch('branch')
631
checkout = branch.create_checkout('reference',lightweight=True)
632
reference_url = _mod_branch.BranchReferenceFormat().get_reference(
634
self.assertFileEqual(reference_url, 'reference/.bzr/branch/location')
635
self.assertEqual(smart_req.SuccessfulSmartServerResponse(
636
('ref', reference_url)),
637
request.execute('reference'))
639
def test_stacked_branch(self):
640
"""Opening a stacked branch does not open the stacked-on branch."""
641
trunk = self.make_branch('trunk')
642
feature = self.make_branch('feature')
643
feature.set_stacked_on_url(trunk.base)
645
_mod_branch.Branch.hooks.install_named_hook(
646
'open', opened_branches.append, None)
647
backing = self.get_transport()
648
request = smart_dir.SmartServerRequestOpenBranchV3(backing)
651
response = request.execute('feature')
653
request.teardown_jail()
654
expected_format = feature._format.network_name()
655
self.assertEqual(smart_req.SuccessfulSmartServerResponse(
656
('branch', expected_format)),
658
self.assertLength(1, opened_branches)
660
def test_notification_on_branch_from_repository(self):
661
"""When there is a repository, the error should return details."""
662
backing = self.get_transport()
663
request = smart_dir.SmartServerRequestOpenBranchV3(backing)
664
repo = self.make_repository('.')
665
self.assertEqual(smart_req.SmartServerResponse(
666
('nobranch', 'location is a repository')),
670
class TestSmartServerRequestRevisionHistory(tests.TestCaseWithMemoryTransport):
185
self.assertEqual(SmartServerResponse(('ok', '')),
186
request.execute(backing.local_abspath('')))
188
def test_branch_reference(self):
189
"""When there is a branch reference, the reference URL is returned."""
190
backing = self.get_transport()
191
request = smart.bzrdir.SmartServerRequestOpenBranch(backing)
192
branch = self.make_branch('branch')
193
checkout = branch.create_checkout('reference',lightweight=True)
194
# TODO: once we have an API to probe for references of any sort, we
196
reference_url = backing.abspath('branch') + '/'
197
self.assertFileEqual(reference_url, 'reference/.bzr/branch/location')
198
self.assertEqual(SmartServerResponse(('ok', reference_url)),
199
request.execute(backing.local_abspath('reference')))
202
class TestSmartServerRequestRevisionHistory(tests.TestCaseWithTransport):
672
204
def test_empty(self):
673
205
"""For an empty branch, the body is empty."""
674
206
backing = self.get_transport()
675
request = smart_branch.SmartServerRequestRevisionHistory(backing)
207
request = smart.branch.SmartServerRequestRevisionHistory(backing)
676
208
self.make_branch('.')
677
self.assertEqual(smart_req.SmartServerResponse(('ok', ), ''),
209
self.assertEqual(SmartServerResponse(('ok', ), ''),
210
request.execute(backing.local_abspath('')))
680
212
def test_not_empty(self):
681
213
"""For a non-empty branch, the body is empty."""
682
214
backing = self.get_transport()
683
request = smart_branch.SmartServerRequestRevisionHistory(backing)
215
request = smart.branch.SmartServerRequestRevisionHistory(backing)
684
216
tree = self.make_branch_and_memory_tree('.')
685
217
tree.lock_write()
735
266
r2 = tree.commit('2nd commit', rev_id=rev_id_utf8)
737
268
self.assertEqual(
738
smart_req.SmartServerResponse(('ok', '2', rev_id_utf8)),
742
class TestSmartServerBranchRequestGetConfigFile(
743
tests.TestCaseWithMemoryTransport):
269
SmartServerResponse(('ok', '2', rev_id_utf8)),
270
request.execute(backing.local_abspath('')))
273
class TestSmartServerBranchRequestGetConfigFile(tests.TestCaseWithTransport):
745
275
def test_default(self):
746
276
"""With no file, we get empty content."""
747
277
backing = self.get_transport()
748
request = smart_branch.SmartServerBranchGetConfigFile(backing)
278
request = smart.branch.SmartServerBranchGetConfigFile(backing)
749
279
branch = self.make_branch('.')
750
280
# there should be no file by default
752
self.assertEqual(smart_req.SmartServerResponse(('ok', ), content),
282
self.assertEqual(SmartServerResponse(('ok', ), content),
283
request.execute(backing.local_abspath('')))
755
285
def test_with_content(self):
756
286
# SmartServerBranchGetConfigFile should return the content from
757
287
# branch.control_files.get('branch.conf') for now - in the future it may
758
# perform more complex processing.
759
backing = self.get_transport()
760
request = smart_branch.SmartServerBranchGetConfigFile(backing)
761
branch = self.make_branch('.')
762
branch._transport.put_bytes('branch.conf', 'foo bar baz')
763
self.assertEqual(smart_req.SmartServerResponse(('ok', ), 'foo bar baz'),
767
class TestLockedBranch(tests.TestCaseWithMemoryTransport):
769
def get_lock_tokens(self, branch):
770
branch_token = branch.lock_write().branch_token
771
repo_token = branch.repository.lock_write().repository_token
772
branch.repository.unlock()
773
return branch_token, repo_token
776
class TestSmartServerBranchRequestSetConfigOption(TestLockedBranch):
778
def test_value_name(self):
779
branch = self.make_branch('.')
780
request = smart_branch.SmartServerBranchRequestSetConfigOption(
781
branch.bzrdir.root_transport)
782
branch_token, repo_token = self.get_lock_tokens(branch)
783
config = branch._get_config()
784
result = request.execute('', branch_token, repo_token, 'bar', 'foo',
786
self.assertEqual(smart_req.SuccessfulSmartServerResponse(()), result)
787
self.assertEqual('bar', config.get_option('foo'))
791
def test_value_name_section(self):
792
branch = self.make_branch('.')
793
request = smart_branch.SmartServerBranchRequestSetConfigOption(
794
branch.bzrdir.root_transport)
795
branch_token, repo_token = self.get_lock_tokens(branch)
796
config = branch._get_config()
797
result = request.execute('', branch_token, repo_token, 'bar', 'foo',
799
self.assertEqual(smart_req.SuccessfulSmartServerResponse(()), result)
800
self.assertEqual('bar', config.get_option('foo', 'gam'))
805
class TestSmartServerBranchRequestSetTagsBytes(TestLockedBranch):
806
# Only called when the branch format and tags match [yay factory
807
# methods] so only need to test straight forward cases.
809
def test_set_bytes(self):
810
base_branch = self.make_branch('base')
811
tag_bytes = base_branch._get_tags_bytes()
812
# get_lock_tokens takes out a lock.
813
branch_token, repo_token = self.get_lock_tokens(base_branch)
814
request = smart_branch.SmartServerBranchSetTagsBytes(
815
self.get_transport())
816
response = request.execute('base', branch_token, repo_token)
817
self.assertEqual(None, response)
818
response = request.do_chunk(tag_bytes)
819
self.assertEqual(None, response)
820
response = request.do_end()
822
smart_req.SuccessfulSmartServerResponse(()), response)
825
def test_lock_failed(self):
826
base_branch = self.make_branch('base')
827
base_branch.lock_write()
828
tag_bytes = base_branch._get_tags_bytes()
829
request = smart_branch.SmartServerBranchSetTagsBytes(
830
self.get_transport())
831
self.assertRaises(errors.TokenMismatch, request.execute,
832
'base', 'wrong token', 'wrong token')
833
# The request handler will keep processing the message parts, so even
834
# if the request fails immediately do_chunk and do_end are still
836
request.do_chunk(tag_bytes)
842
class SetLastRevisionTestBase(TestLockedBranch):
843
"""Base test case for verbs that implement set_last_revision."""
846
tests.TestCaseWithMemoryTransport.setUp(self)
847
backing_transport = self.get_transport()
848
self.request = self.request_class(backing_transport)
849
self.tree = self.make_branch_and_memory_tree('.')
851
def lock_branch(self):
852
return self.get_lock_tokens(self.tree.branch)
854
def unlock_branch(self):
855
self.tree.branch.unlock()
857
def set_last_revision(self, revision_id, revno):
858
branch_token, repo_token = self.lock_branch()
859
response = self._set_last_revision(
860
revision_id, revno, branch_token, repo_token)
864
def assertRequestSucceeds(self, revision_id, revno):
865
response = self.set_last_revision(revision_id, revno)
866
self.assertEqual(smart_req.SuccessfulSmartServerResponse(('ok',)),
870
class TestSetLastRevisionVerbMixin(object):
871
"""Mixin test case for verbs that implement set_last_revision."""
873
def test_set_null_to_null(self):
874
"""An empty branch can have its last revision set to 'null:'."""
875
self.assertRequestSucceeds('null:', 0)
877
def test_NoSuchRevision(self):
878
"""If the revision_id is not present, the verb returns NoSuchRevision.
880
revision_id = 'non-existent revision'
881
self.assertEqual(smart_req.FailedSmartServerResponse(('NoSuchRevision',
883
self.set_last_revision(revision_id, 1))
885
def make_tree_with_two_commits(self):
886
self.tree.lock_write()
888
rev_id_utf8 = u'\xc8'.encode('utf-8')
889
r1 = self.tree.commit('1st commit', rev_id=rev_id_utf8)
890
r2 = self.tree.commit('2nd commit', rev_id='rev-2')
893
def test_branch_last_revision_info_is_updated(self):
894
"""A branch's tip can be set to a revision that is present in its
897
# Make a branch with an empty revision history, but two revisions in
899
self.make_tree_with_two_commits()
900
rev_id_utf8 = u'\xc8'.encode('utf-8')
901
self.tree.branch.set_revision_history([])
903
(0, 'null:'), self.tree.branch.last_revision_info())
904
# We can update the branch to a revision that is present in the
906
self.assertRequestSucceeds(rev_id_utf8, 1)
908
(1, rev_id_utf8), self.tree.branch.last_revision_info())
910
def test_branch_last_revision_info_rewind(self):
911
"""A branch's tip can be set to a revision that is an ancestor of the
914
self.make_tree_with_two_commits()
915
rev_id_utf8 = u'\xc8'.encode('utf-8')
917
(2, 'rev-2'), self.tree.branch.last_revision_info())
918
self.assertRequestSucceeds(rev_id_utf8, 1)
920
(1, rev_id_utf8), self.tree.branch.last_revision_info())
922
def test_TipChangeRejected(self):
923
"""If a pre_change_branch_tip hook raises TipChangeRejected, the verb
924
returns TipChangeRejected.
926
rejection_message = u'rejection message\N{INTERROBANG}'
927
def hook_that_rejects(params):
928
raise errors.TipChangeRejected(rejection_message)
929
_mod_branch.Branch.hooks.install_named_hook(
930
'pre_change_branch_tip', hook_that_rejects, None)
932
smart_req.FailedSmartServerResponse(
933
('TipChangeRejected', rejection_message.encode('utf-8'))),
934
self.set_last_revision('null:', 0))
937
class TestSmartServerBranchRequestSetLastRevision(
938
SetLastRevisionTestBase, TestSetLastRevisionVerbMixin):
939
"""Tests for Branch.set_last_revision verb."""
941
request_class = smart_branch.SmartServerBranchRequestSetLastRevision
943
def _set_last_revision(self, revision_id, revno, branch_token, repo_token):
944
return self.request.execute(
945
'', branch_token, repo_token, revision_id)
948
class TestSmartServerBranchRequestSetLastRevisionInfo(
949
SetLastRevisionTestBase, TestSetLastRevisionVerbMixin):
950
"""Tests for Branch.set_last_revision_info verb."""
952
request_class = smart_branch.SmartServerBranchRequestSetLastRevisionInfo
954
def _set_last_revision(self, revision_id, revno, branch_token, repo_token):
955
return self.request.execute(
956
'', branch_token, repo_token, revno, revision_id)
958
def test_NoSuchRevision(self):
959
"""Branch.set_last_revision_info does not have to return
960
NoSuchRevision if the revision_id is absent.
962
raise tests.TestNotApplicable()
965
class TestSmartServerBranchRequestSetLastRevisionEx(
966
SetLastRevisionTestBase, TestSetLastRevisionVerbMixin):
967
"""Tests for Branch.set_last_revision_ex verb."""
969
request_class = smart_branch.SmartServerBranchRequestSetLastRevisionEx
971
def _set_last_revision(self, revision_id, revno, branch_token, repo_token):
972
return self.request.execute(
973
'', branch_token, repo_token, revision_id, 0, 0)
975
def assertRequestSucceeds(self, revision_id, revno):
976
response = self.set_last_revision(revision_id, revno)
978
smart_req.SuccessfulSmartServerResponse(('ok', revno, revision_id)),
981
def test_branch_last_revision_info_rewind(self):
982
"""A branch's tip can be set to a revision that is an ancestor of the
983
current tip, but only if allow_overwrite_descendant is passed.
985
self.make_tree_with_two_commits()
986
rev_id_utf8 = u'\xc8'.encode('utf-8')
988
(2, 'rev-2'), self.tree.branch.last_revision_info())
989
# If allow_overwrite_descendant flag is 0, then trying to set the tip
990
# to an older revision ID has no effect.
991
branch_token, repo_token = self.lock_branch()
992
response = self.request.execute(
993
'', branch_token, repo_token, rev_id_utf8, 0, 0)
995
smart_req.SuccessfulSmartServerResponse(('ok', 2, 'rev-2')),
998
(2, 'rev-2'), self.tree.branch.last_revision_info())
1000
# If allow_overwrite_descendant flag is 1, then setting the tip to an
1002
response = self.request.execute(
1003
'', branch_token, repo_token, rev_id_utf8, 0, 1)
1005
smart_req.SuccessfulSmartServerResponse(('ok', 1, rev_id_utf8)),
1007
self.unlock_branch()
1009
(1, rev_id_utf8), self.tree.branch.last_revision_info())
1011
def make_branch_with_divergent_history(self):
1012
"""Make a branch with divergent history in its repo.
1014
The branch's tip will be 'child-2', and the repo will also contain
1015
'child-1', which diverges from a common base revision.
1017
self.tree.lock_write()
1019
r1 = self.tree.commit('1st commit')
1020
revno_1, revid_1 = self.tree.branch.last_revision_info()
1021
r2 = self.tree.commit('2nd commit', rev_id='child-1')
1022
# Undo the second commit
1023
self.tree.branch.set_last_revision_info(revno_1, revid_1)
1024
self.tree.set_parent_ids([revid_1])
1025
# Make a new second commit, child-2. child-2 has diverged from
1027
new_r2 = self.tree.commit('2nd commit', rev_id='child-2')
1030
def test_not_allow_diverged(self):
1031
"""If allow_diverged is not passed, then setting a divergent history
1032
returns a Diverged error.
1034
self.make_branch_with_divergent_history()
1036
smart_req.FailedSmartServerResponse(('Diverged',)),
1037
self.set_last_revision('child-1', 2))
1038
# The branch tip was not changed.
1039
self.assertEqual('child-2', self.tree.branch.last_revision())
1041
def test_allow_diverged(self):
1042
"""If allow_diverged is passed, then setting a divergent history
1045
self.make_branch_with_divergent_history()
1046
branch_token, repo_token = self.lock_branch()
1047
response = self.request.execute(
1048
'', branch_token, repo_token, 'child-1', 1, 0)
1050
smart_req.SuccessfulSmartServerResponse(('ok', 2, 'child-1')),
1052
self.unlock_branch()
1053
# The branch tip was changed.
1054
self.assertEqual('child-1', self.tree.branch.last_revision())
1057
class TestSmartServerBranchRequestGetParent(tests.TestCaseWithMemoryTransport):
1059
def test_get_parent_none(self):
1060
base_branch = self.make_branch('base')
1061
request = smart_branch.SmartServerBranchGetParent(self.get_transport())
1062
response = request.execute('base')
1064
smart_req.SuccessfulSmartServerResponse(('',)), response)
1066
def test_get_parent_something(self):
1067
base_branch = self.make_branch('base')
1068
base_branch.set_parent(self.get_url('foo'))
1069
request = smart_branch.SmartServerBranchGetParent(self.get_transport())
1070
response = request.execute('base')
1072
smart_req.SuccessfulSmartServerResponse(("../foo",)),
1076
class TestSmartServerBranchRequestSetParent(TestLockedBranch):
1078
def test_set_parent_none(self):
1079
branch = self.make_branch('base', format="1.9")
1081
branch._set_parent_location('foo')
1083
request = smart_branch.SmartServerBranchRequestSetParentLocation(
1084
self.get_transport())
1085
branch_token, repo_token = self.get_lock_tokens(branch)
1087
response = request.execute('base', branch_token, repo_token, '')
1090
self.assertEqual(smart_req.SuccessfulSmartServerResponse(()), response)
1091
self.assertEqual(None, branch.get_parent())
1093
def test_set_parent_something(self):
1094
branch = self.make_branch('base', format="1.9")
1095
request = smart_branch.SmartServerBranchRequestSetParentLocation(
1096
self.get_transport())
1097
branch_token, repo_token = self.get_lock_tokens(branch)
1099
response = request.execute('base', branch_token, repo_token,
1103
self.assertEqual(smart_req.SuccessfulSmartServerResponse(()), response)
1104
self.assertEqual('http://bar/', branch.get_parent())
1107
class TestSmartServerBranchRequestGetTagsBytes(
1108
tests.TestCaseWithMemoryTransport):
1109
# Only called when the branch format and tags match [yay factory
1110
# methods] so only need to test straight forward cases.
1112
def test_get_bytes(self):
1113
base_branch = self.make_branch('base')
1114
request = smart_branch.SmartServerBranchGetTagsBytes(
1115
self.get_transport())
1116
response = request.execute('base')
1118
smart_req.SuccessfulSmartServerResponse(('',)), response)
1121
class TestSmartServerBranchRequestGetStackedOnURL(tests.TestCaseWithMemoryTransport):
1123
def test_get_stacked_on_url(self):
1124
base_branch = self.make_branch('base', format='1.6')
1125
stacked_branch = self.make_branch('stacked', format='1.6')
1126
# typically should be relative
1127
stacked_branch.set_stacked_on_url('../base')
1128
request = smart_branch.SmartServerBranchRequestGetStackedOnURL(
1129
self.get_transport())
1130
response = request.execute('stacked')
1132
smart_req.SmartServerResponse(('ok', '../base')),
1136
class TestSmartServerBranchRequestLockWrite(TestLockedBranch):
1139
tests.TestCaseWithMemoryTransport.setUp(self)
288
# perform more complex processing.
289
backing = self.get_transport()
290
request = smart.branch.SmartServerBranchGetConfigFile(backing)
291
branch = self.make_branch('.')
292
branch.control_files.put_utf8('branch.conf', 'foo bar baz')
293
self.assertEqual(SmartServerResponse(('ok', ), 'foo bar baz'),
294
request.execute(backing.local_abspath('')))
297
class TestSmartServerBranchRequestSetLastRevision(tests.TestCaseWithTransport):
299
def test_empty(self):
300
backing = self.get_transport()
301
request = smart.branch.SmartServerBranchRequestSetLastRevision(backing)
302
b = self.make_branch('.')
303
branch_token = b.lock_write()
304
repo_token = b.repository.lock_write()
305
b.repository.unlock()
307
self.assertEqual(SmartServerResponse(('ok',)),
309
backing.local_abspath(''), branch_token, repo_token,
314
def test_not_present_revision_id(self):
315
backing = self.get_transport()
316
request = smart.branch.SmartServerBranchRequestSetLastRevision(backing)
317
b = self.make_branch('.')
318
branch_token = b.lock_write()
319
repo_token = b.repository.lock_write()
320
b.repository.unlock()
322
revision_id = 'non-existent revision'
324
SmartServerResponse(('NoSuchRevision', revision_id)),
326
backing.local_abspath(''), branch_token, repo_token,
331
def test_revision_id_present(self):
332
backing = self.get_transport()
333
request = smart.branch.SmartServerBranchRequestSetLastRevision(backing)
334
tree = self.make_branch_and_memory_tree('.')
337
rev_id_utf8 = u'\xc8'.encode('utf-8')
338
r1 = tree.commit('1st commit', rev_id=rev_id_utf8)
339
r2 = tree.commit('2nd commit')
341
branch_token = tree.branch.lock_write()
342
repo_token = tree.branch.repository.lock_write()
343
tree.branch.repository.unlock()
346
SmartServerResponse(('ok',)),
348
backing.local_abspath(''), branch_token, repo_token,
350
self.assertEqual([rev_id_utf8], tree.branch.revision_history())
354
def test_revision_id_present2(self):
355
backing = self.get_transport()
356
request = smart.branch.SmartServerBranchRequestSetLastRevision(backing)
357
tree = self.make_branch_and_memory_tree('.')
360
rev_id_utf8 = u'\xc8'.encode('utf-8')
361
r1 = tree.commit('1st commit', rev_id=rev_id_utf8)
362
r2 = tree.commit('2nd commit')
364
tree.branch.set_revision_history([])
365
branch_token = tree.branch.lock_write()
366
repo_token = tree.branch.repository.lock_write()
367
tree.branch.repository.unlock()
370
SmartServerResponse(('ok',)),
372
backing.local_abspath(''), branch_token, repo_token,
374
self.assertEqual([rev_id_utf8], tree.branch.revision_history())
379
class TestSmartServerBranchRequestLockWrite(tests.TestCaseWithTransport):
382
tests.TestCaseWithTransport.setUp(self)
383
self.reduceLockdirTimeout()
1141
385
def test_lock_write_on_unlocked_branch(self):
1142
386
backing = self.get_transport()
1143
request = smart_branch.SmartServerBranchRequestLockWrite(backing)
387
request = smart.branch.SmartServerBranchRequestLockWrite(backing)
1144
388
branch = self.make_branch('.', format='knit')
1145
389
repository = branch.repository
1146
response = request.execute('')
390
response = request.execute(backing.local_abspath(''))
1147
391
branch_nonce = branch.control_files._lock.peek().get('nonce')
1148
392
repository_nonce = repository.control_files._lock.peek().get('nonce')
1149
self.assertEqual(smart_req.SmartServerResponse(
1150
('ok', branch_nonce, repository_nonce)),
394
SmartServerResponse(('ok', branch_nonce, repository_nonce)),
1152
396
# The branch (and associated repository) is now locked. Verify that
1153
397
# with a new branch object.
1154
398
new_branch = repository.bzrdir.open_branch()
1155
399
self.assertRaises(errors.LockContention, new_branch.lock_write)
1157
request = smart_branch.SmartServerBranchRequestUnlock(backing)
1158
response = request.execute('', branch_nonce, repository_nonce)
1160
401
def test_lock_write_on_locked_branch(self):
1161
402
backing = self.get_transport()
1162
request = smart_branch.SmartServerBranchRequestLockWrite(backing)
403
request = smart.branch.SmartServerBranchRequestLockWrite(backing)
1163
404
branch = self.make_branch('.')
1164
branch_token = branch.lock_write().branch_token
1165
406
branch.leave_lock_in_place()
1167
response = request.execute('')
408
response = request.execute(backing.local_abspath(''))
1168
409
self.assertEqual(
1169
smart_req.SmartServerResponse(('LockContention',)), response)
1171
branch.lock_write(branch_token)
1172
branch.dont_leave_lock_in_place()
410
SmartServerResponse(('LockContention',)), response)
1175
412
def test_lock_write_with_tokens_on_locked_branch(self):
1176
413
backing = self.get_transport()
1177
request = smart_branch.SmartServerBranchRequestLockWrite(backing)
414
request = smart.branch.SmartServerBranchRequestLockWrite(backing)
1178
415
branch = self.make_branch('.', format='knit')
1179
branch_token, repo_token = self.get_lock_tokens(branch)
416
branch_token = branch.lock_write()
417
repo_token = branch.repository.lock_write()
418
branch.repository.unlock()
1180
419
branch.leave_lock_in_place()
1181
420
branch.repository.leave_lock_in_place()
1183
response = request.execute('',
422
response = request.execute(backing.local_abspath(''),
1184
423
branch_token, repo_token)
1185
424
self.assertEqual(
1186
smart_req.SmartServerResponse(('ok', branch_token, repo_token)),
1189
branch.repository.lock_write(repo_token)
1190
branch.repository.dont_leave_lock_in_place()
1191
branch.repository.unlock()
1192
branch.lock_write(branch_token)
1193
branch.dont_leave_lock_in_place()
425
SmartServerResponse(('ok', branch_token, repo_token)), response)
1196
427
def test_lock_write_with_mismatched_tokens_on_locked_branch(self):
1197
428
backing = self.get_transport()
1198
request = smart_branch.SmartServerBranchRequestLockWrite(backing)
429
request = smart.branch.SmartServerBranchRequestLockWrite(backing)
1199
430
branch = self.make_branch('.', format='knit')
1200
branch_token, repo_token = self.get_lock_tokens(branch)
431
branch_token = branch.lock_write()
432
repo_token = branch.repository.lock_write()
433
branch.repository.unlock()
1201
434
branch.leave_lock_in_place()
1202
435
branch.repository.leave_lock_in_place()
1204
response = request.execute('',
437
response = request.execute(backing.local_abspath(''),
1205
438
branch_token+'xxx', repo_token)
1206
439
self.assertEqual(
1207
smart_req.SmartServerResponse(('TokenMismatch',)), response)
1209
branch.repository.lock_write(repo_token)
1210
branch.repository.dont_leave_lock_in_place()
1211
branch.repository.unlock()
1212
branch.lock_write(branch_token)
1213
branch.dont_leave_lock_in_place()
440
SmartServerResponse(('TokenMismatch',)), response)
1216
442
def test_lock_write_on_locked_repo(self):
1217
443
backing = self.get_transport()
1218
request = smart_branch.SmartServerBranchRequestLockWrite(backing)
444
request = smart.branch.SmartServerBranchRequestLockWrite(backing)
1219
445
branch = self.make_branch('.', format='knit')
1220
repo = branch.repository
1221
repo_token = repo.lock_write().repository_token
1222
repo.leave_lock_in_place()
1224
response = request.execute('')
446
branch.repository.lock_write()
447
branch.repository.leave_lock_in_place()
448
branch.repository.unlock()
449
response = request.execute(backing.local_abspath(''))
1225
450
self.assertEqual(
1226
smart_req.SmartServerResponse(('LockContention',)), response)
1228
repo.lock_write(repo_token)
1229
repo.dont_leave_lock_in_place()
451
SmartServerResponse(('LockContention',)), response)
1232
453
def test_lock_write_on_readonly_transport(self):
1233
454
backing = self.get_readonly_transport()
1234
request = smart_branch.SmartServerBranchRequestLockWrite(backing)
455
request = smart.branch.SmartServerBranchRequestLockWrite(backing)
1235
456
branch = self.make_branch('.')
1236
root = self.get_transport().clone('/')
1237
path = urlutils.relative_url(root.base, self.get_transport().base)
1238
response = request.execute(path)
457
response = request.execute('')
1239
458
error_name, lock_str, why_str = response.args
1240
459
self.assertFalse(response.is_successful())
1241
460
self.assertEqual('LockFailed', error_name)
1244
class TestSmartServerBranchRequestUnlock(TestLockedBranch):
463
class TestSmartServerBranchRequestUnlock(tests.TestCaseWithTransport):
1246
465
def setUp(self):
1247
tests.TestCaseWithMemoryTransport.setUp(self)
466
tests.TestCaseWithTransport.setUp(self)
467
self.reduceLockdirTimeout()
1249
469
def test_unlock_on_locked_branch_and_repo(self):
1250
470
backing = self.get_transport()
1251
request = smart_branch.SmartServerBranchRequestUnlock(backing)
471
request = smart.branch.SmartServerBranchRequestUnlock(backing)
1252
472
branch = self.make_branch('.', format='knit')
1253
473
# Lock the branch
1254
branch_token, repo_token = self.get_lock_tokens(branch)
474
branch_token = branch.lock_write()
475
repo_token = branch.repository.lock_write()
476
branch.repository.unlock()
1255
477
# Unlock the branch (and repo) object, leaving the physical locks
1257
479
branch.leave_lock_in_place()
1258
480
branch.repository.leave_lock_in_place()
1260
response = request.execute('',
482
response = request.execute(backing.local_abspath(''),
1261
483
branch_token, repo_token)
1262
484
self.assertEqual(
1263
smart_req.SmartServerResponse(('ok',)), response)
485
SmartServerResponse(('ok',)), response)
1264
486
# The branch is now unlocked. Verify that with a new branch
1266
488
new_branch = branch.bzrdir.open_branch()
1387
593
# Note that it still returns body (of zero bytes).
1388
self.assertEqual(smart_req.SmartServerResponse(
1389
('nosuchrevision', 'missingrevision', ), ''),
1390
request.execute('', 'missingrevision'))
1393
class TestSmartServerRepositoryGetRevIdForRevno(
1394
tests.TestCaseWithMemoryTransport):
1396
def test_revno_found(self):
1397
backing = self.get_transport()
1398
request = smart_repo.SmartServerRepositoryGetRevIdForRevno(backing)
1399
tree = self.make_branch_and_memory_tree('.')
1402
rev1_id_utf8 = u'\xc8'.encode('utf-8')
1403
rev2_id_utf8 = u'\xc9'.encode('utf-8')
1404
tree.commit('1st commit', rev_id=rev1_id_utf8)
1405
tree.commit('2nd commit', rev_id=rev2_id_utf8)
1408
self.assertEqual(smart_req.SmartServerResponse(('ok', rev1_id_utf8)),
1409
request.execute('', 1, (2, rev2_id_utf8)))
1411
def test_known_revid_missing(self):
1412
backing = self.get_transport()
1413
request = smart_repo.SmartServerRepositoryGetRevIdForRevno(backing)
1414
repo = self.make_repository('.')
1416
smart_req.FailedSmartServerResponse(('nosuchrevision', 'ghost')),
1417
request.execute('', 1, (2, 'ghost')))
1419
def test_history_incomplete(self):
1420
backing = self.get_transport()
1421
request = smart_repo.SmartServerRepositoryGetRevIdForRevno(backing)
1422
parent = self.make_branch_and_memory_tree('parent', format='1.9')
1424
parent.add([''], ['TREE_ROOT'])
1425
r1 = parent.commit(message='first commit')
1426
r2 = parent.commit(message='second commit')
1428
local = self.make_branch_and_memory_tree('local', format='1.9')
1429
local.branch.pull(parent.branch)
1430
local.set_parent_ids([r2])
1431
r3 = local.commit(message='local commit')
1432
local.branch.create_clone_on_transport(
1433
self.get_transport('stacked'), stacked_on=self.get_url('parent'))
1435
smart_req.SmartServerResponse(('history-incomplete', 2, r2)),
1436
request.execute('stacked', 1, (3, r3)))
1439
class TestSmartServerRepositoryGetStream(tests.TestCaseWithMemoryTransport):
1441
def make_two_commit_repo(self):
1442
tree = self.make_branch_and_memory_tree('.')
1445
r1 = tree.commit('1st commit')
1446
r2 = tree.commit('2nd commit', rev_id=u'\xc8'.encode('utf-8'))
1448
repo = tree.branch.repository
1451
def test_ancestry_of(self):
1452
"""The search argument may be a 'ancestry-of' some heads'."""
1453
backing = self.get_transport()
1454
request = smart_repo.SmartServerRepositoryGetStream(backing)
1455
repo, r1, r2 = self.make_two_commit_repo()
1456
fetch_spec = ['ancestry-of', r2]
1457
lines = '\n'.join(fetch_spec)
1458
request.execute('', repo._format.network_name())
1459
response = request.do_body(lines)
1460
self.assertEqual(('ok',), response.args)
1461
stream_bytes = ''.join(response.body_stream)
1462
self.assertStartsWith(stream_bytes, 'Bazaar pack format 1')
1464
def test_search(self):
1465
"""The search argument may be a 'search' of some explicit keys."""
1466
backing = self.get_transport()
1467
request = smart_repo.SmartServerRepositoryGetStream(backing)
1468
repo, r1, r2 = self.make_two_commit_repo()
1469
fetch_spec = ['search', '%s %s' % (r1, r2), 'null:', '2']
1470
lines = '\n'.join(fetch_spec)
1471
request.execute('', repo._format.network_name())
1472
response = request.do_body(lines)
1473
self.assertEqual(('ok',), response.args)
1474
stream_bytes = ''.join(response.body_stream)
1475
self.assertStartsWith(stream_bytes, 'Bazaar pack format 1')
1478
class TestSmartServerRequestHasRevision(tests.TestCaseWithMemoryTransport):
595
SmartServerResponse(('nosuchrevision', 'missingrevision', ), ''),
596
request.execute(backing.local_abspath(''), 'missingrevision'))
599
class TestSmartServerRequestHasRevision(tests.TestCaseWithTransport):
1480
601
def test_missing_revision(self):
1481
602
"""For a missing revision, ('no', ) is returned."""
1482
603
backing = self.get_transport()
1483
request = smart_repo.SmartServerRequestHasRevision(backing)
604
request = smart.repository.SmartServerRequestHasRevision(backing)
1484
605
self.make_repository('.')
1485
self.assertEqual(smart_req.SmartServerResponse(('no', )),
1486
request.execute('', 'revid'))
606
self.assertEqual(SmartServerResponse(('no', )),
607
request.execute(backing.local_abspath(''), 'revid'))
1488
609
def test_present_revision(self):
1489
610
"""For a present revision, ('yes', ) is returned."""
1490
611
backing = self.get_transport()
1491
request = smart_repo.SmartServerRequestHasRevision(backing)
612
request = smart.repository.SmartServerRequestHasRevision(backing)
1492
613
tree = self.make_branch_and_memory_tree('.')
1493
614
tree.lock_write()
1551
675
stats = tree.branch.repository.gather_stats()
1553
678
expected_body = ('committers: 2\n'
1554
679
'firstrev: 123456.200 3600\n'
1555
680
'latestrev: 654321.400 0\n'
1557
self.assertEqual(smart_req.SmartServerResponse(('ok', ), expected_body),
683
self.assertEqual(SmartServerResponse(('ok', ), expected_body),
684
request.execute(backing.local_abspath(''),
1559
685
rev_id_utf8, 'yes'))
1562
class TestSmartServerRepositoryIsShared(tests.TestCaseWithMemoryTransport):
688
class TestSmartServerRepositoryIsShared(tests.TestCaseWithTransport):
1564
690
def test_is_shared(self):
1565
691
"""For a shared repository, ('yes', ) is returned."""
1566
692
backing = self.get_transport()
1567
request = smart_repo.SmartServerRepositoryIsShared(backing)
693
request = smart.repository.SmartServerRepositoryIsShared(backing)
1568
694
self.make_repository('.', shared=True)
1569
self.assertEqual(smart_req.SmartServerResponse(('yes', )),
1570
request.execute('', ))
695
self.assertEqual(SmartServerResponse(('yes', )),
696
request.execute(backing.local_abspath(''), ))
1572
698
def test_is_not_shared(self):
1573
699
"""For a shared repository, ('no', ) is returned."""
1574
700
backing = self.get_transport()
1575
request = smart_repo.SmartServerRepositoryIsShared(backing)
701
request = smart.repository.SmartServerRepositoryIsShared(backing)
1576
702
self.make_repository('.', shared=False)
1577
self.assertEqual(smart_req.SmartServerResponse(('no', )),
1578
request.execute('', ))
1581
class TestSmartServerRepositoryLockWrite(tests.TestCaseWithMemoryTransport):
703
self.assertEqual(SmartServerResponse(('no', )),
704
request.execute(backing.local_abspath(''), ))
707
class TestSmartServerRepositoryLockWrite(tests.TestCaseWithTransport):
710
tests.TestCaseWithTransport.setUp(self)
711
self.reduceLockdirTimeout()
1583
713
def test_lock_write_on_unlocked_repo(self):
1584
714
backing = self.get_transport()
1585
request = smart_repo.SmartServerRepositoryLockWrite(backing)
715
request = smart.repository.SmartServerRepositoryLockWrite(backing)
1586
716
repository = self.make_repository('.', format='knit')
1587
response = request.execute('')
717
response = request.execute(backing.local_abspath(''))
1588
718
nonce = repository.control_files._lock.peek().get('nonce')
1589
self.assertEqual(smart_req.SmartServerResponse(('ok', nonce)), response)
719
self.assertEqual(SmartServerResponse(('ok', nonce)), response)
1590
720
# The repository is now locked. Verify that with a new repository
1592
722
new_repo = repository.bzrdir.open_repository()
1593
723
self.assertRaises(errors.LockContention, new_repo.lock_write)
1595
request = smart_repo.SmartServerRepositoryUnlock(backing)
1596
response = request.execute('', nonce)
1598
725
def test_lock_write_on_locked_repo(self):
1599
726
backing = self.get_transport()
1600
request = smart_repo.SmartServerRepositoryLockWrite(backing)
727
request = smart.repository.SmartServerRepositoryLockWrite(backing)
1601
728
repository = self.make_repository('.', format='knit')
1602
repo_token = repository.lock_write().repository_token
729
repository.lock_write()
1603
730
repository.leave_lock_in_place()
1604
731
repository.unlock()
1605
response = request.execute('')
732
response = request.execute(backing.local_abspath(''))
1606
733
self.assertEqual(
1607
smart_req.SmartServerResponse(('LockContention',)), response)
1609
repository.lock_write(repo_token)
1610
repository.dont_leave_lock_in_place()
734
SmartServerResponse(('LockContention',)), response)
1613
736
def test_lock_write_on_readonly_transport(self):
1614
737
backing = self.get_readonly_transport()
1615
request = smart_repo.SmartServerRepositoryLockWrite(backing)
738
request = smart.repository.SmartServerRepositoryLockWrite(backing)
1616
739
repository = self.make_repository('.', format='knit')
1617
740
response = request.execute('')
1618
741
self.assertFalse(response.is_successful())
1619
742
self.assertEqual('LockFailed', response.args[0])
1622
class TestInsertStreamBase(tests.TestCaseWithMemoryTransport):
1624
def make_empty_byte_stream(self, repo):
1625
byte_stream = smart_repo._stream_to_byte_stream([], repo._format)
1626
return ''.join(byte_stream)
1629
class TestSmartServerRepositoryInsertStream(TestInsertStreamBase):
1631
def test_insert_stream_empty(self):
1632
backing = self.get_transport()
1633
request = smart_repo.SmartServerRepositoryInsertStream(backing)
1634
repository = self.make_repository('.')
1635
response = request.execute('', '')
1636
self.assertEqual(None, response)
1637
response = request.do_chunk(self.make_empty_byte_stream(repository))
1638
self.assertEqual(None, response)
1639
response = request.do_end()
1640
self.assertEqual(smart_req.SmartServerResponse(('ok', )), response)
1643
class TestSmartServerRepositoryInsertStreamLocked(TestInsertStreamBase):
1645
def test_insert_stream_empty(self):
1646
backing = self.get_transport()
1647
request = smart_repo.SmartServerRepositoryInsertStreamLocked(
1649
repository = self.make_repository('.', format='knit')
1650
lock_token = repository.lock_write().repository_token
1651
response = request.execute('', '', lock_token)
1652
self.assertEqual(None, response)
1653
response = request.do_chunk(self.make_empty_byte_stream(repository))
1654
self.assertEqual(None, response)
1655
response = request.do_end()
1656
self.assertEqual(smart_req.SmartServerResponse(('ok', )), response)
1659
def test_insert_stream_with_wrong_lock_token(self):
1660
backing = self.get_transport()
1661
request = smart_repo.SmartServerRepositoryInsertStreamLocked(
1663
repository = self.make_repository('.', format='knit')
1664
lock_token = repository.lock_write().repository_token
1666
errors.TokenMismatch, request.execute, '', '', 'wrong-token')
1670
class TestSmartServerRepositoryUnlock(tests.TestCaseWithMemoryTransport):
745
class TestSmartServerRepositoryUnlock(tests.TestCaseWithTransport):
1672
747
def setUp(self):
1673
tests.TestCaseWithMemoryTransport.setUp(self)
748
tests.TestCaseWithTransport.setUp(self)
749
self.reduceLockdirTimeout()
1675
751
def test_unlock_on_locked_repo(self):
1676
752
backing = self.get_transport()
1677
request = smart_repo.SmartServerRepositoryUnlock(backing)
753
request = smart.repository.SmartServerRepositoryUnlock(backing)
1678
754
repository = self.make_repository('.', format='knit')
1679
token = repository.lock_write().repository_token
755
token = repository.lock_write()
1680
756
repository.leave_lock_in_place()
1681
757
repository.unlock()
1682
response = request.execute('', token)
758
response = request.execute(backing.local_abspath(''), token)
1683
759
self.assertEqual(
1684
smart_req.SmartServerResponse(('ok',)), response)
760
SmartServerResponse(('ok',)), response)
1685
761
# The repository is now unlocked. Verify that with a new repository
1687
763
new_repo = repository.bzrdir.open_repository()
1691
767
def test_unlock_on_unlocked_repo(self):
1692
768
backing = self.get_transport()
1693
request = smart_repo.SmartServerRepositoryUnlock(backing)
769
request = smart.repository.SmartServerRepositoryUnlock(backing)
1694
770
repository = self.make_repository('.', format='knit')
1695
response = request.execute('', 'some token')
1697
smart_req.SmartServerResponse(('TokenMismatch',)), response)
1700
class TestSmartServerIsReadonly(tests.TestCaseWithMemoryTransport):
771
response = request.execute(backing.local_abspath(''), 'some token')
773
SmartServerResponse(('TokenMismatch',)), response)
776
class TestSmartServerRepositoryTarball(tests.TestCaseWithTransport):
778
def test_repository_tarball(self):
779
backing = self.get_transport()
780
request = smart.repository.SmartServerRepositoryTarball(backing)
781
repository = self.make_repository('.')
782
# make some extraneous junk in the repository directory which should
784
self.build_tree(['.bzr/repository/extra-junk'])
785
response = request.execute(backing.local_abspath(''), 'bz2')
786
self.assertEqual(('ok',), response.args)
787
# body should be a tbz2
788
body_file = StringIO(response.body)
789
body_tar = tarfile.open('body_tar.tbz2', fileobj=body_file,
791
# let's make sure there are some key repository components inside it.
792
# the tarfile returns directories with trailing slashes...
793
names = set([n.rstrip('/') for n in body_tar.getnames()])
794
self.assertTrue('.bzr/repository/lock' in names)
795
self.assertTrue('.bzr/repository/format' in names)
796
self.assertTrue('.bzr/repository/extra-junk' not in names,
797
"extraneous file present in tar file")
800
class TestSmartServerRepositoryStreamKnitData(tests.TestCaseWithTransport):
802
def test_fetch_revisions(self):
803
backing = self.get_transport()
804
request = smart.repository.SmartServerRepositoryStreamKnitDataForRevisions(backing)
805
tree = self.make_branch_and_memory_tree('.')
808
rev_id1_utf8 = u'\xc8'.encode('utf-8')
809
rev_id2_utf8 = u'\xc9'.encode('utf-8')
810
r1 = tree.commit('1st commit', rev_id=rev_id1_utf8)
811
r1 = tree.commit('2nd commit', rev_id=rev_id2_utf8)
814
response = request.execute(backing.local_abspath(''), rev_id2_utf8)
815
self.assertEqual(('ok',), response.args)
816
from cStringIO import StringIO
817
unpacker = pack.ContainerReader(StringIO(response.body))
819
for [name], read_bytes in unpacker.iter_records():
821
bytes = read_bytes(None)
822
# The bytes should be a valid bencoded string.
823
bencode.bdecode(bytes)
824
# XXX: assert that the bencoded knit records have the right
827
def test_no_such_revision_error(self):
828
backing = self.get_transport()
829
request = smart.repository.SmartServerRepositoryStreamKnitDataForRevisions(backing)
830
repo = self.make_repository('.')
831
rev_id1_utf8 = u'\xc8'.encode('utf-8')
832
response = request.execute(backing.local_abspath(''), rev_id1_utf8)
834
SmartServerResponse(('NoSuchRevision', rev_id1_utf8)),
838
class TestSmartServerRepositoryStreamRevisionsChunked(tests.TestCaseWithTransport):
840
def test_fetch_revisions(self):
841
backing = self.get_transport()
842
request = smart.repository.SmartServerRepositoryStreamRevisionsChunked(
844
tree = self.make_branch_and_memory_tree('.')
847
rev_id1_utf8 = u'\xc8'.encode('utf-8')
848
rev_id2_utf8 = u'\xc9'.encode('utf-8')
849
tree.commit('1st commit', rev_id=rev_id1_utf8)
850
tree.commit('2nd commit', rev_id=rev_id2_utf8)
853
response = request.execute(backing.local_abspath(''))
854
self.assertEqual(None, response)
855
response = request.do_body("%s\n%s\n1" % (rev_id2_utf8, rev_id1_utf8))
856
self.assertEqual(('ok',), response.args)
857
from cStringIO import StringIO
858
parser = pack.ContainerPushParser()
860
for stream_bytes in response.body_stream:
861
parser.accept_bytes(stream_bytes)
862
for [name], record_bytes in parser.read_pending_records():
864
# The bytes should be a valid bencoded string.
865
bencode.bdecode(record_bytes)
866
# XXX: assert that the bencoded knit records have the right
869
def test_no_such_revision_error(self):
870
backing = self.get_transport()
871
request = smart.repository.SmartServerRepositoryStreamRevisionsChunked(
873
repo = self.make_repository('.')
874
rev_id1_utf8 = u'\xc8'.encode('utf-8')
875
response = request.execute(backing.local_abspath(''))
876
self.assertEqual(None, response)
877
response = request.do_body("%s\n\n1" % (rev_id1_utf8,))
879
FailedSmartServerResponse(('NoSuchRevision', )),
883
class TestSmartServerIsReadonly(tests.TestCaseWithTransport):
1702
885
def test_is_readonly_no(self):
1703
886
backing = self.get_transport()
1704
request = smart_req.SmartServerIsReadonly(backing)
887
request = smart.request.SmartServerIsReadonly(backing)
1705
888
response = request.execute()
1706
889
self.assertEqual(
1707
smart_req.SmartServerResponse(('no',)), response)
890
SmartServerResponse(('no',)), response)
1709
892
def test_is_readonly_yes(self):
1710
893
backing = self.get_readonly_transport()
1711
request = smart_req.SmartServerIsReadonly(backing)
894
request = smart.request.SmartServerIsReadonly(backing)
1712
895
response = request.execute()
1713
896
self.assertEqual(
1714
smart_req.SmartServerResponse(('yes',)), response)
1717
class TestSmartServerRepositorySetMakeWorkingTrees(
1718
tests.TestCaseWithMemoryTransport):
1720
def test_set_false(self):
1721
backing = self.get_transport()
1722
repo = self.make_repository('.', shared=True)
1723
repo.set_make_working_trees(True)
1724
request_class = smart_repo.SmartServerRepositorySetMakeWorkingTrees
1725
request = request_class(backing)
1726
self.assertEqual(smart_req.SuccessfulSmartServerResponse(('ok',)),
1727
request.execute('', 'False'))
1728
repo = repo.bzrdir.open_repository()
1729
self.assertFalse(repo.make_working_trees())
1731
def test_set_true(self):
1732
backing = self.get_transport()
1733
repo = self.make_repository('.', shared=True)
1734
repo.set_make_working_trees(False)
1735
request_class = smart_repo.SmartServerRepositorySetMakeWorkingTrees
1736
request = request_class(backing)
1737
self.assertEqual(smart_req.SuccessfulSmartServerResponse(('ok',)),
1738
request.execute('', 'True'))
1739
repo = repo.bzrdir.open_repository()
1740
self.assertTrue(repo.make_working_trees())
1743
class TestSmartServerPackRepositoryAutopack(tests.TestCaseWithTransport):
1745
def make_repo_needing_autopacking(self, path='.'):
1746
# Make a repo in need of autopacking.
1747
tree = self.make_branch_and_tree('.', format='pack-0.92')
1748
repo = tree.branch.repository
1749
# monkey-patch the pack collection to disable autopacking
1750
repo._pack_collection._max_pack_count = lambda count: count
1752
tree.commit('commit %s' % x)
1753
self.assertEqual(10, len(repo._pack_collection.names()))
1754
del repo._pack_collection._max_pack_count
1757
def test_autopack_needed(self):
1758
repo = self.make_repo_needing_autopacking()
1760
self.addCleanup(repo.unlock)
1761
backing = self.get_transport()
1762
request = smart_packrepo.SmartServerPackRepositoryAutopack(
1764
response = request.execute('')
1765
self.assertEqual(smart_req.SmartServerResponse(('ok',)), response)
1766
repo._pack_collection.reload_pack_names()
1767
self.assertEqual(1, len(repo._pack_collection.names()))
1769
def test_autopack_not_needed(self):
1770
tree = self.make_branch_and_tree('.', format='pack-0.92')
1771
repo = tree.branch.repository
1773
self.addCleanup(repo.unlock)
1775
tree.commit('commit %s' % x)
1776
backing = self.get_transport()
1777
request = smart_packrepo.SmartServerPackRepositoryAutopack(
1779
response = request.execute('')
1780
self.assertEqual(smart_req.SmartServerResponse(('ok',)), response)
1781
repo._pack_collection.reload_pack_names()
1782
self.assertEqual(9, len(repo._pack_collection.names()))
1784
def test_autopack_on_nonpack_format(self):
1785
"""A request to autopack a non-pack repo is a no-op."""
1786
repo = self.make_repository('.', format='knit')
1787
backing = self.get_transport()
1788
request = smart_packrepo.SmartServerPackRepositoryAutopack(
1790
response = request.execute('')
1791
self.assertEqual(smart_req.SmartServerResponse(('ok',)), response)
1794
class TestSmartServerVfsGet(tests.TestCaseWithMemoryTransport):
1796
def test_unicode_path(self):
1797
"""VFS requests expect unicode paths to be escaped."""
1798
filename = u'foo\N{INTERROBANG}'
1799
filename_escaped = urlutils.escape(filename)
1800
backing = self.get_transport()
1801
request = vfs.GetRequest(backing)
1802
backing.put_bytes_non_atomic(filename_escaped, 'contents')
1803
self.assertEqual(smart_req.SmartServerResponse(('ok', ), 'contents'),
1804
request.execute(filename_escaped))
897
SmartServerResponse(('yes',)), response)
1807
900
class TestHandlers(tests.TestCase):
1808
901
"""Tests for the request.request_handlers object."""
1810
def test_all_registrations_exist(self):
1811
"""All registered request_handlers can be found."""
1812
# If there's a typo in a register_lazy call, this loop will fail with
1813
# an AttributeError.
1814
for key, item in smart_req.request_handlers.iteritems():
1817
def assertHandlerEqual(self, verb, handler):
1818
self.assertEqual(smart_req.request_handlers.get(verb), handler)
1820
903
def test_registered_methods(self):
1821
904
"""Test that known methods are registered to the correct object."""
1822
self.assertHandlerEqual('Branch.get_config_file',
1823
smart_branch.SmartServerBranchGetConfigFile)
1824
self.assertHandlerEqual('Branch.get_parent',
1825
smart_branch.SmartServerBranchGetParent)
1826
self.assertHandlerEqual('Branch.get_tags_bytes',
1827
smart_branch.SmartServerBranchGetTagsBytes)
1828
self.assertHandlerEqual('Branch.lock_write',
1829
smart_branch.SmartServerBranchRequestLockWrite)
1830
self.assertHandlerEqual('Branch.last_revision_info',
1831
smart_branch.SmartServerBranchRequestLastRevisionInfo)
1832
self.assertHandlerEqual('Branch.revision_history',
1833
smart_branch.SmartServerRequestRevisionHistory)
1834
self.assertHandlerEqual('Branch.set_config_option',
1835
smart_branch.SmartServerBranchRequestSetConfigOption)
1836
self.assertHandlerEqual('Branch.set_last_revision',
1837
smart_branch.SmartServerBranchRequestSetLastRevision)
1838
self.assertHandlerEqual('Branch.set_last_revision_info',
1839
smart_branch.SmartServerBranchRequestSetLastRevisionInfo)
1840
self.assertHandlerEqual('Branch.set_last_revision_ex',
1841
smart_branch.SmartServerBranchRequestSetLastRevisionEx)
1842
self.assertHandlerEqual('Branch.set_parent_location',
1843
smart_branch.SmartServerBranchRequestSetParentLocation)
1844
self.assertHandlerEqual('Branch.unlock',
1845
smart_branch.SmartServerBranchRequestUnlock)
1846
self.assertHandlerEqual('BzrDir.find_repository',
1847
smart_dir.SmartServerRequestFindRepositoryV1)
1848
self.assertHandlerEqual('BzrDir.find_repositoryV2',
1849
smart_dir.SmartServerRequestFindRepositoryV2)
1850
self.assertHandlerEqual('BzrDirFormat.initialize',
1851
smart_dir.SmartServerRequestInitializeBzrDir)
1852
self.assertHandlerEqual('BzrDirFormat.initialize_ex_1.16',
1853
smart_dir.SmartServerRequestBzrDirInitializeEx)
1854
self.assertHandlerEqual('BzrDir.cloning_metadir',
1855
smart_dir.SmartServerBzrDirRequestCloningMetaDir)
1856
self.assertHandlerEqual('BzrDir.get_config_file',
1857
smart_dir.SmartServerBzrDirRequestConfigFile)
1858
self.assertHandlerEqual('BzrDir.open_branch',
1859
smart_dir.SmartServerRequestOpenBranch)
1860
self.assertHandlerEqual('BzrDir.open_branchV2',
1861
smart_dir.SmartServerRequestOpenBranchV2)
1862
self.assertHandlerEqual('BzrDir.open_branchV3',
1863
smart_dir.SmartServerRequestOpenBranchV3)
1864
self.assertHandlerEqual('PackRepository.autopack',
1865
smart_packrepo.SmartServerPackRepositoryAutopack)
1866
self.assertHandlerEqual('Repository.gather_stats',
1867
smart_repo.SmartServerRepositoryGatherStats)
1868
self.assertHandlerEqual('Repository.get_parent_map',
1869
smart_repo.SmartServerRepositoryGetParentMap)
1870
self.assertHandlerEqual('Repository.get_rev_id_for_revno',
1871
smart_repo.SmartServerRepositoryGetRevIdForRevno)
1872
self.assertHandlerEqual('Repository.get_revision_graph',
1873
smart_repo.SmartServerRepositoryGetRevisionGraph)
1874
self.assertHandlerEqual('Repository.get_stream',
1875
smart_repo.SmartServerRepositoryGetStream)
1876
self.assertHandlerEqual('Repository.has_revision',
1877
smart_repo.SmartServerRequestHasRevision)
1878
self.assertHandlerEqual('Repository.insert_stream',
1879
smart_repo.SmartServerRepositoryInsertStream)
1880
self.assertHandlerEqual('Repository.insert_stream_locked',
1881
smart_repo.SmartServerRepositoryInsertStreamLocked)
1882
self.assertHandlerEqual('Repository.is_shared',
1883
smart_repo.SmartServerRepositoryIsShared)
1884
self.assertHandlerEqual('Repository.lock_write',
1885
smart_repo.SmartServerRepositoryLockWrite)
1886
self.assertHandlerEqual('Repository.tarball',
1887
smart_repo.SmartServerRepositoryTarball)
1888
self.assertHandlerEqual('Repository.unlock',
1889
smart_repo.SmartServerRepositoryUnlock)
1890
self.assertHandlerEqual('Transport.is_readonly',
1891
smart_req.SmartServerIsReadonly)
906
smart.request.request_handlers.get('Branch.get_config_file'),
907
smart.branch.SmartServerBranchGetConfigFile)
909
smart.request.request_handlers.get('Branch.lock_write'),
910
smart.branch.SmartServerBranchRequestLockWrite)
912
smart.request.request_handlers.get('Branch.last_revision_info'),
913
smart.branch.SmartServerBranchRequestLastRevisionInfo)
915
smart.request.request_handlers.get('Branch.revision_history'),
916
smart.branch.SmartServerRequestRevisionHistory)
918
smart.request.request_handlers.get('Branch.set_last_revision'),
919
smart.branch.SmartServerBranchRequestSetLastRevision)
921
smart.request.request_handlers.get('Branch.unlock'),
922
smart.branch.SmartServerBranchRequestUnlock)
924
smart.request.request_handlers.get('BzrDir.find_repository'),
925
smart.bzrdir.SmartServerRequestFindRepository)
927
smart.request.request_handlers.get('BzrDirFormat.initialize'),
928
smart.bzrdir.SmartServerRequestInitializeBzrDir)
930
smart.request.request_handlers.get('BzrDir.open_branch'),
931
smart.bzrdir.SmartServerRequestOpenBranch)
933
smart.request.request_handlers.get('Repository.gather_stats'),
934
smart.repository.SmartServerRepositoryGatherStats)
936
smart.request.request_handlers.get('Repository.get_parent_map'),
937
smart.repository.SmartServerRepositoryGetParentMap)
939
smart.request.request_handlers.get(
940
'Repository.get_revision_graph'),
941
smart.repository.SmartServerRepositoryGetRevisionGraph)
943
smart.request.request_handlers.get('Repository.has_revision'),
944
smart.repository.SmartServerRequestHasRevision)
946
smart.request.request_handlers.get('Repository.is_shared'),
947
smart.repository.SmartServerRepositoryIsShared)
949
smart.request.request_handlers.get('Repository.lock_write'),
950
smart.repository.SmartServerRepositoryLockWrite)
952
smart.request.request_handlers.get('Repository.tarball'),
953
smart.repository.SmartServerRepositoryTarball)
955
smart.request.request_handlers.get('Repository.unlock'),
956
smart.repository.SmartServerRepositoryUnlock)
958
smart.request.request_handlers.get('Transport.is_readonly'),
959
smart.request.SmartServerIsReadonly)
added
removed
bzrlib/tests/test_smart.py
