wim/fflib-apex-extensions
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add Factory method for records based on an Id field

Browse Source 
Signed-off-by: WimVelzeboer <wimvelzeboer@protonmail.com>
This commit is contained in:
WimVelzeboer
2022-03-16 21:16:25 +00:00
parent ac1f7bbac0
commit ad75a9b790
2 changed files with 48 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
sfdx-source/apex-extensions/main/default/classes/factories/fflib_SObjectFactory.cls
+13
View File
@@ -40,5 +40,18 @@ public abstract inherited sharing class fflib_SObjectFactory implements fflib_IS
return result; return result;
} }
public virtual List<SObject> generateSObjectByIdField(Schema.SObjectField sObjectField, Set<Id> ids)
{
SObjectType sObjectType = getSObjectType();
List<SObject> result = new List<SObject>();
for (Id recordId : ids)
{
SObject newSObject = sObjectType.newSObject();
newSObject.put(sObjectField, recordId);
result.add(newSObject);
}
return result;
}
abstract Schema.SObjectType getSObjectType(); abstract Schema.SObjectType getSObjectType();
} }
sfdx-source/apex-extensions/tests/classes/factories/fflib_SObjectFactoryTest.cls
+35 -9
View File
@@ -32,7 +32,26 @@
private class fflib_SObjectFactoryTest private class fflib_SObjectFactoryTest
{ {
@IsTest @IsTest
static void itShouldGenerateAccountRecords() static void itShouldGenerateContactRecords()
{
// GIVEN a set of Ids
final Set<Id> ids = new Set<Id>
{
fflib_IDGenerator.generate(Schema.Contact.SObjectType),
fflib_IDGenerator.generate(Schema.Contact.SObjectType),
fflib_IDGenerator.generate(Schema.Contact.SObjectType)
};
// WHEN we generate Contact records
List<Contact> result = (List<Contact>) new ContactsFactory().generateById(ids);
// THEN there should be three Contact records
System.assertEquals(3, result.size(), 'Not enough records created');
System.assert(new Map<Id, Contact>(result).keySet().containsAll(ids), 'Records should contain the given ids');
}
@IsTest
static void itShouldGenerateContactRecordsWithAccountId()
{ {
// GIVEN a set of Ids // GIVEN a set of Ids
final Set<Id> ids = new Set<Id> final Set<Id> ids = new Set<Id>
@@ -42,24 +61,31 @@ private class fflib_SObjectFactoryTest
fflib_IDGenerator.generate(Schema.Account.SObjectType) fflib_IDGenerator.generate(Schema.Account.SObjectType)
}; };
// WHEN we generate Account records // WHEN we generate Contact records
List<Account> result = (List<Account>) new AccountsFactory().generateById(ids); List<Contact> result = (List<Contact>) new ContactsFactory().generateByAccountId(ids);
// THEN there should be three Account records // THEN there should be three Contact records
System.assertEquals(3, result.size(), 'Not enough records created'); System.assertEquals(3, result.size(), 'Not enough records created');
System.assert(new Map<Id, Account>(result).keySet().containsAll(ids), 'Records should contain the given ids'); Set<Id> accountIds = new Set<Id>();
for (Contact record : result) accountIds.add(record.AccountId);
System.assert(accountIds.containsAll(ids), 'Records should contain the given ids');
} }
private class AccountsFactory extends fflib_SObjectFactory private class ContactsFactory extends fflib_SObjectFactory
{ {
public List<Account> generateById(Set<Id> accountIds) public List<Contact> generateById(Set<Id> ContactIds)
{ {
return (List<Account>) generateSObjectById(accountIds); return (List<Contact>) generateSObjectById(ContactIds);
}
public List<Contact> generateByAccountId(Set<Id> ContactIds)
{
return (List<Contact>) generateSObjectByIdField(Schema.Contact.AccountId, ContactIds);
} }
public SObjectType getSObjectType() public SObjectType getSObjectType()
{ {
return Schema.Account.SObjectType; return Schema.Contact.SObjectType;
} }
} }
} }