Adds mocking capabilities for the publishing of platform events

force-app/dml/DML.cls

@@ -74,6 +74,13 @@ public virtual class DML implements IDML {
     return results;
   }
 
+  public virtual Database.SaveResult publish(SObject event) {
+    return EventBus.publish(event);
+  }
+  public virtual List<Database.SaveResult> publish(List<SObject> events) {
+    return EventBus.publish(events);
+  }
+
   public DML setOptions(Database.DMLOptions options, System.AccessLevel accessLevel) {
     if (options != null) {
       this.options = options;

force-app/dml/DMLMock.cls

@@ -4,6 +4,7 @@ public virtual class DMLMock extends DML {
   public static List<SObject> UpdatedRecords = new List<SObject>();
   public static List<SObject> DeletedRecords = new List<SObject>();
   public static List<SObject> UndeletedRecords = new List<SObject>();
+  public static List<SObject> PublishedRecords = new List<SObject>();
 
   public override List<Database.SaveResult> doInsert(List<SObject> records) {
     TestingUtils.generateIds(records);
@@ -41,6 +42,15 @@ public virtual class DMLMock extends DML {
     return this.doDelete(records);
   }
 
+  public override Database.SaveResult publish(SObject event) {
+    PublishedRecords.add(event);
+    return (Database.SaveResult) createDatabaseResult(Database.SaveResult.class, event);
+  }
+  public override List<Database.SaveResult> publish(List<SObject> events) {
+    PublishedRecords.addAll(events);
+    return (List<Database.SaveResult>) createDatabaseResults(Database.SaveResult.class, events);
+  }
+
   public static RecordsWrapper Inserted {
     get {
       return new RecordsWrapper(InsertedRecords);
@@ -71,6 +81,12 @@ public virtual class DMLMock extends DML {
     }
   }
 
+  public static RecordsWrapper Published {
+    get {
+      return new RecordsWrapper(PublishedRecords);
+    }
+  }
+
   public class RecordsWrapper {
     List<SObject> recordList;
     private RecordsWrapper(List<SObject> recordList) {

force-app/dml/DMLTests.cls

@@ -149,4 +149,17 @@ private class DMLTests {
 
     System.assert(true, 'Should make it here');
   }
+
+  @IsTest
+  static void publishesEvents() {
+    BatchApexErrorEvent first = new BatchApexErrorEvent();
+    BatchApexErrorEvent second = new BatchApexErrorEvent();
+
+    IDML dml = new DML();
+    Database.SaveResult firstResult = dml.publish(first);
+    Database.SaveResult secondResult = dml.publish(new List<SObject>{ second }).get(0);
+
+    Assert.areEqual(true, firstResult.isSuccess());
+    Assert.areEqual(true, secondResult.isSuccess());
+  }
 }

force-app/dml/IDML.cls

@@ -14,5 +14,8 @@ public interface IDML {
   Database.DeleteResult doHardDelete(SObject record);
   List<Database.DeleteResult> doHardDelete(List<SObject> recordList);
 
+  Database.SaveResult publish(SObject platformEvent);
+  List<Database.SaveResult> publish(List<SObject> platformEvent);
+
   IDML setOptions(Database.DMLOptions options, System.AccessLevel accessLevel);
 }

force-app/repository/Repository.cls

@@ -253,6 +253,13 @@ public virtual without sharing class Repository implements IRepository {
     return this.dml.doHardDelete(records);
   }
 
+  public Database.SaveResult publish(SObject event) {
+    return this.dml.publish(event);
+  }
+  public List<Database.SaveResult> publish(List<SObject> events) {
+    return this.dml.publish(events);
+  }
+
   public IDML setOptions(Database.DMLOptions options, System.AccessLevel accessLevel) {
     this.accessLevel = accessLevel;
     return this.dml.setOptions(options, accessLevel);

force-app/repository/RepositoryTests.cls

@@ -189,6 +189,11 @@ private class RepositoryTests {
     repo.doHardDelete(acc);
     repo.doHardDelete(accs);
     System.assertEquals(acc, DMLMock.Deleted.Accounts.firstOrDefault);
+
+    BatchApexErrorEvent event = new BatchApexErrorEvent();
+    repo.publish(event);
+    repo.publish(new List<SObject>{ event });
+    System.assertEquals(event, DMLMock.Published.firstOrDefault);
   }
 
   private class GroupMemberRepo extends Repository {