From 9478745ab4312d1c593f2d8e9f5114e7e505fe48 Mon Sep 17 00:00:00 2001
From: Artem Vysochyn <artem.vysochyn@gmail.com>
Date: Wed, 17 Jul 2024 11:46:25 +0300
Subject: [PATCH] Fininshed MembershipProtocolTest

---
 .../membership/MembershipProtocolTest.java    | 21 ++++++++++++++-----
 1 file changed, 16 insertions(+), 5 deletions(-)

diff --git a/cluster2/src/test/java/io/scalecube/cluster2/membership/MembershipProtocolTest.java b/cluster2/src/test/java/io/scalecube/cluster2/membership/MembershipProtocolTest.java
index 6d202a88..4084a481 100644
--- a/cluster2/src/test/java/io/scalecube/cluster2/membership/MembershipProtocolTest.java
+++ b/cluster2/src/test/java/io/scalecube/cluster2/membership/MembershipProtocolTest.java
@@ -5,10 +5,10 @@
 import static org.hamcrest.MatcherAssert.assertThat;
 import static org.hamcrest.Matchers.isOneOf;
 import static org.junit.jupiter.api.Assertions.assertEquals;
-import static org.junit.jupiter.api.Assertions.fail;
 import static org.mockito.ArgumentMatchers.any;
 import static org.mockito.ArgumentMatchers.anyInt;
 import static org.mockito.ArgumentMatchers.argThat;
+import static org.mockito.ArgumentMatchers.eq;
 import static org.mockito.Mockito.doAnswer;
 import static org.mockito.Mockito.mock;
 import static org.mockito.Mockito.never;
@@ -22,6 +22,7 @@
 import io.scalecube.cluster2.fdetector.FailureDetectorCodec;
 import io.scalecube.cluster2.fdetector.FailureDetectorConfig;
 import io.scalecube.cluster2.gossip.GossipMessageCodec;
+import io.scalecube.cluster2.sbe.MemberStatus;
 import java.lang.reflect.Field;
 import java.util.UUID;
 import java.util.function.Consumer;
@@ -159,11 +160,10 @@ void testOnSyncAckPeriodMismatch() {
   @Test
   void testOnGossipMessage() {
     emitGossipInputMessage(
-        gossipMessageCodec -> {
+        codec -> {
           final MembershipRecordCodec membershipRecordCodec = new MembershipRecordCodec();
           final MutableDirectBuffer buffer = membershipRecordCodec.encode(localRecord);
-          return gossipMessageCodec.encodeInputMessage(
-              buffer, 0, membershipRecordCodec.encodedLength());
+          return codec.encodeInputMessage(buffer, 0, membershipRecordCodec.encodedLength());
         });
 
     verify(membershipTable)
@@ -177,7 +177,18 @@ void testOnGossipMessage() {
 
   @Test
   void testOnFailureDetectorEvent() {
-    fail("Implemnent");
+    final Member member = localRecord.member();
+    final MemberStatus memberStatus = ALIVE;
+    emitFailureDetectorEvent(codec -> codec.encodeFailureDetectorEvent(member, memberStatus));
+
+    verify(membershipTable)
+        .put(
+            argThat(
+                arg -> {
+                  assertEquals(member, arg, "member");
+                  return true;
+                }),
+            eq(memberStatus));
   }
 
   private void advanceClock(final long millis) {