Cyclone-Team/gramps
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add more database unit tests

Browse Source
This commit is contained in:
Nick Hall 2016-10-29 22:12:28 +01:00
parent 5bbce108fd
commit ab4e147021
1 changed files with 197 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

200
gramps/plugins/db/dbapi/test/db_test.py
View File

@ -32,14 +32,17 @@ import unittest
#-------------------------------------------------------------------------
from gramps.gen.db import make_database, DbTxn
from gramps.gen.lib import (Person, Family, Event, Place, Repository, Source,
Citation, Media, Note, Tag)
Citation, Media, Note, Tag, Researcher, Surname)
#-------------------------------------------------------------------------
#
# DbTest class
# DbRandomTest class
#
#-------------------------------------------------------------------------
class DbTest(unittest.TestCase):
class DbRandomTest(unittest.TestCase):
'''
Tests with random objects.
'''
@classmethod
def setUpClass(cls):
@ -91,6 +94,8 @@ class DbTest(unittest.TestCase):
def __add_object(self, obj_class, add_func, trans):
obj = obj_class()
if obj_class == Tag:
obj.name = 'Tag%s' % len(self.handles['Tag'])
handle = add_func(obj, trans)
self.handles[obj_class.__name__].append(handle)
if obj_class != Tag:
@ -369,6 +374,10 @@ class DbTest(unittest.TestCase):
self.db.get_note_gramps_ids,
self.db.get_note_from_gramps_id)
def test_get_tag_from_name(self):
tag = self.db.get_tag_from_name('Tag0')
self.assertEqual(tag.handle, self.handles['Tag'][0])
################################################################
#
# Test has_*_handle methods
@ -606,6 +615,191 @@ class DbTest(unittest.TestCase):
self.__iter_objects_test(Tag,
self.db.iter_tags)
################################################################
#
# Test default and initial person methods
#
################################################################
def test_no_default_handle(self):
self.db.set_default_person_handle(None)
handle = self.db.get_default_handle()
self.assertIsNone(handle)
person = self.db.get_default_person()
self.assertIsNone(person)
person = self.db.find_initial_person()
self.assertIsInstance(person, Person)
def test_default_handle(self):
default_handle = self.handles['Person'][0]
self.db.set_default_person_handle(default_handle)
handle = self.db.get_default_handle()
self.assertEqual(handle, default_handle)
person = self.db.get_default_person()
self.assertEqual(person.handle, default_handle)
person = self.db.find_initial_person()
self.assertEqual(person.handle, default_handle)
#-------------------------------------------------------------------------
#
# DbEmptyTest class
#
#-------------------------------------------------------------------------
class DbEmptyTest(unittest.TestCase):
'''
Tests with an empty database.
'''
@classmethod
def setUpClass(cls):
cls.db = make_database("inmemorydb")
cls.db.load(None)
################################################################
#
# Test metadata methods
#
################################################################
def test_metadata(self):
self.db.set_metadata('test-key', 'test-value')
value = self.db.get_metadata('test-key')
self.assertEqual(value, 'test-value')
def test_metadata_missing(self):
value = self.db.get_metadata('missing-key')
self.assertEqual(value, [])
def test_metadata_default(self):
value = self.db.get_metadata('missing-key', default='default-value')
self.assertEqual(value, 'default-value')
################################################################
#
# Test default and initial person methods
#
################################################################
def test_no_default_handle(self):
handle = self.db.get_default_handle()
self.assertIsNone(handle)
person = self.db.get_default_person()
self.assertIsNone(person)
person = self.db.find_initial_person()
self.assertIsNone(person)
################################################################
#
# Test researcher methods
#
################################################################
def test_researcher(self):
res1 = Researcher()
res1.street = 'street'
res1.locality = 'locality'
res1.city = 'city'
res1.county = 'county'
res1.state = 'state'
res1.country = 'country'
res1.postal = 'postal'
res1.phone = 'phone'
res1.name = 'name'
res1.addr = 'addr'
res1.email = 'email'
self.db.set_researcher(res1)
res2 = self.db.get_researcher()
self.assertEqual(res1.serialize(), res2.serialize())
################################################################
#
# Test name group mapping
#
################################################################
def test_name_group_mapping(self):
self.db.set_name_group_mapping('Clark', 'Clarke')
self.assertTrue(self.db.has_name_group_key('Clark'))
self.assertFalse(self.db.has_name_group_key('Baker'))
for key in self.db.get_name_group_keys():
self.assertTrue(self.db.has_name_group_key(key))
mapping = self.db.get_name_group_mapping('Clark')
self.assertEqual(mapping, 'Clarke')
#-------------------------------------------------------------------------
#
# DbPersonTest class
#
#-------------------------------------------------------------------------
class DbPersonTest(unittest.TestCase):
'''
Tests with some sample people.
'''
@classmethod
def setUpClass(cls):
cls.db = make_database("inmemorydb")
cls.db.load(None)
def __add_person(self, gender, first_name, surname, trans):
person = Person()
person.gender = gender
name = person.primary_name
name.first_name = first_name
surname1 = Surname()
surname1.surname = surname
name.set_surname_list([surname1])
self.all_surnames.append(surname)
self.db.add_person(person, trans)
def setUp(self):
self.all_surnames = []
with DbTxn('Add test objects', self.db) as trans:
self.__add_person(Person.MALE, 'John', 'Allen', trans)
self.__add_person(Person.MALE, 'John', 'Baker', trans)
self.__add_person(Person.MALE, 'John', 'Clark', trans)
self.__add_person(Person.FEMALE, 'John', 'Davis', trans)
self.__add_person(Person.UNKNOWN, 'John', 'Evans', trans)
self.__add_person(Person.FEMALE, 'Mary', 'Allen', trans)
self.__add_person(Person.FEMALE, 'Mary', 'Baker', trans)
self.__add_person(Person.FEMALE, 'Mary', 'Clark', trans)
self.__add_person(Person.MALE, 'Mary', 'Davis', trans)
self.__add_person(Person.FEMALE, 'Mary', 'Evans', trans)
def tearDown(self):
with DbTxn('Remove test objects', self.db) as trans:
for handle in self.db.get_person_handles():
self.db.remove_person(handle, trans)
################################################################
#
# Test surname list
#
################################################################
def test_surname_list(self):
surname_list = self.db.get_surname_list()
for surname in surname_list:
self.assertIn(surname, self.all_surnames)
################################################################
#
# Test gender stats
#
################################################################
def test_gender_stats(self):
stats = self.db.genderStats
self.assertEqual(stats.name_stats('John'), (3, 1, 1))
self.assertEqual(stats.name_stats('Mary'), (1, 4, 0))
self.db.save_gender_stats(stats)
saved = self.db.get_gender_stats()
self.assertEqual(saved['John'], (3, 1, 1))
self.assertEqual(saved['Mary'], (1, 4, 0))
if __name__ == "__main__":
unittest.main()