protected void mockUserDatabase(String userId, UaaUser user) { userDatabase = mock(UaaUserDatabase.class); when(userDatabase.retrieveUserById(eq(userId))).thenReturn(user); when(userDatabase.retrieveUserById(not(eq(userId)))) .thenThrow(new UsernameNotFoundException("mock")); tokenServices.setUserDatabase(userDatabase); }