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

Add database unit tests for sorted handles

Browse Source
This commit is contained in:
Nick Hall 2016-11-29 22:30:11 +00:00
parent 98950a89a3
commit 11942d8975
1 changed files with 42 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -144,8 +144,12 @@ class DbRandomTest(unittest.TestCase):
# Test get_*_handles methods
#
################################################################
def __get_handles_test(self, obj_type, handles_func, number_func):
handles = handles_func()
def __get_handles_test(self, obj_type, handles_func, number_func,
sort_handles=False):
if sort_handles:
handles = handles_func(sort_handles=True)
else:
handles = handles_func()
self.assertEqual(len(handles), number_func())
for handle in handles:
self.assertIn(handle.decode('utf8'), self.handles[obj_type])
@ -200,6 +204,42 @@ class DbRandomTest(unittest.TestCase):
self.db.get_tag_handles,
self.db.get_number_of_tags)
def test_get_person_handles_sort(self):
self.__get_handles_test('Person',
self.db.get_person_handles,
self.db.get_number_of_people,
sort_handles=True)
def test_get_family_handles_sort(self):
self.__get_handles_test('Family',
self.db.get_family_handles,
self.db.get_number_of_families,
sort_handles=True)
def test_get_source_handles_sort(self):
self.__get_handles_test('Source',
self.db.get_source_handles,
self.db.get_number_of_sources,
sort_handles=True)
def test_get_citation_handles_sort(self):
self.__get_handles_test('Citation',
self.db.get_citation_handles,
self.db.get_number_of_citations,
sort_handles=True)
def test_get_media_handles_sort(self):
self.__get_handles_test('Media',
self.db.get_media_handles,
self.db.get_number_of_media,
sort_handles=True)
def test_get_tag_handles_sort(self):
self.__get_handles_test('Tag',
self.db.get_tag_handles,
self.db.get_number_of_tags,
sort_handles=True)
################################################################
#
# Test get_*_gramps_ids methods