← 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-24 08:59:32 UTC
  • Revision ID: mbp@sourcefrog.net-20050824085932-c61f1f1f1c930e13
- Add a simple UIFactory 

  The idea of this is to let a client of bzrlib set some 
  policy about how output is displayed.

  In this revision all that's done is that progress bars
  are constructed by a policy established by the application
  rather than being randomly constructed in the library 
  or passed down the calls.  This avoids progress bars
  popping up while running the test suite and cleans up
  some code.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/fetch.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
def greedy_fetch(from_branch, to_branch, last_revision=None):
 
24
    from_history = from_branch.revision_history()
 
25
    if last_revision is not None:
 
26
        from_history = from_history[:from_history.index(last_revision)+1]
 
27
    to_history = to_branch.revision_history()
 
28
    missing = []
 
29
    for rev_id in from_history:
 
30
        if not has_revision(to_branch, rev_id):
 
31
            missing.append(rev_id)
 
32
 
 
33
    while len(missing) > 0:
 
34
        to_branch.install_revisions(from_branch, revision_ids=missing)
 
35
        new_missing = []
 
36
        for rev_id in missing:
 
37
            try:
 
38
                revision = from_branch.get_revision(rev_id)
 
39
            except bzrlib.errors.NoSuchRevision:
 
40
                if revision in from_history:
 
41
                    raise
 
42
                else:
 
43
                    continue
 
44
            for parent in [p.revision_id for p in revision.parents]:
 
45
                if not has_revision(to_branch, parent):
 
46
                    new_missing.append(parent)
 
47
        missing = new_missing
 
48
 
 
49
 
 
50
from testsweet import InTempDir
 
51
def has_revision(branch, revision_id):
 
52
    try:
 
53
        branch.get_revision_xml(revision_id)
 
54
        return True
 
55
    except bzrlib.errors.NoSuchRevision:
 
56
        return False
 
57
 
 
58
class TestFetch(InTempDir):
 
59
    def runTest(self):
 
60
        def new_branch(name):
 
61
            os.mkdir(name)
 
62
            return Branch(name, init=True)
 
63
            
 
64
        #highest indices a: 3, b: 4
 
65
        br_a, br_b = make_branches()
 
66
        assert not has_revision(br_b, br_a.revision_history()[3])
 
67
        assert has_revision(br_b, br_a.revision_history()[2])
 
68
        assert len(br_b.revision_history()) == 5
 
69
        greedy_fetch(br_a, br_b, br_a.revision_history()[2])
 
70
 
 
71
        # greedy_fetch is not supposed to alter the revision history
 
72
        assert len(br_b.revision_history()) == 5
 
73
        assert not has_revision(br_b, br_a.revision_history()[3])
 
74
 
 
75
        assert len(br_b.revision_history()) == 5
 
76
        greedy_fetch(br_a, br_b, br_a.revision_history()[3])
 
77
        assert has_revision(br_b, br_a.revision_history()[3])
 
78
        assert not has_revision(br_a, br_b.revision_history()[3])
 
79
        assert not has_revision(br_a, br_b.revision_history()[4])
 
80
        greedy_fetch(br_b, br_a)
 
81
        assert has_revision(br_a, br_b.revision_history()[3])
 
82
        assert has_revision(br_a, br_b.revision_history()[4])
 
83
 
 
84
        br_b2 = new_branch('br_b2')
 
85
        greedy_fetch(br_b, br_b2)
 
86
        assert has_revision(br_b2, br_b.revision_history()[4])
 
87
        assert has_revision(br_b2, br_a.revision_history()[2])
 
88
        assert not has_revision(br_b2, br_a.revision_history()[3])
 
89
 
 
90
        br_a2 = new_branch('br_a2')
 
91
        greedy_fetch(br_a, br_a2)
 
92
        assert has_revision(br_a2, br_b.revision_history()[4])
 
93
        assert has_revision(br_a2, br_a.revision_history()[3])
 
94
 
 
95
 
 
96
 
 
97
if __name__ == '__main__':
 
98
    import sys
 
99
    sys.exit(run_suite(unittest.makeSuite()))