/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_remote.py

  • Committer: Andrew Bennetts
  • Date: 2008-04-08 06:38:34 UTC
  • mfrom: (2892.2.1 smart-set-last-revision)
  • mto: This revision was merged to the branch mainline in revision 3355.
  • Revision ID: andrew.bennetts@canonical.com-20080408063834-o4mid7woclibs6yj
Merge 'Add Branch.set_last_revision_info smart method'.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test_remote.py
494
494
        branch.unlock()
495
495
 
496
496
 
 
497
class TestBranchSetLastRevisionInfo(tests.TestCase):
 
498
 
 
499
    def test_set_empty(self):
 
500
        # set_last_revision_info(num, 'rev-id') is translated to calling
 
501
        # Branch.set_last_revision_info(num, 'rev-id') on the wire.
 
502
        transport = MemoryTransport()
 
503
        transport.mkdir('branch')
 
504
        transport = transport.clone('branch')
 
505
        client = FakeClient([
 
506
            # lock_write
 
507
            (('ok', 'branch token', 'repo token'), ),
 
508
            # set_last_revision_info
 
509
            (('ok',), ),
 
510
            # unlock
 
511
            (('ok',), )], transport.base)
 
512
 
 
513
        bzrdir = RemoteBzrDir(transport, _client=False)
 
514
        branch = RemoteBranch(bzrdir, None, _client=client)
 
515
        # This is a hack to work around the problem that RemoteBranch currently
 
516
        # unnecessarily invokes _ensure_real upon a call to lock_write.
 
517
        branch._ensure_real = lambda: None
 
518
        # Lock the branch, reset the record of remote calls.
 
519
        branch.lock_write()
 
520
        client._calls = []
 
521
        result = branch.set_last_revision_info(1234, 'a-revision-id')
 
522
        self.assertEqual(
 
523
            [('call', 'Branch.set_last_revision_info',
 
524
                ('branch/', 'branch token', 'repo token',
 
525
                 '1234', 'a-revision-id'))],
 
526
            client._calls)
 
527
        self.assertEqual(None, result)
 
528
 
 
529
    def test_no_such_revision(self):
 
530
        # A response of 'NoSuchRevision' is translated into an exception.
 
531
        client = FakeClient([
 
532
            # lock_write
 
533
            (('ok', 'branch token', 'repo token'), ),
 
534
            # set_last_revision_info
 
535
            (('NoSuchRevision', 'revid'), ),
 
536
            # unlock
 
537
            (('ok',), ),
 
538
            ])
 
539
        transport = MemoryTransport()
 
540
        transport.mkdir('branch')
 
541
        transport = transport.clone('branch')
 
542
 
 
543
        bzrdir = RemoteBzrDir(transport, _client=False)
 
544
        branch = RemoteBranch(bzrdir, None, _client=client)
 
545
        # This is a hack to work around the problem that RemoteBranch currently
 
546
        # unnecessarily invokes _ensure_real upon a call to lock_write.
 
547
        branch._ensure_real = lambda: None
 
548
        # Lock the branch, reset the record of remote calls.
 
549
        branch.lock_write()
 
550
        client._calls = []
 
551
 
 
552
        self.assertRaises(
 
553
            errors.NoSuchRevision, branch.set_last_revision_info, 123, 'revid')
 
554
        branch.unlock()
 
555
 
 
556
 
497
557
class TestBranchControlGetBranchConf(tests.TestCaseWithMemoryTransport):
498
558
    """Test branch.control_files api munging...
499
559