Esempio n. 1
0
  public ConnectionFactory<String, String> newConnectionFactoryMock() {
    @SuppressWarnings("unchecked")
    final ConnectionFactory<String, String> cfMock = Mockito.mock(ConnectionFactory.class);

    List<Instance<SocketAddress>> instances = hosts.toList().toBlocking().single();

    for (Instance<SocketAddress> instance : instances) {
      EmbeddedChannel channel = new EmbeddedChannel();
      final TcpClientEventPublisher eventPublisher = new TcpClientEventPublisher();
      final Connection<String, String> mockConnection =
          ConnectionImpl.create(channel, eventPublisher, eventPublisher);

      Mockito.when(cfMock.newConnection(instance.getValue()))
          .thenReturn(
              ConnectionObservable.createNew(
                  new OnSubcribeFunc<String, String>() {
                    @Override
                    public Subscription subscribeForEvents(TcpClientEventListener eventListener) {
                      return eventPublisher.subscribe(eventListener);
                    }

                    @Override
                    public void call(Subscriber<? super Connection<String, String>> subscriber) {
                      subscriber.onNext(mockConnection);
                      subscriber.onCompleted();
                    }
                  }));
    }

    return cfMock;
  }
Esempio n. 2
0
  public Connection<String, String> connect(
      ConnectionObservable<String, String> connectionObservable) {

    TestSubscriber<Connection<String, String>> testSub = new TestSubscriber<>();
    connectionObservable.subscribe(testSub);

    testSub.awaitTerminalEvent();
    testSub.assertNoErrors();

    testSub.assertValueCount(1);

    return testSub.getOnNextEvents().get(0);
  }