From 86bca816447b02582479abd99537e57420d2e89f Mon Sep 17 00:00:00 2001
From: Patrick Pimentel <ppimentel@bitwarden.com>
Date: Tue, 2 Dec 2025 16:59:52 -0500
Subject: [PATCH] fix(auth-validator): [PM-22975] Client Version Validator -
 Changed some minor things in identity server sso tests.

---
 .../Endpoints/IdentityServerSsoTests.cs                    | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/test/Identity.IntegrationTest/Endpoints/IdentityServerSsoTests.cs b/test/Identity.IntegrationTest/Endpoints/IdentityServerSsoTests.cs
index 0ef761a112..010ac70d27 100644
--- a/test/Identity.IntegrationTest/Endpoints/IdentityServerSsoTests.cs
+++ b/test/Identity.IntegrationTest/Endpoints/IdentityServerSsoTests.cs
@@ -612,6 +612,9 @@ public class IdentityServerSsoTests
             Assert.Fail(message);
         }
 
+        // Only calls that result in a 200 OK should call this helper
+        Assert.Equal(StatusCodes.Status200OK, context.Response.StatusCode);
+
         return await AssertHelper.AssertResponseTypeIs<JsonDocument>(context);
     }
 
@@ -645,10 +648,8 @@ public class IdentityServerSsoTests
         factory.SubstituteService<IAuthorizationCodeStore>(service =>
         {
             // Return our pre-built authorization code regardless of handle representation
-            service.GetAuthorizationCodeAsync(Arg.Any<string>())
+            service.GetAuthorizationCodeAsync("test_code")
                 .Returns(authorizationCode);
-            service.RemoveAuthorizationCodeAsync(Arg.Any<string>())
-                .Returns(Task.CompletedTask);
         });
 
         var user = await factory.RegisterNewIdentityFactoryUserAsync(