Add database unit tests for sorted handles
This commit is contained in:
parent
98950a89a3
commit
11942d8975
@ -144,8 +144,12 @@ class DbRandomTest(unittest.TestCase):
|
|||||||
# Test get_*_handles methods
|
# Test get_*_handles methods
|
||||||
#
|
#
|
||||||
################################################################
|
################################################################
|
||||||
def __get_handles_test(self, obj_type, handles_func, number_func):
|
def __get_handles_test(self, obj_type, handles_func, number_func,
|
||||||
handles = handles_func()
|
sort_handles=False):
|
||||||
|
if sort_handles:
|
||||||
|
handles = handles_func(sort_handles=True)
|
||||||
|
else:
|
||||||
|
handles = handles_func()
|
||||||
self.assertEqual(len(handles), number_func())
|
self.assertEqual(len(handles), number_func())
|
||||||
for handle in handles:
|
for handle in handles:
|
||||||
self.assertIn(handle.decode('utf8'), self.handles[obj_type])
|
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_tag_handles,
|
||||||
self.db.get_number_of_tags)
|
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
|
# Test get_*_gramps_ids methods
|
||||||
|
Loading…
x
Reference in New Issue
Block a user