Update pre issue access token password grant test.

modules/integration/tests-integration/tests-backend/src/test/java/org/wso2/identity/integration/test/actions/PreIssueAccessTokenClientCredentialsGrantTestCase.java

@@ -43,7 +43,7 @@
 import org.testng.annotations.BeforeClass;
 import org.testng.annotations.Test;
 import org.wso2.carbon.automation.engine.context.TestUserMode;
-import org.wso2.identity.integration.test.mocks.MockServer;
+import org.wso2.identity.integration.test.actions.mockserver.ActionsMockServer;
 import org.wso2.identity.integration.test.rest.api.server.action.management.v1.model.ActionModel;
 import org.wso2.identity.integration.test.rest.api.server.action.management.v1.model.AuthenticationType;
 import org.wso2.identity.integration.test.rest.api.server.action.management.v1.model.Endpoint;
@@ -141,6 +141,7 @@ public class PreIssueAccessTokenClientCredentialsGrantTestCase extends ActionsBa
     private String userId;
     private String roleId;
     private JWTClaimsSet jwtClaims;
+    private ActionsMockServer actionsMockServer;
 
     /**
      * Initializes Test environment and sets up necessary configurations.
@@ -187,7 +188,9 @@ protected boolean isRedirectable(String method) {
 
         addUserWithRole(applicationId, customScopes);
 
-        MockServer.createMockServer(MOCK_SERVER_ENDPOINT);
+        actionsMockServer = new ActionsMockServer();
+
+        actionsMockServer.createMockServer(MOCK_SERVER_ENDPOINT);
         actionId = createPreIssueAccessTokenAction();
     }
 
@@ -199,7 +202,7 @@ public void atEnd() throws Exception {
         deleteApp(applicationId);
         deleteDomainAPI(domainAPIId);
         scim2RestClient.deleteUser(userId);
-        MockServer.shutDownMockServer();
+        actionsMockServer.shutDownMockServer();
         restClient.closeHttpClient();
         scim2RestClient.closeHttpClient();
         actionsRestClient.closeHttpClient();

modules/integration/tests-integration/tests-backend/src/test/java/org/wso2/identity/integration/test/mocks/MockServer.java → modules/integration/tests-integration/tests-backend/src/test/java/org/wso2/identity/integration/test/actions/mockserver/ActionsMockServer.java

@@ -1,11 +1,37 @@
-package org.wso2.identity.integration.test.mocks;
+/*
+ * Copyright (c) 2024, WSO2 LLC. (http://www.wso2.com).
+ *
+ * WSO2 LLC. licenses this file to you under the Apache License,
+ * Version 2.0 (the "License"); you may not use this file except
+ * in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied.  See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ */
+
+package org.wso2.identity.integration.test.actions.mockserver;
 
 import com.github.tomakehurst.wiremock.WireMockServer;
+import com.github.tomakehurst.wiremock.client.WireMock;
+import com.github.tomakehurst.wiremock.core.WireMockConfiguration;
+import com.github.tomakehurst.wiremock.matching.RequestPatternBuilder;
+import com.github.tomakehurst.wiremock.verification.LoggedRequest;
+import org.apache.commons.lang.StringUtils;
+
+import java.util.List;
 
 import static com.github.tomakehurst.wiremock.client.WireMock.aResponse;
-import static com.github.tomakehurst.wiremock.client.WireMock.equalTo;
-import static com.github.tomakehurst.wiremock.client.WireMock.matchingJsonPath;
+import static com.github.tomakehurst.wiremock.client.WireMock.equalToJson;
+import static com.github.tomakehurst.wiremock.client.WireMock.matching;
 import static com.github.tomakehurst.wiremock.client.WireMock.post;
+import static com.github.tomakehurst.wiremock.client.WireMock.postRequestedFor;
 import static com.github.tomakehurst.wiremock.client.WireMock.urlEqualTo;
 import static com.github.tomakehurst.wiremock.core.WireMockConfiguration.wireMockConfig;
 
@@ -14,16 +40,49 @@
  * This class starts a mock server on a specified port and sets up predefined
  * responses for POST requests to simulate various operations relation to action execution.
  */
-public class MockServer {
+public class ActionsMockServer {
+
+    private WireMockServer wireMockServer;
+
+    public void startServer() {
 
-    private static WireMockServer wireMockServer;
+        wireMockServer = new WireMockServer(WireMockConfiguration.wireMockConfig().port(8587));
+        wireMockServer.start();
+    }
+
+    public void stopServer() {
+
+        if (wireMockServer != null && wireMockServer.isRunning()) {
+            wireMockServer.stop();
+        }
+    }
+
+    public void setupStub(String url, String authMethod, String responseBody) {
+
+        wireMockServer.stubFor(post(urlEqualTo(url))
+                .withHeader("Authorization", matching(authMethod))
+                .willReturn(aResponse()
+                        .withStatus(200)
+                        .withHeader("Content-Type", "application/json")
+                        .withBody(responseBody)));
+    }
+
+    public String getReceivedRequestPayload(String url) {
+
+        List<LoggedRequest> requestList = wireMockServer.findAll(postRequestedFor(urlEqualTo(url)));
+        if (requestList == null || requestList.isEmpty()) {
+            return StringUtils.EMPTY;
+        }
+
+        return requestList.get(0).getBodyAsString();
+    }
 
     /**
      * Create a mock server with wiremock.
      *
      * @throws Exception If an error occurred while creating the server
      */
-    public static void createMockServer(String mockEndpoint) throws Exception {
+    public void createMockServer(String mockEndpoint) throws Exception {
 
         wireMockServer = new WireMockServer(wireMockConfig().port(8587));
 
@@ -142,7 +201,7 @@ public static void createMockServer(String mockEndpoint) throws Exception {
     /**
      * Shut down the wiremock server instance.
      */
-    public static void shutDownMockServer() {
+    public void shutDownMockServer() {
 
         wireMockServer.stop();
     }

modules/integration/tests-integration/tests-backend/src/test/java/org/wso2/identity/integration/test/actions/model/AccessToken.java

@@ -0,0 +1,255 @@
+/*
+ * Copyright (c) 2024, WSO2 LLC. (http://www.wso2.com).
+ *
+ * WSO2 LLC. licenses this file to you under the Apache License,
+ * Version 2.0 (the "License"); you may not use this file except
+ * in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied.  See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ */
+
+package org.wso2.identity.integration.test.actions.model;
+
+import com.fasterxml.jackson.databind.annotation.JsonDeserialize;
+import com.fasterxml.jackson.databind.annotation.JsonPOJOBuilder;
+
+import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.List;
+import java.util.Objects;
+
+/**
+ * This class represents the model of the access token sent in the request payload
+ * to the API endpoint of the pre issue access token action.
+ */
+@JsonDeserialize(builder = AccessToken.Builder.class)
+public class AccessToken {
+
+    private final String tokenType;
+    List<String> scopes;
+    List<Claim> claims;
+
+    private AccessToken(Builder builder) {
+
+        this.tokenType = builder.tokenType;
+        this.scopes = builder.scopes;
+        this.claims = builder.claims;
+    }
+
+    public String getTokenType() {
+
+        return tokenType;
+    }
+
+    public List<String> getScopes() {
+
+        return scopes;
+    }
+
+    public List<Claim> getClaims() {
+
+        return claims;
+    }
+
+    public Claim getClaim(String name) {
+
+        if (claims != null) {
+            for (Claim claim : claims) {
+                if (claim.getName().equals(name)) {
+                    return claim;
+                }
+            }
+        }
+
+        return null;
+    }
+
+    @Override
+    public boolean equals(Object o) {
+
+        if (this == o) return true;
+        if (o == null || getClass() != o.getClass()) return false;
+        AccessToken that = (AccessToken) o;
+        return Objects.equals(tokenType, that.tokenType) && (scopes == null ? that.scopes == null :
+                scopes.size() == that.scopes.size() && scopes.containsAll(that.scopes)) &&
+                (claims == null ? that.claims == null :
+                        claims.size() == that.claims.size() && claims.containsAll(that.claims));
+    }
+
+    @Override
+    public int hashCode() {
+
+        return Objects.hash(tokenType, scopes, claims);
+    }
+
+    /**
+     * Enum for standard claim names.
+     */
+    public enum ClaimNames {
+
+        SUB("sub"),
+        ISS("iss"),
+        AUD("aud"),
+        CLIENT_ID("client_id"),
+        AUTHORIZED_USER_TYPE("aut"),
+        EXPIRES_IN("expires_in"),
+
+        TOKEN_BINDING_REF("binding_ref"),
+        TOKEN_BINDING_TYPE("binding_type"),
+        SUBJECT_TYPE("subject_type");
+
+        private final String name;
+
+        ClaimNames(String name) {
+
+            this.name = name;
+        }
+
+        public static boolean contains(String name) {
+
+            return Arrays.stream(ClaimNames.values())
+                    .anyMatch(claimName -> claimName.name.equals(name));
+        }
+
+        public String getName() {
+
+            return name;
+        }
+    }
+
+    /**
+     * Model class for claims.
+     */
+    public static class Claim {
+
+        private String name;
+        private Object value;
+
+        public Claim() {
+
+        }
+
+        public Claim(String name, Object value) {
+
+            this.name = name;
+            this.value = value;
+        }
+
+        public String getName() {
+
+            return name;
+        }
+
+        public void setName(String name) {
+
+            this.name = name;
+        }
+
+        public Object getValue() {
+
+            return value;
+        }
+
+        public void setValue(Object value) {
+
+            this.value = value;
+        }
+
+        @Override
+        public boolean equals(Object o) {
+
+            if (this == o) return true;
+            if (o == null || getClass() != o.getClass()) return false;
+            Claim claim = (Claim) o;
+            return Objects.equals(name, claim.name) && Objects.equals(value, claim.value);
+        }
+
+        @Override
+        public int hashCode() {
+
+            return Objects.hash(name, value);
+        }
+    }
+
+    /**
+     * Builder for AccessToken.
+     */
+    @JsonPOJOBuilder(withPrefix = "")
+    public static class Builder {
+
+        private String tokenType;
+        private List<String> scopes = new ArrayList<>();
+        private List<Claim> claims = new ArrayList<>();
+
+        public Builder tokenType(String tokenType) {
+
+            this.tokenType = tokenType;
+            return this;
+        }
+
+        public Builder scopes(List<String> scopes) {
+
+            this.scopes = scopes;
+            return this;
+        }
+
+        public Builder claims(List<Claim> claims) {
+
+            this.claims = claims;
+            return this;
+        }
+
+        public Builder addClaim(String name, Object value) {
+
+            this.claims.add(new Claim(name, value));
+            return this;
+        }
+
+        public Builder addScope(String scope) {
+
+            this.scopes.add(scope);
+            return this;
+        }
+
+        public String getTokenType() {
+
+            return tokenType;
+        }
+
+        public List<String> getScopes() {
+
+            return scopes;
+        }
+
+        public List<Claim> getClaims() {
+
+            return claims;
+        }
+
+        public Claim getClaim(String name) {
+
+            if (claims != null) {
+                for (Claim claim : claims) {
+                    if (claim.getName().equals(name)) {
+                        return claim;
+                    }
+                }
+            }
+
+            return null;
+        }
+
+        public AccessToken build() {
+
+            return new AccessToken(this);
+        }
+    }
+}