Modify integration tests adding the newly introduced 'policyType' fie…

…ld in OperationPolicyDTO
wso2 · Feb 4, 2025 · 14a33f9 · 14a33f9
1 parent 85176a0
commit 14a33f9
Showing 1 changed file with 10 additions and 7 deletions.
diff --git a/...nd/src/test/java/org/wso2/am/integration/tests/apiproduct/APIProductCreationTestCase.java b/...nd/src/test/java/org/wso2/am/integration/tests/apiproduct/APIProductCreationTestCase.java
@@ -84,6 +84,7 @@ public class APIProductCreationTestCase extends APIManagerLifecycleBaseTest {
     private static final String RESTRICTED_SUBSCRIBER = "restricted_user";
     private static final String STANDARD_SUBSCRIBER = "standard_user";
     private static final String PASSWORD = "$3213#@sd";
+    private static final String POLICY_TYPE_COMMON = "common";
     private static final String RESTRICTED_ROLE = "restricted_role";
     private static final String SCOPE = "restricted_scope";
     private ApiTestHelper apiTestHelper;
@@ -469,8 +470,8 @@ public void testCreateAndInvokeApiProductWithOperationPoliciesInRequestApi() thr
         List<APIDTO> apisToBeUsed = new ArrayList<>();
         APIDTO api = apiTestHelper.createAnApi(getBackendEndServiceEndPointHttp("wildcard/resources"));
         APIOperationPoliciesDTO apiOperationPoliciesDTO = new APIOperationPoliciesDTO();
-        apiOperationPoliciesDTO.setRequest(getPolicyList("jsonToXML", null));
-        apiOperationPoliciesDTO.setFault(getPolicyList("jsonFault", null));
+        apiOperationPoliciesDTO.setRequest(getPolicyList("jsonToXML", POLICY_TYPE_COMMON, null));
+        apiOperationPoliciesDTO.setFault(getPolicyList("jsonFault", POLICY_TYPE_COMMON, null));
         for (APIOperationsDTO operationsDTO : api.getOperations()) {
             operationsDTO.setOperationPolicies(apiOperationPoliciesDTO);
         }
@@ -538,7 +539,7 @@ public void testCreateAndInvokeApiProductWithOperationPoliciesInRequestApi() thr
                 expectedResponse, headers);
         // Step 9 : Change In mediation Sequence in base API and verify change reflect in APIProduct.
         Assert.assertNotNull(api);
-        apiOperationPoliciesDTO.setRequest(getPolicyList("xmlToJson", null));
+        apiOperationPoliciesDTO.setRequest(getPolicyList("xmlToJson", POLICY_TYPE_COMMON, null));
         for (APIOperationsDTO operationsDTO : api.getOperations()) {
             operationsDTO.setOperationPolicies(apiOperationPoliciesDTO);
         }
@@ -562,8 +563,8 @@ public void testCreateAndInvokeApiProductWithOperationPoliciesInResponseApi() th
         List<APIDTO> apisToBeUsed = new ArrayList<>();
         APIDTO api = apiTestHelper.createAnApi(getBackendEndServiceEndPointHttp("wildcard/resources"));
         APIOperationPoliciesDTO apiOperationPoliciesDTO = new APIOperationPoliciesDTO();
-        apiOperationPoliciesDTO.setResponse(getPolicyList("xmlToJson", null));
-        apiOperationPoliciesDTO.setFault(getPolicyList("jsonFault", null));
+        apiOperationPoliciesDTO.setResponse(getPolicyList("xmlToJson", POLICY_TYPE_COMMON, null));
+        apiOperationPoliciesDTO.setFault(getPolicyList("jsonFault", POLICY_TYPE_COMMON, null));
         for (APIOperationsDTO operationsDTO : api.getOperations()) {
             operationsDTO.setOperationPolicies(apiOperationPoliciesDTO);
         }
@@ -633,7 +634,7 @@ public void testCreateAndInvokeApiProductWithOperationPoliciesInResponseApi() th
                 jsonResponseBody, headers);
         // Step 9 : Change In mediation Sequence in base API and verify change reflect in APIProduct.
         Assert.assertNotNull(api);
-        apiOperationPoliciesDTO.setRequest(getPolicyList("jsonToXML", null));
+        apiOperationPoliciesDTO.setRequest(getPolicyList("jsonToXML", POLICY_TYPE_COMMON, null));
         for (APIOperationsDTO operationsDTO : api.getOperations()) {
             operationsDTO.setOperationPolicies(apiOperationPoliciesDTO);
         }
@@ -849,11 +850,13 @@ private APIProductDTO publishAPIProduct(String uuid) throws ApiException, APIMan
         return restAPIPublisher.getApiProduct(uuid);
     }
 
-    public List<OperationPolicyDTO> getPolicyList(String policyName, Map<String, Object> attributeMap) {
+    public List<OperationPolicyDTO> getPolicyList(String policyName, String policyType, Map<String,
+            Object> attributeMap) {
 
         List<OperationPolicyDTO> policyList = new ArrayList<>();
         OperationPolicyDTO policyDTO = new OperationPolicyDTO();
         policyDTO.setPolicyName(policyName);
+        policyDTO.setPolicyType(policyType);
         policyDTO.setParameters(attributeMap);
         policyList.add(policyDTO);