diff --git a/fflib/src/classes/fflib_SObjectMocks.cls b/fflib/src/classes/fflib_SObjectMocks.cls index 634285dc666..6685fc465da 100644 --- a/fflib/src/classes/fflib_SObjectMocks.cls +++ b/fflib/src/classes/fflib_SObjectMocks.cls @@ -186,98 +186,47 @@ public class fflib_SObjectMocks public void registerNew(SObject record) { - if (mocks.Verifying) - { - mocks.verifyMethodCall(this, 'registerNew', new List {record}); - } - else - { - mocks.recordMethod(this, 'registerNew', new List {record}); - } + mocks.mockVoidMethod(this, 'registerNew', new List {record}); } public void registerNew(List records) { - for(SObject record : records) - { - registerNew(record); - } + mocks.mockVoidMethod(this, 'registerNew', new List {records}); } public void registerNew(SObject record, Schema.sObjectField relatedToParentField, SObject relatedToParentRecord) { - if (mocks.Verifying) - { - mocks.verifyMethodCall(this, 'registerNew', new List {record, relatedToParentField, relatedToParentRecord}); - } - else - { - mocks.recordMethod(this, 'registerNew', new List {record, relatedToParentField, relatedToParentRecord}); - } + mocks.mockVoidMethod(this, 'registerNew', new List {record, relatedToParentField, relatedToParentRecord}); } public void registerRelationship(SObject record, Schema.sObjectField relatedToField, SObject relatedTo) { - if (mocks.Verifying) - { - mocks.verifyMethodCall(this, 'registerRelationship', new List {record, relatedToField, relatedTo}); - } - else - { - mocks.recordMethod(this, 'registerRelationship', new List {record, relatedToField, relatedTo}); - } + mocks.mockVoidMethod(this, 'registerRelationship', new List {record, relatedToField, relatedTo}); } public void registerDirty(SObject record) { - if (mocks.Verifying) - { - mocks.verifyMethodCall(this, 'registerDirty', new List {record}); - } - else - { - mocks.recordMethod(this, 'registerDirty', new List {record}); - } + mocks.mockVoidMethod(this, 'registerDirty', new List {record}); } public void registerDirty(List records) { - for(SObject record : records) - { - registerDirty(record); - } + mocks.mockVoidMethod(this, 'registerDirty', new List {records}); } public void registerDeleted(SObject record) { - if (mocks.Verifying) - { - mocks.verifyMethodCall(this, 'registerDeleted', new List {record}); - } - else - { - mocks.recordMethod(this, 'registerDeleted', new List {record}); - } + mocks.mockVoidMethod(this, 'registerDeleted', new List {record}); } public void registerDeleted(List records) { - for(SObject record : records) - { - registerDeleted(record); - } + mocks.mockVoidMethod(this, 'registerDeleted', new List {record}); } public void commitWork() { - if (mocks.Verifying) - { - mocks.verifyMethodCall(this, 'commitWork', new List {}); - } - else - { - mocks.recordMethod(this, 'commitWork', new List {}); - } + mocks.mockVoidMethod(this, 'commitWork', new List {}); } } }