use of io.scalecube.services.auth.Authenticator.AUTH_CONTEXT_KEY in project scalecube by scalecube.
the class ServiceMethodInvokerTest method testInvokeBidirectionalWhenReturnNull.
@Test
@DisplayName("invokeBidirectional should return empty response when service returns null")
void testInvokeBidirectionalWhenReturnNull() throws Exception {
final String methodName = "returnNull3";
final Class<? extends StubService> serviceClass = stubService.getClass();
final Method method = serviceClass.getMethod(methodName, Flux.class);
final MethodInfo methodInfo = new MethodInfo(serviceClass.getName(), methodName, method.getReturnType(), IS_RETURN_TYPE_SERVICE_MESSAGE, CommunicationMode.REQUEST_CHANNEL, method.getParameterCount(), Void.TYPE, IS_REQUEST_TYPE_SERVICE_MESSAGE, AUTH);
serviceMethodInvoker = new ServiceMethodInvoker(method, stubService, methodInfo, DefaultErrorMapper.INSTANCE, dataDecoder, nullAuthenticator, nullPrincipalMapper);
ServiceMessage message = ServiceMessage.builder().qualifier(qualifierPrefix + methodName).build();
StepVerifier.create(Flux.deferContextual(context -> serviceMethodInvoker.invokeBidirectional(Flux.just(message))).contextWrite(context -> context.put(AUTH_CONTEXT_KEY, AUTH_DATA))).verifyComplete();
}
use of io.scalecube.services.auth.Authenticator.AUTH_CONTEXT_KEY in project scalecube by scalecube.
the class ServiceMethodInvokerTest method testInvokeManyWhenThrowException.
@Test
@DisplayName("invokeMany should return error response when service throws exception")
void testInvokeManyWhenThrowException() throws Exception {
final String methodName = "throwException2";
final Class<? extends StubService> serviceClass = stubService.getClass();
final Method method = serviceClass.getMethod(methodName);
final MethodInfo methodInfo = new MethodInfo(serviceClass.getName(), methodName, method.getReturnType(), IS_RETURN_TYPE_SERVICE_MESSAGE, CommunicationMode.REQUEST_STREAM, method.getParameterCount(), Void.TYPE, IS_REQUEST_TYPE_SERVICE_MESSAGE, AUTH);
serviceMethodInvoker = new ServiceMethodInvoker(method, stubService, methodInfo, DefaultErrorMapper.INSTANCE, dataDecoder, nullAuthenticator, nullPrincipalMapper);
ServiceMessage message = ServiceMessage.builder().qualifier(qualifierPrefix + methodName).build();
final Flux<ServiceMessage> invokeOne = Flux.deferContextual(context -> serviceMethodInvoker.invokeMany(message)).contextWrite(context -> context.put(AUTH_CONTEXT_KEY, AUTH_DATA));
StepVerifier.create(invokeOne).assertNext(serviceMessage -> Assertions.assertTrue(serviceMessage.isError())).verifyComplete();
}
use of io.scalecube.services.auth.Authenticator.AUTH_CONTEXT_KEY in project scalecube by scalecube.
the class ServiceMethodInvokerTest method testAuthMethodWhenThereIsContextAndNoAuthenticator.
@Test
@DisplayName("invocation of auth method should return empty response " + "if auth.context exists and no authenticator")
void testAuthMethodWhenThereIsContextAndNoAuthenticator() throws Exception {
final String methodName = "helloAuthContext";
final Class<? extends StubService> serviceClass = stubService.getClass();
final Method method = serviceClass.getMethod(methodName);
final MethodInfo methodInfo = new MethodInfo(serviceClass.getName(), methodName, method.getReturnType(), IS_RETURN_TYPE_SERVICE_MESSAGE, CommunicationMode.REQUEST_RESPONSE, method.getParameterCount(), Void.TYPE, IS_REQUEST_TYPE_SERVICE_MESSAGE, AUTH);
serviceMethodInvoker = new ServiceMethodInvoker(method, stubService, methodInfo, DefaultErrorMapper.INSTANCE, dataDecoder, nullAuthenticator, principalMapper);
ServiceMessage message = ServiceMessage.builder().qualifier(qualifierPrefix + methodName).build();
StepVerifier.create(Mono.deferContextual(context -> serviceMethodInvoker.invokeOne(message)).contextWrite(context -> context.put(AUTH_CONTEXT_KEY, AUTH_DATA))).verifyComplete();
}
use of io.scalecube.services.auth.Authenticator.AUTH_CONTEXT_KEY in project scalecube by scalecube.
the class ServiceMethodInvokerTest method testInvokeManyWhenReturnNull.
@Test
@DisplayName("invokeMany should return empty response when service returns null")
void testInvokeManyWhenReturnNull() throws Exception {
final String methodName = "returnNull2";
final Class<? extends StubService> serviceClass = stubService.getClass();
final Method method = serviceClass.getMethod(methodName);
final MethodInfo methodInfo = new MethodInfo(serviceClass.getName(), methodName, method.getReturnType(), IS_RETURN_TYPE_SERVICE_MESSAGE, CommunicationMode.REQUEST_STREAM, method.getParameterCount(), Void.TYPE, IS_REQUEST_TYPE_SERVICE_MESSAGE, AUTH);
serviceMethodInvoker = new ServiceMethodInvoker(method, stubService, methodInfo, DefaultErrorMapper.INSTANCE, dataDecoder, nullAuthenticator, nullPrincipalMapper);
ServiceMessage message = ServiceMessage.builder().qualifier(qualifierPrefix + methodName).build();
StepVerifier.create(Flux.deferContextual(context -> serviceMethodInvoker.invokeMany(message)).contextWrite(context -> context.put(AUTH_CONTEXT_KEY, AUTH_DATA))).verifyComplete();
}
use of io.scalecube.services.auth.Authenticator.AUTH_CONTEXT_KEY in project scalecube by scalecube.
the class ServiceMethodInvokerTest method testInvokeBidirectionalWhenThrowException.
@Test
@DisplayName("invokeBidirectional should return error response when service throws exception")
void testInvokeBidirectionalWhenThrowException() throws Exception {
final String methodName = "throwException3";
final Class<? extends StubService> serviceClass = stubService.getClass();
final Method method = serviceClass.getMethod(methodName, Flux.class);
final MethodInfo methodInfo = new MethodInfo(serviceClass.getName(), methodName, method.getReturnType(), IS_RETURN_TYPE_SERVICE_MESSAGE, CommunicationMode.REQUEST_CHANNEL, method.getParameterCount(), Void.TYPE, IS_REQUEST_TYPE_SERVICE_MESSAGE, AUTH);
serviceMethodInvoker = new ServiceMethodInvoker(method, stubService, methodInfo, DefaultErrorMapper.INSTANCE, dataDecoder, nullAuthenticator, nullPrincipalMapper);
ServiceMessage message = ServiceMessage.builder().qualifier(qualifierPrefix + methodName).build();
// invokeOne
final Flux<ServiceMessage> invokeOne = Flux.deferContextual(context -> serviceMethodInvoker.invokeBidirectional(Flux.just(message))).contextWrite(context -> context.put(AUTH_CONTEXT_KEY, AUTH_DATA));
StepVerifier.create(invokeOne).assertNext(serviceMessage -> Assertions.assertTrue(serviceMessage.isError())).verifyComplete();
}
Aggregations