From e87de0ea1f8347b2587dd0b748aa9f4ffeeabed7 Mon Sep 17 00:00:00 2001
From: Peter <34331512+pmaytak@users.noreply.github.com>
Date: Fri, 19 Sep 2025 12:39:54 -0700
Subject: [PATCH] Update test for IDX10400.

---
 .../Base64UrlEncodingTests.cs                 | 19 +++++++++++++++----
 1 file changed, 15 insertions(+), 4 deletions(-)

diff --git a/test/Microsoft.IdentityModel.Tokens.Tests/Base64UrlEncodingTests.cs b/test/Microsoft.IdentityModel.Tokens.Tests/Base64UrlEncodingTests.cs
index 861da717d2..cbe58877fb 100644
--- a/test/Microsoft.IdentityModel.Tokens.Tests/Base64UrlEncodingTests.cs
+++ b/test/Microsoft.IdentityModel.Tokens.Tests/Base64UrlEncodingTests.cs
@@ -4,6 +4,7 @@
 using System;
 using System.Reflection;
 using System.Text;
+using Microsoft.IdentityModel.Logging;
 using Microsoft.IdentityModel.TestUtils;
 using Xunit;
 
@@ -314,15 +315,25 @@ public Base64UrlEncoderTheoryData(string testId) : base(testId) { }
         }
 
         [Fact]
-        public void ValidateAndGetOutputSizeTests()
+        public void ValidateAndGetOutputSizeTests_InvalidInput_Throws()
         {
-            string input = string.Empty;
-            Assert.Throws<ArgumentNullException>(() => Base64UrlEncoding.ValidateAndGetOutputSize(input.AsSpan(), 0, 0));
+            Assert.Throws<ArgumentNullException>(() => Base64UrlEncoding.ValidateAndGetOutputSize(string.Empty.AsSpan(), 0, 0));
             Assert.Throws<ArgumentException>(() => Base64UrlEncoding.ValidateAndGetOutputSize("abc".AsSpan(), -1, 3));
             Assert.Throws<ArgumentException>(() => Base64UrlEncoding.ValidateAndGetOutputSize("abc".AsSpan(), 0, -1));
             Assert.Throws<ArgumentException>(() => Base64UrlEncoding.ValidateAndGetOutputSize("abc".AsSpan(), 0, 4));
-            Assert.Throws<FormatException>(() => Base64UrlEncoding.ValidateAndGetOutputSize("abcde".AsSpan(), 0, 5));
 
+            IdentityModelEventSource.ShowPII = true;
+            var ex = Assert.Throws<FormatException>(() => Base64UrlEncoding.ValidateAndGetOutputSize("abcde".AsSpan(), 0, 5));
+            Assert.Equal(string.Format(LogMessages.IDX10400, "abcde"), ex.Message);
+
+            IdentityModelEventSource.ShowPII = false;
+            ex = Assert.Throws<FormatException>(() => Base64UrlEncoding.ValidateAndGetOutputSize("abcde".AsSpan(), 0, 5));
+            Assert.Equal(string.Format(LogMessages.IDX10400, string.Format(IdentityModelEventSource.HiddenPIIString, "abcde".GetType())), ex.Message);
+        }
+
+        [Fact]
+        public void ValidateAndGetOutputSizeTests_ValidInput_ReturnsSize()
+        {
             int actualOutputSize = Base64UrlEncoding.ValidateAndGetOutputSize("abc".AsSpan(), 0, 0);
             Assert.Equal(0, actualOutputSize);