← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_smart.py

  • Committer: Vincent Ladeuil
  • Date: 2009-06-22 12:52:39 UTC
  • mto: (4471.1.1 integration)
  • mto: This revision was merged to the branch mainline in revision 4472.
  • Revision ID: v.ladeuil+lp@free.fr-20090622125239-kabo9smxt9c3vnir
Use a consistent scheme for naming pyrex source files.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test_smart.py
12
12
#
13
13
# You should have received a copy of the GNU General Public License
14
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
 
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
16
 
17
17
"""Tests for the smart wire/domain protocol.
18
18
 
29
29
import tarfile
30
30
 
31
31
from bzrlib import (
 
32
    bencode,
32
33
    bzrdir,
33
34
    errors,
34
35
    pack,
36
37
    tests,
37
38
    urlutils,
38
39
    )
39
 
from bzrlib.branch import BranchReferenceFormat
 
40
from bzrlib.branch import Branch, BranchReferenceFormat
40
41
import bzrlib.smart.branch
41
 
import bzrlib.smart.bzrdir
 
42
import bzrlib.smart.bzrdir, bzrlib.smart.bzrdir as smart_dir
 
43
import bzrlib.smart.packrepository
42
44
import bzrlib.smart.repository
43
45
from bzrlib.smart.request import (
44
46
    FailedSmartServerResponse,
47
49
    SuccessfulSmartServerResponse,
48
50
    )
49
51
from bzrlib.tests import (
50
 
    iter_suite_tests,
51
52
    split_suite_by_re,
52
 
    TestScenarioApplier,
53
53
    )
54
54
from bzrlib.transport import chroot, get_transport
55
 
from bzrlib.util import bencode
56
55
 
57
56
 
58
57
def load_tests(standard_tests, module, loader):
59
58
    """Multiply tests version and protocol consistency."""
60
59
    # FindRepository tests.
61
60
    bzrdir_mod = bzrlib.smart.bzrdir
62
 
    applier = TestScenarioApplier()
63
 
    applier.scenarios = [
 
61
    scenarios = [
64
62
        ("find_repository", {
65
63
            "_request_class":bzrdir_mod.SmartServerRequestFindRepositoryV1}),
66
64
        ("find_repositoryV2", {
67
65
            "_request_class":bzrdir_mod.SmartServerRequestFindRepositoryV2}),
 
66
        ("find_repositoryV3", {
 
67
            "_request_class":bzrdir_mod.SmartServerRequestFindRepositoryV3}),
68
68
        ]
69
69
    to_adapt, result = split_suite_by_re(standard_tests,
70
70
        "TestSmartServerRequestFindRepository")
71
71
    v2_only, v1_and_2 = split_suite_by_re(to_adapt,
72
72
        "_v2")
73
 
    for test in iter_suite_tests(v1_and_2):
74
 
        result.addTests(applier.adapt(test))
75
 
    del applier.scenarios[0]
76
 
    for test in iter_suite_tests(v2_only):
77
 
        result.addTests(applier.adapt(test))
 
73
    tests.multiply_tests(v1_and_2, scenarios, result)
 
74
    # The first scenario is only applicable to v1 protocols, it is deleted
 
75
    # since.
 
76
    tests.multiply_tests(v2_only, scenarios[1:], result)
78
77
    return result
79
78
 
80
79
 
131
130
    def test__str__(self):
132
131
        """SmartServerResponses can be stringified."""
133
132
        self.assertEqual(
134
 
            "<SmartServerResponse status=OK args=('args',) body='body'>",
 
133
            "<SuccessfulSmartServerResponse args=('args',) body='body'>",
135
134
            str(SuccessfulSmartServerResponse(('args',), 'body')))
136
135
        self.assertEqual(
137
 
            "<SmartServerResponse status=ERR args=('args',) body='body'>",
 
136
            "<FailedSmartServerResponse args=('args',) body='body'>",
138
137
            str(FailedSmartServerResponse(('args',), 'body')))
139
138
 
140
139
 
160
159
            request.transport_from_client_path('foo/').base)
161
160
 
162
161
 
 
162
class TestSmartServerBzrDirRequestCloningMetaDir(
 
163
    tests.TestCaseWithMemoryTransport):
 
164
    """Tests for BzrDir.cloning_metadir."""
 
165
 
 
166
    def test_cloning_metadir(self):
 
167
        """When there is a bzrdir present, the call succeeds."""
 
168
        backing = self.get_transport()
 
169
        dir = self.make_bzrdir('.')
 
170
        local_result = dir.cloning_metadir()
 
171
        request_class = smart_dir.SmartServerBzrDirRequestCloningMetaDir
 
172
        request = request_class(backing)
 
173
        expected = SuccessfulSmartServerResponse(
 
174
            (local_result.network_name(),
 
175
            local_result.repository_format.network_name(),
 
176
            ('branch', local_result.get_branch_format().network_name())))
 
177
        self.assertEqual(expected, request.execute('', 'False'))
 
178
 
 
179
    def test_cloning_metadir_reference(self):
 
180
        """The request fails when bzrdir contains a branch reference."""
 
181
        backing = self.get_transport()
 
182
        referenced_branch = self.make_branch('referenced')
 
183
        dir = self.make_bzrdir('.')
 
184
        local_result = dir.cloning_metadir()
 
185
        reference = BranchReferenceFormat().initialize(dir, referenced_branch)
 
186
        reference_url = BranchReferenceFormat().get_reference(dir)
 
187
        # The server shouldn't try to follow the branch reference, so it's fine
 
188
        # if the referenced branch isn't reachable.
 
189
        backing.rename('referenced', 'moved')
 
190
        request_class = smart_dir.SmartServerBzrDirRequestCloningMetaDir
 
191
        request = request_class(backing)
 
192
        expected = FailedSmartServerResponse(('BranchReference',))
 
193
        self.assertEqual(expected, request.execute('', 'False'))
 
194
 
 
195
 
 
196
class TestSmartServerRequestCreateRepository(tests.TestCaseWithMemoryTransport):
 
197
    """Tests for BzrDir.create_repository."""
 
198
 
 
199
    def test_makes_repository(self):
 
200
        """When there is a bzrdir present, the call succeeds."""
 
201
        backing = self.get_transport()
 
202
        self.make_bzrdir('.')
 
203
        request_class = bzrlib.smart.bzrdir.SmartServerRequestCreateRepository
 
204
        request = request_class(backing)
 
205
        reference_bzrdir_format = bzrdir.format_registry.get('default')()
 
206
        reference_format = reference_bzrdir_format.repository_format
 
207
        network_name = reference_format.network_name()
 
208
        expected = SuccessfulSmartServerResponse(
 
209
            ('ok', 'no', 'no', 'no', network_name))
 
210
        self.assertEqual(expected, request.execute('', network_name, 'True'))
 
211
 
 
212
 
163
213
class TestSmartServerRequestFindRepository(tests.TestCaseWithMemoryTransport):
164
214
    """Tests for BzrDir.find_repository."""
165
215
 
172
222
            request.execute(''))
173
223
 
174
224
    def test_nonshared_repository(self):
175
 
        # nonshared repositorys only allow 'find' to return a handle when the 
176
 
        # path the repository is being searched on is the same as that that 
 
225
        # nonshared repositorys only allow 'find' to return a handle when the
 
226
        # path the repository is being searched on is the same as that that
177
227
        # the repository is at.
178
228
        backing = self.get_transport()
179
229
        request = self._request_class(backing)
197
247
            subtrees = 'yes'
198
248
        else:
199
249
            subtrees = 'no'
200
 
        if (smart.bzrdir.SmartServerRequestFindRepositoryV2 ==
 
250
        if (smart.bzrdir.SmartServerRequestFindRepositoryV3 ==
 
251
            self._request_class):
 
252
            return SuccessfulSmartServerResponse(
 
253
                ('ok', '', rich_root, subtrees, 'no',
 
254
                 repo._format.network_name()))
 
255
        elif (smart.bzrdir.SmartServerRequestFindRepositoryV2 ==
201
256
            self._request_class):
202
257
            # All tests so far are on formats, and for non-external
203
258
            # repositories.
241
296
        self.assertEqual(result, request.execute(''))
242
297
 
243
298
 
 
299
class TestSmartServerBzrDirRequestGetConfigFile(
 
300
    tests.TestCaseWithMemoryTransport):
 
301
    """Tests for BzrDir.get_config_file."""
 
302
 
 
303
    def test_present(self):
 
304
        backing = self.get_transport()
 
305
        dir = self.make_bzrdir('.')
 
306
        dir.get_config().set_default_stack_on("/")
 
307
        local_result = dir._get_config()._get_config_file().read()
 
308
        request_class = smart_dir.SmartServerBzrDirRequestConfigFile
 
309
        request = request_class(backing)
 
310
        expected = SuccessfulSmartServerResponse((), local_result)
 
311
        self.assertEqual(expected, request.execute(''))
 
312
 
 
313
    def test_missing(self):
 
314
        backing = self.get_transport()
 
315
        dir = self.make_bzrdir('.')
 
316
        request_class = smart_dir.SmartServerBzrDirRequestConfigFile
 
317
        request = request_class(backing)
 
318
        expected = SuccessfulSmartServerResponse((), '')
 
319
        self.assertEqual(expected, request.execute(''))
 
320
 
 
321
 
244
322
class TestSmartServerRequestInitializeBzrDir(tests.TestCaseWithMemoryTransport):
245
323
 
246
324
    def test_empty_dir(self):
250
328
        self.assertEqual(SmartServerResponse(('ok', )),
251
329
            request.execute(''))
252
330
        made_dir = bzrdir.BzrDir.open_from_transport(backing)
253
 
        # no branch, tree or repository is expected with the current 
 
331
        # no branch, tree or repository is expected with the current
254
332
        # default formart.
255
333
        self.assertRaises(errors.NoWorkingTree, made_dir.open_workingtree)
256
334
        self.assertRaises(errors.NotBranchError, made_dir.open_branch)
272
350
            request.execute, 'subdir')
273
351
 
274
352
 
 
353
class TestSmartServerRequestBzrDirInitializeEx(tests.TestCaseWithMemoryTransport):
 
354
    """Basic tests for BzrDir.initialize_ex_1.16 in the smart server.
 
355
 
 
356
    The main unit tests in test_bzrdir exercise the API comprehensively.
 
357
    """
 
358
 
 
359
    def test_empty_dir(self):
 
360
        """Initializing an empty dir should succeed and do it."""
 
361
        backing = self.get_transport()
 
362
        name = self.make_bzrdir('reference')._format.network_name()
 
363
        request = smart.bzrdir.SmartServerRequestBzrDirInitializeEx(backing)
 
364
        self.assertEqual(SmartServerResponse(('', '', '', '', '', '', name,
 
365
            'False', '', '', '')),
 
366
            request.execute(name, '', 'True', 'False', 'False', '', '', '', '',
 
367
            'False'))
 
368
        made_dir = bzrdir.BzrDir.open_from_transport(backing)
 
369
        # no branch, tree or repository is expected with the current
 
370
        # default format.
 
371
        self.assertRaises(errors.NoWorkingTree, made_dir.open_workingtree)
 
372
        self.assertRaises(errors.NotBranchError, made_dir.open_branch)
 
373
        self.assertRaises(errors.NoRepositoryPresent, made_dir.open_repository)
 
374
 
 
375
    def test_missing_dir(self):
 
376
        """Initializing a missing directory should fail like the bzrdir api."""
 
377
        backing = self.get_transport()
 
378
        name = self.make_bzrdir('reference')._format.network_name()
 
379
        request = smart.bzrdir.SmartServerRequestBzrDirInitializeEx(backing)
 
380
        self.assertRaises(errors.NoSuchFile, request.execute, name,
 
381
            'subdir/dir', 'False', 'False', 'False', '', '', '', '', 'False')
 
382
 
 
383
    def test_initialized_dir(self):
 
384
        """Initializing an extant directory should fail like the bzrdir api."""
 
385
        backing = self.get_transport()
 
386
        name = self.make_bzrdir('reference')._format.network_name()
 
387
        request = smart.bzrdir.SmartServerRequestBzrDirInitializeEx(backing)
 
388
        self.make_bzrdir('subdir')
 
389
        self.assertRaises(errors.FileExists, request.execute, name, 'subdir',
 
390
            'False', 'False', 'False', '', '', '', '', 'False')
 
391
 
 
392
 
275
393
class TestSmartServerRequestOpenBranch(TestCaseWithChrootedTransport):
276
394
 
277
395
    def test_no_branch(self):
302
420
            request.execute('reference'))
303
421
 
304
422
 
 
423
class TestSmartServerRequestOpenBranchV2(TestCaseWithChrootedTransport):
 
424
 
 
425
    def test_no_branch(self):
 
426
        """When there is no branch, ('nobranch', ) is returned."""
 
427
        backing = self.get_transport()
 
428
        self.make_bzrdir('.')
 
429
        request = smart.bzrdir.SmartServerRequestOpenBranchV2(backing)
 
430
        self.assertEqual(SmartServerResponse(('nobranch', )),
 
431
            request.execute(''))
 
432
 
 
433
    def test_branch(self):
 
434
        """When there is a branch, 'ok' is returned."""
 
435
        backing = self.get_transport()
 
436
        expected = self.make_branch('.')._format.network_name()
 
437
        request = smart.bzrdir.SmartServerRequestOpenBranchV2(backing)
 
438
        self.assertEqual(SuccessfulSmartServerResponse(('branch', expected)),
 
439
            request.execute(''))
 
440
 
 
441
    def test_branch_reference(self):
 
442
        """When there is a branch reference, the reference URL is returned."""
 
443
        backing = self.get_transport()
 
444
        request = smart.bzrdir.SmartServerRequestOpenBranchV2(backing)
 
445
        branch = self.make_branch('branch')
 
446
        checkout = branch.create_checkout('reference',lightweight=True)
 
447
        reference_url = BranchReferenceFormat().get_reference(checkout.bzrdir)
 
448
        self.assertFileEqual(reference_url, 'reference/.bzr/branch/location')
 
449
        self.assertEqual(SuccessfulSmartServerResponse(('ref', reference_url)),
 
450
            request.execute('reference'))
 
451
 
 
452
    def test_stacked_branch(self):
 
453
        """Opening a stacked branch does not open the stacked-on branch."""
 
454
        trunk = self.make_branch('trunk')
 
455
        feature = self.make_branch('feature', format='1.9')
 
456
        feature.set_stacked_on_url(trunk.base)
 
457
        opened_branches = []
 
458
        Branch.hooks.install_named_hook('open', opened_branches.append, None)
 
459
        backing = self.get_transport()
 
460
        request = smart.bzrdir.SmartServerRequestOpenBranchV2(backing)
 
461
        request.setup_jail()
 
462
        try:
 
463
            response = request.execute('feature')
 
464
        finally:
 
465
            request.teardown_jail()
 
466
        expected_format = feature._format.network_name()
 
467
        self.assertEqual(
 
468
            SuccessfulSmartServerResponse(('branch', expected_format)),
 
469
            response)
 
470
        self.assertLength(1, opened_branches)
 
471
 
 
472
 
305
473
class TestSmartServerRequestRevisionHistory(tests.TestCaseWithMemoryTransport):
306
474
 
307
475
    def test_empty(self):
388
556
    def test_with_content(self):
389
557
        # SmartServerBranchGetConfigFile should return the content from
390
558
        # branch.control_files.get('branch.conf') for now - in the future it may
391
 
        # perform more complex processing. 
 
559
        # perform more complex processing.
392
560
        backing = self.get_transport()
393
561
        request = smart.branch.SmartServerBranchGetConfigFile(backing)
394
562
        branch = self.make_branch('.')
395
 
        branch.control_files.put_utf8('branch.conf', 'foo bar baz')
 
563
        branch._transport.put_bytes('branch.conf', 'foo bar baz')
396
564
        self.assertEqual(SmartServerResponse(('ok', ), 'foo bar baz'),
397
565
            request.execute(''))
398
566
 
399
567
 
400
 
class TestSmartServerBranchRequestSetLastRevision(tests.TestCaseWithMemoryTransport):
401
 
 
402
 
    def test_empty(self):
403
 
        backing = self.get_transport()
404
 
        request = smart.branch.SmartServerBranchRequestSetLastRevision(backing)
405
 
        b = self.make_branch('.')
406
 
        branch_token = b.lock_write()
407
 
        repo_token = b.repository.lock_write()
408
 
        b.repository.unlock()
409
 
        try:
410
 
            self.assertEqual(SmartServerResponse(('ok',)),
411
 
                request.execute(
412
 
                    '', branch_token, repo_token,
413
 
                    'null:'))
414
 
        finally:
415
 
            b.unlock()
416
 
 
417
 
    def test_not_present_revision_id(self):
418
 
        backing = self.get_transport()
419
 
        request = smart.branch.SmartServerBranchRequestSetLastRevision(backing)
420
 
        b = self.make_branch('.')
421
 
        branch_token = b.lock_write()
422
 
        repo_token = b.repository.lock_write()
423
 
        b.repository.unlock()
424
 
        try:
425
 
            revision_id = 'non-existent revision'
426
 
            self.assertEqual(
427
 
                SmartServerResponse(('NoSuchRevision', revision_id)),
428
 
                request.execute(
429
 
                    '', branch_token, repo_token,
430
 
                    revision_id))
431
 
        finally:
432
 
            b.unlock()
433
 
 
434
 
    def test_revision_id_present(self):
435
 
        backing = self.get_transport()
436
 
        request = smart.branch.SmartServerBranchRequestSetLastRevision(backing)
437
 
        tree = self.make_branch_and_memory_tree('.')
438
 
        tree.lock_write()
439
 
        tree.add('')
440
 
        rev_id_utf8 = u'\xc8'.encode('utf-8')
441
 
        r1 = tree.commit('1st commit', rev_id=rev_id_utf8)
442
 
        r2 = tree.commit('2nd commit')
443
 
        tree.unlock()
444
 
        branch_token = tree.branch.lock_write()
445
 
        repo_token = tree.branch.repository.lock_write()
446
 
        tree.branch.repository.unlock()
447
 
        try:
448
 
            self.assertEqual(
449
 
                SmartServerResponse(('ok',)),
450
 
                request.execute(
451
 
                    '', branch_token, repo_token,
452
 
                    rev_id_utf8))
453
 
            self.assertEqual([rev_id_utf8], tree.branch.revision_history())
454
 
        finally:
455
 
            tree.branch.unlock()
456
 
 
457
 
    def test_revision_id_present2(self):
458
 
        backing = self.get_transport()
459
 
        request = smart.branch.SmartServerBranchRequestSetLastRevision(backing)
460
 
        tree = self.make_branch_and_memory_tree('.')
461
 
        tree.lock_write()
462
 
        tree.add('')
463
 
        rev_id_utf8 = u'\xc8'.encode('utf-8')
464
 
        r1 = tree.commit('1st commit', rev_id=rev_id_utf8)
465
 
        r2 = tree.commit('2nd commit')
466
 
        tree.unlock()
467
 
        tree.branch.set_revision_history([])
468
 
        branch_token = tree.branch.lock_write()
469
 
        repo_token = tree.branch.repository.lock_write()
470
 
        tree.branch.repository.unlock()
471
 
        try:
472
 
            self.assertEqual(
473
 
                SmartServerResponse(('ok',)),
474
 
                request.execute(
475
 
                    '', branch_token, repo_token,
476
 
                    rev_id_utf8))
477
 
            self.assertEqual([rev_id_utf8], tree.branch.revision_history())
478
 
        finally:
479
 
            tree.branch.unlock()
480
 
 
481
 
 
482
 
class TestSmartServerBranchRequestSetLastRevisionInfo(tests.TestCaseWithTransport):
483
 
 
484
 
    def lock_branch(self, branch):
 
568
class TestLockedBranch(tests.TestCaseWithMemoryTransport):
 
569
 
 
570
    def get_lock_tokens(self, branch):
485
571
        branch_token = branch.lock_write()
486
572
        repo_token = branch.repository.lock_write()
487
573
        branch.repository.unlock()
488
 
        self.addCleanup(branch.unlock)
489
574
        return branch_token, repo_token
490
575
 
491
 
    def make_locked_branch(self, format=None):
492
 
        branch = self.make_branch('.', format=format)
493
 
        branch_token, repo_token = self.lock_branch(branch)
494
 
        return branch, branch_token, repo_token
495
 
 
496
 
    def test_empty(self):
 
576
 
 
577
class TestSmartServerBranchRequestSetConfigOption(TestLockedBranch):
 
578
 
 
579
    def test_value_name(self):
 
580
        branch = self.make_branch('.')
 
581
        request = smart.branch.SmartServerBranchRequestSetConfigOption(
 
582
            branch.bzrdir.root_transport)
 
583
        branch_token, repo_token = self.get_lock_tokens(branch)
 
584
        config = branch._get_config()
 
585
        result = request.execute('', branch_token, repo_token, 'bar', 'foo',
 
586
            '')
 
587
        self.assertEqual(SuccessfulSmartServerResponse(()), result)
 
588
        self.assertEqual('bar', config.get_option('foo'))
 
589
        # Cleanup
 
590
        branch.unlock()
 
591
 
 
592
    def test_value_name_section(self):
 
593
        branch = self.make_branch('.')
 
594
        request = smart.branch.SmartServerBranchRequestSetConfigOption(
 
595
            branch.bzrdir.root_transport)
 
596
        branch_token, repo_token = self.get_lock_tokens(branch)
 
597
        config = branch._get_config()
 
598
        result = request.execute('', branch_token, repo_token, 'bar', 'foo',
 
599
            'gam')
 
600
        self.assertEqual(SuccessfulSmartServerResponse(()), result)
 
601
        self.assertEqual('bar', config.get_option('foo', 'gam'))
 
602
        # Cleanup
 
603
        branch.unlock()
 
604
 
 
605
 
 
606
class SetLastRevisionTestBase(TestLockedBranch):
 
607
    """Base test case for verbs that implement set_last_revision."""
 
608
 
 
609
    def setUp(self):
 
610
        tests.TestCaseWithMemoryTransport.setUp(self)
 
611
        backing_transport = self.get_transport()
 
612
        self.request = self.request_class(backing_transport)
 
613
        self.tree = self.make_branch_and_memory_tree('.')
 
614
 
 
615
    def lock_branch(self):
 
616
        return self.get_lock_tokens(self.tree.branch)
 
617
 
 
618
    def unlock_branch(self):
 
619
        self.tree.branch.unlock()
 
620
 
 
621
    def set_last_revision(self, revision_id, revno):
 
622
        branch_token, repo_token = self.lock_branch()
 
623
        response = self._set_last_revision(
 
624
            revision_id, revno, branch_token, repo_token)
 
625
        self.unlock_branch()
 
626
        return response
 
627
 
 
628
    def assertRequestSucceeds(self, revision_id, revno):
 
629
        response = self.set_last_revision(revision_id, revno)
 
630
        self.assertEqual(SuccessfulSmartServerResponse(('ok',)), response)
 
631
 
 
632
 
 
633
class TestSetLastRevisionVerbMixin(object):
 
634
    """Mixin test case for verbs that implement set_last_revision."""
 
635
 
 
636
    def test_set_null_to_null(self):
497
637
        """An empty branch can have its last revision set to 'null:'."""
498
 
        b, branch_token, repo_token = self.make_locked_branch()
499
 
        backing = self.get_transport()
500
 
        request = smart.branch.SmartServerBranchRequestSetLastRevisionInfo(
501
 
            backing)
502
 
        response = request.execute('', branch_token, repo_token, '0', 'null:')
503
 
        self.assertEqual(SmartServerResponse(('ok',)), response)
504
 
 
505
 
    def assertBranchLastRevisionInfo(self, expected_info, branch_relpath):
506
 
        branch = bzrdir.BzrDir.open(branch_relpath).open_branch()
507
 
        self.assertEqual(expected_info, branch.last_revision_info())
508
 
 
509
 
    def test_branch_revision_info_is_updated(self):
510
 
        """This method really does update the branch last revision info."""
511
 
        tree = self.make_branch_and_memory_tree('.')
512
 
        tree.lock_write()
513
 
        tree.add('')
514
 
        tree.commit('First commit', rev_id='revision-1')
515
 
        tree.commit('Second commit', rev_id='revision-2')
516
 
        tree.unlock()
517
 
        branch = tree.branch
518
 
 
519
 
        branch_token, repo_token = self.lock_branch(branch)
520
 
        backing = self.get_transport()
521
 
        request = smart.branch.SmartServerBranchRequestSetLastRevisionInfo(
522
 
            backing)
523
 
        self.assertBranchLastRevisionInfo((2, 'revision-2'), '.')
524
 
        response = request.execute(
525
 
            '', branch_token, repo_token, '1', 'revision-1')
526
 
        self.assertEqual(SmartServerResponse(('ok',)), response)
527
 
        self.assertBranchLastRevisionInfo((1, 'revision-1'), '.')
528
 
 
529
 
    def test_not_present_revid(self):
530
 
        """Some branch formats will check that the revision is present in the
531
 
        repository.  When that check fails, a NoSuchRevision error is returned
532
 
        to the client.
533
 
        """
534
 
        # Make a knit format branch, because that format checks the values
535
 
        # given to set_last_revision_info.
536
 
        b, branch_token, repo_token = self.make_locked_branch(format='knit')
537
 
        backing = self.get_transport()
538
 
        request = smart.branch.SmartServerBranchRequestSetLastRevisionInfo(
539
 
            backing)
540
 
        response = request.execute(
541
 
            '', branch_token, repo_token, '1', 'not-present')
542
 
        self.assertEqual(
543
 
            SmartServerResponse(('NoSuchRevision', 'not-present')), response)
 
638
        self.assertRequestSucceeds('null:', 0)
 
639
 
 
640
    def test_NoSuchRevision(self):
 
641
        """If the revision_id is not present, the verb returns NoSuchRevision.
 
642
        """
 
643
        revision_id = 'non-existent revision'
 
644
        self.assertEqual(
 
645
            FailedSmartServerResponse(('NoSuchRevision', revision_id)),
 
646
            self.set_last_revision(revision_id, 1))
 
647
 
 
648
    def make_tree_with_two_commits(self):
 
649
        self.tree.lock_write()
 
650
        self.tree.add('')
 
651
        rev_id_utf8 = u'\xc8'.encode('utf-8')
 
652
        r1 = self.tree.commit('1st commit', rev_id=rev_id_utf8)
 
653
        r2 = self.tree.commit('2nd commit', rev_id='rev-2')
 
654
        self.tree.unlock()
 
655
 
 
656
    def test_branch_last_revision_info_is_updated(self):
 
657
        """A branch's tip can be set to a revision that is present in its
 
658
        repository.
 
659
        """
 
660
        # Make a branch with an empty revision history, but two revisions in
 
661
        # its repository.
 
662
        self.make_tree_with_two_commits()
 
663
        rev_id_utf8 = u'\xc8'.encode('utf-8')
 
664
        self.tree.branch.set_revision_history([])
 
665
        self.assertEqual(
 
666
            (0, 'null:'), self.tree.branch.last_revision_info())
 
667
        # We can update the branch to a revision that is present in the
 
668
        # repository.
 
669
        self.assertRequestSucceeds(rev_id_utf8, 1)
 
670
        self.assertEqual(
 
671
            (1, rev_id_utf8), self.tree.branch.last_revision_info())
 
672
 
 
673
    def test_branch_last_revision_info_rewind(self):
 
674
        """A branch's tip can be set to a revision that is an ancestor of the
 
675
        current tip.
 
676
        """
 
677
        self.make_tree_with_two_commits()
 
678
        rev_id_utf8 = u'\xc8'.encode('utf-8')
 
679
        self.assertEqual(
 
680
            (2, 'rev-2'), self.tree.branch.last_revision_info())
 
681
        self.assertRequestSucceeds(rev_id_utf8, 1)
 
682
        self.assertEqual(
 
683
            (1, rev_id_utf8), self.tree.branch.last_revision_info())
 
684
 
 
685
    def test_TipChangeRejected(self):
 
686
        """If a pre_change_branch_tip hook raises TipChangeRejected, the verb
 
687
        returns TipChangeRejected.
 
688
        """
 
689
        rejection_message = u'rejection message\N{INTERROBANG}'
 
690
        def hook_that_rejects(params):
 
691
            raise errors.TipChangeRejected(rejection_message)
 
692
        Branch.hooks.install_named_hook(
 
693
            'pre_change_branch_tip', hook_that_rejects, None)
 
694
        self.assertEqual(
 
695
            FailedSmartServerResponse(
 
696
                ('TipChangeRejected', rejection_message.encode('utf-8'))),
 
697
            self.set_last_revision('null:', 0))
 
698
 
 
699
 
 
700
class TestSmartServerBranchRequestSetLastRevision(
 
701
        SetLastRevisionTestBase, TestSetLastRevisionVerbMixin):
 
702
    """Tests for Branch.set_last_revision verb."""
 
703
 
 
704
    request_class = smart.branch.SmartServerBranchRequestSetLastRevision
 
705
 
 
706
    def _set_last_revision(self, revision_id, revno, branch_token, repo_token):
 
707
        return self.request.execute(
 
708
            '', branch_token, repo_token, revision_id)
 
709
 
 
710
 
 
711
class TestSmartServerBranchRequestSetLastRevisionInfo(
 
712
        SetLastRevisionTestBase, TestSetLastRevisionVerbMixin):
 
713
    """Tests for Branch.set_last_revision_info verb."""
 
714
 
 
715
    request_class = smart.branch.SmartServerBranchRequestSetLastRevisionInfo
 
716
 
 
717
    def _set_last_revision(self, revision_id, revno, branch_token, repo_token):
 
718
        return self.request.execute(
 
719
            '', branch_token, repo_token, revno, revision_id)
 
720
 
 
721
    def test_NoSuchRevision(self):
 
722
        """Branch.set_last_revision_info does not have to return
 
723
        NoSuchRevision if the revision_id is absent.
 
724
        """
 
725
        raise tests.TestNotApplicable()
 
726
 
 
727
 
 
728
class TestSmartServerBranchRequestSetLastRevisionEx(
 
729
        SetLastRevisionTestBase, TestSetLastRevisionVerbMixin):
 
730
    """Tests for Branch.set_last_revision_ex verb."""
 
731
 
 
732
    request_class = smart.branch.SmartServerBranchRequestSetLastRevisionEx
 
733
 
 
734
    def _set_last_revision(self, revision_id, revno, branch_token, repo_token):
 
735
        return self.request.execute(
 
736
            '', branch_token, repo_token, revision_id, 0, 0)
 
737
 
 
738
    def assertRequestSucceeds(self, revision_id, revno):
 
739
        response = self.set_last_revision(revision_id, revno)
 
740
        self.assertEqual(
 
741
            SuccessfulSmartServerResponse(('ok', revno, revision_id)),
 
742
            response)
 
743
 
 
744
    def test_branch_last_revision_info_rewind(self):
 
745
        """A branch's tip can be set to a revision that is an ancestor of the
 
746
        current tip, but only if allow_overwrite_descendant is passed.
 
747
        """
 
748
        self.make_tree_with_two_commits()
 
749
        rev_id_utf8 = u'\xc8'.encode('utf-8')
 
750
        self.assertEqual(
 
751
            (2, 'rev-2'), self.tree.branch.last_revision_info())
 
752
        # If allow_overwrite_descendant flag is 0, then trying to set the tip
 
753
        # to an older revision ID has no effect.
 
754
        branch_token, repo_token = self.lock_branch()
 
755
        response = self.request.execute(
 
756
            '', branch_token, repo_token, rev_id_utf8, 0, 0)
 
757
        self.assertEqual(
 
758
            SuccessfulSmartServerResponse(('ok', 2, 'rev-2')),
 
759
            response)
 
760
        self.assertEqual(
 
761
            (2, 'rev-2'), self.tree.branch.last_revision_info())
 
762
 
 
763
        # If allow_overwrite_descendant flag is 1, then setting the tip to an
 
764
        # ancestor works.
 
765
        response = self.request.execute(
 
766
            '', branch_token, repo_token, rev_id_utf8, 0, 1)
 
767
        self.assertEqual(
 
768
            SuccessfulSmartServerResponse(('ok', 1, rev_id_utf8)),
 
769
            response)
 
770
        self.unlock_branch()
 
771
        self.assertEqual(
 
772
            (1, rev_id_utf8), self.tree.branch.last_revision_info())
 
773
 
 
774
    def make_branch_with_divergent_history(self):
 
775
        """Make a branch with divergent history in its repo.
 
776
 
 
777
        The branch's tip will be 'child-2', and the repo will also contain
 
778
        'child-1', which diverges from a common base revision.
 
779
        """
 
780
        self.tree.lock_write()
 
781
        self.tree.add('')
 
782
        r1 = self.tree.commit('1st commit')
 
783
        revno_1, revid_1 = self.tree.branch.last_revision_info()
 
784
        r2 = self.tree.commit('2nd commit', rev_id='child-1')
 
785
        # Undo the second commit
 
786
        self.tree.branch.set_last_revision_info(revno_1, revid_1)
 
787
        self.tree.set_parent_ids([revid_1])
 
788
        # Make a new second commit, child-2.  child-2 has diverged from
 
789
        # child-1.
 
790
        new_r2 = self.tree.commit('2nd commit', rev_id='child-2')
 
791
        self.tree.unlock()
 
792
 
 
793
    def test_not_allow_diverged(self):
 
794
        """If allow_diverged is not passed, then setting a divergent history
 
795
        returns a Diverged error.
 
796
        """
 
797
        self.make_branch_with_divergent_history()
 
798
        self.assertEqual(
 
799
            FailedSmartServerResponse(('Diverged',)),
 
800
            self.set_last_revision('child-1', 2))
 
801
        # The branch tip was not changed.
 
802
        self.assertEqual('child-2', self.tree.branch.last_revision())
 
803
 
 
804
    def test_allow_diverged(self):
 
805
        """If allow_diverged is passed, then setting a divergent history
 
806
        succeeds.
 
807
        """
 
808
        self.make_branch_with_divergent_history()
 
809
        branch_token, repo_token = self.lock_branch()
 
810
        response = self.request.execute(
 
811
            '', branch_token, repo_token, 'child-1', 1, 0)
 
812
        self.assertEqual(
 
813
            SuccessfulSmartServerResponse(('ok', 2, 'child-1')),
 
814
            response)
 
815
        self.unlock_branch()
 
816
        # The branch tip was changed.
 
817
        self.assertEqual('child-1', self.tree.branch.last_revision())
 
818
 
 
819
 
 
820
class TestSmartServerBranchRequestGetParent(tests.TestCaseWithMemoryTransport):
 
821
 
 
822
    def test_get_parent_none(self):
 
823
        base_branch = self.make_branch('base')
 
824
        request = smart.branch.SmartServerBranchGetParent(self.get_transport())
 
825
        response = request.execute('base')
 
826
        self.assertEquals(
 
827
            SuccessfulSmartServerResponse(('',)), response)
 
828
 
 
829
    def test_get_parent_something(self):
 
830
        base_branch = self.make_branch('base')
 
831
        base_branch.set_parent(self.get_url('foo'))
 
832
        request = smart.branch.SmartServerBranchGetParent(self.get_transport())
 
833
        response = request.execute('base')
 
834
        self.assertEquals(
 
835
            SuccessfulSmartServerResponse(("../foo",)),
 
836
            response)
 
837
 
 
838
 
 
839
class TestSmartServerBranchRequestSetParent(tests.TestCaseWithMemoryTransport):
 
840
 
 
841
    def test_set_parent_none(self):
 
842
        branch = self.make_branch('base', format="1.9")
 
843
        branch.lock_write()
 
844
        branch._set_parent_location('foo')
 
845
        branch.unlock()
 
846
        request = smart.branch.SmartServerBranchRequestSetParentLocation(
 
847
            self.get_transport())
 
848
        branch_token = branch.lock_write()
 
849
        repo_token = branch.repository.lock_write()
 
850
        try:
 
851
            response = request.execute('base', branch_token, repo_token, '')
 
852
        finally:
 
853
            branch.repository.unlock()
 
854
            branch.unlock()
 
855
        self.assertEqual(SuccessfulSmartServerResponse(()), response)
 
856
        self.assertEqual(None, branch.get_parent())
 
857
 
 
858
    def test_set_parent_something(self):
 
859
        branch = self.make_branch('base', format="1.9")
 
860
        request = smart.branch.SmartServerBranchRequestSetParentLocation(
 
861
            self.get_transport())
 
862
        branch_token = branch.lock_write()
 
863
        repo_token = branch.repository.lock_write()
 
864
        try:
 
865
            response = request.execute('base', branch_token, repo_token,
 
866
            'http://bar/')
 
867
        finally:
 
868
            branch.repository.unlock()
 
869
            branch.unlock()
 
870
        self.assertEqual(SuccessfulSmartServerResponse(()), response)
 
871
        self.assertEqual('http://bar/', branch.get_parent())
 
872
 
 
873
 
 
874
class TestSmartServerBranchRequestGetTagsBytes(tests.TestCaseWithMemoryTransport):
 
875
# Only called when the branch format and tags match [yay factory
 
876
# methods] so only need to test straight forward cases.
 
877
 
 
878
    def test_get_bytes(self):
 
879
        base_branch = self.make_branch('base')
 
880
        request = smart.branch.SmartServerBranchGetTagsBytes(
 
881
            self.get_transport())
 
882
        response = request.execute('base')
 
883
        self.assertEquals(
 
884
            SuccessfulSmartServerResponse(('',)), response)
 
885
 
 
886
 
 
887
class TestSmartServerBranchRequestGetStackedOnURL(tests.TestCaseWithMemoryTransport):
 
888
 
 
889
    def test_get_stacked_on_url(self):
 
890
        base_branch = self.make_branch('base', format='1.6')
 
891
        stacked_branch = self.make_branch('stacked', format='1.6')
 
892
        # typically should be relative
 
893
        stacked_branch.set_stacked_on_url('../base')
 
894
        request = smart.branch.SmartServerBranchRequestGetStackedOnURL(
 
895
            self.get_transport())
 
896
        response = request.execute('stacked')
 
897
        self.assertEquals(
 
898
            SmartServerResponse(('ok', '../base')),
 
899
            response)
544
900
 
545
901
 
546
902
class TestSmartServerBranchRequestLockWrite(tests.TestCaseWithMemoryTransport):
563
919
        # with a new branch object.
564
920
        new_branch = repository.bzrdir.open_branch()
565
921
        self.assertRaises(errors.LockContention, new_branch.lock_write)
 
922
        # Cleanup
 
923
        request = smart.branch.SmartServerBranchRequestUnlock(backing)
 
924
        response = request.execute('', branch_nonce, repository_nonce)
566
925
 
567
926
    def test_lock_write_on_locked_branch(self):
568
927
        backing = self.get_transport()
569
928
        request = smart.branch.SmartServerBranchRequestLockWrite(backing)
570
929
        branch = self.make_branch('.')
571
 
        branch.lock_write()
 
930
        branch_token = branch.lock_write()
572
931
        branch.leave_lock_in_place()
573
932
        branch.unlock()
574
933
        response = request.execute('')
575
934
        self.assertEqual(
576
935
            SmartServerResponse(('LockContention',)), response)
 
936
        # Cleanup
 
937
        branch.lock_write(branch_token)
 
938
        branch.dont_leave_lock_in_place()
 
939
        branch.unlock()
577
940
 
578
941
    def test_lock_write_with_tokens_on_locked_branch(self):
579
942
        backing = self.get_transport()
589
952
                                   branch_token, repo_token)
590
953
        self.assertEqual(
591
954
            SmartServerResponse(('ok', branch_token, repo_token)), response)
 
955
        # Cleanup
 
956
        branch.repository.lock_write(repo_token)
 
957
        branch.repository.dont_leave_lock_in_place()
 
958
        branch.repository.unlock()
 
959
        branch.lock_write(branch_token)
 
960
        branch.dont_leave_lock_in_place()
 
961
        branch.unlock()
592
962
 
593
963
    def test_lock_write_with_mismatched_tokens_on_locked_branch(self):
594
964
        backing = self.get_transport()
604
974
                                   branch_token+'xxx', repo_token)
605
975
        self.assertEqual(
606
976
            SmartServerResponse(('TokenMismatch',)), response)
 
977
        # Cleanup
 
978
        branch.repository.lock_write(repo_token)
 
979
        branch.repository.dont_leave_lock_in_place()
 
980
        branch.repository.unlock()
 
981
        branch.lock_write(branch_token)
 
982
        branch.dont_leave_lock_in_place()
 
983
        branch.unlock()
607
984
 
608
985
    def test_lock_write_on_locked_repo(self):
609
986
        backing = self.get_transport()
610
987
        request = smart.branch.SmartServerBranchRequestLockWrite(backing)
611
988
        branch = self.make_branch('.', format='knit')
612
 
        branch.repository.lock_write()
613
 
        branch.repository.leave_lock_in_place()
614
 
        branch.repository.unlock()
 
989
        repo = branch.repository
 
990
        repo_token = repo.lock_write()
 
991
        repo.leave_lock_in_place()
 
992
        repo.unlock()
615
993
        response = request.execute('')
616
994
        self.assertEqual(
617
995
            SmartServerResponse(('LockContention',)), response)
 
996
        # Cleanup
 
997
        repo.lock_write(repo_token)
 
998
        repo.dont_leave_lock_in_place()
 
999
        repo.unlock()
618
1000
 
619
1001
    def test_lock_write_on_readonly_transport(self):
620
1002
        backing = self.get_readonly_transport()
679
1061
            '', 'branch token', repo_token)
680
1062
        self.assertEqual(
681
1063
            SmartServerResponse(('TokenMismatch',)), response)
 
1064
        # Cleanup
 
1065
        branch.repository.lock_write(repo_token)
 
1066
        branch.repository.dont_leave_lock_in_place()
 
1067
        branch.repository.unlock()
682
1068
 
683
1069
 
684
1070
class TestSmartServerRepositoryRequest(tests.TestCaseWithMemoryTransport):
697
1083
            request.execute, 'subdir')
698
1084
 
699
1085
 
700
 
class TestSmartServerRepositoryGetParentMap(tests.TestCaseWithTransport):
 
1086
class TestSmartServerRepositoryGetParentMap(tests.TestCaseWithMemoryTransport):
701
1087
 
702
1088
    def test_trivial_bzipped(self):
703
1089
        # This tests that the wire encoding is actually bzipped
707
1093
 
708
1094
        self.assertEqual(None,
709
1095
            request.execute('', 'missing-id'))
710
 
        # Note that it returns a body (of '' bzipped).
 
1096
        # Note that it returns a body that is bzipped.
711
1097
        self.assertEqual(
712
1098
            SuccessfulSmartServerResponse(('ok', ), bz2.compress('')),
713
1099
            request.do_body('\n\n0\n'))
714
1100
 
 
1101
    def test_trivial_include_missing(self):
 
1102
        backing = self.get_transport()
 
1103
        request = smart.repository.SmartServerRepositoryGetParentMap(backing)
 
1104
        tree = self.make_branch_and_memory_tree('.')
 
1105
 
 
1106
        self.assertEqual(None,
 
1107
            request.execute('', 'missing-id', 'include-missing:'))
 
1108
        self.assertEqual(
 
1109
            SuccessfulSmartServerResponse(('ok', ),
 
1110
                bz2.compress('missing:missing-id')),
 
1111
            request.do_body('\n\n0\n'))
 
1112
 
715
1113
 
716
1114
class TestSmartServerRepositoryGetRevisionGraph(tests.TestCaseWithMemoryTransport):
717
1115
 
747
1145
 
748
1146
        self.assertEqual(SmartServerResponse(('ok', ), rev_id_utf8),
749
1147
            request.execute('', rev_id_utf8))
750
 
    
 
1148
 
751
1149
    def test_no_such_revision(self):
752
1150
        backing = self.get_transport()
753
1151
        request = smart.repository.SmartServerRepositoryGetRevisionGraph(backing)
763
1161
            request.execute('', 'missingrevision'))
764
1162
 
765
1163
 
 
1164
class TestSmartServerRepositoryGetRevIdForRevno(tests.TestCaseWithMemoryTransport):
 
1165
 
 
1166
    def test_revno_found(self):
 
1167
        backing = self.get_transport()
 
1168
        request = smart.repository.SmartServerRepositoryGetRevIdForRevno(backing)
 
1169
        tree = self.make_branch_and_memory_tree('.')
 
1170
        tree.lock_write()
 
1171
        tree.add('')
 
1172
        rev1_id_utf8 = u'\xc8'.encode('utf-8')
 
1173
        rev2_id_utf8 = u'\xc9'.encode('utf-8')
 
1174
        tree.commit('1st commit', rev_id=rev1_id_utf8)
 
1175
        tree.commit('2nd commit', rev_id=rev2_id_utf8)
 
1176
        tree.unlock()
 
1177
 
 
1178
        self.assertEqual(SmartServerResponse(('ok', rev1_id_utf8)),
 
1179
            request.execute('', 1, (2, rev2_id_utf8)))
 
1180
 
 
1181
    def test_known_revid_missing(self):
 
1182
        backing = self.get_transport()
 
1183
        request = smart.repository.SmartServerRepositoryGetRevIdForRevno(backing)
 
1184
        repo = self.make_repository('.')
 
1185
        self.assertEqual(
 
1186
            FailedSmartServerResponse(('nosuchrevision', 'ghost')),
 
1187
            request.execute('', 1, (2, 'ghost')))
 
1188
 
 
1189
    def test_history_incomplete(self):
 
1190
        backing = self.get_transport()
 
1191
        request = smart.repository.SmartServerRepositoryGetRevIdForRevno(backing)
 
1192
        parent = self.make_branch_and_memory_tree('parent', format='1.9')
 
1193
        r1 = parent.commit(message='first commit')
 
1194
        r2 = parent.commit(message='second commit')
 
1195
        local = self.make_branch_and_memory_tree('local', format='1.9')
 
1196
        local.branch.pull(parent.branch)
 
1197
        local.set_parent_ids([r2])
 
1198
        r3 = local.commit(message='local commit')
 
1199
        local.branch.create_clone_on_transport(
 
1200
            self.get_transport('stacked'), stacked_on=self.get_url('parent'))
 
1201
        self.assertEqual(
 
1202
            SmartServerResponse(('history-incomplete', 2, r2)),
 
1203
            request.execute('stacked', 1, (3, r3)))
 
1204
 
 
1205
class TestSmartServerRepositoryGetStream(tests.TestCaseWithMemoryTransport):
 
1206
 
 
1207
    def make_two_commit_repo(self):
 
1208
        tree = self.make_branch_and_memory_tree('.')
 
1209
        tree.lock_write()
 
1210
        tree.add('')
 
1211
        r1 = tree.commit('1st commit')
 
1212
        r2 = tree.commit('2nd commit', rev_id=u'\xc8'.encode('utf-8'))
 
1213
        tree.unlock()
 
1214
        repo = tree.branch.repository
 
1215
        return repo, r1, r2
 
1216
 
 
1217
    def test_ancestry_of(self):
 
1218
        """The search argument may be a 'ancestry-of' some heads'."""
 
1219
        backing = self.get_transport()
 
1220
        request = smart.repository.SmartServerRepositoryGetStream(backing)
 
1221
        repo, r1, r2 = self.make_two_commit_repo()
 
1222
        fetch_spec = ['ancestry-of', r2]
 
1223
        lines = '\n'.join(fetch_spec)
 
1224
        request.execute('', repo._format.network_name())
 
1225
        response = request.do_body(lines)
 
1226
        self.assertEqual(('ok',), response.args)
 
1227
        stream_bytes = ''.join(response.body_stream)
 
1228
        self.assertStartsWith(stream_bytes, 'Bazaar pack format 1')
 
1229
 
 
1230
    def test_search(self):
 
1231
        """The search argument may be a 'search' of some explicit keys."""
 
1232
        backing = self.get_transport()
 
1233
        request = smart.repository.SmartServerRepositoryGetStream(backing)
 
1234
        repo, r1, r2 = self.make_two_commit_repo()
 
1235
        fetch_spec = ['search', '%s %s' % (r1, r2), 'null:', '2']
 
1236
        lines = '\n'.join(fetch_spec)
 
1237
        request.execute('', repo._format.network_name())
 
1238
        response = request.do_body(lines)
 
1239
        self.assertEqual(('ok',), response.args)
 
1240
        stream_bytes = ''.join(response.body_stream)
 
1241
        self.assertStartsWith(stream_bytes, 'Bazaar pack format 1')
 
1242
 
 
1243
 
766
1244
class TestSmartServerRequestHasRevision(tests.TestCaseWithMemoryTransport):
767
1245
 
768
1246
    def test_missing_revision(self):
796
1274
        request = smart.repository.SmartServerRepositoryGatherStats(backing)
797
1275
        repository = self.make_repository('.')
798
1276
        stats = repository.gather_stats()
799
 
        size = stats['size']
800
 
        expected_body = 'revisions: 0\nsize: %d\n' % size
 
1277
        expected_body = 'revisions: 0\n'
801
1278
        self.assertEqual(SmartServerResponse(('ok', ), expected_body),
802
1279
                         request.execute('', '', 'no'))
803
1280
 
816
1293
        tree.unlock()
817
1294
 
818
1295
        stats = tree.branch.repository.gather_stats()
819
 
        size = stats['size']
820
1296
        expected_body = ('firstrev: 123456.200 3600\n'
821
1297
                         'latestrev: 654321.400 0\n'
822
 
                         'revisions: 2\n'
823
 
                         'size: %d\n' % size)
 
1298
                         'revisions: 2\n')
824
1299
        self.assertEqual(SmartServerResponse(('ok', ), expected_body),
825
1300
                         request.execute('',
826
1301
                                         rev_id_utf8, 'no'))
841
1316
        tree.unlock()
842
1317
        stats = tree.branch.repository.gather_stats()
843
1318
 
844
 
        size = stats['size']
845
1319
        expected_body = ('committers: 2\n'
846
1320
                         'firstrev: 123456.200 3600\n'
847
1321
                         'latestrev: 654321.400 0\n'
848
 
                         'revisions: 2\n'
849
 
                         'size: %d\n' % size)
 
1322
                         'revisions: 2\n')
850
1323
        self.assertEqual(SmartServerResponse(('ok', ), expected_body),
851
1324
                         request.execute('',
852
1325
                                         rev_id_utf8, 'yes'))
873
1346
 
874
1347
class TestSmartServerRepositoryLockWrite(tests.TestCaseWithMemoryTransport):
875
1348
 
876
 
    def setUp(self):
877
 
        tests.TestCaseWithMemoryTransport.setUp(self)
878
 
 
879
1349
    def test_lock_write_on_unlocked_repo(self):
880
1350
        backing = self.get_transport()
881
1351
        request = smart.repository.SmartServerRepositoryLockWrite(backing)
887
1357
        # object.
888
1358
        new_repo = repository.bzrdir.open_repository()
889
1359
        self.assertRaises(errors.LockContention, new_repo.lock_write)
 
1360
        # Cleanup
 
1361
        request = smart.repository.SmartServerRepositoryUnlock(backing)
 
1362
        response = request.execute('', nonce)
890
1363
 
891
1364
    def test_lock_write_on_locked_repo(self):
892
1365
        backing = self.get_transport()
893
1366
        request = smart.repository.SmartServerRepositoryLockWrite(backing)
894
1367
        repository = self.make_repository('.', format='knit')
895
 
        repository.lock_write()
 
1368
        repo_token = repository.lock_write()
896
1369
        repository.leave_lock_in_place()
897
1370
        repository.unlock()
898
1371
        response = request.execute('')
899
1372
        self.assertEqual(
900
1373
            SmartServerResponse(('LockContention',)), response)
 
1374
        # Cleanup
 
1375
        repository.lock_write(repo_token)
 
1376
        repository.dont_leave_lock_in_place()
 
1377
        repository.unlock()
901
1378
 
902
1379
    def test_lock_write_on_readonly_transport(self):
903
1380
        backing = self.get_readonly_transport()
908
1385
        self.assertEqual('LockFailed', response.args[0])
909
1386
 
910
1387
 
 
1388
class TestInsertStreamBase(tests.TestCaseWithMemoryTransport):
 
1389
 
 
1390
    def make_empty_byte_stream(self, repo):
 
1391
        byte_stream = smart.repository._stream_to_byte_stream([], repo._format)
 
1392
        return ''.join(byte_stream)
 
1393
 
 
1394
 
 
1395
class TestSmartServerRepositoryInsertStream(TestInsertStreamBase):
 
1396
 
 
1397
    def test_insert_stream_empty(self):
 
1398
        backing = self.get_transport()
 
1399
        request = smart.repository.SmartServerRepositoryInsertStream(backing)
 
1400
        repository = self.make_repository('.')
 
1401
        response = request.execute('', '')
 
1402
        self.assertEqual(None, response)
 
1403
        response = request.do_chunk(self.make_empty_byte_stream(repository))
 
1404
        self.assertEqual(None, response)
 
1405
        response = request.do_end()
 
1406
        self.assertEqual(SmartServerResponse(('ok', )), response)
 
1407
        
 
1408
 
 
1409
class TestSmartServerRepositoryInsertStreamLocked(TestInsertStreamBase):
 
1410
 
 
1411
    def test_insert_stream_empty(self):
 
1412
        backing = self.get_transport()
 
1413
        request = smart.repository.SmartServerRepositoryInsertStreamLocked(
 
1414
            backing)
 
1415
        repository = self.make_repository('.', format='knit')
 
1416
        lock_token = repository.lock_write()
 
1417
        response = request.execute('', '', lock_token)
 
1418
        self.assertEqual(None, response)
 
1419
        response = request.do_chunk(self.make_empty_byte_stream(repository))
 
1420
        self.assertEqual(None, response)
 
1421
        response = request.do_end()
 
1422
        self.assertEqual(SmartServerResponse(('ok', )), response)
 
1423
        repository.unlock()
 
1424
 
 
1425
    def test_insert_stream_with_wrong_lock_token(self):
 
1426
        backing = self.get_transport()
 
1427
        request = smart.repository.SmartServerRepositoryInsertStreamLocked(
 
1428
            backing)
 
1429
        repository = self.make_repository('.', format='knit')
 
1430
        lock_token = repository.lock_write()
 
1431
        self.assertRaises(
 
1432
            errors.TokenMismatch, request.execute, '', '', 'wrong-token')
 
1433
        repository.unlock()
 
1434
 
 
1435
 
911
1436
class TestSmartServerRepositoryUnlock(tests.TestCaseWithMemoryTransport):
912
1437
 
913
1438
    def setUp(self):
938
1463
            SmartServerResponse(('TokenMismatch',)), response)
939
1464
 
940
1465
 
941
 
class TestSmartServerRepositoryTarball(tests.TestCaseWithTransport):
942
 
 
943
 
    def test_repository_tarball(self):
944
 
        backing = self.get_transport()
945
 
        request = smart.repository.SmartServerRepositoryTarball(backing)
946
 
        repository = self.make_repository('.')
947
 
        # make some extraneous junk in the repository directory which should
948
 
        # not be copied
949
 
        self.build_tree(['.bzr/repository/extra-junk'])
950
 
        response = request.execute('', 'bz2')
951
 
        self.assertEqual(('ok',), response.args)
952
 
        # body should be a tbz2
953
 
        body_file = StringIO(response.body)
954
 
        body_tar = tarfile.open('body_tar.tbz2', fileobj=body_file,
955
 
            mode='r|bz2')
956
 
        # let's make sure there are some key repository components inside it.
957
 
        # the tarfile returns directories with trailing slashes...
958
 
        names = set([n.rstrip('/') for n in body_tar.getnames()])
959
 
        self.assertTrue('.bzr/repository/lock' in names)
960
 
        self.assertTrue('.bzr/repository/format' in names)
961
 
        self.assertTrue('.bzr/repository/extra-junk' not in names,
962
 
            "extraneous file present in tar file")
963
 
 
964
 
 
965
 
class TestSmartServerRepositoryStreamKnitData(tests.TestCaseWithMemoryTransport):
966
 
 
967
 
    def test_fetch_revisions(self):
968
 
        backing = self.get_transport()
969
 
        request = smart.repository.SmartServerRepositoryStreamKnitDataForRevisions(backing)
970
 
        tree = self.make_branch_and_memory_tree('.')
971
 
        tree.lock_write()
972
 
        tree.add('')
973
 
        rev_id1_utf8 = u'\xc8'.encode('utf-8')
974
 
        rev_id2_utf8 = u'\xc9'.encode('utf-8')
975
 
        r1 = tree.commit('1st commit', rev_id=rev_id1_utf8)
976
 
        r1 = tree.commit('2nd commit', rev_id=rev_id2_utf8)
977
 
        tree.unlock()
978
 
 
979
 
        response = request.execute('', rev_id2_utf8)
980
 
        self.assertEqual(('ok',), response.args)
981
 
        unpacker = pack.ContainerReader(StringIO(response.body))
982
 
        names = []
983
 
        for [name], read_bytes in unpacker.iter_records():
984
 
            names.append(name)
985
 
            bytes = read_bytes(None)
986
 
            # The bytes should be a valid bencoded string.
987
 
            bencode.bdecode(bytes)
988
 
            # XXX: assert that the bencoded knit records have the right
989
 
            # contents?
990
 
        
991
 
    def test_no_such_revision_error(self):
992
 
        backing = self.get_transport()
993
 
        request = smart.repository.SmartServerRepositoryStreamKnitDataForRevisions(backing)
994
 
        repo = self.make_repository('.')
995
 
        rev_id1_utf8 = u'\xc8'.encode('utf-8')
996
 
        response = request.execute('', rev_id1_utf8)
997
 
        self.assertEqual(
998
 
            SmartServerResponse(('NoSuchRevision', rev_id1_utf8)),
999
 
            response)
1000
 
 
1001
 
 
1002
 
class TestSmartServerRepositoryStreamRevisionsChunked(tests.TestCaseWithMemoryTransport):
1003
 
 
1004
 
    def test_fetch_revisions(self):
1005
 
        backing = self.get_transport()
1006
 
        request = smart.repository.SmartServerRepositoryStreamRevisionsChunked(
1007
 
            backing)
1008
 
        tree = self.make_branch_and_memory_tree('.')
1009
 
        tree.lock_write()
1010
 
        tree.add('')
1011
 
        rev_id1_utf8 = u'\xc8'.encode('utf-8')
1012
 
        rev_id2_utf8 = u'\xc9'.encode('utf-8')
1013
 
        tree.commit('1st commit', rev_id=rev_id1_utf8)
1014
 
        tree.commit('2nd commit', rev_id=rev_id2_utf8)
1015
 
        tree.unlock()
1016
 
 
1017
 
        response = request.execute('')
1018
 
        self.assertEqual(None, response)
1019
 
        response = request.do_body("%s\n%s\n1" % (rev_id2_utf8, rev_id1_utf8))
1020
 
        self.assertEqual(('ok',), response.args)
1021
 
        parser = pack.ContainerPushParser()
1022
 
        names = []
1023
 
        for stream_bytes in response.body_stream:
1024
 
            parser.accept_bytes(stream_bytes)
1025
 
            for [name], record_bytes in parser.read_pending_records():
1026
 
                names.append(name)
1027
 
                # The bytes should be a valid bencoded string.
1028
 
                bencode.bdecode(record_bytes)
1029
 
                # XXX: assert that the bencoded knit records have the right
1030
 
                # contents?
1031
 
        
1032
 
    def test_no_such_revision_error(self):
1033
 
        backing = self.get_transport()
1034
 
        request = smart.repository.SmartServerRepositoryStreamRevisionsChunked(
1035
 
            backing)
1036
 
        repo = self.make_repository('.')
1037
 
        rev_id1_utf8 = u'\xc8'.encode('utf-8')
1038
 
        response = request.execute('')
1039
 
        self.assertEqual(None, response)
1040
 
        response = request.do_body("%s\n\n1" % (rev_id1_utf8,))
1041
 
        self.assertEqual(
1042
 
            FailedSmartServerResponse(('NoSuchRevision', )),
1043
 
            response)
1044
 
 
1045
 
 
1046
1466
class TestSmartServerIsReadonly(tests.TestCaseWithMemoryTransport):
1047
1467
 
1048
1468
    def test_is_readonly_no(self):
1060
1480
            SmartServerResponse(('yes',)), response)
1061
1481
 
1062
1482
 
 
1483
class TestSmartServerRepositorySetMakeWorkingTrees(tests.TestCaseWithMemoryTransport):
 
1484
 
 
1485
    def test_set_false(self):
 
1486
        backing = self.get_transport()
 
1487
        repo = self.make_repository('.', shared=True)
 
1488
        repo.set_make_working_trees(True)
 
1489
        request_class = smart.repository.SmartServerRepositorySetMakeWorkingTrees
 
1490
        request = request_class(backing)
 
1491
        self.assertEqual(SuccessfulSmartServerResponse(('ok',)),
 
1492
            request.execute('', 'False'))
 
1493
        repo = repo.bzrdir.open_repository()
 
1494
        self.assertFalse(repo.make_working_trees())
 
1495
 
 
1496
    def test_set_true(self):
 
1497
        backing = self.get_transport()
 
1498
        repo = self.make_repository('.', shared=True)
 
1499
        repo.set_make_working_trees(False)
 
1500
        request_class = smart.repository.SmartServerRepositorySetMakeWorkingTrees
 
1501
        request = request_class(backing)
 
1502
        self.assertEqual(SuccessfulSmartServerResponse(('ok',)),
 
1503
            request.execute('', 'True'))
 
1504
        repo = repo.bzrdir.open_repository()
 
1505
        self.assertTrue(repo.make_working_trees())
 
1506
 
 
1507
 
 
1508
class TestSmartServerPackRepositoryAutopack(tests.TestCaseWithTransport):
 
1509
 
 
1510
    def make_repo_needing_autopacking(self, path='.'):
 
1511
        # Make a repo in need of autopacking.
 
1512
        tree = self.make_branch_and_tree('.', format='pack-0.92')
 
1513
        repo = tree.branch.repository
 
1514
        # monkey-patch the pack collection to disable autopacking
 
1515
        repo._pack_collection._max_pack_count = lambda count: count
 
1516
        for x in range(10):
 
1517
            tree.commit('commit %s' % x)
 
1518
        self.assertEqual(10, len(repo._pack_collection.names()))
 
1519
        del repo._pack_collection._max_pack_count
 
1520
        return repo
 
1521
 
 
1522
    def test_autopack_needed(self):
 
1523
        repo = self.make_repo_needing_autopacking()
 
1524
        repo.lock_write()
 
1525
        self.addCleanup(repo.unlock)
 
1526
        backing = self.get_transport()
 
1527
        request = smart.packrepository.SmartServerPackRepositoryAutopack(
 
1528
            backing)
 
1529
        response = request.execute('')
 
1530
        self.assertEqual(SmartServerResponse(('ok',)), response)
 
1531
        repo._pack_collection.reload_pack_names()
 
1532
        self.assertEqual(1, len(repo._pack_collection.names()))
 
1533
 
 
1534
    def test_autopack_not_needed(self):
 
1535
        tree = self.make_branch_and_tree('.', format='pack-0.92')
 
1536
        repo = tree.branch.repository
 
1537
        repo.lock_write()
 
1538
        self.addCleanup(repo.unlock)
 
1539
        for x in range(9):
 
1540
            tree.commit('commit %s' % x)
 
1541
        backing = self.get_transport()
 
1542
        request = smart.packrepository.SmartServerPackRepositoryAutopack(
 
1543
            backing)
 
1544
        response = request.execute('')
 
1545
        self.assertEqual(SmartServerResponse(('ok',)), response)
 
1546
        repo._pack_collection.reload_pack_names()
 
1547
        self.assertEqual(9, len(repo._pack_collection.names()))
 
1548
 
 
1549
    def test_autopack_on_nonpack_format(self):
 
1550
        """A request to autopack a non-pack repo is a no-op."""
 
1551
        repo = self.make_repository('.', format='knit')
 
1552
        backing = self.get_transport()
 
1553
        request = smart.packrepository.SmartServerPackRepositoryAutopack(
 
1554
            backing)
 
1555
        response = request.execute('')
 
1556
        self.assertEqual(SmartServerResponse(('ok',)), response)
 
1557
 
 
1558
 
1063
1559
class TestHandlers(tests.TestCase):
1064
1560
    """Tests for the request.request_handlers object."""
1065
1561
 
 
1562
    def test_all_registrations_exist(self):
 
1563
        """All registered request_handlers can be found."""
 
1564
        # If there's a typo in a register_lazy call, this loop will fail with
 
1565
        # an AttributeError.
 
1566
        for key, item in smart.request.request_handlers.iteritems():
 
1567
            pass
 
1568
 
 
1569
    def assertHandlerEqual(self, verb, handler):
 
1570
        self.assertEqual(smart.request.request_handlers.get(verb), handler)
 
1571
 
1066
1572
    def test_registered_methods(self):
1067
1573
        """Test that known methods are registered to the correct object."""
1068
 
        self.assertEqual(
1069
 
            smart.request.request_handlers.get('Branch.get_config_file'),
 
1574
        self.assertHandlerEqual('Branch.get_config_file',
1070
1575
            smart.branch.SmartServerBranchGetConfigFile)
1071
 
        self.assertEqual(
1072
 
            smart.request.request_handlers.get('Branch.lock_write'),
 
1576
        self.assertHandlerEqual('Branch.get_parent',
 
1577
            smart.branch.SmartServerBranchGetParent)
 
1578
        self.assertHandlerEqual('Branch.get_tags_bytes',
 
1579
            smart.branch.SmartServerBranchGetTagsBytes)
 
1580
        self.assertHandlerEqual('Branch.lock_write',
1073
1581
            smart.branch.SmartServerBranchRequestLockWrite)
1074
 
        self.assertEqual(
1075
 
            smart.request.request_handlers.get('Branch.last_revision_info'),
 
1582
        self.assertHandlerEqual('Branch.last_revision_info',
1076
1583
            smart.branch.SmartServerBranchRequestLastRevisionInfo)
1077
 
        self.assertEqual(
1078
 
            smart.request.request_handlers.get('Branch.revision_history'),
 
1584
        self.assertHandlerEqual('Branch.revision_history',
1079
1585
            smart.branch.SmartServerRequestRevisionHistory)
1080
 
        self.assertEqual(
1081
 
            smart.request.request_handlers.get('Branch.set_last_revision'),
 
1586
        self.assertHandlerEqual('Branch.set_config_option',
 
1587
            smart.branch.SmartServerBranchRequestSetConfigOption)
 
1588
        self.assertHandlerEqual('Branch.set_last_revision',
1082
1589
            smart.branch.SmartServerBranchRequestSetLastRevision)
1083
 
        self.assertEqual(
1084
 
            smart.request.request_handlers.get('Branch.set_last_revision_info'),
 
1590
        self.assertHandlerEqual('Branch.set_last_revision_info',
1085
1591
            smart.branch.SmartServerBranchRequestSetLastRevisionInfo)
1086
 
        self.assertEqual(
1087
 
            smart.request.request_handlers.get('Branch.unlock'),
 
1592
        self.assertHandlerEqual('Branch.set_last_revision_ex',
 
1593
            smart.branch.SmartServerBranchRequestSetLastRevisionEx)
 
1594
        self.assertHandlerEqual('Branch.set_parent_location',
 
1595
            smart.branch.SmartServerBranchRequestSetParentLocation)
 
1596
        self.assertHandlerEqual('Branch.unlock',
1088
1597
            smart.branch.SmartServerBranchRequestUnlock)
1089
 
        self.assertEqual(
1090
 
            smart.request.request_handlers.get('BzrDir.find_repository'),
 
1598
        self.assertHandlerEqual('BzrDir.find_repository',
1091
1599
            smart.bzrdir.SmartServerRequestFindRepositoryV1)
1092
 
        self.assertEqual(
1093
 
            smart.request.request_handlers.get('BzrDir.find_repositoryV2'),
 
1600
        self.assertHandlerEqual('BzrDir.find_repositoryV2',
1094
1601
            smart.bzrdir.SmartServerRequestFindRepositoryV2)
1095
 
        self.assertEqual(
1096
 
            smart.request.request_handlers.get('BzrDirFormat.initialize'),
 
1602
        self.assertHandlerEqual('BzrDirFormat.initialize',
1097
1603
            smart.bzrdir.SmartServerRequestInitializeBzrDir)
1098
 
        self.assertEqual(
1099
 
            smart.request.request_handlers.get('BzrDir.open_branch'),
 
1604
        self.assertHandlerEqual('BzrDirFormat.initialize_ex_1.16',
 
1605
            smart.bzrdir.SmartServerRequestBzrDirInitializeEx)
 
1606
        self.assertHandlerEqual('BzrDir.cloning_metadir',
 
1607
            smart.bzrdir.SmartServerBzrDirRequestCloningMetaDir)
 
1608
        self.assertHandlerEqual('BzrDir.get_config_file',
 
1609
            smart.bzrdir.SmartServerBzrDirRequestConfigFile)
 
1610
        self.assertHandlerEqual('BzrDir.open_branch',
1100
1611
            smart.bzrdir.SmartServerRequestOpenBranch)
1101
 
        self.assertEqual(
1102
 
            smart.request.request_handlers.get('Repository.gather_stats'),
 
1612
        self.assertHandlerEqual('BzrDir.open_branchV2',
 
1613
            smart.bzrdir.SmartServerRequestOpenBranchV2)
 
1614
        self.assertHandlerEqual('PackRepository.autopack',
 
1615
            smart.packrepository.SmartServerPackRepositoryAutopack)
 
1616
        self.assertHandlerEqual('Repository.gather_stats',
1103
1617
            smart.repository.SmartServerRepositoryGatherStats)
1104
 
        self.assertEqual(
1105
 
            smart.request.request_handlers.get('Repository.get_parent_map'),
 
1618
        self.assertHandlerEqual('Repository.get_parent_map',
1106
1619
            smart.repository.SmartServerRepositoryGetParentMap)
1107
 
        self.assertEqual(
1108
 
            smart.request.request_handlers.get(
1109
 
                'Repository.get_revision_graph'),
 
1620
        self.assertHandlerEqual('Repository.get_rev_id_for_revno',
 
1621
            smart.repository.SmartServerRepositoryGetRevIdForRevno)
 
1622
        self.assertHandlerEqual('Repository.get_revision_graph',
1110
1623
            smart.repository.SmartServerRepositoryGetRevisionGraph)
1111
 
        self.assertEqual(
1112
 
            smart.request.request_handlers.get('Repository.has_revision'),
 
1624
        self.assertHandlerEqual('Repository.get_stream',
 
1625
            smart.repository.SmartServerRepositoryGetStream)
 
1626
        self.assertHandlerEqual('Repository.has_revision',
1113
1627
            smart.repository.SmartServerRequestHasRevision)
1114
 
        self.assertEqual(
1115
 
            smart.request.request_handlers.get('Repository.is_shared'),
 
1628
        self.assertHandlerEqual('Repository.insert_stream',
 
1629
            smart.repository.SmartServerRepositoryInsertStream)
 
1630
        self.assertHandlerEqual('Repository.insert_stream_locked',
 
1631
            smart.repository.SmartServerRepositoryInsertStreamLocked)
 
1632
        self.assertHandlerEqual('Repository.is_shared',
1116
1633
            smart.repository.SmartServerRepositoryIsShared)
1117
 
        self.assertEqual(
1118
 
            smart.request.request_handlers.get('Repository.lock_write'),
 
1634
        self.assertHandlerEqual('Repository.lock_write',
1119
1635
            smart.repository.SmartServerRepositoryLockWrite)
1120
 
        self.assertEqual(
1121
 
            smart.request.request_handlers.get('Repository.tarball'),
 
1636
        self.assertHandlerEqual('Repository.tarball',
1122
1637
            smart.repository.SmartServerRepositoryTarball)
1123
 
        self.assertEqual(
1124
 
            smart.request.request_handlers.get('Repository.unlock'),
 
1638
        self.assertHandlerEqual('Repository.unlock',
1125
1639
            smart.repository.SmartServerRepositoryUnlock)
1126
 
        self.assertEqual(
1127
 
            smart.request.request_handlers.get('Transport.is_readonly'),
 
1640
        self.assertHandlerEqual('Transport.is_readonly',
1128
1641
            smart.request.SmartServerIsReadonly)