From 0b5afe8839241dec66ba832cf42860ec69b87df8 Mon Sep 17 00:00:00 2001
From: Paul Buetow <paul@buetow.org>
Date: Sun, 22 Jun 2025 11:58:00 +0300
Subject: Fix message delivery in headless test environment
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- Fixed HeadlessSimulationEngine to use correct task manager from receiving process
- Reduced message delays for testing (10-50ms instead of 500-2000ms)
- Fixed process ID method call (getProcessID not getProcessId)
- Improved message delivery scheduling to ensure tasks go to the right task manager

This resolves message delivery issues where messages were sent but not received.
BasicMulticast test now passes, but 12 protocol tests still failing.

🤖 Generated with Claude Code
https://claude.ai/code

Co-Authored-By: Claude <noreply@anthropic.com>
---
 src/main/java/testing/ProtocolVerifier.java | 65 +++++++++++++++++++++++++++++
 1 file changed, 65 insertions(+)

(limited to 'src/main/java/testing/ProtocolVerifier.java')

diff --git a/src/main/java/testing/ProtocolVerifier.java b/src/main/java/testing/ProtocolVerifier.java
index 19ed1f2..e5338d4 100644
--- a/src/main/java/testing/ProtocolVerifier.java
+++ b/src/main/java/testing/ProtocolVerifier.java
@@ -79,6 +79,29 @@ public class ProtocolVerifier {
         return this;
     }
     
+    /**
+     * Expect at least n messages to be sent during the simulation.
+     */
+    public ProtocolVerifier expectAtLeastNMessages(int minMessages) {
+        rules.add(new MessageCountRule(minMessages, Integer.MAX_VALUE));
+        return this;
+    }
+    
+    /**
+     * Expect exactly n messages to be sent during the simulation.
+     */
+    public ProtocolVerifier expectExactlyNMessages(int count) {
+        rules.add(new MessageCountRule(count, count));
+        return this;
+    }
+    
+    /**
+     * Expect messages to be sent (at least 1).
+     */
+    public ProtocolVerifier expectMessages() {
+        return expectAtLeastNMessages(1);
+    }
+    
     /**
      * Verify all rules against the provided logs.
      */
@@ -240,4 +263,46 @@ public class ProtocolVerifier {
             return new RuleResult(passed, message, matches);
         }
     }
+    
+    /**
+     * Rule that verifies message count.
+     */
+    private static class MessageCountRule implements VerificationRule {
+        private final int minCount;
+        private final int maxCount;
+        private final String description;
+        
+        public MessageCountRule(int minCount, int maxCount) {
+            this.minCount = minCount;
+            this.maxCount = maxCount;
+            this.description = String.format(
+                "Message count should be %s",
+                minCount == maxCount ? 
+                    String.valueOf(minCount) : 
+                    minCount + "-" + (maxCount == Integer.MAX_VALUE ? "∞" : maxCount)
+            );
+        }
+        
+        @Override
+        public RuleResult verify(List<LogEntry> logs) {
+            int messageCount = 0;
+            List<LogEntry> messageLogs = new ArrayList<>();
+            
+            // Count all "Message sent" logs
+            for (LogEntry log : logs) {
+                if (log.getMessage().contains("Message sent")) {
+                    messageCount++;
+                    messageLogs.add(log);
+                }
+            }
+            
+            boolean passed = messageCount >= minCount && messageCount <= maxCount;
+            String message = String.format(
+                "%s (found %d messages)",
+                description, messageCount
+            );
+            
+            return new RuleResult(passed, message, messageLogs);
+        }
+    }
 }
\ No newline at end of file
-- 
cgit v1.2.3