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

Fix bugs in the test case generator

Browse Source
This commit is contained in:
Nick Hall 2014-01-21 00:05:03 +00:00
parent c574f3a767
commit 7b05c1944b
1 changed files with 34 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

53
gramps/plugins/tool/testcasegenerator.py
View File

@ -58,6 +58,19 @@ from gramps.gen.lib import (Address, Attribute, AttributeType, ChildRef,
Person, PersonRef, Place, RepoRef, Repository, RepositoryType, Person, PersonRef, Place, RepoRef, Repository, RepositoryType,
Source, SourceMediaType, SrcAttribute, SrcAttributeType, Source, SourceMediaType, SrcAttribute, SrcAttributeType,
Surname, Tag, Url, UrlType) Surname, Tag, Url, UrlType)
from gramps.gen.lib.addressbase import AddressBase
from gramps.gen.lib.attrbase import AttributeBase
from gramps.gen.lib.primaryobj import BasicPrimaryObject
from gramps.gen.lib.citationbase import CitationBase
from gramps.gen.lib.datebase import DateBase
from gramps.gen.lib.ldsordbase import LdsOrdBase
from gramps.gen.lib.locationbase import LocationBase
from gramps.gen.lib.mediabase import MediaBase
from gramps.gen.lib.notebase import NoteBase
from gramps.gen.lib.placebase import PlaceBase
from gramps.gen.lib.privacybase import PrivacyBase
from gramps.gen.lib.tagbase import TagBase
from gramps.gen.lib.urlbase import UrlBase
from gramps.gen.lib import StyledText, StyledTextTag, StyledTextTagType from gramps.gen.lib import StyledText, StyledTextTag, StyledTextTagType
from gramps.gen.db import DbTxn from gramps.gen.db import DbTxn
from gramps.gen.mime import get_type from gramps.gen.mime import get_type
@ -1507,7 +1520,7 @@ class TestcaseGenerator(tool.BatchTool):
def fill_object( self, o): def fill_object( self, o):
if issubclass(o.__class__,gen.lib.addressbase.AddressBase): if issubclass(o.__class__, AddressBase):
while randint(0,1) == 1: while randint(0,1) == 1:
a = Address() a = Address()
self.fill_object(a) self.fill_object(a)
@ -1517,7 +1530,7 @@ class TestcaseGenerator(tool.BatchTool):
o.set_type( self.rand_type(AttributeType())) o.set_type( self.rand_type(AttributeType()))
o.set_value( self.rand_text(self.SHORT)) o.set_value( self.rand_text(self.SHORT))
if issubclass(o.__class__,gen.lib.attrbase.AttributeBase): if issubclass(o.__class__, AttributeBase):
while randint(0,1) == 1: while randint(0,1) == 1:
a = Attribute() a = Attribute()
self.fill_object(a) self.fill_object(a)
@ -1529,7 +1542,7 @@ class TestcaseGenerator(tool.BatchTool):
if randint(0,3) == 1: if randint(0,3) == 1:
o.set_father_relation( self.rand_type( ChildRefType())) o.set_father_relation( self.rand_type( ChildRefType()))
if issubclass(o.__class__,gen.lib.datebase.DateBase): if issubclass(o.__class__, DateBase):
if randint(0,1) == 1: if randint(0,1) == 1:
(y,d) = self.rand_date() (y,d) = self.rand_date()
o.set_date_object( d) o.set_date_object( d)
@ -1538,7 +1551,7 @@ class TestcaseGenerator(tool.BatchTool):
if randint(0,1) == 1: if randint(0,1) == 1:
o.set_description( self.rand_text(self.LONG)) o.set_description( self.rand_text(self.LONG))
if issubclass(o.__class__,gen.lib.eventref.EventRef): if issubclass(o.__class__, EventRef):
o.set_role( self.rand_type(EventRoleType())) o.set_role( self.rand_type(EventRoleType()))
if isinstance(o,Family): if isinstance(o,Family):
@ -1551,7 +1564,7 @@ class TestcaseGenerator(tool.BatchTool):
if randint(0,1) == 1: if randint(0,1) == 1:
o.set_temple( choice(TEMPLES.name_code_data())[1]) o.set_temple( choice(TEMPLES.name_code_data())[1])
if issubclass(o.__class__,gen.lib.ldsordbase.LdsOrdBase): if issubclass(o.__class__, LdsOrdBase):
while randint(0,1) == 1: while randint(0,1) == 1:
ldsord = LdsOrd() ldsord = LdsOrd()
self.fill_object( ldsord) self.fill_object( ldsord)
@ -1569,7 +1582,7 @@ class TestcaseGenerator(tool.BatchTool):
if randint(0,1) == 1: if randint(0,1) == 1:
o.set_parish( self.rand_text(self.SHORT)) o.set_parish( self.rand_text(self.SHORT))
if issubclass(o.__class__,gen.lib.locationbase.LocationBase): if issubclass(o.__class__, LocationBase):
if randint(0,1) == 1: if randint(0,1) == 1:
o.set_street( self.rand_text(self.SHORT)) o.set_street( self.rand_text(self.SHORT))
if randint(0,1) == 1: if randint(0,1) == 1:
@ -1585,7 +1598,7 @@ class TestcaseGenerator(tool.BatchTool):
if randint(0,1) == 1: if randint(0,1) == 1:
o.set_county( self.rand_text(self.SHORT)) o.set_county( self.rand_text(self.SHORT))
if issubclass(o.__class__,gen.lib.mediabase.MediaBase): if issubclass(o.__class__, MediaBase):
# FIXME: frequency changed to prevent recursion # FIXME: frequency changed to prevent recursion
while randint(0,10) == 1: while randint(0,10) == 1:
o.add_media_reference( self.fill_object( MediaRef())) o.add_media_reference( self.fill_object( MediaRef()))
@ -1641,7 +1654,7 @@ class TestcaseGenerator(tool.BatchTool):
o.set_format( choice( (Note.FLOWED,Note.FORMATTED))) o.set_format( choice( (Note.FLOWED,Note.FORMATTED)))
o.set_type(type) o.set_type(type)
if issubclass(o.__class__,gen.lib.notebase.NoteBase): if issubclass(o.__class__, NoteBase):
while randint(0,1) == 1: while randint(0,1) == 1:
if not self.generated_notes or randint(0,10) == 1: if not self.generated_notes or randint(0,10) == 1:
n = Note() n = Note()
@ -1667,15 +1680,15 @@ class TestcaseGenerator(tool.BatchTool):
while randint(0,1) == 1: while randint(0,1) == 1:
o.add_alternate_locations( self.fill_object( Location())) o.add_alternate_locations( self.fill_object( Location()))
if issubclass(o.__class__,gen.lib.placebase.PlaceBase): if issubclass(o.__class__, PlaceBase):
if randint(0,1) == 1: if randint(0,1) == 1:
o.set_place_handle( self.rand_place()) o.set_place_handle( self.rand_place())
if issubclass(o.__class__,gen.lib.primaryobj.BasicPrimaryObject): if issubclass(o.__class__, BasicPrimaryObject):
if randint(0,1) == 1: if randint(0,1) == 1:
o.set_gramps_id( self.rand_text(self.SHORT)) o.set_gramps_id( self.rand_text(self.SHORT))
if issubclass(o.__class__,gen.lib.privacybase.PrivacyBase): if issubclass(o.__class__, PrivacyBase):
o.set_privacy( randint(0,5) == 1) o.set_privacy( randint(0,5) == 1)
if isinstance(o,RepoRef): if isinstance(o,RepoRef):
@ -1711,7 +1724,7 @@ class TestcaseGenerator(tool.BatchTool):
self.fill_object(r) self.fill_object(r)
o.add_repo_reference( r) o.add_repo_reference( r)
if issubclass(o.__class__,gen.lib.citationbase.CitationBase): if issubclass(o.__class__, CitationBase):
while randint(0,1) == 1: while randint(0,1) == 1:
if not self.generated_citations or randint(1,10) == 1: if not self.generated_citations or randint(1,10) == 1:
s = Citation() s = Citation()
@ -1737,11 +1750,11 @@ class TestcaseGenerator(tool.BatchTool):
# o.set_date_object( d) # o.set_date_object( d)
o.set_confidence_level(choice(list(conf_strings.keys()))) o.set_confidence_level(choice(list(conf_strings.keys())))
if issubclass(o.__class__,gen.lib.tagbase.TagBase): if issubclass(o.__class__, TagBase):
if randint(0,1) == 1: if randint(0,1) == 1:
o.set_tag_list(self.rand_tags()) o.set_tag_list(self.rand_tags())
if issubclass(o.__class__,gen.lib.urlbase.UrlBase): if issubclass(o.__class__, UrlBase):
while randint(0,1) == 1: while randint(0,1) == 1:
u = Url() u = Url()
self.fill_object(u) self.fill_object(u)
@ -1783,16 +1796,16 @@ class TestcaseGenerator(tool.BatchTool):
event_ref.set_reference_handle(event_h) event_ref.set_reference_handle(event_h)
return (year, event_ref) return (year, event_ref)
def rand_type( self, list): def rand_type(self, gtype):
if issubclass( list.__class__, GrampsType): if issubclass(gtype.__class__, GrampsType):
map = list.get_map() map = gtype.get_map()
key = choice(list(map.keys())) key = choice(list(map.keys()))
if key == list.get_custom(): if key == gtype.get_custom():
value = self.rand_text(self.SHORT) value = self.rand_text(self.SHORT)
else: else:
value = '' value = ''
list.set( (key,value)) gtype.set((key, value))
return list return gtype
def rand_place( self): def rand_place( self):
if not self.generated_places or randint(0,10) == 1: if not self.generated_places or randint(0,10) == 1: