io.rsocket.test.ClientSetupRule Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of rsocket-test Show documentation
Test utilities for RSocket projects
There is a newer version: 1.1.4
/*
 * Copyright 2015-2020 the original author or authors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package io.rsocket.test;

import io.rsocket.Closeable;
import io.rsocket.RSocket;
import io.rsocket.core.RSocketConnector;
import io.rsocket.core.RSocketServer;
import io.rsocket.transport.ClientTransport;
import io.rsocket.transport.ServerTransport;
import java.util.function.BiFunction;
import java.util.function.Function;
import java.util.function.Supplier;
import org.junit.rules.ExternalResource;
import org.junit.runner.Description;
import org.junit.runners.model.Statement;
import reactor.core.publisher.Mono;

public class ClientSetupRule extends ExternalResource {
  private static final String data = "hello world";
  private static final String metadata = "metadata";

  private Supplier addressSupplier;
  private BiFunction clientConnector;
  private Function serverInit;

  private RSocket client;

  public ClientSetupRule(
      Supplier addressSupplier,
      BiFunction clientTransportSupplier,
      Function> serverTransportSupplier) {
    this.addressSupplier = addressSupplier;

    this.serverInit =
        address ->
            RSocketServer.create((setup, rsocket) -> Mono.just(new TestRSocket(data, metadata)))
                .bind(serverTransportSupplier.apply(address))
                .block();

    this.clientConnector =
        (address, server) ->
            RSocketConnector.connectWith(clientTransportSupplier.apply(address, server))
                .doOnError(Throwable::printStackTrace)
                .block();
  }

  @Override
  public Statement apply(Statement base, Description description) {
    return new Statement() {
      @Override
      public void evaluate() throws Throwable {
        T address = addressSupplier.get();
        S server = serverInit.apply(address);
        client = clientConnector.apply(address, server);
        base.evaluate();
        server.dispose();
      }
    };
  }

  public RSocket getRSocket() {
    return client;
  }

  public String expectedPayloadData() {
    return data;
  }

  public String expectedPayloadMetadata() {
    return metadata;
  }
}