Resolve the relationships on persist of fabricated sobjects

Author: rob-baillie-ortoo

framework/default/sobject-fabricator-extensions/classes/ortoo_FabricatedSObjectRegister.cls

@@ -45,7 +45,6 @@ public class ortoo_FabricatedSObjectRegister {
 
             sobjectsByFabricated.put( thisFabricatedObject, objectToStore );
         }
-        system.debug( sobjectsByFabricated.values() );
 
         // work out the order to do things in, using the directed graph
         List<Object> childToParentTypes = graph.generateSorted();
@@ -102,21 +101,19 @@ public class ortoo_FabricatedSObjectRegister {
             Sobject parentSobject = sobjectsByFabricated.get( parent );
             Sobject childSobject  = sobjectsByFabricated.get( child );
 
-            system.debug( 'registering the relationship: ' + childSobject + ' : ' + relationship + ' : ' + parentSobject );
-
             uow.registerRelationship( childSobject, relationship, parentSobject );
         }
     }
 
     public static Relationship buildChildOfRelationship( sfab_FabricatedSObject child, String relationship, sfab_FabricatedSObject parent )
     {
-        // TODO: resolve the passed in relationship
-        return new Relationship( child, Contact.AccountId, parent );
+        SobjectField relationshipField = new sfab_ObjectDescriber().getFieldForParentRelationship( child.getSobjectName(), relationship );
+        return new Relationship( child, relationshipField, parent );
     }
 
     public static Relationship buildParentOfRelationship( sfab_FabricatedSObject parent, String relationship, sfab_FabricatedSObject child )
     {
-        // TODO: resolve the passed in relationship
-        return new Relationship( child, Contact.AccountId, parent );
+        SobjectField relationshipField = new sfab_ObjectDescriber().getFieldForChildRelationship( parent.getSobjectName(), relationship );
+        return new Relationship( child, relationshipField, parent );
     }
 }

framework/default/sobject-fabricator/classes/sfab_ObjectDescriber.cls

@@ -10,6 +10,16 @@ public with sharing class sfab_ObjectDescriber {
         return getObjectDescriber( objectName )?.getField( fieldName );
     }
 
+    public SObjectField getFieldForChildRelationship( String objectName, String relationshipName ) {
+        ensureSingleObjectDescriberIsInitialised( objectName );
+        return getObjectDescriber( objectName )?.getChildRelationshipNamed( relationshipName )?.getField();
+    }
+
+    public SObjectField getFieldForParentRelationship( String objectName, String relationshipName ) {
+        ensureSingleObjectDescriberIsInitialised( objectName );
+        return getObjectDescriber( objectName )?.getParentRelationshipNamed( relationshipName )?.getField();
+    }
+
     public Boolean objectHasSimpleFieldNamed( String objectName, String fieldName ) {
         ensureSingleObjectDescriberIsInitialised( objectName );
         return getObjectDescriber( objectName )?.hasSimpleFieldNamed( fieldName );
@@ -235,6 +245,10 @@ public with sharing class sfab_ObjectDescriber {
             return fieldDescribe.getName();
         }
 
+        public SobjectField getField() {
+            return fieldDescribe.getSobjectField();
+        }
+
         public List<String> getTargetObjectTypes() {
 
             if ( !isValid() ) {
@@ -295,6 +309,9 @@ public with sharing class sfab_ObjectDescriber {
             return String.valueOf( schemaRelationship.getChildSObject() );
         }
 
+        public SobjectField getField() {
+            return schemaRelationship.getField();
+        }
         // Some don't have names - and if they don't have names they can't possibly be
         // set by the fabricator, so must be invalid.
         public Boolean isValid() {