← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/selftest/testfetch.py

First programmatic generation of completions.

The list of commands is generated, as is the list of possible switches for
each command. Commands requiring arguments aren't treated specially yet.

For every option name there is a list of switches, and if any switch for the
option is given, then the others will be suppressed.  This might be
incorrect in some cases, but closely mimics the old, static function.  The
list of these switch suppressions might be incomplete, as some switches
might have different alternative forms for different commands, in which
cases only the intersection of all alternatives gets suppressed.  A future
version might do this suppression based on the current command, or drop it
completely.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/selftest/testfetch.py
1
 
# Copyright (C) 2005 by Canonical Ltd
2
 
 
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.
7
 
 
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.
12
 
 
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
16
 
import bzrlib.errors
17
 
from bzrlib.selftest.testrevision import make_branches
18
 
from bzrlib.trace import mutter
19
 
from bzrlib.branch import Branch
20
 
import sys
21
 
import os
22
 
 
23
 
from bzrlib.selftest import TestCaseInTempDir
24
 
        
25
 
 
26
 
class TestFetch(TestCaseInTempDir):
27
 
    def runTest(self):
28
 
        from bzrlib.fetch import greedy_fetch, has_revision
29
 
 
30
 
        def new_branch(name):
31
 
            os.mkdir(name)
32
 
            return Branch(name, init=True)
33
 
            
34
 
        #highest indices a: 5, b: 7
35
 
        br_a, br_b = make_branches()
36
 
        assert not has_revision(br_b, br_a.revision_history()[3])
37
 
        assert has_revision(br_b, br_a.revision_history()[2])
38
 
        assert len(br_b.revision_history()) == 7
39
 
        assert greedy_fetch(br_b, br_a, br_a.revision_history()[2])[0] == 0
40
 
 
41
 
        # greedy_fetch is not supposed to alter the revision history
42
 
        assert len(br_b.revision_history()) == 7
43
 
        assert not has_revision(br_b, br_a.revision_history()[3])
44
 
 
45
 
        assert len(br_b.revision_history()) == 7
46
 
        assert greedy_fetch(br_b, br_a, br_a.revision_history()[3])[0] == 1
47
 
        assert has_revision(br_b, br_a.revision_history()[3])
48
 
        assert not has_revision(br_a, br_b.revision_history()[3])
49
 
        assert not has_revision(br_a, br_b.revision_history()[4])
50
 
 
51
 
        # When a non-branch ancestor is missing, it should be a failure, not
52
 
        # exception
53
 
        br_a4 = new_branch('br_a4')
54
 
        count, failures = greedy_fetch(br_a4, br_a)
55
 
        assert count == 6
56
 
        assert failures == set((br_b.revision_history()[4],
57
 
                                br_b.revision_history()[5])) 
58
 
 
59
 
        assert greedy_fetch(br_a, br_b)[0] == 4
60
 
        assert has_revision(br_a, br_b.revision_history()[3])
61
 
        assert has_revision(br_a, br_b.revision_history()[4])
62
 
 
63
 
        br_b2 = new_branch('br_b2')
64
 
        assert greedy_fetch(br_b2, br_b)[0] == 7
65
 
        assert has_revision(br_b2, br_b.revision_history()[4])
66
 
        assert has_revision(br_b2, br_a.revision_history()[2])
67
 
        assert not has_revision(br_b2, br_a.revision_history()[3])
68
 
 
69
 
        br_a2 = new_branch('br_a2')
70
 
        assert greedy_fetch(br_a2, br_a)[0] == 9
71
 
        assert has_revision(br_a2, br_b.revision_history()[4])
72
 
        assert has_revision(br_a2, br_a.revision_history()[3])
73
 
 
74
 
        br_a3 = new_branch('br_a3')
75
 
        assert greedy_fetch(br_a3, br_a2)[0] == 0
76
 
        for revno in range(4):
77
 
            assert not has_revision(br_a3, br_a.revision_history()[revno])
78
 
        assert greedy_fetch(br_a3, br_a2, br_a.revision_history()[2])[0] == 3
79
 
        fetched = greedy_fetch(br_a3, br_a2, br_a.revision_history()[3])[0]
80
 
        assert fetched == 3, "fetched %d instead of 3" % fetched
81
 
        # InstallFailed should be raised if the branch is missing the revision
82
 
        # that was requested.
83
 
        self.assertRaises(bzrlib.errors.InstallFailed, greedy_fetch, br_a3,
84
 
                          br_a2, 'pizza')
85
 
        # InstallFailed should be raised if the branch is missing a revision
86
 
        # from its own revision history
87
 
        br_a2.append_revision('a-b-c')
88
 
        self.assertRaises(bzrlib.errors.InstallFailed, greedy_fetch, br_a3,
89
 
                          br_a2)
90
 
 
91
 
 
92
 
 
93
 
if __name__ == '__main__':
94
 
    import unittest
95
 
    sys.exit(unittest.run_suite(unittest.makeSuite()))