calculate total number of queries issued [DRAFT] (#134)

* calculate total number of queries issued

Signed-off-by: Piotr PG Gajek <[email protected]>

* queryIssued

Signed-off-by: Piotr PG Gajek <[email protected]>

* QueryException

Signed-off-by: Piotr PG Gajek <[email protected]>

* synchronous query issued count

Signed-off-by: Piotr PG Gajek <[email protected]>

---------

Signed-off-by: Piotr PG Gajek <[email protected]>

Author: pgajek2

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

@@ -334,6 +334,7 @@ public virtual inherited sharing class SOQL implements Queryable {
 
     private static Mock mock = new Mock();
     private static Binder binder = new Binder();
+    private static Integer syncQueriesIssued = 0;
 
     private SoqlBuilder builder;
     private Executor executor;
@@ -2216,6 +2217,8 @@ public virtual inherited sharing class SOQL implements Queryable {
         }
 
         public List<SObject> toList() {
+            incrementQueryIssued();
+
             if (mock.hasMock(mockId)) {
                 return mock.getSObjectsMock(mockId);
             }
@@ -2231,6 +2234,8 @@ public virtual inherited sharing class SOQL implements Queryable {
         }
 
         public Integer toInteger() {
+            incrementQueryIssued();
+
             if (mock.hasCountMock(mockId)) {
                 return mock.getCountMock(mockId);
             }
@@ -2239,8 +2244,22 @@ public virtual inherited sharing class SOQL implements Queryable {
         }
 
         public Database.QueryLocator toQueryLocator() {
+            incrementQueryIssued();
+
             return sharingExecutor.toQueryLocator(builder.toString(), binder.getBindingMap(), accessMode);
         }
+
+        private void incrementQueryIssued() {
+            if (!Test.isRunningTest() || System.isBatch() || System.isFuture() || System.isQueueable() || System.isScheduled()) {
+                return;
+            }
+
+            SOQL.syncQueriesIssued++;
+
+            if (SOQL.syncQueriesIssued > 100) {
+                throw new QueryException('Too many SOQL queries.');
+            }
+        }
     }
 
     private interface DatabaseQuery {

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

@@ -3480,6 +3480,103 @@ private class SOQL_Test {
         }
     }
 
+    @IsTest
+    static void querySynchronousCloseToLimitWithMocking() {
+        // Setup
+        Exception queryException = null;
+
+        SOQL.setMock('mockingQuery', new List<Account>{
+            new Account(Name = 'Test 1'),
+            new Account(Name = 'Test 2')
+        });
+
+        // Test
+        try {
+            Test.startTest();
+                for (Integer i = 0 ; i < 100 ; i++) {
+                    SOQL.of(Account.SObjectType).mockId('mockingQuery').toList();
+                }
+            Test.stopTest();
+        } catch (QueryException e) {
+            queryException = e;
+        }
+
+        // Verify
+        Assert.isNull(queryException, 'The synchronous query should not have exceeded the limit.');
+    }
+
+    @IsTest
+    static void querySynchronousLimitExceptionWithMocking() {
+        // Setup
+        Exception queryException = null;
+
+        SOQL.setMock('mockingQuery', new List<Account>{
+            new Account(Name = 'Test 1'),
+            new Account(Name = 'Test 2')
+        });
+
+        // Test
+        try {
+            Test.startTest();
+                for (Integer i = 0 ; i < 101 ; i++) {
+                    SOQL.of(Account.SObjectType).mockId('mockingQuery').toList();
+                }
+            Test.stopTest();
+        } catch (QueryException e) {
+            queryException = e;
+        }
+
+        // Verify
+        Assert.isNotNull(queryException, 'The synchronous query should have exceeded the limit.');
+        Assert.areEqual('Too many SOQL queries.', queryException.getMessage(), 'The synchronous query should not have been exceeded.');
+    }
+
+    @IsTest
+    static void querySynchronousLimitExceptionWithoutMocking() {
+        // Setup
+        Exception queryException = null;
+
+        // Test
+        try {
+            for (Integer i = 0 ; i < 102 ; i++) {
+                SOQL.of(Account.SObjectType).toList();
+            }
+        } catch (QueryException e) {
+            queryException = e;
+        }
+
+        // Verify
+        Assert.isNotNull(queryException, 'The synchronous query should have exceeded the limit.');
+        Assert.areEqual('Too many SOQL queries.', queryException.getMessage(), 'The synchronous query should have been exceeded.');
+    }
+
+    @SuppressWarnings('PMD.ApexUnitTestClassShouldHaveAsserts')
+    @IsTest
+    static void preview() {
+        // Test
+        SOQL.of(Account.SObjectType).preview().toList();
+    }
+
+    @SuppressWarnings('PMD.ApexUnitTestClassShouldHaveAsserts')
+    @IsTest
+    static void previewWithConditions() {
+        // Test
+        SOQL.of(Account.SObjectType)
+            .whereAre(SOQL.FilterGroup
+                .add(SOQL.Filter.with(Account.Name).equal('Test'))
+                .add(SOQL.Filter.with(Account.Industry).equal('IT'))
+            )
+            .preview()
+            .toList();
+    }
+
+    @SuppressWarnings('PMD.ApexUnitTestClassShouldHaveAsserts')
+    @IsTest
+    static void previewCount() {
+        // Test
+        SOQL.of(Account.SObjectType).count().preview().toInteger();
+    }
+
     static List<Account> insertAccounts() {
         List<Account> accounts = new List<Account>{
             new Account(Name = 'Test 1'),
@@ -3534,31 +3631,4 @@ private class SOQL_Test {
             UserName = '[email protected]'
         );
     }
-
-    @SuppressWarnings('PMD.ApexUnitTestClassShouldHaveAsserts')
-    @IsTest
-    static void preview() {
-        // Test
-        SOQL.of(Account.SObjectType).preview().toList();
-    }
-
-    @SuppressWarnings('PMD.ApexUnitTestClassShouldHaveAsserts')
-    @IsTest
-    static void previewWithConditions() {
-        // Test
-        SOQL.of(Account.SObjectType)
-            .whereAre(SOQL.FilterGroup
-                .add(SOQL.Filter.with(Account.Name).equal('Test'))
-                .add(SOQL.Filter.with(Account.Industry).equal('IT'))
-            )
-            .preview()
-            .toList();
-    }
-
-    @SuppressWarnings('PMD.ApexUnitTestClassShouldHaveAsserts')
-    @IsTest
-    static void previewCount() {
-        // Test
-        SOQL.of(Account.SObjectType).count().preview().toInteger();
-    }
 }