1
# Copyright (C) 2006, 2007 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., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
17
"""Tests for the smart wire/domain protocol.
19
This module contains tests for the domain-level smart requests and responses,
20
such as the 'Branch.lock_write' request. Many of these use specific disk
21
formats to exercise calls that only make sense for formats with specific
24
Tests for low-level protocol encoding are found in test_smart_transport.
28
from StringIO import StringIO
32
from bzrlib import bzrdir, errors, pack, smart, tests
33
from bzrlib.smart.request import (
34
FailedSmartServerResponse,
36
SuccessfulSmartServerResponse,
38
import bzrlib.smart.bzrdir
39
import bzrlib.smart.branch
40
import bzrlib.smart.repository
41
from bzrlib.util import bencode
44
class TestCaseWithSmartMedium(tests.TestCaseWithTransport):
47
super(TestCaseWithSmartMedium, self).setUp()
48
# We're allowed to set the transport class here, so that we don't use
49
# the default or a parameterized class, but rather use the
50
# TestCaseWithTransport infrastructure to set up a smart server and
52
self.transport_server = smart.server.SmartTCPServer_for_testing
54
def get_smart_medium(self):
55
"""Get a smart medium to use in tests."""
56
return self.get_transport().get_smart_medium()
59
class TestSmartServerResponse(tests.TestCase):
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', )))
70
self.assertNotEqual(None,
71
SmartServerResponse(('ok', )))
74
class TestSmartServerRequestFindRepository(tests.TestCaseWithTransport):
75
"""Tests for BzrDir.find_repository."""
77
def test_no_repository(self):
78
"""When there is no repository to be found, ('norepository', ) is returned."""
79
backing = self.get_transport()
80
request = smart.bzrdir.SmartServerRequestFindRepository(backing)
82
self.assertEqual(SmartServerResponse(('norepository', )),
83
request.execute(backing.local_abspath('')))
85
def test_nonshared_repository(self):
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
88
# the repository is at.
89
backing = self.get_transport()
90
request = smart.bzrdir.SmartServerRequestFindRepository(backing)
91
result = self._make_repository_and_result()
92
self.assertEqual(result, request.execute(backing.local_abspath('')))
93
self.make_bzrdir('subdir')
94
self.assertEqual(SmartServerResponse(('norepository', )),
95
request.execute(backing.local_abspath('subdir')))
97
def _make_repository_and_result(self, shared=False, format=None):
98
"""Convenience function to setup a repository.
100
:result: The SmartServerResponse to expect when opening it.
102
repo = self.make_repository('.', shared=shared, format=format)
103
if repo.supports_rich_root():
107
if repo._format.supports_tree_reference:
111
return SmartServerResponse(('ok', '', rich_root, subtrees))
113
def test_shared_repository(self):
114
"""When there is a shared repository, we get 'ok', 'relpath-to-repo'."""
115
backing = self.get_transport()
116
request = smart.bzrdir.SmartServerRequestFindRepository(backing)
117
result = self._make_repository_and_result(shared=True)
118
self.assertEqual(result, request.execute(backing.local_abspath('')))
119
self.make_bzrdir('subdir')
120
result2 = SmartServerResponse(result.args[0:1] + ('..', ) + result.args[2:])
121
self.assertEqual(result2,
122
request.execute(backing.local_abspath('subdir')))
123
self.make_bzrdir('subdir/deeper')
124
result3 = SmartServerResponse(result.args[0:1] + ('../..', ) + result.args[2:])
125
self.assertEqual(result3,
126
request.execute(backing.local_abspath('subdir/deeper')))
128
def test_rich_root_and_subtree_encoding(self):
129
"""Test for the format attributes for rich root and subtree support."""
130
backing = self.get_transport()
131
request = smart.bzrdir.SmartServerRequestFindRepository(backing)
132
result = self._make_repository_and_result(format='dirstate-with-subtree')
133
# check the test will be valid
134
self.assertEqual('yes', result.args[2])
135
self.assertEqual('yes', result.args[3])
136
self.assertEqual(result, request.execute(backing.local_abspath('')))
139
class TestSmartServerRequestInitializeBzrDir(tests.TestCaseWithTransport):
141
def test_empty_dir(self):
142
"""Initializing an empty dir should succeed and do it."""
143
backing = self.get_transport()
144
request = smart.bzrdir.SmartServerRequestInitializeBzrDir(backing)
145
self.assertEqual(SmartServerResponse(('ok', )),
146
request.execute(backing.local_abspath('.')))
147
made_dir = bzrdir.BzrDir.open_from_transport(backing)
148
# no branch, tree or repository is expected with the current
150
self.assertRaises(errors.NoWorkingTree, made_dir.open_workingtree)
151
self.assertRaises(errors.NotBranchError, made_dir.open_branch)
152
self.assertRaises(errors.NoRepositoryPresent, made_dir.open_repository)
154
def test_missing_dir(self):
155
"""Initializing a missing directory should fail like the bzrdir api."""
156
backing = self.get_transport()
157
request = smart.bzrdir.SmartServerRequestInitializeBzrDir(backing)
158
self.assertRaises(errors.NoSuchFile,
159
request.execute, backing.local_abspath('subdir'))
161
def test_initialized_dir(self):
162
"""Initializing an extant bzrdir should fail like the bzrdir api."""
163
backing = self.get_transport()
164
request = smart.bzrdir.SmartServerRequestInitializeBzrDir(backing)
165
self.make_bzrdir('subdir')
166
self.assertRaises(errors.FileExists,
167
request.execute, backing.local_abspath('subdir'))
170
class TestSmartServerRequestOpenBranch(tests.TestCaseWithTransport):
172
def test_no_branch(self):
173
"""When there is no branch, ('nobranch', ) is returned."""
174
backing = self.get_transport()
175
request = smart.bzrdir.SmartServerRequestOpenBranch(backing)
176
self.make_bzrdir('.')
177
self.assertEqual(SmartServerResponse(('nobranch', )),
178
request.execute(backing.local_abspath('')))
180
def test_branch(self):
181
"""When there is a branch, 'ok' is returned."""
182
backing = self.get_transport()
183
request = smart.bzrdir.SmartServerRequestOpenBranch(backing)
184
self.make_branch('.')
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):
204
def test_empty(self):
205
"""For an empty branch, the body is empty."""
206
backing = self.get_transport()
207
request = smart.branch.SmartServerRequestRevisionHistory(backing)
208
self.make_branch('.')
209
self.assertEqual(SmartServerResponse(('ok', ), ''),
210
request.execute(backing.local_abspath('')))
212
def test_not_empty(self):
213
"""For a non-empty branch, the body is empty."""
214
backing = self.get_transport()
215
request = smart.branch.SmartServerRequestRevisionHistory(backing)
216
tree = self.make_branch_and_memory_tree('.')
219
r1 = tree.commit('1st commit')
220
r2 = tree.commit('2nd commit', rev_id=u'\xc8'.encode('utf-8'))
223
SmartServerResponse(('ok', ), ('\x00'.join([r1, r2]))),
224
request.execute(backing.local_abspath('')))
227
class TestSmartServerBranchRequest(tests.TestCaseWithTransport):
229
def test_no_branch(self):
230
"""When there is a bzrdir and no branch, NotBranchError is raised."""
231
backing = self.get_transport()
232
request = smart.branch.SmartServerBranchRequest(backing)
233
self.make_bzrdir('.')
234
self.assertRaises(errors.NotBranchError,
235
request.execute, backing.local_abspath(''))
237
def test_branch_reference(self):
238
"""When there is a branch reference, NotBranchError is raised."""
239
backing = self.get_transport()
240
request = smart.branch.SmartServerBranchRequest(backing)
241
branch = self.make_branch('branch')
242
checkout = branch.create_checkout('reference',lightweight=True)
243
self.assertRaises(errors.NotBranchError,
244
request.execute, backing.local_abspath('checkout'))
247
class TestSmartServerBranchRequestLastRevisionInfo(tests.TestCaseWithTransport):
249
def test_empty(self):
250
"""For an empty branch, the result is ('ok', '0', 'null:')."""
251
backing = self.get_transport()
252
request = smart.branch.SmartServerBranchRequestLastRevisionInfo(backing)
253
self.make_branch('.')
254
self.assertEqual(SmartServerResponse(('ok', '0', 'null:')),
255
request.execute(backing.local_abspath('')))
257
def test_not_empty(self):
258
"""For a non-empty branch, the result is ('ok', 'revno', 'revid')."""
259
backing = self.get_transport()
260
request = smart.branch.SmartServerBranchRequestLastRevisionInfo(backing)
261
tree = self.make_branch_and_memory_tree('.')
264
rev_id_utf8 = u'\xc8'.encode('utf-8')
265
r1 = tree.commit('1st commit')
266
r2 = tree.commit('2nd commit', rev_id=rev_id_utf8)
269
SmartServerResponse(('ok', '2', rev_id_utf8)),
270
request.execute(backing.local_abspath('')))
273
class TestSmartServerBranchRequestGetConfigFile(tests.TestCaseWithTransport):
275
def test_default(self):
276
"""With no file, we get empty content."""
277
backing = self.get_transport()
278
request = smart.branch.SmartServerBranchGetConfigFile(backing)
279
branch = self.make_branch('.')
280
# there should be no file by default
282
self.assertEqual(SmartServerResponse(('ok', ), content),
283
request.execute(backing.local_abspath('')))
285
def test_with_content(self):
286
# SmartServerBranchGetConfigFile should return the content from
287
# branch.control_files.get('branch.conf') for now - in the future it may
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()
385
def test_lock_write_on_unlocked_branch(self):
386
backing = self.get_transport()
387
request = smart.branch.SmartServerBranchRequestLockWrite(backing)
388
branch = self.make_branch('.', format='knit')
389
repository = branch.repository
390
response = request.execute(backing.local_abspath(''))
391
branch_nonce = branch.control_files._lock.peek().get('nonce')
392
repository_nonce = repository.control_files._lock.peek().get('nonce')
394
SmartServerResponse(('ok', branch_nonce, repository_nonce)),
396
# The branch (and associated repository) is now locked. Verify that
397
# with a new branch object.
398
new_branch = repository.bzrdir.open_branch()
399
self.assertRaises(errors.LockContention, new_branch.lock_write)
401
def test_lock_write_on_locked_branch(self):
402
backing = self.get_transport()
403
request = smart.branch.SmartServerBranchRequestLockWrite(backing)
404
branch = self.make_branch('.')
406
branch.leave_lock_in_place()
408
response = request.execute(backing.local_abspath(''))
410
SmartServerResponse(('LockContention',)), response)
412
def test_lock_write_with_tokens_on_locked_branch(self):
413
backing = self.get_transport()
414
request = smart.branch.SmartServerBranchRequestLockWrite(backing)
415
branch = self.make_branch('.', format='knit')
416
branch_token = branch.lock_write()
417
repo_token = branch.repository.lock_write()
418
branch.repository.unlock()
419
branch.leave_lock_in_place()
420
branch.repository.leave_lock_in_place()
422
response = request.execute(backing.local_abspath(''),
423
branch_token, repo_token)
425
SmartServerResponse(('ok', branch_token, repo_token)), response)
427
def test_lock_write_with_mismatched_tokens_on_locked_branch(self):
428
backing = self.get_transport()
429
request = smart.branch.SmartServerBranchRequestLockWrite(backing)
430
branch = self.make_branch('.', format='knit')
431
branch_token = branch.lock_write()
432
repo_token = branch.repository.lock_write()
433
branch.repository.unlock()
434
branch.leave_lock_in_place()
435
branch.repository.leave_lock_in_place()
437
response = request.execute(backing.local_abspath(''),
438
branch_token+'xxx', repo_token)
440
SmartServerResponse(('TokenMismatch',)), response)
442
def test_lock_write_on_locked_repo(self):
443
backing = self.get_transport()
444
request = smart.branch.SmartServerBranchRequestLockWrite(backing)
445
branch = self.make_branch('.', format='knit')
446
branch.repository.lock_write()
447
branch.repository.leave_lock_in_place()
448
branch.repository.unlock()
449
response = request.execute(backing.local_abspath(''))
451
SmartServerResponse(('LockContention',)), response)
453
def test_lock_write_on_readonly_transport(self):
454
backing = self.get_readonly_transport()
455
request = smart.branch.SmartServerBranchRequestLockWrite(backing)
456
branch = self.make_branch('.')
457
response = request.execute('')
458
error_name, lock_str, why_str = response.args
459
self.assertFalse(response.is_successful())
460
self.assertEqual('LockFailed', error_name)
463
class TestSmartServerBranchRequestUnlock(tests.TestCaseWithTransport):
466
tests.TestCaseWithTransport.setUp(self)
467
self.reduceLockdirTimeout()
469
def test_unlock_on_locked_branch_and_repo(self):
470
backing = self.get_transport()
471
request = smart.branch.SmartServerBranchRequestUnlock(backing)
472
branch = self.make_branch('.', format='knit')
474
branch_token = branch.lock_write()
475
repo_token = branch.repository.lock_write()
476
branch.repository.unlock()
477
# Unlock the branch (and repo) object, leaving the physical locks
479
branch.leave_lock_in_place()
480
branch.repository.leave_lock_in_place()
482
response = request.execute(backing.local_abspath(''),
483
branch_token, repo_token)
485
SmartServerResponse(('ok',)), response)
486
# The branch is now unlocked. Verify that with a new branch
488
new_branch = branch.bzrdir.open_branch()
489
new_branch.lock_write()
492
def test_unlock_on_unlocked_branch_unlocked_repo(self):
493
backing = self.get_transport()
494
request = smart.branch.SmartServerBranchRequestUnlock(backing)
495
branch = self.make_branch('.', format='knit')
496
response = request.execute(
497
backing.local_abspath(''), 'branch token', 'repo token')
499
SmartServerResponse(('TokenMismatch',)), response)
501
def test_unlock_on_unlocked_branch_locked_repo(self):
502
backing = self.get_transport()
503
request = smart.branch.SmartServerBranchRequestUnlock(backing)
504
branch = self.make_branch('.', format='knit')
505
# Lock the repository.
506
repo_token = branch.repository.lock_write()
507
branch.repository.leave_lock_in_place()
508
branch.repository.unlock()
509
# Issue branch lock_write request on the unlocked branch (with locked
511
response = request.execute(
512
backing.local_abspath(''), 'branch token', repo_token)
514
SmartServerResponse(('TokenMismatch',)), response)
517
class TestSmartServerRepositoryRequest(tests.TestCaseWithTransport):
519
def test_no_repository(self):
520
"""Raise NoRepositoryPresent when there is a bzrdir and no repo."""
521
# we test this using a shared repository above the named path,
522
# thus checking the right search logic is used - that is, that
523
# its the exact path being looked at and the server is not
525
backing = self.get_transport()
526
request = smart.repository.SmartServerRepositoryRequest(backing)
527
self.make_repository('.', shared=True)
528
self.make_bzrdir('subdir')
529
self.assertRaises(errors.NoRepositoryPresent,
530
request.execute, backing.local_abspath('subdir'))
533
class TestSmartServerRepositoryGetParentMap(tests.TestCaseWithTransport):
535
def test_trivial_bzipped(self):
536
# This tests that the wire encoding is actually bzipped
537
backing = self.get_transport()
538
request = smart.repository.SmartServerRepositoryGetParentMap(backing)
539
tree = self.make_branch_and_memory_tree('.')
541
self.assertEqual(None,
542
request.execute(backing.local_abspath(''), 'missing-id'))
543
# Note that it returns a body (of '' bzipped).
545
SuccessfulSmartServerResponse(('ok', ), bz2.compress('')),
546
request.do_body('\n\n0\n'))
549
class TestSmartServerRepositoryGetRevisionGraph(tests.TestCaseWithTransport):
551
def test_none_argument(self):
552
backing = self.get_transport()
553
request = smart.repository.SmartServerRepositoryGetRevisionGraph(backing)
554
tree = self.make_branch_and_memory_tree('.')
557
r1 = tree.commit('1st commit')
558
r2 = tree.commit('2nd commit', rev_id=u'\xc8'.encode('utf-8'))
561
# the lines of revision_id->revision_parent_list has no guaranteed
562
# order coming out of a dict, so sort both our test and response
563
lines = sorted([' '.join([r2, r1]), r1])
564
response = request.execute(backing.local_abspath(''), '')
565
response.body = '\n'.join(sorted(response.body.split('\n')))
568
SmartServerResponse(('ok', ), '\n'.join(lines)), response)
570
def test_specific_revision_argument(self):
571
backing = self.get_transport()
572
request = smart.repository.SmartServerRepositoryGetRevisionGraph(backing)
573
tree = self.make_branch_and_memory_tree('.')
576
rev_id_utf8 = u'\xc9'.encode('utf-8')
577
r1 = tree.commit('1st commit', rev_id=rev_id_utf8)
578
r2 = tree.commit('2nd commit', rev_id=u'\xc8'.encode('utf-8'))
581
self.assertEqual(SmartServerResponse(('ok', ), rev_id_utf8),
582
request.execute(backing.local_abspath(''), rev_id_utf8))
584
def test_no_such_revision(self):
585
backing = self.get_transport()
586
request = smart.repository.SmartServerRepositoryGetRevisionGraph(backing)
587
tree = self.make_branch_and_memory_tree('.')
590
r1 = tree.commit('1st commit')
593
# Note that it still returns body (of zero bytes).
595
SmartServerResponse(('nosuchrevision', 'missingrevision', ), ''),
596
request.execute(backing.local_abspath(''), 'missingrevision'))
599
class TestSmartServerRequestHasRevision(tests.TestCaseWithTransport):
601
def test_missing_revision(self):
602
"""For a missing revision, ('no', ) is returned."""
603
backing = self.get_transport()
604
request = smart.repository.SmartServerRequestHasRevision(backing)
605
self.make_repository('.')
606
self.assertEqual(SmartServerResponse(('no', )),
607
request.execute(backing.local_abspath(''), 'revid'))
609
def test_present_revision(self):
610
"""For a present revision, ('yes', ) is returned."""
611
backing = self.get_transport()
612
request = smart.repository.SmartServerRequestHasRevision(backing)
613
tree = self.make_branch_and_memory_tree('.')
616
rev_id_utf8 = u'\xc8abc'.encode('utf-8')
617
r1 = tree.commit('a commit', rev_id=rev_id_utf8)
619
self.assertTrue(tree.branch.repository.has_revision(rev_id_utf8))
620
self.assertEqual(SmartServerResponse(('yes', )),
621
request.execute(backing.local_abspath(''), rev_id_utf8))
624
class TestSmartServerRepositoryGatherStats(tests.TestCaseWithTransport):
626
def test_empty_revid(self):
627
"""With an empty revid, we get only size an number and revisions"""
628
backing = self.get_transport()
629
request = smart.repository.SmartServerRepositoryGatherStats(backing)
630
repository = self.make_repository('.')
631
stats = repository.gather_stats()
633
expected_body = 'revisions: 0\nsize: %d\n' % size
634
self.assertEqual(SmartServerResponse(('ok', ), expected_body),
635
request.execute(backing.local_abspath(''), '', 'no'))
637
def test_revid_with_committers(self):
638
"""For a revid we get more infos."""
639
backing = self.get_transport()
640
rev_id_utf8 = u'\xc8abc'.encode('utf-8')
641
request = smart.repository.SmartServerRepositoryGatherStats(backing)
642
tree = self.make_branch_and_memory_tree('.')
645
# Let's build a predictable result
646
tree.commit('a commit', timestamp=123456.2, timezone=3600)
647
tree.commit('a commit', timestamp=654321.4, timezone=0,
651
stats = tree.branch.repository.gather_stats()
653
expected_body = ('firstrev: 123456.200 3600\n'
654
'latestrev: 654321.400 0\n'
657
self.assertEqual(SmartServerResponse(('ok', ), expected_body),
658
request.execute(backing.local_abspath(''),
661
def test_not_empty_repository_with_committers(self):
662
"""For a revid and requesting committers we get the whole thing."""
663
backing = self.get_transport()
664
rev_id_utf8 = u'\xc8abc'.encode('utf-8')
665
request = smart.repository.SmartServerRepositoryGatherStats(backing)
666
tree = self.make_branch_and_memory_tree('.')
669
# Let's build a predictable result
670
tree.commit('a commit', timestamp=123456.2, timezone=3600,
672
tree.commit('a commit', timestamp=654321.4, timezone=0,
673
committer='bar', rev_id=rev_id_utf8)
675
stats = tree.branch.repository.gather_stats()
678
expected_body = ('committers: 2\n'
679
'firstrev: 123456.200 3600\n'
680
'latestrev: 654321.400 0\n'
683
self.assertEqual(SmartServerResponse(('ok', ), expected_body),
684
request.execute(backing.local_abspath(''),
688
class TestSmartServerRepositoryIsShared(tests.TestCaseWithTransport):
690
def test_is_shared(self):
691
"""For a shared repository, ('yes', ) is returned."""
692
backing = self.get_transport()
693
request = smart.repository.SmartServerRepositoryIsShared(backing)
694
self.make_repository('.', shared=True)
695
self.assertEqual(SmartServerResponse(('yes', )),
696
request.execute(backing.local_abspath(''), ))
698
def test_is_not_shared(self):
699
"""For a shared repository, ('no', ) is returned."""
700
backing = self.get_transport()
701
request = smart.repository.SmartServerRepositoryIsShared(backing)
702
self.make_repository('.', shared=False)
703
self.assertEqual(SmartServerResponse(('no', )),
704
request.execute(backing.local_abspath(''), ))
707
class TestSmartServerRepositoryLockWrite(tests.TestCaseWithTransport):
710
tests.TestCaseWithTransport.setUp(self)
711
self.reduceLockdirTimeout()
713
def test_lock_write_on_unlocked_repo(self):
714
backing = self.get_transport()
715
request = smart.repository.SmartServerRepositoryLockWrite(backing)
716
repository = self.make_repository('.', format='knit')
717
response = request.execute(backing.local_abspath(''))
718
nonce = repository.control_files._lock.peek().get('nonce')
719
self.assertEqual(SmartServerResponse(('ok', nonce)), response)
720
# The repository is now locked. Verify that with a new repository
722
new_repo = repository.bzrdir.open_repository()
723
self.assertRaises(errors.LockContention, new_repo.lock_write)
725
def test_lock_write_on_locked_repo(self):
726
backing = self.get_transport()
727
request = smart.repository.SmartServerRepositoryLockWrite(backing)
728
repository = self.make_repository('.', format='knit')
729
repository.lock_write()
730
repository.leave_lock_in_place()
732
response = request.execute(backing.local_abspath(''))
734
SmartServerResponse(('LockContention',)), response)
736
def test_lock_write_on_readonly_transport(self):
737
backing = self.get_readonly_transport()
738
request = smart.repository.SmartServerRepositoryLockWrite(backing)
739
repository = self.make_repository('.', format='knit')
740
response = request.execute('')
741
self.assertFalse(response.is_successful())
742
self.assertEqual('LockFailed', response.args[0])
745
class TestSmartServerRepositoryUnlock(tests.TestCaseWithTransport):
748
tests.TestCaseWithTransport.setUp(self)
749
self.reduceLockdirTimeout()
751
def test_unlock_on_locked_repo(self):
752
backing = self.get_transport()
753
request = smart.repository.SmartServerRepositoryUnlock(backing)
754
repository = self.make_repository('.', format='knit')
755
token = repository.lock_write()
756
repository.leave_lock_in_place()
758
response = request.execute(backing.local_abspath(''), token)
760
SmartServerResponse(('ok',)), response)
761
# The repository is now unlocked. Verify that with a new repository
763
new_repo = repository.bzrdir.open_repository()
764
new_repo.lock_write()
767
def test_unlock_on_unlocked_repo(self):
768
backing = self.get_transport()
769
request = smart.repository.SmartServerRepositoryUnlock(backing)
770
repository = self.make_repository('.', format='knit')
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):
885
def test_is_readonly_no(self):
886
backing = self.get_transport()
887
request = smart.request.SmartServerIsReadonly(backing)
888
response = request.execute()
890
SmartServerResponse(('no',)), response)
892
def test_is_readonly_yes(self):
893
backing = self.get_readonly_transport()
894
request = smart.request.SmartServerIsReadonly(backing)
895
response = request.execute()
897
SmartServerResponse(('yes',)), response)
900
class TestHandlers(tests.TestCase):
901
"""Tests for the request.request_handlers object."""
903
def test_registered_methods(self):
904
"""Test that known methods are registered to the correct object."""
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
