Fixing test failures on branch update

wso2-extensions · Jan 27, 2025 · aa3e9f5 · aa3e9f5
1 parent 3d61b76
commit aa3e9f5
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/...o2/carbon/identity/oauth2/token/handlers/grant/AbstractAuthorizationGrantHandlerTest.java b/...o2/carbon/identity/oauth2/token/handlers/grant/AbstractAuthorizationGrantHandlerTest.java
@@ -519,8 +519,9 @@ public void testIssueExistingAccessTokensWithoutConsent(boolean idpIdColumnEnabl
             OauthTokenIssuer oauthTokenIssuer = mock(OauthTokenIssuerImpl.class);
             when(oauthTokenIssuer.getAccessTokenType()).thenReturn(OAuth2Constants.TokenTypes.OPAQUE);
 
-            oauth2Util.when(() -> OAuth2Util.getOAuthTokenIssuerForOAuthApp(clientId)).thenReturn(oauthTokenIssuer);
-            oauth2Util.when(() -> OAuth2Util.getAppInformationByClientId(clientId)).thenReturn(oAuthAppDO);
+            oauth2Util.when(() -> OAuth2Util.getOAuthTokenIssuerForOAuthApp(eq(clientId))).thenReturn(oauthTokenIssuer);
+            oauth2Util.when(() -> OAuth2Util.getAppInformationByClientId(eq(clientId))).thenReturn(oAuthAppDO);
+            oauth2Util.when(() -> OAuth2Util.getAppInformationByClientId(eq(clientId), any())).thenReturn(oAuthAppDO);
             oauth2Util.when(() -> OAuth2Util.isOrganizationValidAndActive(anyString())).thenReturn(true);
             oauth2Util.when(() -> OAuth2Util.buildScopeString(any())).thenCallRealMethod();
             oauth2Util.when(() -> OAuth2Util.getTokenPartitionedSqlByUserStore(anyString(), any()))