From 966a2a55fc84764f5d38447bee7b782f9cbbce69 Mon Sep 17 00:00:00 2001
From: Chris Canal <chris@ocd.codes>
Date: Mon, 13 Nov 2023 14:33:04 -0600
Subject: [PATCH] Tweaks test to better test if IEvent is handled

---
 .../PersistenceTests/Marten/event_streaming.cs       | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/src/Persistence/PersistenceTests/Marten/event_streaming.cs b/src/Persistence/PersistenceTests/Marten/event_streaming.cs
index 140c71f29..96067be4d 100644
--- a/src/Persistence/PersistenceTests/Marten/event_streaming.cs
+++ b/src/Persistence/PersistenceTests/Marten/event_streaming.cs
@@ -88,6 +88,9 @@ public async Task event_should_be_published_from_sender_to_receiver()
         
         results.Received.SingleMessage<SecondMessage>()
             .Sequence.ShouldBeGreaterThan(0);
+        
+        TriggerHandler.ThirdEventHandled.ShouldBeTrue();
+        TriggerHandler.ThirdIEventHandled.ShouldBeTrue();
     }
 }
 
@@ -99,6 +102,8 @@ public class TriggerCommand
 
 public class TriggerHandler
 {
+    public static bool ThirdIEventHandled;
+    public static bool ThirdEventHandled;
     [Transactional]
     public void Handle(TriggerCommand command, IDocumentSession session)
     {
@@ -107,7 +112,12 @@ public void Handle(TriggerCommand command, IDocumentSession session)
     
     public void Handle(IEvent<ThirdEvent> e)
     {
-        
+        ThirdIEventHandled = true;
+    }
+    
+    public void Handle(ThirdEvent e)
+    {
+        ThirdEventHandled = true;
     }
 }