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) {