/brz/remove-bazaar

# Copyright (C) 2007 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

"""Benchmarks for Knit performance"""

import os

from bzrlib import (

    generate_ids,

    knit,

    tests,

    )

from bzrlib.benchmarks import Benchmark

from bzrlib.tests.test_knit import CompiledKnitFeature

class BenchKnitIndex(Benchmark):

    """Benchmark Knit index performance."""

    def create_50k_index(self):

        """Create an knit index file with 50,000 entries.

        This isn't super realistic, but it *is* big :)

        The file 'test.kndx' will be created.

        """

        rev_id = generate_ids.gen_revision_id('long.name@this.example.com')

        versions = [(rev_id, ('fulltext',), 0, 200, [])]

        pos = 200

        alt_parent = None

        for i in xrange(49999):

            if alt_parent is not None:

                parent_ids = [rev_id, alt_parent]

            else:

                parent_ids = [rev_id]

            if i % 8 == 0:

                # The *next* entry will be a merge

                alt_parent = rev_id

            else:

                alt_parent = None

            rev_id = generate_ids.gen_revision_id('long.name@this.example.com')

            versions.append((rev_id, ('line-delta',), pos, 200, parent_ids))

            pos += 200

        t = self.get_transport()

        kndx = knit._KnitIndex(t, 'test.kndx', 'w', create=True,

                               delay_create=True)

        kndx.add_versions(versions)

    def setup_load_data_c(self):

        self.requireFeature(CompiledKnitFeature)

        orig = knit._load_data

        def reset():

            knit._load_data = orig

        self.addCleanup(reset)

        from bzrlib._knit_load_data_pyx import _load_data_c

        knit._load_data = _load_data_c

    def setup_load_data_py(self):

        orig = knit._load_data

        def reset():

            knit._load_data = orig

        self.addCleanup(reset)

        from bzrlib._knit_load_data_py import _load_data_py

        knit._load_data = _load_data_py

    def test_read_50k_index_c(self):

        self.setup_load_data_c()

        self.create_50k_index()

        t = self.get_transport()

        kndx = self.time(knit._KnitIndex, t, 'test.kndx', 'r')

    def test_read_50k_index_c_again(self):

        self.setup_load_data_c()

        self.create_50k_index()

        t = self.get_transport()

        kndx = self.time(knit._KnitIndex, t, 'test.kndx', 'r')

    def test_read_50k_index_py(self):

        self.create_50k_index()

        self.setup_load_data_py()

        t = self.get_transport()

        kndx = self.time(knit._KnitIndex, t, 'test.kndx', 'r')

    def test_read_50k_index_py_again(self):

        self.create_50k_index()

        self.setup_load_data_py()

        t = self.get_transport()

        kndx = self.time(knit._KnitIndex, t, 'test.kndx', 'r')