Examples with RSocket io.rsocket.RSocket used on opensource projects

Example 6 with RSocket

use of io.rsocket.RSocket in project spring-security by spring-projects.

the class PayloadInterceptorRSocketTests method requestStreamWhenSecurityContextThenDelegateContext.

@Test
public void requestStreamWhenSecurityContextThenDelegateContext() {
    TestingAuthenticationToken authentication = new TestingAuthenticationToken("user", "password");
    given(this.interceptor.intercept(any(), any())).willAnswer(withAuthenticated(authentication));
    given(this.delegate.requestStream(any())).willReturn(this.payloadResult.flux());
    RSocket assertAuthentication = new RSocketProxy(this.delegate) {

        @Override
        public Flux<Payload> requestStream(Payload payload) {
            return assertAuthentication(authentication).flatMapMany((a) -> super.requestStream(payload));
        }
    };
    PayloadInterceptorRSocket interceptor = new PayloadInterceptorRSocket(assertAuthentication, Arrays.asList(this.interceptor), this.metadataMimeType, this.dataMimeType);
    StepVerifier.create(interceptor.requestStream(this.payload)).then(() -> this.payloadResult.assertSubscribers()).then(() -> this.payloadResult.emit(this.payload)).expectNext(this.payload).verifyComplete();
    verify(this.interceptor).intercept(this.exchange.capture(), any());
    assertThat(this.exchange.getValue().getPayload()).isEqualTo(this.payload);
    verify(this.delegate).requestStream(this.payload);
}

Example 7 with RSocket

use of io.rsocket.RSocket in project spring-security by spring-projects.

the class PayloadInterceptorRSocketTests method requestChannelWhenSecurityContextThenDelegateContext.

@Test
public void requestChannelWhenSecurityContextThenDelegateContext() {
    Mono<Payload> payload = Mono.just(this.payload);
    TestingAuthenticationToken authentication = new TestingAuthenticationToken("user", "password");
    given(this.interceptor.intercept(any(), any())).willAnswer(withAuthenticated(authentication));
    given(this.delegate.requestChannel(any())).willReturn(this.payloadResult.flux());
    RSocket assertAuthentication = new RSocketProxy(this.delegate) {

        @Override
        public Flux<Payload> requestChannel(Publisher<Payload> payload) {
            return assertAuthentication(authentication).flatMapMany((a) -> super.requestChannel(payload));
        }
    };
    PayloadInterceptorRSocket interceptor = new PayloadInterceptorRSocket(assertAuthentication, Arrays.asList(this.interceptor), this.metadataMimeType, this.dataMimeType);
    StepVerifier.create(interceptor.requestChannel(payload)).then(() -> this.payloadResult.assertSubscribers()).then(() -> this.payloadResult.emit(this.payload)).expectNext(this.payload).verifyComplete();
    verify(this.interceptor).intercept(this.exchange.capture(), any());
    assertThat(this.exchange.getValue().getPayload()).isEqualTo(this.payload);
    verify(this.delegate).requestChannel(any());
}