Merge pull request #17605 from Yasasr1/update-token-exchange-tests

Update token exchange grant integration tests
wso2 · Nov 27, 2023 · a8d889e · a8d889e
2 parents 0357c8f + 51543b1
commit a8d889e
Show file tree

Hide file tree

Showing 3 changed files with 311 additions and 12 deletions.
diff --git a/...rg/wso2/identity/integration/test/application/mgt/AbstractIdentityFederationTestCase.java b/...rg/wso2/identity/integration/test/application/mgt/AbstractIdentityFederationTestCase.java
@@ -40,7 +40,9 @@
 import org.wso2.carbon.identity.sso.saml.stub.types.SAMLSSOServiceProviderDTO;
 import org.wso2.carbon.identity.sso.saml.stub.types.SAMLSSOServiceProviderInfoDTO;
 import org.wso2.carbon.integration.common.admin.client.AuthenticatorClient;
+import org.wso2.carbon.user.mgt.stub.types.carbon.ClaimValue;
 import org.wso2.identity.integration.common.clients.Idp.IdentityProviderMgtServiceClient;
+import org.wso2.identity.integration.common.clients.UserManagementClient;
 import org.wso2.identity.integration.common.clients.application.mgt.ApplicationManagementServiceClient;
 import org.wso2.identity.integration.common.clients.oauth.OauthAdminClient;
 import org.wso2.identity.integration.common.clients.sso.saml.SAMLSSOConfigServiceClient;
@@ -62,6 +64,7 @@
 import java.io.IOException;
 import java.io.InputStreamReader;
 import java.util.HashMap;
+import java.util.HashSet;
 import java.util.Iterator;
 import java.util.Map;
 
@@ -73,6 +76,7 @@ public abstract class AbstractIdentityFederationTestCase extends ISIntegrationTe
  private Map<Integer, OauthAdminClient> oauthAdminClients;
  private Map<Integer, OAuth2RestClient> applicationManagementRestClients;
  private Map<Integer, IdpMgtRestClient> identityProviderMgtRestClients;
+ private Map<Integer, UserManagementClient> userManagementClients;
  protected Map<Integer, AutomationContext> automationContextMap;
  private MultipleServersManager manager;
  protected static final int DEFAULT_PORT = CommonConstants.IS_DEFAULT_HTTPS_PORT;
@@ -87,6 +91,7 @@ public void initTest() throws Exception {
  identityProviderMgtServiceClients = new HashMap<>();
  samlSSOConfigServiceClients = new HashMap<>();
  oauthAdminClients = new HashMap<>();
+ userManagementClients = new HashMap<>();
  automationContextMap = testDataHolder.getAutomationContextMap();
  manager = testDataHolder.getManager();
 
@@ -144,6 +149,8 @@ public void createServiceClients(int portOffset, String sessionCookie,
  samlSSOConfigServiceClients.put(portOffset, new SAMLSSOConfigServiceClient(serviceUrl, sessionCookie));
  } else if (IdentityConstants.ServiceClientType.OAUTH_ADMIN.equals(clientType)) {
  oauthAdminClients.put(portOffset, new OauthAdminClient(serviceUrl, sessionCookie));
+ } else if (IdentityConstants.ServiceClientType.USER_MGT.equals(clientType)) {
+ userManagementClients.put(portOffset, new UserManagementClient(serviceUrl, sessionCookie));
  }
  }
  }
@@ -169,6 +176,22 @@ public void createServiceClients(int portOffset, IdentityConstants.ServiceClient
  }
  }
 
+ public void addUser(int portOffset, String username, String password, String[] roles,
+ String profileName, ClaimValue[] claims) throws Exception {
+
+ userManagementClients.get(portOffset).addUser(username, password, roles, profileName, claims);
+ }
+
+ public void deleteUser(int portOffset, String username) throws Exception {
+
+ userManagementClients.get(portOffset).deleteUser(username);
+ }
+
+ public HashSet<String> getUserList(int portOffset) throws Exception {
+
+ return userManagementClients.get(portOffset).getUserList();
+ }
+
  public void addServiceProvider(int portOffset, String applicationName) throws Exception {
 
  ServiceProvider serviceProvider = new ServiceProvider();