# Copyright (C) 2006 Canonical Ltd
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
"""Tests for bzr bundle performance."""
from cStringIO import StringIO
import os
import shutil
from bzrlib import bzrdir
from bzrlib.benchmarks import Benchmark
from bzrlib.branch import Branch
from bzrlib.bundle.apply_bundle import install_bundle
from bzrlib.bundle.serializer import read_bundle,write_bundle
from bzrlib.revision import NULL_REVISION
from bzrlib.revisionspec import RevisionSpec
from bzrlib.workingtree import WorkingTree
class BundleBenchmark(Benchmark):
"""Benchmarks for bzr bundle performance and bzr merge with a bundle."""
def test_create_bundle_known_kernel_like_tree(self):
"""Create a bundle for a kernel sized tree with no ignored, unknowns,
or added and one commit.
"""
self.make_kernel_like_committed_tree()
self.time(self.run_bzr, ['bundle', '--revision', '..-1'])
def test_create_bundle_many_commit_tree (self):
"""Create a bundle for a tree with many commits but no changes."""
self.make_many_commit_tree()
self.time(self.run_bzr, ['bundle', '--revision', '..-1'])
def test_create_bundle_heavily_merged_tree(self):
"""Create a bundle for a heavily merged tree."""
self.make_heavily_merged_tree()
self.time(self.run_bzr, ['bundle', '--revision', '..-1'])
def test_apply_bundle_known_kernel_like_tree(self):
"""Create a bundle for a kernel sized tree with no ignored, unknowns,
or added and one commit.
"""
tree = self.make_kernel_like_committed_tree('tree')
f = open('bundle', 'wb')
try:
write_bundle(tree.branch.repository, tree.last_revision(),
NULL_REVISION, f)
finally:
f.close()
tree2 = self.make_branch_and_tree('branch_a')
os.chdir('branch_a')
self.time(self.run_bzr, ['merge', '../bundle'])
class BundleLibraryLevelWriteBenchmark(Benchmark):
""" Benchmarks for the write_bundle library function. """
def _time_read_write(self):
branch, relpath = Branch.open_containing("a")
revision_history = branch.revision_history()
bundle_text = StringIO()
self.time(write_bundle, branch.repository, revision_history[-1],
NULL_REVISION, bundle_text)
bundle_text.seek(0)
target_tree = self.make_branch_and_tree('b')
bundle = self.time(read_bundle, bundle_text)
self.time(install_bundle, target_tree.branch.repository, bundle)
def test_few_files_small_tree_1_revision(self):
os.mkdir("a")
tree, files = self.create_with_commits(5, 1, directory_name="a")
self.commit_some_revisions(tree, files[:5], 1, 1)
self._time_read_write()
def test_few_files_small_tree_100_revision(self):
os.mkdir("a")
tree, files = self.create_with_commits(5, 1, directory_name="a")
self.commit_some_revisions(tree, files[:5], 100, 1)
self._time_read_write()
def test_few_files_moderate_tree_1_revision(self):
os.mkdir("a")
tree, files = self.create_with_commits(100, 1, directory_name="a")
self.commit_some_revisions(tree, files[:5], 1, 1)
self._time_read_write()
def test_few_files_moderate_tree_100_revision(self):
os.mkdir("a")
tree, files = self.create_with_commits(100, 1, directory_name="a")
self.commit_some_revisions(tree, files[:5], 100, 1)
self._time_read_write()
def test_some_files_moderate_tree_1_revision(self):
os.mkdir("a")
tree, files = self.create_with_commits(100, 1, directory_name="a")
self.commit_some_revisions(tree, files[:100], 1, 1)
self._time_read_write()
def test_few_files_big_tree_1_revision(self):
os.mkdir("a")
tree, files = self.create_with_commits(1000, 1, directory_name="a")
self.commit_some_revisions(tree, files[:5], 1, 1)
self._time_read_write()
def test_some_files_big_tree_1_revision(self):
os.mkdir("a")
tree, files = self.create_with_commits(1000, 1, directory_name="a")
self.commit_some_revisions(tree, files[:100], 1, 1)
self._time_read_write()
class BundleLibraryLevelInstallBenchmark(Benchmark):
""" Benchmarks for the install_bundle library function. """
def _time_read_write(self):
branch, relpath = Branch.open_containing("a")
revision_history = branch.revision_history()
bundle_text = StringIO()
write_bundle(branch.repository, revision_history[-1],
NULL_REVISION, bundle_text)
bundle_text.seek(0)
target_tree = self.make_branch_and_tree('b')
bundle = self.time(read_bundle, bundle_text)
self.time(install_bundle, target_tree.branch.repository, bundle)
def test_few_files_small_tree_1_revision(self):
os.mkdir("a")
tree, files = self.create_with_commits(5, 1, directory_name="a")
self.commit_some_revisions(tree, files[:5], 1, 1)
self._time_read_write()
def test_few_files_small_tree_100_revision(self):
os.mkdir("a")
tree, files = self.create_with_commits(5, 1, directory_name="a")
self.commit_some_revisions(tree, files[:5], 100, 1)
self._time_read_write()
def test_few_files_moderate_tree_1_revision(self):
os.mkdir("a")
tree, files = self.create_with_commits(100, 1, directory_name="a")
self.commit_some_revisions(tree, files[:5], 1, 1)
self._time_read_write()
def test_few_files_moderate_tree_100_revision(self):
os.mkdir("a")
tree, files = self.create_with_commits(100, 1, directory_name="a")
self.commit_some_revisions(tree, files[:5], 100, 1)
self._time_read_write()
def test_some_files_moderate_tree_1_revision(self):
os.mkdir("a")
tree, files = self.create_with_commits(100, 1, directory_name="a")
self.commit_some_revisions(tree, files[:100], 1, 1)
self._time_read_write()
def test_few_files_big_tree_1_revision(self):
os.mkdir("a")
tree, files = self.create_with_commits(1000, 1, directory_name="a")
self.commit_some_revisions(tree, files[:5], 1, 1)
self._time_read_write()
def test_some_files_big_tree_1_revision(self):
os.mkdir("a")
tree, files = self.create_with_commits(1000, 1, directory_name="a")
self.commit_some_revisions(tree, files[:100], 1, 1)
self._time_read_write()
if __name__ == '__main__':
# USE the following if you want to regenerate the above test functions
for treesize, treesize_h in [(5, "small"), (100, "moderate"),
(1000, "big")]:
for bundlefiles, bundlefiles_h in [(5, "few"), (100, "some")]:
if bundlefiles > treesize:
continue
for num_revisions in [1, 100]:
if (num_revisions >= 100 and
(bundlefiles >= 100 or treesize >= 1000)):
# Skip the 100x100x? tests.
# And the 100x?x1000
continue
code = """\
def test_%s_files_%s_tree_%s_revision(self):
os.mkdir("a")
tree, files = self.create_with_commits(%s, 1, directory_name="a")
self.commit_some_revisions(tree, files[:%s], %s, 1)
self._time_read_write()
""" % (bundlefiles_h, treesize_h, num_revisions,
treesize, bundlefiles, num_revisions)
print code
|