bugfix for Issue 221

sfdx-source/apex-common/main/classes/fflib_SObjectUnitOfWork.cls

@@ -707,6 +707,7 @@ public virtual class fflib_SObjectUnitOfWork
 	{
 		for (Schema.SObjectType sObjectType : m_sObjectTypes)
 		{
+            m_relationships.get(sObjectType.getDescribe().getName()).resolve();
 			m_dml.dmlUpdate(m_dirtyMapByType.get(sObjectType.getDescribe().getName()).values());
 		}
 	}

sfdx-source/apex-common/test/classes/fflib_SObjectUnitOfWorkTest.cls

@@ -608,6 +608,63 @@ private with sharing class fflib_SObjectUnitOfWorkTest
         );
     }
 
+    @IsTest
+    private static void testRegisterDirtyWithRelationship() {
+        // GIVEN an existing opportunity
+        Opportunity existingOpp = new Opportunity(
+            Id = fflib_IDGenerator.generate(Schema.Opportunity.SObjectType),
+            Name = 'Existing Opportunity',
+            StageName = 'Closed',
+            CloseDate = System.today()
+        );
+        // AND an existing Account to which the existing opportunity will be related to
+        Account newAccount = new Account(
+            Name = 'New Account'
+        );
+
+        // WHEN
+        Test.startTest();
+        MockDMLWithInsertIds mockDML = new MockDMLWithInsertIds();
+        List<Schema.SObjectType> mySobjects = new List<Schema.SObjectType>{
+            Opportunity.SObjectType,
+            Account.SObjectType
+        };
+        fflib_SObjectUnitOfWork uow = new fflib_SObjectUnitOfWork(mySobjects, mockDML);
+        uow.registerNew(newAccount);
+        uow.registerDirty(existingOpp, Opportunity.AccountId, newAccount);
+        uow.commitWork();
+        Test.stopTest();
+
+        // THEN
+        System.Assert.isTrue(
+            new fflib_MatcherDefinitions.SObjectsWith(
+                new List<Map<SObjectField, Object>>{
+                    new Map<SObjectField, Object>
+                    {
+                        Account.Id => newAccount.Id,
+                        Account.Name => 'New Account'
+                    }
+                }
+            ).matches(mockDML.recordsForInsert),
+            'The new account record does not match'
+        );
+
+        // AND
+        System.Assert.isTrue(
+            new fflib_MatcherDefinitions.SObjectsWith(
+                new List<Map<SObjectField, Object>>{
+                    new Map<SObjectField, Object>
+                    {
+                        Opportunity.Id => existingOpp.Id,
+                        Opportunity.Name => 'Existing Opportunity',
+                        Opportunity.StageName => 'Closed',
+                        Opportunity.AccountId => newAccount.Id
+                    }
+                }
+            ).matches(mockDML.recordsForUpdate),
+            'The opportunity record should be related to the new Account'
+        );
+    }
     @IsTest
     private static void testRegisterUpsert() {
         Opportunity existingOpp = new Opportunity(
@@ -848,6 +905,43 @@ private with sharing class fflib_SObjectUnitOfWorkTest
         }
     }
 
+    private class MockDMLWithInsertIds implements fflib_SObjectUnitOfWork.IDML
+    {
+        public List<SObject> recordsForInsert = new List<SObject>();
+        public List<SObject> recordsForUpdate = new List<SObject>();
+        public List<SObject> recordsForDelete = new List<SObject>();
+        public List<SObject> recordsForRecycleBin = new List<SObject>();
+        public List<SObject> recordsForEventPublish = new List<SObject>();
+
+        public void dmlInsert(List<SObject> objList)
+        {
+            for (SObject obj : objList) {
+                obj.Id = fflib_IDGenerator.generate(obj.getSObjectType());
+            }
+            this.recordsForInsert.addAll(objList);
+        }
+
+        public void dmlUpdate(List<SObject> objList)
+        {
+            this.recordsForUpdate.addAll(objList);
+        }
+
+        public void dmlDelete(List<SObject> objList)
+        {
+            this.recordsForDelete.addAll(objList);
+        }
+
+        public void eventPublish(List<SObject> objList)
+        {
+            this.recordsForEventPublish.addAll(objList);
+        }
+
+        public void emptyRecycleBin(List<SObject> objList)
+        {
+            this.recordsForRecycleBin.addAll(objList);
+        }
+    }
+
     public class DerivedUnitOfWorkException extends Exception {}
     public class FailDoingWorkException extends Exception {}
 }