/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/benchmarks/bench_bundle.py

Merge from remote-transport

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/benchmarks/bench_bundle.py
1
 
# Copyright (C) 2006 by Canonical Ltd
 
1
# Copyright (C) 2006 Canonical Ltd
2
2
#
3
3
# This program is free software; you can redistribute it and/or modify
4
 
# it under the terms of the GNU General Public License version 2 as published by
5
 
# the Free Software Foundation.
 
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.
6
7
#
7
8
# This program is distributed in the hope that it will be useful,
8
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
23
24
from bzrlib.add import smart_add
24
25
from bzrlib.benchmarks import Benchmark
25
26
from bzrlib.branch import Branch
26
 
from bzrlib.bundle import read_bundle
27
27
from bzrlib.bundle.apply_bundle import install_bundle
28
 
from bzrlib.bundle.serializer import write_bundle
 
28
from bzrlib.bundle.serializer import read_bundle, write_bundle
29
29
from bzrlib.revision import NULL_REVISION
30
30
from bzrlib.revisionspec import RevisionSpec
31
31
from bzrlib.workingtree import WorkingTree
68
68
        os.chdir('branch_a')
69
69
        self.time(self.run_bzr, 'merge', '../bundle')
70
70
 
 
71
 
71
72
class BundleLibraryLevelWriteBenchmark(Benchmark):
72
73
    """ Benchmarks for the write_bundle library function. """
73
74
 
74
75
    def _time_read_write(self):
75
 
        print "timing"
76
76
        branch, relpath = Branch.open_containing("a")
77
77
        revision_history = branch.revision_history()
78
78
        bundle_text = StringIO()
79
 
        print "starting write bundle"
80
79
        self.time(write_bundle, branch.repository, revision_history[-1],
81
80
                  NULL_REVISION, bundle_text)
82
 
        print "stopped writing bundle"
83
81
        bundle_text.seek(0)
84
82
        target_tree = self.make_branch_and_tree('b')
85
 
        print "starting reading bundle"
86
83
        bundle = self.time(read_bundle, bundle_text)
87
 
        print "starting installing bundle"
88
84
        self.time(install_bundle, target_tree.branch.repository, bundle)
89
85
 
90
86
    def test_few_files_small_tree_1_revision(self):
99
95
        self.commit_some_revisions(tree, files[:5], 100, 1)
100
96
        self._time_read_write()
101
97
 
102
 
    def test_few_files_small_tree_1000_revision(self):
103
 
        os.mkdir("a")
104
 
        tree, files = self.create_with_commits(5, 1, directory_name="a")
105
 
        self.commit_some_revisions(tree, files[:5], 1000, 1)
106
 
        self._time_read_write()
107
 
 
108
98
    def test_few_files_moderate_tree_1_revision(self):
109
99
        os.mkdir("a")
110
100
        tree, files = self.create_with_commits(100, 1, directory_name="a")
117
107
        self.commit_some_revisions(tree, files[:5], 100, 1)
118
108
        self._time_read_write()
119
109
 
120
 
    def test_few_files_moderate_tree_1000_revision(self):
121
 
        os.mkdir("a")
122
 
        tree, files = self.create_with_commits(100, 1, directory_name="a")
123
 
        self.commit_some_revisions(tree, files[:5], 1000, 1)
124
 
        self._time_read_write()
125
 
 
126
110
    def test_some_files_moderate_tree_1_revision(self):
127
111
        os.mkdir("a")
128
112
        tree, files = self.create_with_commits(100, 1, directory_name="a")
129
113
        self.commit_some_revisions(tree, files[:100], 1, 1)
130
114
        self._time_read_write()
131
115
 
132
 
    def test_some_files_moderate_tree_100_revision(self):
133
 
        os.mkdir("a")
134
 
        tree, files = self.create_with_commits(100, 1, directory_name="a")
135
 
        self.commit_some_revisions(tree, files[:100], 100, 1)
136
 
        self._time_read_write()
137
 
 
138
 
    def test_some_files_moderate_tree_1000_revision(self):
139
 
        os.mkdir("a")
140
 
        tree, files = self.create_with_commits(100, 1, directory_name="a")
141
 
        self.commit_some_revisions(tree, files[:100], 1000, 1)
142
 
        self._time_read_write()
143
 
 
144
116
    def test_few_files_big_tree_1_revision(self):
145
117
        os.mkdir("a")
146
118
        tree, files = self.create_with_commits(1000, 1, directory_name="a")
147
119
        self.commit_some_revisions(tree, files[:5], 1, 1)
148
120
        self._time_read_write()
149
121
 
150
 
    def test_few_files_big_tree_100_revision(self):
151
 
        os.mkdir("a")
152
 
        tree, files = self.create_with_commits(1000, 1, directory_name="a")
153
 
        self.commit_some_revisions(tree, files[:5], 100, 1)
154
 
        self._time_read_write()
155
 
 
156
 
    def test_few_files_big_tree_1000_revision(self):
157
 
        os.mkdir("a")
158
 
        tree, files = self.create_with_commits(1000, 1, directory_name="a")
159
 
        self.commit_some_revisions(tree, files[:5], 1000, 1)
160
 
        self._time_read_write()
161
 
 
162
122
    def test_some_files_big_tree_1_revision(self):
163
123
        os.mkdir("a")
164
124
        tree, files = self.create_with_commits(1000, 1, directory_name="a")
165
125
        self.commit_some_revisions(tree, files[:100], 1, 1)
166
126
        self._time_read_write()
167
127
 
168
 
    def test_some_files_big_tree_100_revision(self):
169
 
        os.mkdir("a")
170
 
        tree, files = self.create_with_commits(1000, 1, directory_name="a")
171
 
        self.commit_some_revisions(tree, files[:100], 100, 1)
172
 
        self._time_read_write()
173
 
 
174
 
    def test_some_files_big_tree_1000_revision(self):
175
 
        os.mkdir("a")
176
 
        tree, files = self.create_with_commits(1000, 1, directory_name="a")
177
 
        self.commit_some_revisions(tree, files[:100], 1000, 1)
178
 
        self._time_read_write()
179
 
 
180
 
    def test_many_files_big_tree_1_revision(self):
181
 
        os.mkdir("a")
182
 
        tree, files = self.create_with_commits(1000, 1, directory_name="a")
183
 
        self.commit_some_revisions(tree, files[:1000], 1, 1)
184
 
        self._time_read_write()
185
 
 
186
 
    def test_many_files_big_tree_100_revision(self):
187
 
        os.mkdir("a")
188
 
        tree, files = self.create_with_commits(1000, 1, directory_name="a")
189
 
        self.commit_some_revisions(tree, files[:1000], 100, 1)
190
 
        self._time_read_write()
191
 
 
192
 
    def test_many_files_big_tree_1000_revision(self):
193
 
        os.mkdir("a")
194
 
        tree, files = self.create_with_commits(1000, 1, directory_name="a")
195
 
        self.commit_some_revisions(tree, files[:1000], 1000, 1)
196
 
        self._time_read_write()
197
128
 
198
129
class BundleLibraryLevelInstallBenchmark(Benchmark):
199
130
    """ Benchmarks for the install_bundle library function. """
239
170
        self.commit_some_revisions(tree, files[:100], 1, 1)
240
171
        self._time_read_write()
241
172
 
242
 
    def test_some_files_moderate_tree_100_revision(self):
243
 
        os.mkdir("a")
244
 
        tree, files = self.create_with_commits(100, 1, directory_name="a")
245
 
        self.commit_some_revisions(tree, files[:100], 100, 1)
246
 
        self._time_read_write()
247
 
 
248
173
    def test_few_files_big_tree_1_revision(self):
249
174
        os.mkdir("a")
250
175
        tree, files = self.create_with_commits(1000, 1, directory_name="a")
251
176
        self.commit_some_revisions(tree, files[:5], 1, 1)
252
177
        self._time_read_write()
253
178
 
254
 
    def test_few_files_big_tree_100_revision(self):
255
 
        os.mkdir("a")
256
 
        tree, files = self.create_with_commits(1000, 1, directory_name="a")
257
 
        self.commit_some_revisions(tree, files[:5], 100, 1)
258
 
        self._time_read_write()
259
 
 
260
179
    def test_some_files_big_tree_1_revision(self):
261
180
        os.mkdir("a")
262
181
        tree, files = self.create_with_commits(1000, 1, directory_name="a")
263
182
        self.commit_some_revisions(tree, files[:100], 1, 1)
264
183
        self._time_read_write()
265
184
 
266
 
    def test_some_files_big_tree_100_revision(self):
267
 
        os.mkdir("a")
268
 
        tree, files = self.create_with_commits(1000, 1, directory_name="a")
269
 
        self.commit_some_revisions(tree, files[:100], 100, 1)
270
 
        self._time_read_write()
271
 
 
272
 
    def test_many_files_big_tree_1_revision(self):
273
 
        os.mkdir("a")
274
 
        tree, files = self.create_with_commits(1000, 1, directory_name="a")
275
 
        self.commit_some_revisions(tree, files[:1000], 1, 1)
276
 
        self._time_read_write()
277
 
 
278
 
    def test_many_files_big_tree_100_revision(self):
279
 
        os.mkdir("a")
280
 
        tree, files = self.create_with_commits(1000, 1, directory_name="a")
281
 
        self.commit_some_revisions(tree, files[:1000], 100, 1)
282
 
        self._time_read_write()
283
 
 
284
185
 
285
186
if __name__ == '__main__':
286
187
    # USE the following if you want to regenerate the above test functions 
287
188
    for treesize, treesize_h in [(5, "small"), (100, "moderate"),
288
189
                                 (1000, "big")]:
289
 
        for bundlefiles, bundlefiles_h in [(5, "few"), (100, "some"),
290
 
                                           (1000, "many")]:
 
190
        for bundlefiles, bundlefiles_h in [(5, "few"), (100, "some")]:
291
191
            if bundlefiles > treesize:
292
192
                continue
293
 
            for num_revisions in [1, 100, 1000]:
 
193
            for num_revisions in [1, 100]:
 
194
                if (num_revisions >= 100 and 
 
195
                        (bundlefiles >= 100 or treesize >= 1000)):
 
196
                    # Skip the 100x100x? tests.
 
197
                    # And the 100x?x1000
 
198
                    continue
294
199
                code = """\
295
200
    def test_%s_files_%s_tree_%s_revision(self):
296
201
        os.mkdir("a")
301
206
       treesize, bundlefiles, num_revisions)
302
207
                print code
303
208
 
304
 
 
305