Enhance transport tests

scalecube · Aug 31, 2023 · fca28ce · fca28ce
1 parent 90e2d8d
commit fca28ce
Showing 1 changed file with 14 additions and 20 deletions.
diff --git a/cluster/src/test/java/io/scalecube/cluster/TransportWrapperTest.java b/cluster/src/test/java/io/scalecube/cluster/TransportWrapperTest.java
@@ -51,12 +51,7 @@ void requestResponseShouldWork() {
 
       when(transport.requestResponse(addresses.get(0), request)).thenReturn(Mono.just(response));
 
-      StepVerifier.create(transportWrapper.requestResponse(member, request))
-          .assertNext(message -> Assertions.assertSame(response, message, "response"))
-          .thenCancel()
-          .verify();
-
-      StepVerifier.create(transportWrapper.requestResponse(member, request))
+      StepVerifier.create(transportWrapper.requestResponse(member, request).retry(2))
           .assertNext(message -> Assertions.assertSame(response, message, "response"))
           .thenCancel()
           .verify();
@@ -71,7 +66,7 @@ void requestResponseShouldWorkMemberSingleAddress() {
 
       when(transport.requestResponse(addresses.get(0), request)).thenReturn(Mono.just(response));
 
-      StepVerifier.create(transportWrapper.requestResponse(member, request))
+      StepVerifier.create(transportWrapper.requestResponse(member, request).retry(2))
           .assertNext(message -> Assertions.assertSame(response, message, "response"))
           .thenCancel()
           .verify();
@@ -88,7 +83,7 @@ void requestResponseShouldWorkMemberTwoAddresses() {
           .thenReturn(Mono.error(new RuntimeException("Error")));
       when(transport.requestResponse(addresses.get(1), request)).thenReturn(Mono.just(response));
 
-      StepVerifier.create(transportWrapper.requestResponse(member, request))
+      StepVerifier.create(transportWrapper.requestResponse(member, request).retry(2))
           .assertNext(message -> Assertions.assertSame(response, message, "response"))
           .thenCancel()
           .verify();
@@ -108,7 +103,7 @@ void requestResponseShouldWorkMemberThreeAddresses() {
           .thenReturn(Mono.error(new RuntimeException("Error")));
       when(transport.requestResponse(addresses.get(2), request)).thenReturn(Mono.just(response));
 
-      StepVerifier.create(transportWrapper.requestResponse(member, request))
+      StepVerifier.create(transportWrapper.requestResponse(member, request).retry(2))
           .assertNext(message -> Assertions.assertSame(response, message, "response"))
           .thenCancel()
           .verify();
@@ -124,7 +119,7 @@ void requestResponseShouldFailMemberSingleAddress() {
       when(transport.requestResponse(addresses.get(0), request))
           .thenReturn(Mono.error(new RuntimeException("Error")));
 
-      StepVerifier.create(transportWrapper.requestResponse(member, request))
+      StepVerifier.create(transportWrapper.requestResponse(member, request).retry(2))
           .verifyErrorSatisfies(
               throwable -> Assertions.assertEquals("Error", throwable.getMessage()));
     }
@@ -141,7 +136,7 @@ void requestResponseShouldFailMemberTwoAddresses() {
       when(transport.requestResponse(addresses.get(1), request))
           .thenReturn(Mono.error(new RuntimeException("Error - 1")));
 
-      StepVerifier.create(transportWrapper.requestResponse(member, request))
+      StepVerifier.create(transportWrapper.requestResponse(member, request).retry(2))
           .verifyErrorSatisfies(
               throwable -> Assertions.assertEquals("Error - 1", throwable.getMessage()));
     }
@@ -161,7 +156,7 @@ void requestResponseShouldFailMemberThreeAddresses() {
       when(transport.requestResponse(addresses.get(2), request))
           .thenReturn(Mono.error(new RuntimeException("Error - 2")));
 
-      StepVerifier.create(transportWrapper.requestResponse(member, request))
+      StepVerifier.create(transportWrapper.requestResponse(member, request).retry(2))
           .verifyErrorSatisfies(
               throwable -> Assertions.assertEquals("Error - 2", throwable.getMessage()));
     }
@@ -179,8 +174,7 @@ void sendShouldWork() {
 
       when(transport.send(addresses.get(0), request)).thenReturn(Mono.empty());
 
-      StepVerifier.create(transportWrapper.send(member, request)).verifyComplete();
-      StepVerifier.create(transportWrapper.send(member, request)).verifyComplete();
+      StepVerifier.create(transportWrapper.send(member, request).retry(2)).verifyComplete();
     }
 
     @Test
@@ -192,7 +186,7 @@ void sendShouldWorkMemberSingleAddress() {
 
       when(transport.send(addresses.get(0), request)).thenReturn(Mono.empty());
 
-      StepVerifier.create(transportWrapper.send(member, request)).verifyComplete();
+      StepVerifier.create(transportWrapper.send(member, request).retry(2)).verifyComplete();
     }
 
     @Test
@@ -206,7 +200,7 @@ void sendShouldWorkMemberTwoAddresses() {
           .thenReturn(Mono.error(new RuntimeException("Error")));
       when(transport.send(addresses.get(1), request)).thenReturn(Mono.empty());
 
-      StepVerifier.create(transportWrapper.send(member, request)).verifyComplete();
+      StepVerifier.create(transportWrapper.send(member, request).retry(2)).verifyComplete();
     }
 
     @Test
@@ -223,7 +217,7 @@ void sendShouldWorkMemberThreeAddresses() {
           .thenReturn(Mono.error(new RuntimeException("Error")));
       when(transport.send(addresses.get(2), request)).thenReturn(Mono.empty());
 
-      StepVerifier.create(transportWrapper.send(member, request)).verifyComplete();
+      StepVerifier.create(transportWrapper.send(member, request).retry(2)).verifyComplete();
     }
 
     @Test
@@ -236,7 +230,7 @@ void sendShouldFailMemberSingleAddress() {
       when(transport.send(addresses.get(0), request))
           .thenReturn(Mono.error(new RuntimeException("Error")));
 
-      StepVerifier.create(transportWrapper.send(member, request))
+      StepVerifier.create(transportWrapper.send(member, request).retry(2))
           .verifyErrorSatisfies(
               throwable -> Assertions.assertEquals("Error", throwable.getMessage()));
     }
@@ -253,7 +247,7 @@ void sendShouldFailMemberTwoAddresses() {
       when(transport.send(addresses.get(1), request))
           .thenReturn(Mono.error(new RuntimeException("Error - 1")));
 
-      StepVerifier.create(transportWrapper.send(member, request))
+      StepVerifier.create(transportWrapper.send(member, request).retry(2))
           .verifyErrorSatisfies(
               throwable -> Assertions.assertEquals("Error - 1", throwable.getMessage()));
     }
@@ -273,7 +267,7 @@ void sendShouldFailMemberThreeAddresses() {
       when(transport.send(addresses.get(2), request))
           .thenReturn(Mono.error(new RuntimeException("Error - 2")));
 
-      StepVerifier.create(transportWrapper.send(member, request))
+      StepVerifier.create(transportWrapper.send(member, request).retry(2))
           .verifyErrorSatisfies(
               throwable -> Assertions.assertEquals("Error - 2", throwable.getMessage()));
     }