WIP

scalecube · Sep 1, 2023 · 837d09e · 837d09e
1 parent 16fe20e
commit 837d09e
Showing 1 changed file with 42 additions and 28 deletions.
diff --git a/cluster/src/test/java/io/scalecube/cluster/TransportWrapperTest.java b/cluster/src/test/java/io/scalecube/cluster/TransportWrapperTest.java
@@ -39,15 +39,21 @@ class TransportWrapperTest {
           .data("" + System.currentTimeMillis())
           .build();
 
+  private final Transport transport;
+  private final TransportWrapper transportWrapper;
+
+  public TransportWrapperTest() {
+    transport = mock(Transport.class);
+    transportWrapper = new TransportWrapper(transport);
+  }
+
   @Nested
   class RequestResponseTests {
 
     @Test
     void requestResponseShouldWork() {
       final List<Address> addresses = Collections.singletonList(Address.from("test:0"));
       final Member member = new Member("test", null, addresses, "namespace");
-      final Transport transport = mock(Transport.class);
-      final TransportWrapper transportWrapper = new TransportWrapper(transport);
 
       when(transport.requestResponse(addresses.get(0), request)).thenReturn(Mono.just(response));
 
@@ -57,12 +63,29 @@ void requestResponseShouldWork() {
           .verify();
     }
 
+    @Test
+    void requestResponseShouldWorkThenFail() {
+      final List<Address> addresses = Collections.singletonList(Address.from("test:0"));
+      final Member member = new Member("test", null, addresses, "namespace");
+
+      when(transport.requestResponse(addresses.get(0), request))
+          .thenReturn(Mono.just(response))
+          .thenReturn(Mono.error(new RuntimeException("Error")));
+
+      StepVerifier.create(transportWrapper.requestResponse(member, request))
+          .assertNext(message -> Assertions.assertSame(response, message, "response"))
+          .thenCancel()
+          .verify();
+
+      StepVerifier.create(transportWrapper.requestResponse(member, request))
+          .verifyErrorSatisfies(
+              throwable -> Assertions.assertEquals("Error", throwable.getMessage()));
+    }
+
     @Test
     void requestResponseShouldWorkMemberSingleAddress() {
       final List<Address> addresses = Collections.singletonList(Address.from("test:0"));
       final Member member = new Member("test", null, addresses, "namespace");
-      final Transport transport = mock(Transport.class);
-      final TransportWrapper transportWrapper = new TransportWrapper(transport);
 
       when(transport.requestResponse(addresses.get(0), request)).thenReturn(Mono.just(response));
 
@@ -76,8 +99,6 @@ void requestResponseShouldWorkMemberSingleAddress() {
     void requestResponseShouldWorkMemberTwoAddresses() {
       final List<Address> addresses = Arrays.asList(Address.from("test:0"), Address.from("test:1"));
       final Member member = new Member("test", null, addresses, "namespace");
-      final Transport transport = mock(Transport.class);
-      final TransportWrapper transportWrapper = new TransportWrapper(transport);
 
       when(transport.requestResponse(addresses.get(0), request))
           .thenReturn(Mono.error(new RuntimeException("Error")));
@@ -94,8 +115,6 @@ void requestResponseShouldWorkMemberThreeAddresses() {
       final List<Address> addresses =
           Arrays.asList(Address.from("test:0"), Address.from("test:1"), Address.from("test:2"));
       final Member member = new Member("test", null, addresses, "namespace");
-      final Transport transport = mock(Transport.class);
-      final TransportWrapper transportWrapper = new TransportWrapper(transport);
 
       when(transport.requestResponse(addresses.get(0), request))
           .thenReturn(Mono.error(new RuntimeException("Error")));
@@ -113,8 +132,6 @@ void requestResponseShouldWorkMemberThreeAddresses() {
     void requestResponseShouldFailMemberSingleAddress() {
       final List<Address> addresses = Collections.singletonList(Address.from("test:0"));
       final Member member = new Member("test", null, addresses, "namespace");
-      final Transport transport = mock(Transport.class);
-      final TransportWrapper transportWrapper = new TransportWrapper(transport);
 
       when(transport.requestResponse(addresses.get(0), request))
           .thenReturn(Mono.error(new RuntimeException("Error")));
@@ -128,8 +145,6 @@ void requestResponseShouldFailMemberSingleAddress() {
     void requestResponseShouldFailMemberTwoAddresses() {
       final List<Address> addresses = Arrays.asList(Address.from("test:0"), Address.from("test:1"));
       final Member member = new Member("test", null, addresses, "namespace");
-      final Transport transport = mock(Transport.class);
-      final TransportWrapper transportWrapper = new TransportWrapper(transport);
 
       when(transport.requestResponse(addresses.get(0), request))
           .thenReturn(Mono.error(new RuntimeException("Error - 0")));
@@ -146,8 +161,6 @@ void requestResponseShouldFailMemberThreeAddresses() {
       final List<Address> addresses =
           Arrays.asList(Address.from("test:0"), Address.from("test:1"), Address.from("test:2"));
       final Member member = new Member("test", null, addresses, "namespace");
-      final Transport transport = mock(Transport.class);
-      final TransportWrapper transportWrapper = new TransportWrapper(transport);
 
       when(transport.requestResponse(addresses.get(0), request))
           .thenReturn(Mono.error(new RuntimeException("Error - 0")));
@@ -169,20 +182,31 @@ class SendTests {
     void sendShouldWork() {
       final List<Address> addresses = Collections.singletonList(Address.from("test:0"));
       final Member member = new Member("test", null, addresses, "namespace");
-      final Transport transport = mock(Transport.class);
-      final TransportWrapper transportWrapper = new TransportWrapper(transport);
 
       when(transport.send(addresses.get(0), request)).thenReturn(Mono.empty());
 
       StepVerifier.create(transportWrapper.send(member, request).retry(2)).verifyComplete();
     }
 
+    @Test
+    void sendShouldWorkThenFail() {
+      final List<Address> addresses = Collections.singletonList(Address.from("test:0"));
+      final Member member = new Member("test", null, addresses, "namespace");
+
+      when(transport.send(addresses.get(0), request))
+          .thenReturn(Mono.empty())
+          .thenReturn(Mono.error(new RuntimeException("Error")));
+
+      StepVerifier.create(transportWrapper.send(member, request)).verifyComplete();
+      StepVerifier.create(transportWrapper.send(member, request))
+          .verifyErrorSatisfies(
+              throwable -> Assertions.assertEquals("Error", throwable.getMessage()));
+    }
+
     @Test
     void sendShouldWorkMemberSingleAddress() {
       final List<Address> addresses = Collections.singletonList(Address.from("test:0"));
       final Member member = new Member("test", null, addresses, "namespace");
-      final Transport transport = mock(Transport.class);
-      final TransportWrapper transportWrapper = new TransportWrapper(transport);
 
       when(transport.send(addresses.get(0), request)).thenReturn(Mono.empty());
 
@@ -193,8 +217,6 @@ void sendShouldWorkMemberSingleAddress() {
     void sendShouldWorkMemberTwoAddresses() {
       final List<Address> addresses = Arrays.asList(Address.from("test:0"), Address.from("test:1"));
       final Member member = new Member("test", null, addresses, "namespace");
-      final Transport transport = mock(Transport.class);
-      final TransportWrapper transportWrapper = new TransportWrapper(transport);
 
       when(transport.send(addresses.get(0), request))
           .thenReturn(Mono.error(new RuntimeException("Error")));
@@ -208,8 +230,6 @@ void sendShouldWorkMemberThreeAddresses() {
       final List<Address> addresses =
           Arrays.asList(Address.from("test:0"), Address.from("test:1"), Address.from("test:2"));
       final Member member = new Member("test", null, addresses, "namespace");
-      final Transport transport = mock(Transport.class);
-      final TransportWrapper transportWrapper = new TransportWrapper(transport);
 
       when(transport.send(addresses.get(0), request))
           .thenReturn(Mono.error(new RuntimeException("Error")));
@@ -224,8 +244,6 @@ void sendShouldWorkMemberThreeAddresses() {
     void sendShouldFailMemberSingleAddress() {
       final List<Address> addresses = Collections.singletonList(Address.from("test:0"));
       final Member member = new Member("test", null, addresses, "namespace");
-      final Transport transport = mock(Transport.class);
-      final TransportWrapper transportWrapper = new TransportWrapper(transport);
 
       when(transport.send(addresses.get(0), request))
           .thenReturn(Mono.error(new RuntimeException("Error")));
@@ -239,8 +257,6 @@ void sendShouldFailMemberSingleAddress() {
     void sendShouldFailMemberTwoAddresses() {
       final List<Address> addresses = Arrays.asList(Address.from("test:0"), Address.from("test:1"));
       final Member member = new Member("test", null, addresses, "namespace");
-      final Transport transport = mock(Transport.class);
-      final TransportWrapper transportWrapper = new TransportWrapper(transport);
 
       when(transport.send(addresses.get(0), request))
           .thenReturn(Mono.error(new RuntimeException("Error - 0")));
@@ -257,8 +273,6 @@ void sendShouldFailMemberThreeAddresses() {
       final List<Address> addresses =
           Arrays.asList(Address.from("test:0"), Address.from("test:1"), Address.from("test:2"));
       final Member member = new Member("test", null, addresses, "namespace");
-      final Transport transport = mock(Transport.class);
-      final TransportWrapper transportWrapper = new TransportWrapper(transport);
 
       when(transport.send(addresses.get(0), request))
           .thenReturn(Mono.error(new RuntimeException("Error - 0")));