Fix and move db unit tests
svn: r22767
This commit is contained in:
Nick Hall
@@ -1,184 +0,0 @@
|
||||
#
|
||||
# Gramps - a GTK+/GNOME based genealogy program
|
||||
#
|
||||
# Copyright (C) 2000-2007 Donald N. Allingham
|
||||
#
|
||||
# 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., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
||||
#
|
||||
|
||||
# test/GrampsDb/Cursor_Test.py
|
||||
# $Id$
|
||||
|
||||
import unittest
|
||||
import logging
|
||||
import os
|
||||
import tempfile
|
||||
import shutil
|
||||
import time
|
||||
import traceback
|
||||
import sys
|
||||
from bsddb import dbshelve, db
|
||||
|
||||
sys.path.append('../../gramps')
|
||||
|
||||
try:
|
||||
set()
|
||||
except NameError:
|
||||
from sets import Set as set
|
||||
|
||||
import gramps.gen.const
|
||||
|
||||
logger = logging.getLogger('Gramps.GrampsDbBase_Test')
|
||||
|
||||
from GrampsDbTestBase import GrampsDbBaseTest
|
||||
|
||||
class Data(object):
|
||||
|
||||
def __init__(self, handle,surname, name):
|
||||
self.handle = handle
|
||||
self.surname = surname
|
||||
self.name = name
|
||||
|
||||
## def __repr__(self):
|
||||
## return repr((self.handle,self.surname,self.name))
|
||||
|
||||
class CursorTest(unittest.TestCase):
|
||||
"""Test the cursor handling."""
|
||||
|
||||
def setUp(self):
|
||||
self._tmpdir = tempfile.mkdtemp()
|
||||
self.full_name = os.path.join(self._tmpdir,'test.grdb')
|
||||
self.env = db.DBEnv()
|
||||
self.env.set_cachesize(0,0x2000000)
|
||||
self.env.set_lk_max_locks(25000)
|
||||
self.env.set_lk_max_objects(25000)
|
||||
self.env.set_flags(db.DB_LOG_AUTOREMOVE,1) # clean up unused logs
|
||||
# The DB_PRIVATE flag must go if we ever move to multi-user setup
|
||||
env_flags = db.DB_CREATE|db.DB_RECOVER|db.DB_PRIVATE|\
|
||||
db.DB_INIT_MPOOL|db.DB_INIT_LOCK|\
|
||||
db.DB_INIT_LOG|db.DB_INIT_TXN
|
||||
|
||||
env_name = "%s/env" % (self._tmpdir,)
|
||||
if not os.path.isdir(env_name):
|
||||
os.mkdir(env_name)
|
||||
self.env.open(env_name,env_flags)
|
||||
(self.person_map,self.surnames) = self._open_tables()
|
||||
|
||||
def _open_tables(self):
|
||||
dbmap = dbshelve.DBShelf(self.env)
|
||||
dbmap.db.set_pagesize(16384)
|
||||
dbmap.open(self.full_name, 'person', db.DB_HASH,
|
||||
db.DB_CREATE|db.DB_AUTO_COMMIT, 0666)
|
||||
person_map = dbmap
|
||||
|
||||
table_flags = db.DB_CREATE|db.DB_AUTO_COMMIT
|
||||
|
||||
surnames = db.DB(self.env)
|
||||
surnames.set_flags(db.DB_DUP|db.DB_DUPSORT)
|
||||
surnames.open(self.full_name, "surnames", db.DB_BTREE,
|
||||
flags=table_flags)
|
||||
|
||||
def find_surname(key,data):
|
||||
return data.surname
|
||||
|
||||
person_map.associate(surnames, find_surname, table_flags)
|
||||
|
||||
return (person_map,surnames)
|
||||
|
||||
def tearDown(self):
|
||||
shutil.rmtree(self._tmpdir)
|
||||
|
||||
def test_simple_insert(self):
|
||||
"""test insert and retrieve works."""
|
||||
|
||||
data = Data(str(1),'surname1','name1')
|
||||
the_txn = self.env.txn_begin()
|
||||
self.person_map.put(data.handle,data,txn=the_txn)
|
||||
the_txn.commit()
|
||||
|
||||
v = self.person_map.get(data.handle)
|
||||
|
||||
assert v.handle == data.handle
|
||||
|
||||
def test_insert_with_curor_closed(self):
|
||||
"""test_insert_with_curor_closed"""
|
||||
|
||||
cursor_txn = self.env.txn_begin()
|
||||
|
||||
cursor = self.surnames.cursor(txn=cursor_txn)
|
||||
cursor.first()
|
||||
cursor.next()
|
||||
cursor.close()
|
||||
cursor_txn.commit()
|
||||
|
||||
data = Data(str(2),'surname2','name2')
|
||||
the_txn = self.env.txn_begin()
|
||||
self.person_map.put(data.handle,data,txn=the_txn)
|
||||
the_txn.commit()
|
||||
|
||||
v = self.person_map.get(data.handle)
|
||||
|
||||
assert v.handle == data.handle
|
||||
|
||||
def test_insert_with_curor_open(self):
|
||||
"""test_insert_with_curor_open"""
|
||||
|
||||
cursor_txn = self.env.txn_begin()
|
||||
cursor = self.surnames.cursor(txn=cursor_txn)
|
||||
cursor.first()
|
||||
cursor.next()
|
||||
|
||||
data = Data(str(2),'surname2','name2')
|
||||
the_txn = self.env.txn_begin()
|
||||
self.person_map.put(data.handle,data,txn=the_txn)
|
||||
the_txn.commit()
|
||||
|
||||
cursor.close()
|
||||
cursor_txn.commit()
|
||||
|
||||
v = self.person_map.get(data.handle)
|
||||
|
||||
assert v.handle == data.handle
|
||||
|
||||
def xtest_insert_with_curor_open_and_db_open(self):
|
||||
"""test_insert_with_curor_open_and_db_open"""
|
||||
|
||||
(person2,surnames2) = self._open_tables()
|
||||
|
||||
cursor_txn = self.env.txn_begin()
|
||||
cursor = surnames2.cursor(txn=cursor_txn)
|
||||
cursor.first()
|
||||
cursor.next()
|
||||
|
||||
data = Data(str(2),'surname2','name2')
|
||||
the_txn = self.env.txn_begin()
|
||||
self.person_map.put(data.handle,data,txn=the_txn)
|
||||
the_txn.commit()
|
||||
|
||||
cursor.close()
|
||||
cursor_txn.commit()
|
||||
|
||||
v = self.person_map.get(data.handle)
|
||||
|
||||
assert v.handle == data.handle
|
||||
|
||||
|
||||
def testSuite():
|
||||
suite = unittest.makeSuite(CursorTest,'test')
|
||||
return suite
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
unittest.TextTestRunner().run(testSuite())
|
||||
@@ -1,224 +0,0 @@
|
||||
#
|
||||
# Gramps - a GTK+/GNOME based genealogy program
|
||||
#
|
||||
# Copyright (C) 2000-2007 Donald N. Allingham
|
||||
#
|
||||
# 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., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
||||
#
|
||||
|
||||
# test/GrampsDb/GrampsDbBase_Test.py
|
||||
# $Id$
|
||||
|
||||
import unittest
|
||||
import logging
|
||||
import os
|
||||
import tempfile
|
||||
import shutil
|
||||
import time
|
||||
import traceback
|
||||
import sys
|
||||
|
||||
sys.path.append(os.path.join(os.path.dirname(__file__), '../../gramps'))
|
||||
|
||||
try:
|
||||
set()
|
||||
except NameError:
|
||||
from sets import Set as set
|
||||
|
||||
import gramps.gen.lib
|
||||
|
||||
logger = logging.getLogger('Gramps.GrampsDbBase_Test')
|
||||
|
||||
from GrampsDbTestBase import GrampsDbBaseTest
|
||||
|
||||
class ReferenceMapTest (GrampsDbBaseTest):
|
||||
"""Test methods on the GrampsDbBase class that are related to the reference_map
|
||||
index implementation."""
|
||||
|
||||
def test_simple_lookup(self):
|
||||
"""insert a record and a reference and check that
|
||||
a lookup for the reference returns the original
|
||||
record."""
|
||||
|
||||
source = self._add_source()
|
||||
person = self._add_person_with_sources([source])
|
||||
|
||||
references = list(self._db.find_backlink_handles(source.get_handle()))
|
||||
|
||||
assert len(references) == 1
|
||||
assert references[0] == (gen.lib.Person.__name__,person.get_handle())
|
||||
|
||||
def test_backlink_for_repository(self):
|
||||
"""check that the source / repos backlink lookup works."""
|
||||
|
||||
repos = self._add_repository()
|
||||
source = self._add_source(repos=repos)
|
||||
|
||||
references = list(self._db.find_backlink_handles(repos.get_handle()))
|
||||
|
||||
assert len(references) == 1
|
||||
assert references[0] == (gen.lib.Source.__name__,source.get_handle())
|
||||
|
||||
def test_class_limited_lookup(self):
|
||||
"""check that class limited lookups work."""
|
||||
|
||||
source = self._add_source()
|
||||
person = self._add_person_with_sources([source])
|
||||
|
||||
self._add_family_with_sources([source])
|
||||
self._add_event_with_sources([source])
|
||||
self._add_place_with_sources([source])
|
||||
self._add_media_object_with_sources([source])
|
||||
|
||||
# make sure that we have the correct number of references (one for each object)
|
||||
references = list(self._db.find_backlink_handles(source.get_handle()))
|
||||
|
||||
assert len(references) == 5, "len(references) == %s " % str(len(references))
|
||||
|
||||
# should just return the person reference
|
||||
references = [ ref for ref in self._db.find_backlink_handles(source.get_handle(),(gen.lib.Person.__name__,)) ]
|
||||
assert len(references) == 1, "len(references) == %s " % str(len(references))
|
||||
assert references[0][0] == gen.lib.Person.__name__, "references = %s" % repr(references)
|
||||
|
||||
# should just return the person and event reference
|
||||
references = list(self._db.find_backlink_handles(source.get_handle(),
|
||||
(gen.lib.Person.__name__, gen.lib.Event.__name__)))
|
||||
assert len(references) == 2, "len(references) == %s " % str(len(references))
|
||||
assert references[0][0] == gen.lib.Person.__name__, "references = %s" % repr(references)
|
||||
assert references[1][0] == gen.lib.Event.__name__, "references = %s" % repr(references)
|
||||
|
||||
|
||||
|
||||
def test_delete_primary(self):
|
||||
"""check that deleting a primary will remove the backreferences
|
||||
from the reference_map"""
|
||||
|
||||
source = self._add_source()
|
||||
person = self._add_person_with_sources([source])
|
||||
|
||||
assert self._db.get_person_from_handle(person.get_handle()) is not None
|
||||
|
||||
tran = self._db.transaction_begin()
|
||||
self._db.remove_person(person.get_handle(),tran)
|
||||
self._db.transaction_commit(tran, "Del Person")
|
||||
|
||||
assert self._db.get_person_from_handle(person.get_handle()) is None
|
||||
|
||||
references = list(self._db.find_backlink_handles(source.get_handle()))
|
||||
|
||||
assert len(references) == 0, "len(references) == %s " % str(len(references))
|
||||
|
||||
|
||||
def test_reindex_reference_map(self):
|
||||
"""Test that the reindex function works."""
|
||||
|
||||
def cb(count):
|
||||
pass
|
||||
|
||||
# unhook the reference_map update function so that we
|
||||
# can insert some records without the reference_map being updated.
|
||||
update_method = self._db.update_reference_map
|
||||
self._db._update_reference_map = lambda x,y: 1
|
||||
|
||||
# Insert a person/source pair.
|
||||
source = self._add_source()
|
||||
person = self._add_person_with_sources([source])
|
||||
|
||||
# Check that the reference map does not contain the reference.
|
||||
references = list(self._db.find_backlink_handles(source.get_handle()))
|
||||
|
||||
assert len(references) == 0, "len(references) == %s " % str(len(references))
|
||||
|
||||
# Reinstate the reference_map method and reindex the database
|
||||
self._db._update_reference_map = update_method
|
||||
self._db.reindex_reference_map(cb)
|
||||
|
||||
# Check that the reference now appears in the reference_map
|
||||
references = list(self._db.find_backlink_handles(source.get_handle()))
|
||||
|
||||
assert len(references) == 1, "len(references) == %s " % str(len(references))
|
||||
|
||||
|
||||
|
||||
def perf_simple_search_speed(self):
|
||||
|
||||
num_sources = 100
|
||||
num_persons = 1000
|
||||
num_families = 10
|
||||
num_events = 10
|
||||
num_places = 10
|
||||
num_media_objects = 10
|
||||
num_links = 10
|
||||
|
||||
self._populate_database(num_sources,
|
||||
num_persons,
|
||||
num_families,
|
||||
num_events,
|
||||
num_places,
|
||||
num_media_objects,
|
||||
num_links)
|
||||
|
||||
|
||||
# time searching for source backrefs with and without reference_map
|
||||
cur = self._db.get_source_cursor()
|
||||
handle,data = cur.first()
|
||||
cur.close()
|
||||
|
||||
start = time.time()
|
||||
references = list(self._db.find_backlink_handles(handle))
|
||||
end = time.time()
|
||||
|
||||
with_reference_map = end - start
|
||||
|
||||
remember = self._db.__class__.find_backlink_handles
|
||||
|
||||
self._db.__class__.find_backlink_handles = self._db.__class__.__base__.find_backlink_handles
|
||||
|
||||
start = time.time()
|
||||
references = list(self._db.find_backlink_handles(handle))
|
||||
end = time.time()
|
||||
|
||||
without_reference_map = end - start
|
||||
|
||||
self._db.__class__.find_backlink_handles = remember
|
||||
|
||||
logger.info("search test with following data: \n"
|
||||
"num_sources = %d \n"
|
||||
"num_persons = %d \n"
|
||||
"num_families = %d \n"
|
||||
"num_events = %d \n"
|
||||
"num_places = %d \n"
|
||||
"num_media_objects = %d \n"
|
||||
"num_links = %d" % (num_sources,
|
||||
num_persons,
|
||||
num_families,
|
||||
num_events,
|
||||
num_places,
|
||||
num_media_objects,
|
||||
num_links))
|
||||
logger.info("with refs %s\n", str(with_reference_map))
|
||||
logger.info("without refs %s\n", str(without_reference_map))
|
||||
|
||||
assert with_reference_map < (without_reference_map / 10), "Reference_map should an order of magnitude faster."
|
||||
|
||||
def testSuite():
|
||||
suite = unittest.makeSuite(ReferenceMapTest,'test')
|
||||
return suite
|
||||
|
||||
def perfSuite():
|
||||
return unittest.makeSuite(ReferenceMapTest,'perf')
|
||||
|
||||
if __name__ == '__main__':
|
||||
unittest.TextTestRunner().run(testSuite())
|
||||
@@ -1,185 +0,0 @@
|
||||
#
|
||||
# Gramps - a GTK+/GNOME based genealogy program
|
||||
#
|
||||
# Copyright (C) 2000-2007 Donald N. Allingham
|
||||
#
|
||||
# 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., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
||||
#
|
||||
|
||||
# test/GrampsDb/GrampsDbTestBase.py
|
||||
# $Id$
|
||||
|
||||
import unittest
|
||||
import logging
|
||||
import os
|
||||
import tempfile
|
||||
import shutil
|
||||
import time
|
||||
import traceback
|
||||
import sys
|
||||
|
||||
sys.path.append('../gramps')
|
||||
|
||||
try:
|
||||
set()
|
||||
except NameError:
|
||||
from sets import Set as set
|
||||
|
||||
from gramps.gen.db import DbBsddb
|
||||
from gramps.cli.clidbman import CLIDbManager
|
||||
import gramps.gen.const
|
||||
import gramps.gen.lib
|
||||
|
||||
logger = logging.getLogger('Gramps.GrampsDbTestBase')
|
||||
|
||||
class GrampsDbBaseTest(unittest.TestCase):
|
||||
"""Base class for unittest that need to be able to create
|
||||
test databases."""
|
||||
|
||||
def setUp(self):
|
||||
def dummy_callback(dummy):
|
||||
pass
|
||||
self._tmpdir = tempfile.mkdtemp()
|
||||
#self._filename = os.path.join(self._tmpdir,'test.grdb')
|
||||
|
||||
self._db = DbBsddb()
|
||||
dbman = CLIDbManager(None)
|
||||
self._filename, title = dbman.create_new_db_cli(title="Test")
|
||||
self._db.load(self._filename, dummy_callback, "w")
|
||||
|
||||
def tearDown(self):
|
||||
self._db.close()
|
||||
shutil.rmtree(self._tmpdir)
|
||||
|
||||
def _populate_database(self,
|
||||
num_sources = 1,
|
||||
num_persons = 0,
|
||||
num_families = 0,
|
||||
num_events = 0,
|
||||
num_places = 0,
|
||||
num_media_objects = 0,
|
||||
num_links = 1):
|
||||
|
||||
# start with sources
|
||||
sources = []
|
||||
for i in xrange(0, num_sources):
|
||||
sources.append(self._add_source())
|
||||
|
||||
# now for each of the other tables. Give each entry a link
|
||||
# to num_link sources, sources are chosen on a round robin
|
||||
# basis
|
||||
|
||||
for num, add_func in ((num_persons, self._add_person_with_sources),
|
||||
(num_families, self._add_family_with_sources),
|
||||
(num_events, self._add_event_with_sources),
|
||||
(num_places, self._add_place_with_sources),
|
||||
(num_media_objects, self._add_media_object_with_sources)):
|
||||
|
||||
source_idx = 1
|
||||
for person_idx in xrange(0, num):
|
||||
|
||||
# Get the list of sources to link
|
||||
lnk_sources = set()
|
||||
for i in xrange(0, num_links):
|
||||
lnk_sources.add(sources[source_idx-1])
|
||||
source_idx = (source_idx+1) % len(sources)
|
||||
|
||||
try:
|
||||
add_func(lnk_sources)
|
||||
except:
|
||||
print "person_idx = ", person_idx
|
||||
print "lnk_sources = ", repr(lnk_sources)
|
||||
raise
|
||||
|
||||
return
|
||||
|
||||
def _add_source(self,repos=None):
|
||||
# Add a Source
|
||||
|
||||
tran = self._db.transaction_begin()
|
||||
source = gen.lib.Source()
|
||||
if repos is not None:
|
||||
repo_ref = gen.lib.RepoRef()
|
||||
repo_ref.set_reference_handle(repos.get_handle())
|
||||
source.add_repo_reference(repo_ref)
|
||||
self._db.add_source(source,tran)
|
||||
self._db.commit_source(source,tran)
|
||||
self._db.transaction_commit(tran, "Add Source")
|
||||
|
||||
return source
|
||||
|
||||
def _add_repository(self):
|
||||
# Add a Repository
|
||||
|
||||
tran = self._db.transaction_begin()
|
||||
repos = gen.lib.Repository()
|
||||
self._db.add_repository(repos,tran)
|
||||
self._db.commit_repository(repos,tran)
|
||||
self._db.transaction_commit(tran, "Add Repository")
|
||||
|
||||
return repos
|
||||
|
||||
|
||||
def _add_object_with_source(self,sources, object_class,add_method,commit_method):
|
||||
|
||||
object = object_class()
|
||||
|
||||
for source in sources:
|
||||
src_ref = gen.lib.SourceRef()
|
||||
src_ref.set_reference_handle(source.get_handle())
|
||||
object.add_source_reference(src_ref)
|
||||
|
||||
|
||||
tran = self._db.transaction_begin()
|
||||
add_method(object,tran)
|
||||
commit_method(object,tran)
|
||||
self._db.transaction_commit(tran, "Add Object")
|
||||
|
||||
return object
|
||||
|
||||
def _add_person_with_sources(self,sources):
|
||||
|
||||
return self._add_object_with_source(sources,
|
||||
gen.lib.Person,
|
||||
self._db.add_person,
|
||||
self._db.commit_person)
|
||||
|
||||
def _add_family_with_sources(self,sources):
|
||||
|
||||
return self._add_object_with_source(sources,
|
||||
gen.lib.Family,
|
||||
self._db.add_family,
|
||||
self._db.commit_family)
|
||||
|
||||
def _add_event_with_sources(self,sources):
|
||||
|
||||
return self._add_object_with_source(sources,
|
||||
gen.lib.Event,
|
||||
self._db.add_event,
|
||||
self._db.commit_event)
|
||||
|
||||
def _add_place_with_sources(self,sources):
|
||||
|
||||
return self._add_object_with_source(sources,
|
||||
gen.lib.Place,
|
||||
self._db.add_place,
|
||||
self._db.commit_place)
|
||||
|
||||
def _add_media_object_with_sources(self,sources):
|
||||
|
||||
return self._add_object_with_source(sources,
|
||||
gen.lib.MediaObject,
|
||||
self._db.add_object,
|
||||
self._db.commit_media_object)
|
||||
Reference in New Issue
Block a user