← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_fetch.py

  • Committer: John Arbash Meinel
  • Date: 2006-08-15 15:50:31 UTC
  • mto: This revision was merged to the branch mainline in revision 1927.
  • Revision ID: john@arbash-meinel.com-20060815155031-f1480d692d2cf9d2
There is no strict ordering file addition, other than directories are added before child files

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test_fetch.py
1
1
# Copyright (C) 2005 by Canonical Ltd
2
 
 
 
2
#
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
5
5
# the Free Software Foundation; either version 2 of the License, or
6
6
# (at your option) any later version.
7
 
 
 
7
#
8
8
# This program is distributed in the hope that it will be useful,
9
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11
11
# GNU General Public License for more details.
12
 
 
 
12
#
13
13
# You should have received a copy of the GNU General Public License
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
21
21
from bzrlib.bzrdir import BzrDir
22
22
from bzrlib.builtins import merge
23
23
import bzrlib.errors
24
 
from bzrlib.fetch import greedy_fetch
25
24
from bzrlib.tests import TestCaseWithTransport
26
25
from bzrlib.tests.HTTPTestUtil import TestCaseWithWebserver
27
26
from bzrlib.tests.test_revision import make_branches
40
39
    self.assertFalse(repo_b.has_revision(br_a.revision_history()[3]))
41
40
    self.assertTrue(repo_b.has_revision(br_a.revision_history()[2]))
42
41
    self.assertEquals(len(br_b.revision_history()), 7)
43
 
    self.assertEquals(greedy_fetch(br_b, br_a, br_a.revision_history()[2])[0], 0)
44
 
    # greedy_fetch is not supposed to alter the revision history
 
42
    self.assertEquals(br_b.fetch(br_a, br_a.revision_history()[2])[0], 0)
 
43
    # branch.fetch is not supposed to alter the revision history
45
44
    self.assertEquals(len(br_b.revision_history()), 7)
46
45
    self.assertFalse(repo_b.has_revision(br_a.revision_history()[3]))
47
46
 
48
47
    # fetching the next revision up in sample data copies one revision
49
 
    self.assertEquals(greedy_fetch(br_b, br_a, br_a.revision_history()[3])[0], 1)
 
48
    self.assertEquals(br_b.fetch(br_a, br_a.revision_history()[3])[0], 1)
50
49
    self.assertTrue(repo_b.has_revision(br_a.revision_history()[3]))
51
50
    self.assertFalse(has_revision(br_a, br_b.revision_history()[6]))
52
51
    self.assertTrue(br_a.repository.has_revision(br_b.revision_history()[5]))
54
53
    # When a non-branch ancestor is missing, it should be unlisted...
55
54
    # as its not reference from the inventory weave.
56
55
    br_b4 = self.make_branch('br_4')
57
 
    count, failures = greedy_fetch(br_b4, br_b)
 
56
    count, failures = br_b4.fetch(br_b)
58
57
    self.assertEqual(count, 7)
59
58
    self.assertEqual(failures, [])
60
59
 
61
 
    self.assertEqual(greedy_fetch(writable_a, br_b)[0], 1)
 
60
    self.assertEqual(writable_a.fetch(br_b)[0], 1)
62
61
    self.assertTrue(has_revision(br_a, br_b.revision_history()[3]))
63
62
    self.assertTrue(has_revision(br_a, br_b.revision_history()[4]))
64
63
        
65
64
    br_b2 = self.make_branch('br_b2')
66
 
    self.assertEquals(greedy_fetch(br_b2, br_b)[0], 7)
 
65
    self.assertEquals(br_b2.fetch(br_b)[0], 7)
67
66
    self.assertTrue(has_revision(br_b2, br_b.revision_history()[4]))
68
67
    self.assertTrue(has_revision(br_b2, br_a.revision_history()[2]))
69
68
    self.assertFalse(has_revision(br_b2, br_a.revision_history()[3]))
70
69
 
71
70
    br_a2 = self.make_branch('br_a2')
72
 
    self.assertEquals(greedy_fetch(br_a2, br_a)[0], 9)
 
71
    self.assertEquals(br_a2.fetch(br_a)[0], 9)
73
72
    self.assertTrue(has_revision(br_a2, br_b.revision_history()[4]))
74
73
    self.assertTrue(has_revision(br_a2, br_a.revision_history()[3]))
75
74
    self.assertTrue(has_revision(br_a2, br_a.revision_history()[2]))
77
76
    br_a3 = self.make_branch('br_a3')
78
77
    # pulling a branch with no revisions grabs nothing, regardless of 
79
78
    # whats in the inventory.
80
 
    self.assertEquals(greedy_fetch(br_a3, br_a2)[0], 0)
 
79
    self.assertEquals(br_a3.fetch(br_a2)[0], 0)
81
80
    for revno in range(4):
82
81
        self.assertFalse(
83
82
            br_a3.repository.has_revision(br_a.revision_history()[revno]))
84
 
    self.assertEqual(greedy_fetch(br_a3, br_a2, br_a.revision_history()[2])[0], 3)
 
83
    self.assertEqual(br_a3.fetch(br_a2, br_a.revision_history()[2])[0], 3)
85
84
    # pull the 3 revisions introduced by a@u-0-3
86
 
    fetched = greedy_fetch(br_a3, br_a2, br_a.revision_history()[3])[0]
 
85
    fetched = br_a3.fetch(br_a2, br_a.revision_history()[3])[0]
87
86
    self.assertEquals(fetched, 3, "fetched %d instead of 3" % fetched)
88
87
    # InstallFailed should be raised if the branch is missing the revision
89
88
    # that was requested.
90
 
    self.assertRaises(bzrlib.errors.InstallFailed, greedy_fetch, br_a3,
91
 
                      br_a2, 'pizza')
 
89
    self.assertRaises(bzrlib.errors.InstallFailed, br_a3.fetch, br_a2, 'pizza')
92
90
    # InstallFailed should be raised if the branch is missing a revision
93
91
    # from its own revision history
94
92
    br_a2.append_revision('a-b-c')
95
 
    self.assertRaises(bzrlib.errors.InstallFailed, greedy_fetch, br_a3,
96
 
                      br_a2)
 
93
    self.assertRaises(bzrlib.errors.InstallFailed, br_a3.fetch, br_a2)
 
94
 
 
95
    # TODO: jam 20051218 Branch should no longer allow append_revision for revisions
 
96
    #       which don't exist. So this test needs to be rewritten
 
97
    #       RBC 20060403 the way to do this is to uncommit the revision from the
 
98
    #           repository after the commit
 
99
 
97
100
    #TODO: test that fetch correctly does reweaving when needed. RBC 20051008
98
101
    # Note that this means - updating the weave when ghosts are filled in to 
99
102
    # add the right parents.
106
109
        br_a, br_b = make_branches(self)
107
110
        fetch_steps(self, br_a, br_b, br_a)
108
111
 
 
112
    def test_fetch_self(self):
 
113
        wt = self.make_branch_and_tree('br')
 
114
        self.assertEqual(wt.branch.fetch(wt.branch), (0, []))
 
115
 
109
116
 
110
117
class TestMergeFetch(TestCaseWithTransport):
111
118
 
189
196
        br_rem_a = Branch.open(self.get_readonly_url('branch1'))
190
197
        fetch_steps(self, br_rem_a, br_b, br_a)
191
198
 
 
199
    def _count_log_matches(self, target, logs):
 
200
        """Count the number of times the target file pattern was fetched in an http log"""
 
201
        log_pattern = '%s HTTP/1.1" 200 - "-" "bzr/%s' % \
 
202
            (target, bzrlib.__version__)
 
203
        c = 0
 
204
        for line in logs:
 
205
            # TODO: perhaps use a regexp instead so we can match more
 
206
            # precisely?
 
207
            if line.find(log_pattern) > -1:
 
208
                c += 1
 
209
        return c
 
210
 
192
211
    def test_weaves_are_retrieved_once(self):
193
212
        self.build_tree(("source/", "source/file", "target/"))
194
213
        wt = self.make_branch_and_tree('source')
199
218
        wt.commit("changed file")
200
219
        target = BzrDir.create_branch_and_repo("target/")
201
220
        source = Branch.open(self.get_readonly_url("source/"))
202
 
        self.assertEqual(greedy_fetch(target, source), (2, []))
 
221
        self.assertEqual(target.fetch(source), (2, []))
 
222
        log_pattern = '%%s HTTP/1.1" 200 - "-" "bzr/%s' % bzrlib.__version__
203
223
        # this is the path to the literal file. As format changes 
204
224
        # occur it needs to be updated. FIXME: ask the store for the
205
225
        # path.
206
 
        weave_suffix = 'weaves/ce/id.weave HTTP/1.1" 200 -'
207
 
        self.assertEqual(1,
208
 
            len([log for log in self.get_readonly_server().logs if log.endswith(weave_suffix)]))
209
 
        inventory_weave_suffix = 'inventory.weave HTTP/1.1" 200 -'
210
 
        self.assertEqual(1,
211
 
            len([log for log in self.get_readonly_server().logs if log.endswith(
212
 
                inventory_weave_suffix)]))
 
226
        self.log("web server logs are:")
 
227
        http_logs = self.get_readonly_server().logs
 
228
        self.log('\n'.join(http_logs))
 
229
        # unfortunately this log entry is branch format specific. We could 
 
230
        # factor out the 'what files does this format use' to a method on the 
 
231
        # repository, which would let us to this generically. RBC 20060419
 
232
        self.assertEqual(1, self._count_log_matches('/ce/id.kndx', http_logs))
 
233
        self.assertEqual(1, self._count_log_matches('/ce/id.knit', http_logs))
 
234
        self.assertEqual(1, self._count_log_matches('inventory.kndx', http_logs))
213
235
        # this r-h check test will prevent regressions, but it currently already 
214
236
        # passes, before the patch to cache-rh is applied :[
215
 
        revision_history_suffix = 'revision-history HTTP/1.1" 200 -'
216
 
        self.assertEqual(1,
217
 
            len([log for log in self.get_readonly_server().logs if log.endswith(
218
 
                revision_history_suffix)]))
 
237
        self.assertEqual(1, self._count_log_matches('revision-history', http_logs))
219
238
        # FIXME naughty poking in there.
220
239
        self.get_readonly_server().logs = []
221
240
        # check there is nothing more to fetch
222
241
        source = Branch.open(self.get_readonly_url("source/"))
223
 
        self.assertEqual(greedy_fetch(target, source), (0, []))
224
 
        self.failUnless(self.get_readonly_server().logs[0].endswith('branch-format HTTP/1.1" 200 -'))
225
 
        self.failUnless(self.get_readonly_server().logs[1].endswith('revision-history HTTP/1.1" 200 -'))
226
 
        self.assertEqual(2, len(self.get_readonly_server().logs))
 
242
        self.assertEqual(target.fetch(source), (0, []))
 
243
        # should make just two requests
 
244
        http_logs = self.get_readonly_server().logs
 
245
        self.log("web server logs are:")
 
246
        self.log('\n'.join(http_logs))
 
247
        self.assertEqual(1, self._count_log_matches('branch-format', http_logs))
 
248
        self.assertEqual(1, self._count_log_matches('branch/format', http_logs))
 
249
        self.assertEqual(1, self._count_log_matches('repository/format', http_logs))
 
250
        self.assertEqual(1, self._count_log_matches('revision-history', http_logs))
 
251
        self.assertEqual(4, len(http_logs))