Make sure method return non-null values

apex-enterprise-patterns · May 27, 2022 · 4ad786c · 4ad786c
1 parent a64b4a2
commit 4ad786c
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/sfdx-source/apex-common/main/classes/fflib_SObjects.cls b/sfdx-source/apex-common/main/classes/fflib_SObjects.cls
@@ -161,11 +161,13 @@ public virtual class fflib_SObjects
 	 * @return Return a set with all the Id values of the given field
 	 */
 	@TestVisible
-	protected Set<Id> getIdFieldValues(Schema.SObjectField field)
+	protected virtual Set<Id> getIdFieldValues(Schema.SObjectField field)
 	{
 		Set<Id> result = new Set<Id>();
 		for (SObject record : getRecords())
 		{
+			if (record.get(field) == null) continue;
+
 			result.add((Id) record.get(field));
 		}
 		return result;
@@ -177,11 +179,13 @@ public virtual class fflib_SObjects
 	 * @return Return a set with all the String values of the given field
 	 */
 	@TestVisible
-	protected Set<String> getStringFieldValues(Schema.SObjectField field)
+	protected virtual Set<String> getStringFieldValues(Schema.SObjectField field)
 	{
 		Set<String> result = new Set<String>();
 		for (SObject record : getRecords())
 		{
+			if (record.get(field) == null) continue;
+
 			result.add((String) record.get(field));
 		}
 		return result;