Fix bugs in the test case generator

This commit is contained in:
Nick Hall 2014-01-21 00:05:03 +00:00
parent c574f3a767
commit 7b05c1944b

View File

@ -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: