Enable updating SolrTestDataAssemblers' profiles

This makes a small change to how SolrTestDataAssembler profiles are
generated (adding a helper method `make_profile`) to expose a way to add
new profiles to an existing assembler on the fly.

Author: jthomale

django/sierra/utils/test_helpers/fixture_factories.py

@@ -202,19 +202,33 @@ def __init__(self, solr_types, global_unique_fields, gen_factory,
             arg become the id strings you use to create and access
             records using each profile in the other object methods.
             """
-            self.profiles = {}
-            self.records = {}
             self.gen_factory = gen_factory
-            for rectype, pdef in profile_definitions.items():
-                profile = sf.SolrProfile(
-                    rectype, pdef['conn'], user_fields=pdef['user_fields'],
-                    inclusive=pdef.get('inclusive', True),
-                    unique_fields=global_unique_fields, solr_types=solr_types,
-                    gen_factory=gen_factory,
-                    default_field_gens=pdef.get('field_gens', tuple())
-                )
-                self.profiles[rectype] = profile
-                self.records[rectype] = tuple()
+            self.default_solr_types = solr_types
+            self.default_unique_fields = global_unique_fields
+            self.records = {}
+            self.profiles = {
+                rectype: self.make_profile(rectype, **profile_def)
+                    for rectype, profile_def in profile_definitions.items()
+            }
+
+        def make_profile(self, rectype, conn=None, user_fields=None,
+                         inclusive=True, unique_fields=None, gen_factory=None,
+                         solr_types=None, field_gens=None):
+            """
+            Generate a solr_factories.SolrProfile object using the
+            provided kwargs.
+            """
+            unique_fields = unique_fields or self.default_unique_fields
+            gen_factory = gen_factory or self.gen_factory
+            solr_types = solr_types or self.default_solr_types
+            field_gens = field_gens or tuple()
+            profile = sf.SolrProfile(
+                rectype, conn, user_fields=user_fields, inclusive=inclusive,
+                unique_fields=unique_fields, solr_types=solr_types,
+                gen_factory=gen_factory, default_field_gens=field_gens
+            )
+            self.records[rectype] = self.records.get(rectype, tuple())
+            return profile
 
         def make(self, rectype, number, context=None, **field_gens):
             """
@@ -288,7 +302,7 @@ def load_static_test_data(self, rectype, test_data, id_field='id',
             """
             context = context or tuple()
             records = tuple()
-            gens = self.gen_factory
+            gens = self.profiles[rectype].gen_factory
             for rec_id, record in test_data:
                 datagens = {k: gens.static(v) for k, v in record.items()}
                 datagens[id_field] = gens.static(rec_id)

django/sierra/utils/test_helpers/solr_factories.py

@@ -363,7 +363,7 @@ def __init__(self, name, conn=None, schema=None, user_fields=None,
         """
         unique_fields = unique_fields or []
         solr_types = solr_types or type(self).DEFAULT_SOLR_FIELD_TYPE_MAPPING
-        gen_factory = gen_factory or SolrDataGenFactory()
+        self.gen_factory = gen_factory or SolrDataGenFactory()
         self.conn = conn
         schema = schema or self.fetch_schema(conn)