further work to complete the reference_map implementation.

svn: r5563

test/GrampsDbBase_Test.py

@@ -0,0 +1,118 @@
+import unittest
+import logging
+import os
+import tempfile
+import shutil
+
+import sys
+sys.path.append('../src')
+
+import GrampsBSDDB
+import RelLib
+
+logger = logging.getLogger('Gramps.GrampsDbBase_Test')
+
+class ReferenceMapTest (unittest.TestCase):
+
+    def setUp(self):        
+        self._tmpdir = tempfile.mkdtemp()
+        self._filename = os.path.join(self._tmpdir,'test.grdb')
+        
+        self._db = GrampsBSDDB.GrampsBSDDB()
+        self._db.load(self._filename,
+                      None, # callback
+                      "w")
+
+    def tearDown(self):
+        shutil.rmtree(self._tmpdir)
+
+
+    def _add_person_and_source(self):
+        # Add a Source
+        
+        tran = self._db.transaction_begin()
+        source = RelLib.Source()
+        self._db.add_source(source,tran)
+        self._db.commit_source(source,tran)
+        self._db.transaction_commit(tran, "Add Source")
+
+        src_ref = RelLib.SourceRef()
+        src_ref.set_base_handle(source.get_handle())
+
+        # Add Person with reference to the Source
+
+        tran = self._db.transaction_begin()
+        person = RelLib.Person()
+
+        person.add_source_reference(src_ref)
+        self._db.add_person(person,tran)
+        self._db.commit_person(person,tran)
+        self._db.transaction_commit(tran, "Add Person")
+
+        return (person,source)
+
+    def test_simple_lookup(self):
+        """insert a record and a reference and check that
+        a lookup for the reference returns the original
+        record."""
+
+        person,source = self._add_person_and_source()
+        
+        references = [ ref for ref in self._db.find_backlink_handles(source.get_handle()) ]
+
+        assert len(references) == 1
+        assert references[0] == ('Person',person.get_handle())
+
+    def test_delete_primary(self):
+        """check that deleting a primary will remove the backreferences
+        from the reference_map"""
+
+        person,source = self._add_person_and_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()) == None
+        
+        references = [ ref for ref in 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."""
+
+        # 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.
+        person,source = self._add_person_and_source()
+
+        # Check that the reference map does not contain the reference.
+        references = [ ref for ref in 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()
+
+        # Check that the reference now appears in the reference_map
+        references = [ ref for ref in self._db.find_backlink_handles(source.get_handle()) ]
+
+        assert len(references) == 1, "len(references) == %s " % str(len(references))
+
+        
+                       
+
+        
+def testSuite():
+    return unittest.makeSuite(ReferenceMapTest,'test')
+
+if __name__ == '__main__':
+    unittest.TextTestRunner().run(testSuite())

test/RunAllTests.py

@@ -0,0 +1,54 @@
+"""Copyright QinetiQ Ltd (2005) - See LICENSE.TXT for licensing information.
+
+"""
+
+import logging
+ 
+import os
+import unittest
+from optparse import OptionParser
+
+def make_parser():
+    usage = "usage: %prog [options]"
+    parser = OptionParser(usage)
+    parser.add_option("-v", "--verbosity", type="int", dest="verbose_level", default=0,
+                      help="Level of verboseness")  
+    return parser
+
+
+def getTestSuites():
+    
+    test_modules = [ i for i in os.listdir('.') if i[-8:] == "_Test.py" ]
+
+    test_suites = []
+    for module in test_modules:
+        mod = __import__(module[:-3])
+        test_suites.append(mod.testSuite())
+
+    return test_suites
+
+def allTheTests():
+    return unittest.TestSuite(getTestSuites())
+
+if __name__ == '__main__':
+
+    console = logging.StreamHandler()
+    console.setLevel(logging.INFO)
+    console.setFormatter(logging.Formatter('%(name)-12s: %(levelname)-8s %(message)s'))
+
+    logger = logging.getLogger('Gramps')
+    logger.addHandler(console)
+    
+    (options,args) = make_parser().parse_args()
+
+    if options.verbose_level == 1:
+        logger.setLevel(logging.INFO)
+    elif options.verbose_level >= 2:
+        logger.setLevel(logging.DEBUG)
+        os.environ['GRAMPS_SIGNAL'] = "1"
+    elif options.verbose_level >= 3:
+        logger.setLevel(logging.NOTSET)
+    else:
+        logger.setLevel(logging.ERROR)
+        
+    unittest.TextTestRunner(verbosity=options.verbose_level).run(allTheTests())