Fixed test

scalecube · May 13, 2020 · 10a1393 · 10a1393
1 parent 471b9c6
commit 10a1393
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/services/src/test/java/io/scalecube/services/sut/security/SecuredServiceImpl.java b/services/src/test/java/io/scalecube/services/sut/security/SecuredServiceImpl.java
@@ -1,5 +1,6 @@
 package io.scalecube.services.sut.security;
 
+import io.scalecube.services.auth.Authenticator;
 import io.scalecube.services.exceptions.ForbiddenException;
 import reactor.core.publisher.Mono;
 
@@ -12,7 +13,8 @@ public Mono<String> helloWithRequest(String name) {
 
   @Override
   public Mono<String> helloWithPrincipal() {
-    return Mono.deferWithContext(context -> Mono.just(context.get(UserProfile.class)))
+    return Mono.deferWithContext(context -> Mono.just(context.get(Authenticator.AUTH_CONTEXT_KEY)))
+        .cast(UserProfile.class)
         .flatMap(
             user -> {
               checkPrincipal(user);
@@ -22,7 +24,8 @@ public Mono<String> helloWithPrincipal() {
 
   @Override
   public Mono<String> helloWithRequestAndPrincipal(String name) {
-    return Mono.deferWithContext(context -> Mono.just(context.get(UserProfile.class)))
+    return Mono.deferWithContext(context -> Mono.just(context.get(Authenticator.AUTH_CONTEXT_KEY)))
+        .cast(UserProfile.class)
         .flatMap(
             user -> {
               checkPrincipal(user);