/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/selftest/testfetch.py

  • Committer: Martin Pool
  • Date: 2005-08-29 04:05:28 UTC
  • Revision ID: mbp@sourcefrog.net-20050829040528-3fb016af15c4e4c0
- smart_add routine now doesn't print filenames directly, but rather
  returns them to the caller for appropriate display.  the add command
  displays them in the same way as before.

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