Updates TestingUtils class to properly account for share/history Id g…

…eneration
jamessimone · Sep 25, 2023 · 80eb1ce · 80eb1ce
1 parent 626cbf0
commit 80eb1ce
Showing 1 changed file with 24 additions and 2 deletions.
diff --git a/force-app/utils/TestingUtils.cls b/force-app/utils/TestingUtils.cls
@@ -1,9 +1,21 @@
-public abstract class TestingUtils {
+@IsTest
+public class TestingUtils {
   private static Integer startingNumber = 1;
 
   public static String generateId(Schema.SObjectType sObjectType) {
     String result = String.valueOf(startingNumber++);
-    return sObjectType.getDescribe().getKeyPrefix() + '0'.repeat(12 - result.length()) + result;
+    String keyPrefix = sObjectType.getDescribe().getKeyPrefix();
+    if (keyPrefix == null) {
+      switch on sObjectType.getDescribe().getAssociateEntityType() {
+        when 'History' {
+          keyPrefix = '017';
+        }
+        when 'Share' {
+          keyPrefix = '02c';
+        }
+      }
+    }
+    return keyPrefix + '0'.repeat(12 - result.length()) + result;
   }
 
   public static SObject generateId(SObject objectInstance) {
@@ -20,4 +32,14 @@ public abstract class TestingUtils {
       }
     }
   }
+
+  @IsTest
+  static void generates_fake_history_ids() {
+    Assert.isNotNull(generateId(LeadHistory.SObjectType));
+  }
+
+  @IsTest
+  static void generates_fake_share_records() {
+    Assert.isNotNull(generateId(LeadShare.SObjectType));
+  }
 }