Add unit tests.

AzureAD · Jan 5, 2021 · 7883d6b · 7883d6b
1 parent 672d9f7
commit 7883d6b
Show file tree

Hide file tree

Showing 3 changed files with 52 additions and 1 deletion.
diff --git a/src/client/Microsoft.Identity.Client/TokenCache.cs b/src/client/Microsoft.Identity.Client/TokenCache.cs
@@ -42,7 +42,7 @@ public sealed partial class TokenCache : ITokenCacheInternal
         private ICoreLogger Logger => ServiceBundle.DefaultLogger;
 
         internal IServiceBundle ServiceBundle { get; }
-        internal ILegacyCachePersistence LegacyCachePersistence { get; }
+        internal ILegacyCachePersistence LegacyCachePersistence { get; set; }
         internal string ClientId => ServiceBundle.Config.ClientId;
 
         ITokenCacheAccessor ITokenCacheInternal.Accessor => _accessor;

diff --git a/.../Microsoft.Identity.Test.Unit/AppConfigTests/ConfidentialClientApplicationBuilderTests.cs b/.../Microsoft.Identity.Test.Unit/AppConfigTests/ConfidentialClientApplicationBuilderTests.cs
@@ -338,5 +338,17 @@ public void TestConstructor_BadInstanceMetadata()
 
             Assert.AreEqual(ex.ErrorCode, MsalError.InvalidUserInstanceMetadata);
         }
+
+        [TestMethod]
+        public void TestConstructor_WithAdalCacheCompatibility()
+        {
+            var cca = ConfidentialClientApplicationBuilder
+                      .Create(TestConstants.ClientId)
+                      .WithClientSecret(TestConstants.ClientSecret)
+                      .WithAdalCacheCompatibility(true)
+                      .Build();
+
+            Assert.AreEqual(true, cca.AppConfig.AdalCacheCompatibilityEnabled);
+        }
     }
 }
diff --git a/tests/Microsoft.Identity.Test.Unit/CacheTests/TokenCacheTests.cs b/tests/Microsoft.Identity.Test.Unit/CacheTests/TokenCacheTests.cs
@@ -44,6 +44,45 @@ public override void TestInitialize()
             base.TestInitialize();
         }
 
+        [DataTestMethod]
+        [DataRow(true)]
+        [DataRow(false)]
+        public async Task WithAdalCacheCompatibilityTest_Async(bool enableAdalCacheCompatibility)
+        {
+            // Arrange
+            var legacyCachePersistence = Substitute.For<ILegacyCachePersistence>();
+            var serviceBundle = TestCommon.CreateServiceBundleWithCustomHttpManager(null, isAdalCacheEnabled: enableAdalCacheCompatibility);
+            var requestContext = new RequestContext(serviceBundle, Guid.NewGuid());
+            var response = TestConstants.CreateMsalTokenResponse();
+
+            ITokenCacheInternal cache = new TokenCache(serviceBundle, false);
+            ((TokenCache)cache).LegacyCachePersistence = legacyCachePersistence;      
+
+            var requestParams = TestCommon.CreateAuthenticationRequestParameters(serviceBundle);
+            requestParams.TenantUpdatedCanonicalAuthority = Authority.CreateAuthorityWithTenant(
+                requestParams.AuthorityInfo,
+                TestConstants.Utid);
+            requestParams.Account = new Account(TestConstants.s_userIdentifier, $"1{TestConstants.DisplayableId}", TestConstants.ProductionPrefNetworkEnvironment);
+
+            // Act
+            await cache.FindRefreshTokenAsync(requestParams).ConfigureAwait(true);
+            await cache.SaveTokenResponseAsync(requestParams, response).ConfigureAwait(true);
+            await cache.GetAccountsAsync(requestParams).ConfigureAwait(true);
+            await cache.RemoveAccountAsync(requestParams.Account, requestContext).ConfigureAwait(true);
+
+            // Assert
+            if (enableAdalCacheCompatibility)
+            {
+                legacyCachePersistence.ReceivedWithAnyArgs().LoadCache();
+                legacyCachePersistence.ReceivedWithAnyArgs().WriteCache(Arg.Any<byte[]>());
+            }
+            else
+            {
+                legacyCachePersistence.DidNotReceiveWithAnyArgs().LoadCache();
+                legacyCachePersistence.DidNotReceiveWithAnyArgs().WriteCache(Arg.Any<byte[]>());
+            }
+        }
+
         [TestMethod]
         public void GetExactScopesMatchedAccessTokenTest()
         {