fixed tests

Author: robertroeser

reactivesocket-aeron/src/examples/java/io/reactivesocket/aeron/example/fireandforget/Forget.java

@@ -18,6 +18,7 @@
 import io.reactivesocket.ConnectionSetupHandler;
 import io.reactivesocket.ConnectionSetupPayload;
 import io.reactivesocket.Payload;
+import io.reactivesocket.ReactiveSocket;
 import io.reactivesocket.RequestHandler;
 import io.reactivesocket.aeron.server.ReactiveSocketAeronServer;
 import io.reactivesocket.exceptions.SetupException;
@@ -33,7 +34,7 @@ public static void main(String... args) {
 
         ReactiveSocketAeronServer server = ReactiveSocketAeronServer.create(host, 39790, new ConnectionSetupHandler() {
             @Override
-            public RequestHandler apply(ConnectionSetupPayload setupPayload) throws SetupException {
+            public RequestHandler apply(ConnectionSetupPayload setupPayload, ReactiveSocket rs) throws SetupException {
                 return new RequestHandler() {
                     @Override
                     public Publisher<Payload> handleRequestResponse(Payload payload) {

reactivesocket-aeron/src/examples/java/io/reactivesocket/aeron/example/requestreply/Pong.java

@@ -18,6 +18,7 @@
 import io.reactivesocket.ConnectionSetupHandler;
 import io.reactivesocket.ConnectionSetupPayload;
 import io.reactivesocket.Payload;
+import io.reactivesocket.ReactiveSocket;
 import io.reactivesocket.RequestHandler;
 import io.reactivesocket.aeron.server.ReactiveSocketAeronServer;
 import io.reactivesocket.exceptions.SetupException;
@@ -44,7 +45,7 @@ public static void main(String... args) {
 
         ReactiveSocketAeronServer server = ReactiveSocketAeronServer.create(host, 39790, new ConnectionSetupHandler() {
             @Override
-            public RequestHandler apply(ConnectionSetupPayload setupPayload) throws SetupException {
+            public RequestHandler apply(ConnectionSetupPayload setupPayload, ReactiveSocket rs) throws SetupException {
                 return new RequestHandler() {
                     @Override
                     public Publisher<Payload> handleRequestResponse(Payload payload) {

reactivesocket-aeron/src/test/java/io/reactivesocket/aeron/client/ReactiveSocketAeronTest.java

@@ -112,7 +112,7 @@ public void requestResponseN(int count) throws Exception {
         AtomicLong counter = new AtomicLong();
         ReactiveSocketAeronServer.create(new ConnectionSetupHandler() {
             @Override
-            public RequestHandler apply(ConnectionSetupPayload setupPayload) throws SetupException {
+            public RequestHandler apply(ConnectionSetupPayload setupPayload, ReactiveSocket rs) throws SetupException {
                 return new RequestHandler.Builder()
                   .withRequestResponse(new Function<Payload, Publisher<Payload>>() {
                       Frame frame = Frame.from(ByteBuffer.allocate(1));
@@ -194,7 +194,7 @@ public void onNext(Payload payload) {
     }
 
     public void requestStreamN(int count) throws Exception {
-        ReactiveSocketAeronServer.create(setupPayload ->
+        ReactiveSocketAeronServer.create((setupPayload, rs) ->
           new RequestHandler.Builder()
             .withRequestStream(payload -> {
                 ByteBuffer data = payload.getData();
@@ -264,7 +264,7 @@ public void testReconnection() throws Exception {
 
         ReactiveSocketAeronServer server = ReactiveSocketAeronServer.create(new ConnectionSetupHandler() {
             @Override
-            public RequestHandler apply(ConnectionSetupPayload setupPayload) throws SetupException {
+            public RequestHandler apply(ConnectionSetupPayload setupPayload, ReactiveSocket rs) throws SetupException {
                 return new RequestHandler() {
                     Frame frame = Frame.from(ByteBuffer.allocate(1));
 

reactivesocket-jsr-356/src/test/java/io/reactivesocket/javax/websocket/ClientServerEndpoint.java

@@ -25,7 +25,7 @@
 
 public class ClientServerEndpoint extends ReactiveSocketWebSocketServer {
     public ClientServerEndpoint() {
-        super(setupPayload -> new RequestHandler() {
+        super((setupPayload, rs) -> new RequestHandler() {
             @Override
             public Publisher<Payload> handleRequestResponse(Payload payload) {
                 return s -> {

reactivesocket-jsr-356/src/test/java/io/reactivesocket/javax/websocket/PongEndpoint.java

@@ -34,7 +34,7 @@ public class PongEndpoint extends ReactiveSocketWebSocketServer {
     }
 
     public PongEndpoint() {
-        super(setupPayload -> new RequestHandler() {
+        super((setupPayload, rs) -> new RequestHandler() {
             @Override
             public Publisher<Payload> handleRequestResponse(Payload payload) {
                 return new Publisher<Payload>() {

reactivesocket-local/src/test/java/io/reactivesocket/local/ClientServerTest.java

@@ -41,7 +41,7 @@ public class ClientServerTest {
     public static void setup() throws Exception {
         server = LocalServerReactiveSocketFactory.INSTANCE.callAndWait(new LocalServerReactiveSocketFactory.Config("test", new ConnectionSetupHandler() {
             @Override
-            public RequestHandler apply(ConnectionSetupPayload setupPayload) throws SetupException {
+            public RequestHandler apply(ConnectionSetupPayload setupPayload, ReactiveSocket rs) throws SetupException {
                 return new RequestHandler() {
                     @Override
                     public Publisher<Payload> handleRequestResponse(Payload payload) {