diff --git a/gms/src/test/java/it/inaf/ia2/gms/authn/SessionDataTest.java b/gms/src/test/java/it/inaf/ia2/gms/authn/SessionDataTest.java
index d525b88779193172c722419dcebb568b7f56fd6e..00648021317db5c3704b58e56feb509e7072f644 100644
--- a/gms/src/test/java/it/inaf/ia2/gms/authn/SessionDataTest.java
+++ b/gms/src/test/java/it/inaf/ia2/gms/authn/SessionDataTest.java
@@ -1,18 +1,17 @@
 package it.inaf.ia2.gms.authn;
 
-import java.util.ArrayList;
-import java.util.HashMap;
+import it.inaf.ia2.aa.data.User;
 import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpSession;
 import static org.junit.Assert.assertTrue;
 import org.junit.Test;
 import org.junit.runner.RunWith;
+import static org.mockito.ArgumentMatchers.eq;
 import org.mockito.InjectMocks;
 import org.mockito.Mock;
 import static org.mockito.Mockito.mock;
 import static org.mockito.Mockito.when;
 import org.mockito.junit.MockitoJUnitRunner;
-import org.springframework.security.oauth2.common.OAuth2AccessToken;
-import org.springframework.security.oauth2.provider.OAuth2Authentication;
 
 @RunWith(MockitoJUnitRunner.class)
 public class SessionDataTest {
@@ -26,15 +25,17 @@ public class SessionDataTest {
     @Test
     public void testExpired() {
 
-        OAuth2AccessToken accessToken = mock(OAuth2AccessToken.class);
-        when(accessToken.getExpiresIn()).thenReturn(3600);
+        HttpSession session = mock(HttpSession.class);
+        when(request.getSession(eq(false))).thenReturn(session);
 
-        CustomAuthenticationData data = new CustomAuthenticationData("user",
-                new HashMap<>(), new ArrayList<>(), accessToken, "refresh_token");
+        User user = new User()
+                .setUserId("123")
+                .setUserLabel("Name Surname")
+                .setAccessToken("<access_token>")
+                .setRefreshToken("<refresh_token>")
+                .setExpiresIn(3600);
 
-        OAuth2Authentication auth = mock(OAuth2Authentication.class);
-        when(auth.getUserAuthentication()).thenReturn(data);
-        when(request.getUserPrincipal()).thenReturn(auth);
+        when(session.getAttribute(eq("user_data"))).thenReturn(user);
 
         sessionData.init();
 
diff --git a/gms/src/test/java/it/inaf/ia2/gms/rap/RapClientTest.java b/gms/src/test/java/it/inaf/ia2/gms/rap/RapClientTest.java
index 56651a9b695c4776aa94aa272a26dceb14690807..bd317c09d73c7def90f287ede0cf2c05ab08c584 100644
--- a/gms/src/test/java/it/inaf/ia2/gms/rap/RapClientTest.java
+++ b/gms/src/test/java/it/inaf/ia2/gms/rap/RapClientTest.java
@@ -148,6 +148,6 @@ public class RapClientTest {
 
         // verifies that token is refreshed
         verify(sessionData, times(1)).setAccessToken(eq("<access_token>"));
-        verify(sessionData, times(1)).setExpiresIn(eq(3600));
+        verify(sessionData, times(1)).setExpiresIn(eq(3600l));
     }
 }