public void GetIdToken_Test() { var accessor = new InMemoryPartitionedUserTokenCacheAccessor(new NullLogger(), null); var idt1 = TokenCacheHelper.CreateIdTokenCacheItem("tenant1", "homeAccountId"); var idt2 = TokenCacheHelper.CreateIdTokenCacheItem("tenant1", "homeAccountId2"); var idt3 = TokenCacheHelper.CreateIdTokenCacheItem("tenant2", "homeAccountId"); var at2 = TokenCacheHelper.CreateAccessTokenItem("scope1", "tenant1", "homeAccountId2"); var at3 = TokenCacheHelper.CreateAccessTokenItem("scope1", "tenant2", "homeAccountId"); // Assert: Null non-existing item Assert.IsNull(accessor.GetIdToken(at2)); accessor.SaveIdToken(idt1); accessor.SaveIdToken(idt2); accessor.SaveIdToken(idt3); // Assert: Get token by key Assert.AreEqual(idt2.GetKey(), accessor.GetIdToken(at2).GetKey()); Assert.AreEqual(idt3.GetKey(), accessor.GetIdToken(at3).GetKey()); }
private ITokenCacheAccessor CreateTokenCacheAccessorWithKeyPrefix( string keyPrefix, int numAccessTokens, int numRefreshTokens, int numIdTokens, int numAccounts) { var accessor = new InMemoryPartitionedUserTokenCacheAccessor(Substitute.For <ICoreLogger>(), null); for (int i = 1; i <= numAccessTokens; i++) { var item = CreateAccessTokenItem( cachedAt: DateTimeHelpers.UnixTimestampToDateTime(34567), expiresOn: DateTimeHelpers.UnixTimestampToDateTime(12345), extendedExpiresOn: DateTimeHelpers.UnixTimestampToDateTime(23456)); item.Environment = item.Environment + $"_{keyPrefix}{i}"; // ensure we get unique cache keys accessor.SaveAccessToken(item); } for (int i = 1; i <= numRefreshTokens; i++) { var item = CreateRefreshTokenItem(); item.Environment = item.Environment + $"_{keyPrefix}{i}"; // ensure we get unique cache keys accessor.SaveRefreshToken(item); } // Create an FRT var frt = CreateRefreshTokenItem(true); frt.FamilyId = "1"; accessor.SaveRefreshToken(frt); for (int i = 1; i <= numIdTokens; i++) { var item = CreateIdTokenItem(); item.Environment = item.Environment + $"_{keyPrefix}{i}"; // ensure we get unique cache keys accessor.SaveIdToken(item); } for (int i = 1; i <= numAccounts; i++) { var item = CreateAccountItem(); item.Environment = item.Environment + $"_{keyPrefix}{i}"; // ensure we get unique cache keys accessor.SaveAccount(item); } accessor.SaveAppMetadata(new MsalAppMetadataCacheItem(TestConstants.ClientId, "env_1", "1")); accessor.SaveAppMetadata(new MsalAppMetadataCacheItem(TestConstants.ClientId, "env_2", "")); accessor.SaveAppMetadata(new MsalAppMetadataCacheItem(TestConstants.ClientId2, "env_1", "another_family")); return(accessor); }
public void DeleteIdToken_Test() { var accessor = new InMemoryPartitionedUserTokenCacheAccessor(new NullLogger(), null); var idt1 = TokenCacheHelper.CreateIdTokenCacheItem("tenant1", "homeAccountId"); var idt2 = TokenCacheHelper.CreateIdTokenCacheItem("tenant1", "homeAccountId2"); var idt3 = TokenCacheHelper.CreateIdTokenCacheItem("tenant2", "homeAccountId"); // Assert: Delete on empty collection doesn't throw accessor.DeleteIdToken(idt1); accessor.SaveIdToken(idt1); accessor.SaveIdToken(idt2); accessor.SaveIdToken(idt3); Assert.AreEqual(3, accessor.GetAllIdTokens().Count); // Assert: Delete on existing item accessor.DeleteIdToken(idt1); Assert.AreEqual(2, accessor.GetAllIdTokens().Count); }
public void GetAllIdTokens_Test() { var accessor = new InMemoryPartitionedUserTokenCacheAccessor(new NullLogger(), null); var idt1 = TokenCacheHelper.CreateIdTokenCacheItem("tenant1", "homeAccountId"); var idt2 = TokenCacheHelper.CreateIdTokenCacheItem("tenant1", "homeAccountId2"); var idt3 = TokenCacheHelper.CreateIdTokenCacheItem("tenant2", "homeAccountId"); string partitionKey1 = CacheKeyFactory.GetKeyFromCachedItem(idt1); string partitionKey2 = CacheKeyFactory.GetKeyFromCachedItem(idt2); // Assert: Returns empty collection Assert.AreEqual(0, accessor.GetAllIdTokens().Count); Assert.AreEqual(0, accessor.GetAllIdTokens(partitionKey1).Count); accessor.SaveIdToken(idt1); accessor.SaveIdToken(idt2); accessor.SaveIdToken(idt3); // Assert: Get all tokens and get all tokens by partition key Assert.AreEqual(3, accessor.GetAllIdTokens().Count); Assert.AreEqual(2, accessor.GetAllIdTokens(partitionKey1).Count); Assert.AreEqual(1, accessor.GetAllIdTokens(partitionKey2).Count); }
public void SaveIdToken_Test() { var accessor = new InMemoryPartitionedUserTokenCacheAccessor(new NullLogger(), null); var idt1 = TokenCacheHelper.CreateIdTokenCacheItem("tenant1", "homeAccountId"); // Assert: Saves with new item accessor.SaveIdToken(idt1); Assert.AreEqual(1, accessor.GetAllIdTokens().Count); Assert.AreEqual(1, accessor.IdTokenCacheDictionary.Count); string partitionKey1 = CacheKeyFactory.GetKeyFromCachedItem(idt1); Assert.IsNotNull(accessor.IdTokenCacheDictionary[partitionKey1][idt1.GetKey().ToString()]); var idt2 = TokenCacheHelper.CreateIdTokenCacheItem("tenant2", "homeAccountId"); // Assert: Save under the existing partition accessor.SaveIdToken(idt2); Assert.AreEqual(2, accessor.GetAllIdTokens().Count); Assert.AreEqual(1, accessor.IdTokenCacheDictionary.Count); Assert.IsNotNull(accessor.IdTokenCacheDictionary[partitionKey1][idt2.GetKey().ToString()]); var idt3 = TokenCacheHelper.CreateIdTokenCacheItem("tenant1", "homeAccountId2"); // Assert: Save under a new partition accessor.SaveIdToken(idt3); // Assert: Save overwrites the existing token accessor.SaveIdToken(idt3); Assert.AreEqual(3, accessor.GetAllIdTokens().Count); Assert.AreEqual(2, accessor.IdTokenCacheDictionary.Count); string partitionKey2 = CacheKeyFactory.GetKeyFromCachedItem(idt3); Assert.IsNotNull(accessor.IdTokenCacheDictionary[partitionKey2][idt3.GetKey().ToString()]); }
public void ClearCache_UserCache_Test() { var accessor = new InMemoryPartitionedUserTokenCacheAccessor(new NullLogger(), null); accessor.SaveAccessToken(TokenCacheHelper.CreateAccessTokenItem()); accessor.SaveRefreshToken(TokenCacheHelper.CreateRefreshTokenItem()); accessor.SaveIdToken(TokenCacheHelper.CreateIdTokenCacheItem()); accessor.SaveAccount(TokenCacheHelper.CreateAccountItem()); Assert.AreEqual(1, accessor.AccessTokenCacheDictionary.Count); Assert.AreEqual(1, accessor.RefreshTokenCacheDictionary.Count); Assert.AreEqual(1, accessor.IdTokenCacheDictionary.Count); Assert.AreEqual(1, accessor.AccountCacheDictionary.Count); accessor.Clear(); Assert.AreEqual(0, accessor.AccessTokenCacheDictionary.Count); Assert.AreEqual(0, accessor.RefreshTokenCacheDictionary.Count); Assert.AreEqual(0, accessor.IdTokenCacheDictionary.Count); Assert.AreEqual(0, accessor.AccountCacheDictionary.Count); }