← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/fetch.py

  • Committer: Martin Pool
  • Date: 2005-08-23 00:40:27 UTC
  • Revision ID: mbp@sourcefrog.net-20050823004027-e6e45a392f540bb4
todo

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/fetch.py
14
14
# along with this program; if not, write to the Free Software
15
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
16
16
import bzrlib.errors
17
 
from bzrlib.trace import mutter, note
 
17
from bzrlib.selftest.testrevision import make_branches
 
18
from bzrlib.trace import mutter
18
19
from bzrlib.branch import Branch
19
20
from bzrlib.progress import ProgressBar
20
21
import sys
21
22
import os
22
23
 
23
 
def has_revision(branch, revision_id):
24
 
    try:
25
 
        branch.get_revision_xml(revision_id)
26
 
        return True
27
 
    except bzrlib.errors.NoSuchRevision:
28
 
        return False
29
 
 
30
 
 
31
 
def greedy_fetch(to_branch, from_branch, revision=None, pb=None):
32
 
    """Copy a revision and all available ancestors from one branch to another
33
 
    If no revision is specified, uses the last revision in the source branch's
34
 
    revision history.
35
 
    """
 
24
def greedy_fetch(from_branch, to_branch, last_revision=None):
36
25
    from_history = from_branch.revision_history()
37
 
    required_revisions = set(from_history)
38
 
    all_failed = set()
39
 
    if revision is not None:
40
 
        required_revisions.add(revision)
41
 
        try:
42
 
            rev_index = from_history.index(revision)
43
 
        except ValueError:
44
 
            rev_index = None
45
 
        if rev_index is not None:
46
 
            from_history = from_history[:rev_index + 1]
47
 
        else:
48
 
            from_history = [revision]
 
26
    if last_revision is not None:
 
27
        from_history = from_history[:from_history.index(last_revision)+1]
49
28
    to_history = to_branch.revision_history()
50
29
    missing = []
51
30
    for rev_id in from_history:
52
31
        if not has_revision(to_branch, rev_id):
53
32
            missing.append(rev_id)
54
 
    
55
 
    count = 0
 
33
 
56
34
    while len(missing) > 0:
57
 
        installed, failed = to_branch.install_revisions(from_branch, 
58
 
                                                        revision_ids=missing,
59
 
                                                        pb=pb)
60
 
        count += installed
61
 
        required_failed = failed.intersection(required_revisions)
62
 
        if len(required_failed) > 0:
63
 
            raise bzrlib.errors.InstallFailed(required_failed)
64
 
        for rev_id in failed:
65
 
            note("Failed to install %s" % rev_id)
66
 
        all_failed.update(failed)
67
 
        new_missing = set() 
 
35
        to_branch.install_revisions(from_branch, revision_ids=missing)
 
36
        new_missing = []
68
37
        for rev_id in missing:
69
38
            try:
70
39
                revision = from_branch.get_revision(rev_id)
75
44
                    continue
76
45
            for parent in [p.revision_id for p in revision.parents]:
77
46
                if not has_revision(to_branch, parent):
78
 
                    new_missing.add(parent)
 
47
                    new_missing.append(parent)
79
48
        missing = new_missing
80
 
    return count, all_failed
81
 
 
82
 
 
 
49
 
 
50
 
 
51
from testsweet import InTempDir
 
52
def has_revision(branch, revision_id):
 
53
    try:
 
54
        branch.get_revision_xml(revision_id)
 
55
        return True
 
56
    except bzrlib.errors.NoSuchRevision:
 
57
        return False
 
58
 
 
59
class TestFetch(InTempDir):
 
60
    def runTest(self):
 
61
        def new_branch(name):
 
62
            os.mkdir(name)
 
63
            return Branch(name, init=True)
 
64
            
 
65
        #highest indices a: 3, b: 4
 
66
        br_a, br_b = make_branches()
 
67
        assert not has_revision(br_b, br_a.revision_history()[3])
 
68
        assert has_revision(br_b, br_a.revision_history()[2])
 
69
        assert len(br_b.revision_history()) == 5
 
70
        greedy_fetch(br_a, br_b, br_a.revision_history()[2])
 
71
 
 
72
        # greedy_fetch is not supposed to alter the revision history
 
73
        assert len(br_b.revision_history()) == 5
 
74
        assert not has_revision(br_b, br_a.revision_history()[3])
 
75
 
 
76
        assert len(br_b.revision_history()) == 5
 
77
        greedy_fetch(br_a, br_b, br_a.revision_history()[3])
 
78
        assert has_revision(br_b, br_a.revision_history()[3])
 
79
        assert not has_revision(br_a, br_b.revision_history()[3])
 
80
        assert not has_revision(br_a, br_b.revision_history()[4])
 
81
        greedy_fetch(br_b, br_a)
 
82
        assert has_revision(br_a, br_b.revision_history()[3])
 
83
        assert has_revision(br_a, br_b.revision_history()[4])
 
84
 
 
85
        br_b2 = new_branch('br_b2')
 
86
        greedy_fetch(br_b, br_b2)
 
87
        assert has_revision(br_b2, br_b.revision_history()[4])
 
88
        assert has_revision(br_b2, br_a.revision_history()[2])
 
89
        assert not has_revision(br_b2, br_a.revision_history()[3])
 
90
 
 
91
        br_a2 = new_branch('br_a2')
 
92
        greedy_fetch(br_a, br_a2)
 
93
        assert has_revision(br_a2, br_b.revision_history()[4])
 
94
        assert has_revision(br_a2, br_a.revision_history()[3])
 
95
 
 
96
 
 
97
 
 
98
if __name__ == '__main__':
 
99
    import sys
 
100
    sys.exit(run_suite(unittest.makeSuite()))