Users hierarchy simplification

1 files changed, 9 insertions, 1 deletions

diff --git a/src/test/scala/xyz/driver/core/AuthTest.scala b/src/test/scala/xyz/driver/core/AuthTest.scala
index 9018a3e..9abc7e0 100644
--- a/src/test/scala/xyz/driver/core/AuthTest.scala
+++ b/src/test/scala/xyz/driver/core/AuthTest.scala
@@ -9,6 +9,7 @@ import org.scalatest.mock.MockitoSugar
 import org.scalatest.{FlatSpec, Matchers}
 import pdi.jwt.{Jwt, JwtAlgorithm}
 import xyz.driver.core.auth._
+import xyz.driver.core.domain.Email
 import xyz.driver.core.logging._
 import xyz.driver.core.rest._
 
@@ -51,7 +52,14 @@ class AuthTest extends FlatSpec with Matchers with MockitoSugar with ScalatestRo
     override def authenticatedUser(implicit ctx: ServiceRequestContext): OptionT[Future, User] =
       OptionT.optionT[Future] {
         if (ctx.contextHeaders.keySet.contains(AuthProvider.AuthenticationTokenHeader)) {
-          Future.successful(Some(BasicUser(Id[User]("1"), Set(TestRole))))
+          Future.successful(Some(AuthTokenUserInfo(
+            Id[User]("1"),
+            authUserId = Id[AuthUser]("2"),
+            Email("foo", "bar"),
+            emailVerified = true,
+            audience = "driver",
+            roles = Set(TestRole)
+          )))
         } else {
           Future.successful(Option.empty[User])
         }