@Override
protected void tearDown() throws Exception {
AuthenticationSettings.INSTANCE.setSharedPrefPackageName(null);
DefaultTokenCacheStore store = new DefaultTokenCacheStore(ctx);
store.removeAll();
super.tearDown();
}
public void testSharedCache()
throws NoSuchAlgorithmException, NoSuchPaddingException, InvalidKeyException,
InvalidKeySpecException, KeyStoreException, CertificateException, NoSuchProviderException,
InvalidAlgorithmParameterException, UnrecoverableEntryException, DigestException,
IllegalBlockSizeException, BadPaddingException, IOException, NameNotFoundException,
NoSuchFieldException, IllegalArgumentException, IllegalAccessException {
AuthenticationSettings.INSTANCE.setSharedPrefPackageName("mockpackage");
StorageHelper mockSecure = mock(StorageHelper.class);
Context mockContext = mock(Context.class);
Context packageContext = mock(Context.class);
SharedPreferences prefs = mock(SharedPreferences.class);
when(prefs.contains("testkey")).thenReturn(true);
when(prefs.getString("testkey", "")).thenReturn("test_encrypted");
when(mockSecure.decrypt("test_encrypted")).thenReturn("{\"mClientId\":\"clientId23\"}");
when(mockContext.createPackageContext("mockpackage", Context.MODE_PRIVATE))
.thenReturn(packageContext);
when(packageContext.getSharedPreferences("com.microsoft.aad.adal.cache", Activity.MODE_PRIVATE))
.thenReturn(prefs);
Class<?> c = DefaultTokenCacheStore.class;
Field encryptHelper = c.getDeclaredField("sHelper");
encryptHelper.setAccessible(true);
encryptHelper.set(null, mockSecure);
DefaultTokenCacheStore cache = new DefaultTokenCacheStore(mockContext);
TokenCacheItem item = cache.getItem("testkey");
// Verify returned item
assertEquals("Same item as mock", "clientId23", item.getClientId());
encryptHelper.set(null, null);
}
public void testExpireBuffer() throws NoSuchAlgorithmException, NoSuchPaddingException {
DefaultTokenCacheStore store = (DefaultTokenCacheStore) setupItems();
ArrayList<TokenCacheItem> tokens = store.getTokensForUser("userid1");
Calendar expireTime = Calendar.getInstance();
Logger.d(TAG, "Time now: " + expireTime.getTime());
expireTime.add(Calendar.SECOND, 240);
Logger.d(TAG, "Time modified: " + expireTime.getTime());
// Sets token to expire if less than this buffer
AuthenticationSettings.INSTANCE.setExpirationBuffer(300);
for (TokenCacheItem item : tokens) {
item.setExpiresOn(expireTime.getTime());
assertTrue("Should say expired", TokenCacheItem.isTokenExpired(item.getExpiresOn()));
}
// Set expire time ahead of buffer 240 +100 secs more than 300secs
// buffer
expireTime.add(Calendar.SECOND, 100);
for (TokenCacheItem item : tokens) {
item.setExpiresOn(expireTime.getTime());
assertFalse(
"Should not say expired since time is more than buffer",
TokenCacheItem.isTokenExpired(item.getExpiresOn()));
}
}
