Enhanced polymorph tests with one more usecase

Author: artem-v

services-gateway/src/test/java/io/scalecube/services/gateway/http/HttpGatewayTest.java

@@ -23,6 +23,8 @@
 import io.scalecube.services.gateway.SomeException;
 import io.scalecube.services.gateway.client.http.HttpGatewayClientTransport;
 import io.scalecube.services.gateway.sut.typed.Circle;
+import io.scalecube.services.gateway.sut.typed.Rectangle;
+import io.scalecube.services.gateway.sut.typed.Square;
 import io.scalecube.services.gateway.sut.typed.StartOfDayEvent;
 import io.scalecube.services.routing.StaticAddressRouter;
 import io.scalecube.services.transport.rsocket.RSocketServiceTransport;
@@ -231,6 +233,20 @@ public void shouldReturnPolymorph() {
         .verify();
   }
 
+  @Test
+  public void shouldReturnListPolymorph() {
+    StepVerifier.create(typedGreetingService.helloListPolymorph())
+        .assertNext(
+            shapes -> {
+              assertEquals(1.0, ((Circle) shapes.get(0)).radius());
+              assertEquals(1.0, ((Rectangle) shapes.get(1)).height());
+              assertEquals(1.0, ((Rectangle) shapes.get(1)).width());
+              assertEquals(1.0, ((Square) shapes.get(2)).side());
+            })
+        .thenCancel()
+        .verify();
+  }
+
   @Test
   public void shouldReturnMultitype() {
     StepVerifier.create(typedGreetingService.helloMultitype())

services-gateway/src/test/java/io/scalecube/services/gateway/http/HttpLocalGatewayTest.java

@@ -22,6 +22,8 @@
 import io.scalecube.services.gateway.SomeException;
 import io.scalecube.services.gateway.client.http.HttpGatewayClientTransport;
 import io.scalecube.services.gateway.sut.typed.Circle;
+import io.scalecube.services.gateway.sut.typed.Rectangle;
+import io.scalecube.services.gateway.sut.typed.Square;
 import io.scalecube.services.gateway.sut.typed.StartOfDayEvent;
 import io.scalecube.services.routing.StaticAddressRouter;
 import java.time.Duration;
@@ -205,6 +207,20 @@ public void shouldReturnPolymorph() {
         .verify();
   }
 
+  @Test
+  public void shouldReturnListPolymorph() {
+    StepVerifier.create(typedGreetingService.helloListPolymorph())
+        .assertNext(
+            shapes -> {
+              assertEquals(1.0, ((Circle) shapes.get(0)).radius());
+              assertEquals(1.0, ((Rectangle) shapes.get(1)).height());
+              assertEquals(1.0, ((Rectangle) shapes.get(1)).width());
+              assertEquals(1.0, ((Square) shapes.get(2)).side());
+            })
+        .thenCancel()
+        .verify();
+  }
+
   @Test
   public void shouldReturnMultitype() {
     StepVerifier.create(typedGreetingService.helloMultitype())

services-gateway/src/test/java/io/scalecube/services/gateway/http/TypedGreetingService.java

@@ -3,6 +3,7 @@
 import io.scalecube.services.annotations.Service;
 import io.scalecube.services.annotations.ServiceMethod;
 import io.scalecube.services.gateway.sut.typed.Shape;
+import java.util.List;
 import reactor.core.publisher.Mono;
 
 @Service(TypedGreetingService.SERVICE_NAME)
@@ -13,6 +14,9 @@ public interface TypedGreetingService {
   @ServiceMethod
   Mono<Shape> helloPolymorph();
 
+  @ServiceMethod
+  Mono<List<Shape>> helloListPolymorph();
+
   @ServiceMethod
   Mono<Object> helloMultitype();
 

services-gateway/src/test/java/io/scalecube/services/gateway/http/TypedGreetingServiceImpl.java

@@ -1,10 +1,13 @@
 package io.scalecube.services.gateway.http;
 
 import io.scalecube.services.gateway.sut.typed.Circle;
+import io.scalecube.services.gateway.sut.typed.Rectangle;
 import io.scalecube.services.gateway.sut.typed.Shape;
+import io.scalecube.services.gateway.sut.typed.Square;
 import io.scalecube.services.gateway.sut.typed.StartOfDayEvent;
 import java.time.Clock;
 import java.time.LocalDateTime;
+import java.util.List;
 import reactor.core.publisher.Mono;
 
 public class TypedGreetingServiceImpl implements TypedGreetingService {
@@ -14,6 +17,11 @@ public Mono<Shape> helloPolymorph() {
     return Mono.just(new Circle(1.0));
   }
 
+  @Override
+  public Mono<List<Shape>> helloListPolymorph() {
+    return Mono.just(List.of(new Circle(1.0), new Rectangle(1.0, 1.0), new Square(1.0)));
+  }
+
   @Override
   public Mono<Object> helloMultitype() {
     return Mono.just(new StartOfDayEvent(1, 1, 1, LocalDateTime.now(Clock.systemUTC())));

services-gateway/src/test/java/io/scalecube/services/gateway/websocket/TypedGreetingService.java

@@ -3,7 +3,9 @@
 import io.scalecube.services.annotations.Service;
 import io.scalecube.services.annotations.ServiceMethod;
 import io.scalecube.services.gateway.sut.typed.Shape;
+import java.util.List;
 import reactor.core.publisher.Flux;
+import reactor.core.publisher.Mono;
 
 @Service(TypedGreetingService.SERVICE_NAME)
 public interface TypedGreetingService {
@@ -13,6 +15,9 @@ public interface TypedGreetingService {
   @ServiceMethod
   Flux<Shape> helloPolymorph();
 
+  @ServiceMethod
+  Mono<List<Shape>> helloListPolymorph();
+
   @ServiceMethod
   Flux<Object> helloMultitype();
 

services-gateway/src/test/java/io/scalecube/services/gateway/websocket/TypedGreetingServiceImpl.java

@@ -10,7 +10,9 @@
 import java.math.BigDecimal;
 import java.time.Clock;
 import java.time.LocalDateTime;
+import java.util.List;
 import reactor.core.publisher.Flux;
+import reactor.core.publisher.Mono;
 
 public class TypedGreetingServiceImpl implements TypedGreetingService {
 
@@ -19,6 +21,11 @@ public Flux<Shape> helloPolymorph() {
     return Flux.just(new Circle(1.0), new Rectangle(1.0, 1.0), new Square(1.0));
   }
 
+  @Override
+  public Mono<List<Shape>> helloListPolymorph() {
+    return Mono.just(List.of(new Circle(1.0), new Rectangle(1.0, 1.0), new Square(1.0)));
+  }
+
   @Override
   public Flux<Object> helloMultitype() {
     return Flux.just(

services-gateway/src/test/java/io/scalecube/services/gateway/websocket/WebsocketGatewayTest.java

@@ -275,6 +275,20 @@ public void shouldReturnPolymorph() {
         .verify();
   }
 
+  @Test
+  public void shouldReturnListPolymorph() {
+    StepVerifier.create(typedGreetingService.helloListPolymorph())
+        .assertNext(
+            shapes -> {
+              assertEquals(1.0, ((Circle) shapes.get(0)).radius());
+              assertEquals(1.0, ((Rectangle) shapes.get(1)).height());
+              assertEquals(1.0, ((Rectangle) shapes.get(1)).width());
+              assertEquals(1.0, ((Square) shapes.get(2)).side());
+            })
+        .thenCancel()
+        .verify();
+  }
+
   @Test
   public void shouldReturnMultitype() {
     StepVerifier.create(typedGreetingService.helloMultitype())

services-gateway/src/test/java/io/scalecube/services/gateway/websocket/WebsocketLocalGatewayTest.java

@@ -247,6 +247,20 @@ public void shouldReturnPolymorph() {
         .verify();
   }
 
+  @Test
+  public void shouldReturnListPolymorph() {
+    StepVerifier.create(typedGreetingService.helloListPolymorph())
+        .assertNext(
+            shapes -> {
+              assertEquals(1.0, ((Circle) shapes.get(0)).radius());
+              assertEquals(1.0, ((Rectangle) shapes.get(1)).height());
+              assertEquals(1.0, ((Rectangle) shapes.get(1)).width());
+              assertEquals(1.0, ((Square) shapes.get(2)).side());
+            })
+        .thenCancel()
+        .verify();
+  }
+
   @Test
   public void shouldReturnMultitype() {
     StepVerifier.create(typedGreetingService.helloMultitype())

services/src/test/java/io/scalecube/services/ServiceLocalTest.java

@@ -438,6 +438,22 @@ public void test_polymorph() {
         .verify();
   }
 
+  @Test
+  public void test_list_polymorph() {
+    final var greetingService = api(TypedGreetingService.class);
+
+    StepVerifier.create(greetingService.helloListPolymorph())
+        .assertNext(
+            shapes -> {
+              assertEquals(1.0, ((Circle) shapes.get(0)).radius());
+              assertEquals(1.0, ((Rectangle) shapes.get(1)).height());
+              assertEquals(1.0, ((Rectangle) shapes.get(1)).width());
+              assertEquals(1.0, ((Square) shapes.get(2)).side());
+            })
+        .thenCancel()
+        .verify();
+  }
+
   @Test
   public void test_multitype() {
     final var greetingService = api(TypedGreetingService.class);

services/src/test/java/io/scalecube/services/ServiceRemoteTest.java

@@ -610,6 +610,22 @@ public void test_polymorph() {
         .verify();
   }
 
+  @Test
+  public void test_list_polymorph() {
+    final var greetingService = api(TypedGreetingService.class);
+
+    StepVerifier.create(greetingService.helloListPolymorph())
+        .assertNext(
+            shapes -> {
+              assertEquals(1.0, ((Circle) shapes.get(0)).radius());
+              assertEquals(1.0, ((Rectangle) shapes.get(1)).height());
+              assertEquals(1.0, ((Rectangle) shapes.get(1)).width());
+              assertEquals(1.0, ((Square) shapes.get(2)).side());
+            })
+        .thenCancel()
+        .verify();
+  }
+
   @Test
   public void test_multitype() {
     final var greetingService = api(TypedGreetingService.class);