reset the resident idp cache

Author: DilshanSenarath

modules/integration/tests-integration/tests-backend/src/test/java/org/wso2/identity/integration/test/rest/api/server/session/SessionMgtTest.java

@@ -18,15 +18,11 @@
 
 package org.wso2.identity.integration.test.rest.api.server.session;
 
-import io.restassured.RestAssured;
 import io.restassured.response.Response;
-import org.apache.commons.lang.StringUtils;
 import org.apache.http.HttpStatus;
 import org.apache.http.client.CookieStore;
 import org.testng.annotations.AfterClass;
-import org.testng.annotations.AfterMethod;
 import org.testng.annotations.BeforeClass;
-import org.testng.annotations.BeforeMethod;
 import org.testng.annotations.DataProvider;
 import org.testng.annotations.Factory;
 import org.testng.annotations.Test;
@@ -78,6 +74,7 @@ public static Object[][] restAPIUserConfigProvider() {
     public void init() throws Exception {
 
         super.testInit(API_VERSION, swaggerDefinition, tenant);
+        resetResidentIDPCache();
         restClient = new OAuth2RestClient(serverURL, context.getContextTenant());
 
         // Create test application.

modules/integration/tests-integration/tests-backend/src/test/java/org/wso2/identity/integration/test/rest/api/server/session/SessionMgtTestBase.java

@@ -21,6 +21,7 @@
 import io.restassured.RestAssured;
 import io.restassured.response.Response;
 import org.apache.commons.lang.StringUtils;
+import org.apache.commons.lang.ArrayUtils;
 import org.apache.http.HttpHeaders;
 import org.apache.http.HttpStatus;
 import org.apache.http.NameValuePair;
@@ -44,6 +45,9 @@
 import org.testng.annotations.AfterClass;
 import org.testng.annotations.AfterMethod;
 import org.testng.annotations.BeforeMethod;
+import org.wso2.carbon.identity.application.common.model.idp.xsd.FederatedAuthenticatorConfig;
+import org.wso2.carbon.identity.application.common.model.idp.xsd.IdentityProvider;
+import org.wso2.identity.integration.common.clients.Idp.IdentityProviderMgtServiceClient;
 import org.wso2.identity.integration.test.rest.api.server.common.RESTAPIServerTestBase;
 
 import java.io.IOException;
@@ -408,4 +412,23 @@ private String removeTenantDomain(String username) {
         }
         return username;
     }
+
+    /**
+     * Resets the resident IDP cache to ensure that any changes to the IDP (e.g. enabling
+     * just the SAML SSO federated authenticator) are reflected in the test execution.
+     */
+    protected void resetResidentIDPCache() throws Exception {
+
+        IdentityProviderMgtServiceClient idpMgtClient = new IdentityProviderMgtServiceClient(sessionCookie, backendURL);
+        IdentityProvider residentIdp = idpMgtClient.getResidentIdP();
+
+        FederatedAuthenticatorConfig[] configs = residentIdp.getFederatedAuthenticatorConfigs();
+        for (FederatedAuthenticatorConfig cfg : configs) {
+            if (!cfg.getName().equalsIgnoreCase("samlsso")) {
+                configs = (FederatedAuthenticatorConfig[]) ArrayUtils.removeElement(configs, cfg);
+            }
+        }
+        residentIdp.setFederatedAuthenticatorConfigs(configs);
+        idpMgtClient.updateResidentIdP(residentIdp);
+    }
 }