specify number of hearbeats

dedis · Dec 5, 2023 · 20710ee · 20710ee
1 parent fc8b02c
commit 20710ee
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/tests/karate/src/test/java/be/features/decentralizedCom/heartbeat.feature b/tests/karate/src/test/java/be/features/decentralizedCom/heartbeat.feature
@@ -25,7 +25,7 @@ Feature: Send heartbeats to other servers
 
     When def heartbeatMessages = mockServer.getHeartbeats()
 
-    Then assert heartbeatMessages.length > 1
+    Then assert heartbeatMessages.length == 2
 
   # Check that after receiving a publish message (in this case a create roll call), the server sends a heartbeat containing
   # the message id of that publish message.

diff --git a/tests/karate/src/test/java/common/net/MultiMsgWebSocketClient.java b/tests/karate/src/test/java/common/net/MultiMsgWebSocketClient.java
@@ -61,7 +61,7 @@ public void send(Map<String, Object> jsonRpcMsg){
    */
   public void publish(Map<String, Object> highLevelMessageDataMap, String channel){
     String highLevelMessageData = mapToJsonString(highLevelMessageDataMap);
-    int messageId = new Random().nextInt(Integer.MAX_VALUE);
+    int messageId = new Random().nextInt();
     Json publishMessageJson =  jsonConverter.constructPublishMessage(highLevelMessageData, messageId, channel);
     PublishMessageIds publishMessageIds = new PublishMessageIds(messageId, publishMessageJson.get("params.message.message_id"));
     sentMessages.put(highLevelMessageData, publishMessageIds);
@@ -192,11 +192,11 @@ public synchronized Object listen(long timeout) {
 
   private PublishMessageIds getPublishMessageIds(Map<String, Object> highLevelMessageDataMap){
     String highLevelMessageData = mapToJsonString(highLevelMessageDataMap);
-    PublishMessageIds message_ids = sentMessages.get(highLevelMessageData);
-    if(message_ids == null){
+    PublishMessageIds messageIds = sentMessages.get(highLevelMessageData);
+    if(messageIds == null){
       throw new IllegalArgumentException("Did not send this message");
     }
-    return message_ids;
+    return messageIds;
   }
 
   /**