GEPS045 Fix Testcasegenerator for enhanced places

Author: prculley

gramps/plugins/tool/testcasegenerator.py

@@ -253,6 +253,30 @@ class TestcaseGenerator(tool.BatchTool):
         ]
     )
 
+    PLACETYPES = [
+        "Unknown",  # -1 original value
+        "Country",  # 1
+        "State",  # 2
+        "County",  # 3
+        "City",  # 4
+        "Parish",  # 5
+        "Locality",  # 6
+        "Street",  # 7
+        "Province",  # 8
+        "Region",  # 9
+        "Department",  # 10
+        "Neighborhood",  # 11
+        "District",  # 12
+        "Borough",  # 13
+        "Municipality",  # 14
+        "Town",  # 15
+        "Village",  # 16
+        "Hamlet",  # 17
+        "Farm",  # 18
+        "Building",  # 19
+        "Number",
+    ]  # 20
+
     def __init__(self, dbstate, user, options_class, name, callback=None):
         uistate = user.uistate
         if uistate:
@@ -480,6 +504,33 @@ def run_tool(self, cli=False):
         if not cli:
             self.top.destroy()
 
+    def generate_persons(self):
+        """This creates the persons and families"""
+        with DbTxn(
+            _("Testcase generator step %d") % self.transaction_count, self.db
+        ) as self.trans, self.progress(
+            _("Generating testcases"), _("Generating families"), self.max_person_count
+        ) as self.progress_step:
+            self.person_count = 0
+
+            while True:
+                if not self.persons_todo:
+                    pers_h = self.generate_person(0)
+                    self.persons_todo.append(pers_h)
+                    self.parents_todo.append(pers_h)
+                person_h = self.persons_todo.pop(0)
+                self.generate_family(person_h)
+                if _randint(0, 3) == 0:
+                    self.generate_family(person_h)
+                if _randint(0, 7) == 0:
+                    self.generate_family(person_h)
+                if self.person_count > self.max_person_count:
+                    break
+                for child_h in self.parents_todo:
+                    self.generate_parents(child_h)
+                    if self.person_count > self.max_person_count:
+                        break
+
     def generate_data_errors(self, step):
         """This generates errors in the database to test src/plugins/tool/Check
         The module names correspond to the checking methods in
@@ -1421,13 +1472,17 @@ def create_all_possible_citations(self, c_h_list, name, message):
 
         place = Place()
         place.set_title(message)
+        # Place
         place.add_citation(_choice(c_h_list))
-        pname = PlaceName(value='All Attribute Test')
+        # Place : Name
+        pname = PlaceName(value="All Attribute Test")
         pname.add_citation(_choice(c_h_list))
         place.add_name(pname)
-        ptype = PlaceType(value=PlaceType.BOROUGH)
+        # Place : Type
+        ptype = PlaceType("Borough")
         ptype.add_citation(_choice(c_h_list))
         place.add_type(ptype)
+        # Place : Attribute
         place.add_attribute(att)
         # Place : MediaRef
         mref = MediaRef()
@@ -1440,7 +1495,8 @@ def create_all_possible_citations(self, c_h_list, name, message):
         att.add_citation(_choice(c_h_list))
         mref.add_attribute(att)
         place.add_media_reference(mref)
-        place.add_attribute(att)
+        # Place : EventRef
+        place.add_event_ref(eref)
         self.db.add_place(place, self.trans)
 
         ref = Repository()
@@ -1663,6 +1719,7 @@ def generate_family(self, person1_h):
         if person2_h and _randint(0, 2) > 0:
             self.parents_todo.append(person2_h)
 
+        self.transaction_count += 1
         fam = Family()
         self.add_defaults(fam)
         if person1_h:
@@ -1766,6 +1823,7 @@ def generate_parents(self, child_h):
         if _randint(0, 2) > 1:
             self.parents_todo.append(person2_h)
 
+        self.transaction_count += 1
         fam = Family()
         self.add_defaults(fam)
         fam.set_father_handle(person1_h)
@@ -2200,7 +2258,7 @@ def generate_place(self):
                 # skip some levels in the place hierarchy
                 continue
             place = Place()
-            place.set_type(PlaceType(type_num))
+            place.set_type(self.PLACETYPES[type_num])
             if parent_handle is not None:
                 self.add_parent_place(place, parent_handle)
             if type_num > 1 and _randint(1, 3) == 1:
@@ -2209,6 +2267,7 @@ def generate_place(self):
                 if parent_handle is not None:
                     self.add_parent_place(place, parent_handle)
             self.fill_object(place)
+            place.group = place.get_type().get_probable_group()
             self.db.add_place(place, self.trans)
             parent_handle = place.get_handle()
             self.generated_places.append(place.get_handle())