From 2f336c2d8eb0ca6ac2d5966baf12c3c6d0d01a4d Mon Sep 17 00:00:00 2001
From: Ignacio Inglese <iinglese@microsoft.com>
Date: Sun, 17 Nov 2024 16:26:47 +0000
Subject: [PATCH] Added cusotm audience validation delegates for testing

---
 .../CustomAudienceValidationDelegates.cs      | 137 ++++++++++++++++++
 1 file changed, 137 insertions(+)
 create mode 100644 test/Microsoft.IdentityModel.TestUtils/TokenValidationExtensibility/CustomAudienceValidationDelegates.cs

diff --git a/test/Microsoft.IdentityModel.TestUtils/TokenValidationExtensibility/CustomAudienceValidationDelegates.cs b/test/Microsoft.IdentityModel.TestUtils/TokenValidationExtensibility/CustomAudienceValidationDelegates.cs
new file mode 100644
index 0000000000..3362aea29c
--- /dev/null
+++ b/test/Microsoft.IdentityModel.TestUtils/TokenValidationExtensibility/CustomAudienceValidationDelegates.cs
@@ -0,0 +1,137 @@
+// Copyright (c) Microsoft Corporation. All rights reserved.
+// Licensed under the MIT License.
+
+using System;
+using System.Collections.Generic;
+using Microsoft.IdentityModel.Tokens;
+
+#nullable enable
+namespace Microsoft.IdentityModel.TestUtils
+{
+    internal class CustomAudienceValidationDelegates
+    {
+        internal static ValidationResult<string> CustomAudienceValidatorDelegate(
+            IList<string> tokenAudiences,
+            SecurityToken? securityToken,
+            ValidationParameters validationParameters,
+            CallContext callContext)
+        {
+            // Returns a CustomAudienceValidationError : AudienceValidationError
+            return new CustomAudienceValidationError(
+                new MessageDetail(nameof(CustomAudienceValidatorDelegate), null),
+                typeof(SecurityTokenInvalidAudienceException),
+                ValidationError.GetCurrentStackFrame(),
+                tokenAudiences,
+                null);
+        }
+
+        internal static ValidationResult<string> CustomAudienceValidatorCustomExceptionDelegate(
+            IList<string> tokenAudiences,
+            SecurityToken? securityToken,
+            ValidationParameters validationParameters,
+            CallContext callContext)
+        {
+            return new CustomAudienceValidationError(
+                new MessageDetail(nameof(CustomAudienceValidatorCustomExceptionDelegate), null),
+                typeof(CustomSecurityTokenInvalidAudienceException),
+                ValidationError.GetCurrentStackFrame(),
+                tokenAudiences,
+                null);
+        }
+
+        internal static ValidationResult<string> CustomAudienceValidatorCustomExceptionCustomFailureTypeDelegate(
+            IList<string> tokenAudiences,
+            SecurityToken? securityToken,
+            ValidationParameters validationParameters,
+            CallContext callContext)
+        {
+            return new CustomAudienceValidationError(
+                new MessageDetail(nameof(CustomAudienceValidatorCustomExceptionCustomFailureTypeDelegate), null),
+                typeof(CustomSecurityTokenInvalidAudienceException),
+                ValidationError.GetCurrentStackFrame(),
+                tokenAudiences,
+                null,
+                CustomAudienceValidationError.CustomAudienceValidationFailureType);
+        }
+
+        internal static ValidationResult<string> CustomAudienceValidatorUnknownExceptionDelegate(
+            IList<string> tokenAudiences,
+            SecurityToken? securityToken,
+            ValidationParameters validationParameters,
+            CallContext callContext)
+        {
+            return new CustomAudienceValidationError(
+                new MessageDetail(nameof(CustomAudienceValidatorUnknownExceptionDelegate), null),
+                typeof(NotSupportedException),
+                ValidationError.GetCurrentStackFrame(),
+                tokenAudiences,
+                null);
+        }
+
+        internal static ValidationResult<string> CustomAudienceValidatorWithoutGetExceptionOverrideDelegate(
+            IList<string> tokenAudiences,
+            SecurityToken? securityToken,
+            ValidationParameters validationParameters,
+            CallContext callContext)
+        {
+            return new CustomAudienceWithoutGetExceptionValidationOverrideError(
+                new MessageDetail(nameof(CustomAudienceValidatorWithoutGetExceptionOverrideDelegate), null),
+                typeof(CustomSecurityTokenInvalidAudienceException),
+                ValidationError.GetCurrentStackFrame(),
+                tokenAudiences,
+                null);
+        }
+
+        internal static ValidationResult<string> AudienceValidatorDelegate(
+            IList<string> tokenAudiences,
+            SecurityToken? securityToken,
+            ValidationParameters validationParameters,
+            CallContext callContext)
+        {
+            return new AudienceValidationError(
+                new MessageDetail(nameof(AudienceValidatorDelegate), null),
+                typeof(SecurityTokenInvalidAudienceException),
+                ValidationError.GetCurrentStackFrame(),
+                tokenAudiences,
+                null);
+        }
+
+        internal static ValidationResult<string> AudienceValidatorThrows(
+            IList<string> tokenAudiences,
+            SecurityToken? securityToken,
+            ValidationParameters validationParameters,
+            CallContext callContext)
+        {
+            throw new CustomSecurityTokenInvalidAudienceException(nameof(AudienceValidatorThrows), null);
+        }
+
+        internal static ValidationResult<string> AudienceValidatorCustomAudienceExceptionTypeDelegate(
+            IList<string> tokenAudiences,
+            SecurityToken? securityToken,
+            ValidationParameters validationParameters,
+            CallContext callContext)
+        {
+            return new AudienceValidationError(
+                new MessageDetail(nameof(AudienceValidatorCustomAudienceExceptionTypeDelegate), null),
+                typeof(CustomSecurityTokenInvalidAudienceException),
+                ValidationError.GetCurrentStackFrame(),
+                tokenAudiences,
+                null);
+        }
+
+        internal static ValidationResult<string> AudienceValidatorCustomExceptionTypeDelegate(
+            IList<string> tokenAudiences,
+            SecurityToken? securityToken,
+            ValidationParameters validationParameters,
+            CallContext callContext)
+        {
+            return new AudienceValidationError(
+                new MessageDetail(nameof(AudienceValidatorCustomExceptionTypeDelegate), null),
+                typeof(CustomSecurityTokenException),
+                ValidationError.GetCurrentStackFrame(),
+                tokenAudiences,
+                null);
+        }
+    }
+}
+#nullable restore