Feature/mock stack (#188)

* Mock Stack

* mocking stack

Author: pgajek2

force-app/main/default/classes/main/standard-soql/SOQL.cls

@@ -380,8 +380,11 @@ public virtual inherited sharing class SOQL implements Queryable {
 
     @TestVisible
     private static Mockable mock(String mockId) {
-        queryIdToMock.put(mockId, new SoqlMock());
-        return queryIdToMock.get(mockId);
+        if (!SOQL.queryIdToMock.containsKey(mockId)) {   
+            SOQL.queryIdToMock.put(mockId, new List<SoqlMock>());
+        }
+        SOQL.queryIdToMock.get(mockId).add(new SoqlMock());
+        return SOQL.queryIdToMock.get(mockId).get(SOQL.queryIdToMock.get(mockId).size() - 1);
     }
 
     @TestVisible
@@ -402,25 +405,25 @@ public virtual inherited sharing class SOQL implements Queryable {
 
     @TestVisible // deprecated
     private static void setMock(String mockId, SObject record) {
-        queryIdToMock.put(mockId, new SoqlMock().useLegacyMockingBehavior());
-        queryIdToMock.get(mockId).thenReturn(record);
+        SOQL.queryIdToMock.put(mockId, new List<SoqlMock>{ new SoqlMock().useLegacyMockingBehavior() });
+        SOQL.queryIdToMock.get(mockId).get(SOQL.queryIdToMock.get(mockId).size() - 1).thenReturn(record);
     }
 
     @TestVisible // deprecated
     private static void setMock(String mockId, List<SObject> records) {
-        queryIdToMock.put(mockId, new SoqlMock().useLegacyMockingBehavior());
-        queryIdToMock.get(mockId).thenReturn(records);
+        SOQL.queryIdToMock.put(mockId, new List<SoqlMock>{ new SoqlMock().useLegacyMockingBehavior() });
+        SOQL.queryIdToMock.get(mockId).get(SOQL.queryIdToMock.get(mockId).size() - 1).thenReturn(records);
     }
 
     @TestVisible // deprecated
     private static void setCountMock(String mockId, Integer amount) {
-        queryIdToMock.put(mockId, new SoqlMock().useLegacyMockingBehavior());
-        queryIdToMock.get(mockId).thenReturn(amount);
+        SOQL.queryIdToMock.put(mockId, new List<SoqlMock>{ new SoqlMock().useLegacyMockingBehavior() });
+        SOQL.queryIdToMock.get(mockId).get(SOQL.queryIdToMock.get(mockId).size() - 1).thenReturn(amount);
     }
 
     // Implementation
 
-    private static Map<String, SoqlMock> queryIdToMock = new Map<String, SoqlMock>();
+    private static Map<String, List<SoqlMock>> queryIdToMock = new Map<String, List<SoqlMock>>();
 
     private static Binder binder = new Binder();
     private static Integer syncQueriesIssued = 0;
@@ -896,7 +899,7 @@ public virtual inherited sharing class SOQL implements Queryable {
     }
 
     public Queryable mockId(String queryIdentifier) {
-        this.executor.mock(queryIdToMock.get(queryIdentifier));
+        this.executor.mock(SOQL.queryIdToMock.get(queryIdentifier));
         return this;
     }
 
@@ -2505,7 +2508,7 @@ public virtual inherited sharing class SOQL implements Queryable {
         private AccessLevel accessMode;
         private AccessType accessType;
         private SoqlBuilder builder;
-        private SoqlMock mock;
+        private List<SoqlMock> mocks = new List<SoqlMock>();
 
         public Executor(SoqlBuilder builder) {
             this.builder = builder;
@@ -2529,8 +2532,8 @@ public virtual inherited sharing class SOQL implements Queryable {
             this.accessMode = accessMode;
         }
 
-        public void mock(SoqlMock mock) {
-            this.mock = mock;
+        public void mock(List<SoqlMock> mocks) {
+            this.mocks = mocks;
         }
 
         public SObject toObject() {
@@ -2550,8 +2553,8 @@ public virtual inherited sharing class SOQL implements Queryable {
         public List<SObject> toList() {
             this.incrementQueryIssued();
 
-            if (this.mock != null) {
-                return this.mock.sObjectMock.get(this.builder.fields, this.builder.subQueries);
+            if (!this.mocks.isEmpty()) {
+                return this.getMockedListProxy();
             }
 
             if (this.accessType == null) {
@@ -2564,26 +2567,47 @@ public virtual inherited sharing class SOQL implements Queryable {
             ).getRecords();
         }
 
+        private List<SObject> getMockedListProxy() {
+            if (this.mocks.size() == 1) {
+                return this.mocks[0].sObjectMock.get(this.builder.fields, this.builder.subQueries);
+            }
+            return this.mocks.remove(0).sObjectMock.get(this.builder.fields, this.builder.subQueries);
+        }
+
         public List<AggregateResultProxy> toAggregatedProxy() {
             this.incrementQueryIssued();
 
-            if (this.mock != null) {
-                return this.mock.aggregateResultMock.get();
+            if (!this.mocks.isEmpty()) {
+                return this.getMockedAggregateProxy();
             }
 
             return new AggregateResultProxys().add((List<AggregateResult>) this.toList()).get();
         }
 
+        private List<AggregateResultProxy> getMockedAggregateProxy() {
+            if (this.mocks.size() == 1) {
+                return this.mocks[0].aggregateResultMock.get();
+            }
+            return this.mocks.remove(0).aggregateResultMock.get();
+        }
+
         public Integer toInteger() {
             this.incrementQueryIssued();
 
-            if (this.mock != null) {
-                return this.mock.countMock.get();
+            if (!this.mocks.isEmpty()) {
+                return this.getMockedCount();
             }
 
             return this.sharingExecutor.toInteger(this.builder.toString(), binder.getBindingMap(), this.accessMode);
         }
 
+        private Integer getMockedCount() {
+            if (this.mocks.size() == 1) {
+                return this.mocks[0].countMock.get();
+            }
+            return this.mocks.remove(0).countMock.get();
+        }
+
         public Database.QueryLocator toQueryLocator() {
             this.incrementQueryIssued();
             return this.sharingExecutor.toQueryLocator(this.builder.toString(), binder.getBindingMap(), this.accessMode);

force-app/main/default/classes/main/standard-soql/SOQL_Test.cls

@@ -3854,6 +3854,28 @@ private class SOQL_Test {
         Assert.areEqual(0, account.Opportunities.size(), 'The size of the mocked opportunities should be 0, because it was not requested.');
     }
 
+    @IsTest
+    static void mockStack() {
+        // Setup
+        SOQL.mock('mockingQuery').thenReturn(new Account(Name = 'Test 1'));
+        SOQL.mock('mockingQuery').thenReturn(new Account(Name = 'Test 2'));
+        SOQL.mock('mockingQuery').thenReturn(new Account(Name = 'Test 3'));
+
+        // Test
+        SOQL.Queryable query = SOQL.of(Account.SObjectType).with(Account.Name).mockId('mockingQuery');
+
+        Account acc1 = (Account) query.toObject();
+        Account acc2 = (Account) query.toObject();
+        Account acc3 = (Account) query.toObject();
+        Account acc4 = (Account) query.toObject();
+
+        // Verify
+        Assert.areEqual('Test 1', acc1.Name, 'The returned account name should match the expected one.');
+        Assert.areEqual('Test 2', acc2.Name, 'The returned account name should match the expected one.');
+        Assert.areEqual('Test 3', acc3.Name, 'The returned account name should match the expected one.');
+        Assert.areEqual('Test 3', acc4.Name, 'The returned account name should match the expected one.');
+    }
+
     @IsTest
     static void toId() {
         // Setup