Fix bugs in the test case generator
This commit is contained in:
parent
c574f3a767
commit
7b05c1944b
@ -58,6 +58,19 @@ from gramps.gen.lib import (Address, Attribute, AttributeType, ChildRef,
|
||||
Person, PersonRef, Place, RepoRef, Repository, RepositoryType,
|
||||
Source, SourceMediaType, SrcAttribute, SrcAttributeType,
|
||||
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.db import DbTxn
|
||||
from gramps.gen.mime import get_type
|
||||
@ -1507,7 +1520,7 @@ class TestcaseGenerator(tool.BatchTool):
|
||||
def fill_object( self, o):
|
||||
|
||||
|
||||
if issubclass(o.__class__,gen.lib.addressbase.AddressBase):
|
||||
if issubclass(o.__class__, AddressBase):
|
||||
while randint(0,1) == 1:
|
||||
a = Address()
|
||||
self.fill_object(a)
|
||||
@ -1517,7 +1530,7 @@ class TestcaseGenerator(tool.BatchTool):
|
||||
o.set_type( self.rand_type(AttributeType()))
|
||||
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:
|
||||
a = Attribute()
|
||||
self.fill_object(a)
|
||||
@ -1529,7 +1542,7 @@ class TestcaseGenerator(tool.BatchTool):
|
||||
if randint(0,3) == 1:
|
||||
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:
|
||||
(y,d) = self.rand_date()
|
||||
o.set_date_object( d)
|
||||
@ -1538,7 +1551,7 @@ class TestcaseGenerator(tool.BatchTool):
|
||||
if randint(0,1) == 1:
|
||||
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()))
|
||||
|
||||
if isinstance(o,Family):
|
||||
@ -1551,7 +1564,7 @@ class TestcaseGenerator(tool.BatchTool):
|
||||
if randint(0,1) == 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:
|
||||
ldsord = LdsOrd()
|
||||
self.fill_object( ldsord)
|
||||
@ -1569,7 +1582,7 @@ class TestcaseGenerator(tool.BatchTool):
|
||||
if randint(0,1) == 1:
|
||||
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:
|
||||
o.set_street( self.rand_text(self.SHORT))
|
||||
if randint(0,1) == 1:
|
||||
@ -1585,7 +1598,7 @@ class TestcaseGenerator(tool.BatchTool):
|
||||
if randint(0,1) == 1:
|
||||
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
|
||||
while randint(0,10) == 1:
|
||||
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_type(type)
|
||||
|
||||
if issubclass(o.__class__,gen.lib.notebase.NoteBase):
|
||||
if issubclass(o.__class__, NoteBase):
|
||||
while randint(0,1) == 1:
|
||||
if not self.generated_notes or randint(0,10) == 1:
|
||||
n = Note()
|
||||
@ -1667,15 +1680,15 @@ class TestcaseGenerator(tool.BatchTool):
|
||||
while randint(0,1) == 1:
|
||||
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:
|
||||
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:
|
||||
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)
|
||||
|
||||
if isinstance(o,RepoRef):
|
||||
@ -1711,7 +1724,7 @@ class TestcaseGenerator(tool.BatchTool):
|
||||
self.fill_object(r)
|
||||
o.add_repo_reference( r)
|
||||
|
||||
if issubclass(o.__class__,gen.lib.citationbase.CitationBase):
|
||||
if issubclass(o.__class__, CitationBase):
|
||||
while randint(0,1) == 1:
|
||||
if not self.generated_citations or randint(1,10) == 1:
|
||||
s = Citation()
|
||||
@ -1737,11 +1750,11 @@ class TestcaseGenerator(tool.BatchTool):
|
||||
# o.set_date_object( d)
|
||||
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:
|
||||
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:
|
||||
u = Url()
|
||||
self.fill_object(u)
|
||||
@ -1783,16 +1796,16 @@ class TestcaseGenerator(tool.BatchTool):
|
||||
event_ref.set_reference_handle(event_h)
|
||||
return (year, event_ref)
|
||||
|
||||
def rand_type( self, list):
|
||||
if issubclass( list.__class__, GrampsType):
|
||||
map = list.get_map()
|
||||
key = choice( list(map.keys()))
|
||||
if key == list.get_custom():
|
||||
def rand_type(self, gtype):
|
||||
if issubclass(gtype.__class__, GrampsType):
|
||||
map = gtype.get_map()
|
||||
key = choice(list(map.keys()))
|
||||
if key == gtype.get_custom():
|
||||
value = self.rand_text(self.SHORT)
|
||||
else:
|
||||
value = ''
|
||||
list.set( (key,value))
|
||||
return list
|
||||
gtype.set((key, value))
|
||||
return gtype
|
||||
|
||||
def rand_place( self):
|
||||
if not self.generated_places or randint(0,10) == 1:
|
||||
|
Loading…
Reference in New Issue
Block a user