Integration test for the fix unable to add API specific policy using YAML format via Publisher REST API

modules/integration/tests-integration/tests-backend/src/test/java/org/wso2/am/integration/tests/operationPolicy/OperationPolicyTestCase.java

@@ -613,6 +613,70 @@ public void testAPIInvocationAfterAddingNewMultipleOperationPolicies() throws Ex
  assertEquals(invokeAPIResponse.getHeaders("TestHeader")[0].getValue(), "TestValue");
  }
 
+ @Test(groups = {"wso2.am"}, description = "Add API specific operation policy using YAML Policy Definition",
+ dependsOnMethods = "testAPIInvocationAfterAddingNewMultipleOperationPolicies")
+ public void testAddAPISpecificOperationPolicyYAML() throws Exception {
+
+ HttpResponse addPolicyResponse =
+ addPolicy(apiId, "customAPISpecificLogPolicyForYAMLPolicyDefinitionTesting.yaml",
+ "customAPISpecificLogPolicy.j2");
+ assertNotNull(addPolicyResponse, "Error adding operation policy customAPISpecificLogPolicyYAML");
+ assertEquals(addPolicyResponse.getResponseCode(), 201, "Response code mismatched");
+
+ OperationPolicyDataDTO policyDTO =
+ new Gson().fromJson(addPolicyResponse.getData(), OperationPolicyDataDTO.class);
+ String newPolicyId = policyDTO.getId();
+ assertNotNull(newPolicyId, "Policy Id is null");
+
+ Map<String, String> apiSpecificPolicyMap = restAPIPublisher.getAllAPISpecificOperationPolicies(apiId);
+ Assert.assertNotNull(apiSpecificPolicyMap.get("customAPISpecificLogPolicyYAML"),
+ "Unable to find the newly added API specific policy");
+ policyMap.put("customAPISpecificLogPolicyYAML", newPolicyId);
+ }
+
+ @Test(groups = {"wso2.am"}, description = "Delete API specific operation policy created using YAML Policy Definition",
+ dependsOnMethods = "testAddAPISpecificOperationPolicyYAML")
+ public void testDeleteAPISpecificOperationPolicyYAML() throws Exception {
+
+ int responseCode = deleteOperationPolicy(policyMap.get("customAPISpecificLogPolicyYAML"), apiId);
+ assertEquals(responseCode, 200);
+ Map<String, String> updatedAPISpecificPolicyMap = restAPIPublisher.getAllAPISpecificOperationPolicies(apiId);
+ Assert.assertNull(updatedAPISpecificPolicyMap.get("customAPISpecificLogPolicyYAML"));
+ policyMap.remove("customAPISpecificLogPolicyYAML");
+ }
+
+ @Test(groups = {"wso2.am"}, description = "Add common operation policy using YAML specification file",
+ dependsOnMethods = "testDeleteAPISpecificOperationPolicyYAML")
+ public void testAddNewCommonOperationPolicyYAML() throws Exception {
+
+ HttpResponse addPolicyResponse = addPolicy(null,
+ "customCommonLogPolicyForYAMLPolicyDefinitionTesting.yaml", "customCommonLogPolicy.j2");
+
+ assertNotNull(addPolicyResponse, "Error adding operation policy customCommonLogPolicy");
+ assertEquals(addPolicyResponse.getResponseCode(), 201, "Response code mismatched");
+
+ OperationPolicyDataDTO policyDTO =
+ new Gson().fromJson(addPolicyResponse.getData(), OperationPolicyDataDTO.class);
+ String newPolicyId = policyDTO.getId();
+ assertNotNull(newPolicyId, "Policy Id is null");
+
+ Map<String, String> updatedCommonPolicyMap = restAPIPublisher.getAllCommonOperationPolicies();
+ Assert.assertNotNull(updatedCommonPolicyMap.get("customCommonLogPolicyYAML"),
+ "Unable to find the newly added common policy");
+ policyMap.put("customCommonLogPolicyYAML", newPolicyId);
+ }
+
+ @Test(groups = {"wso2.am"}, description = "Delete common operation policy created using YAML specification file",
+ dependsOnMethods = "testAddNewCommonOperationPolicyYAML")
+ public void testDeleteCommonOperationPolicyYAML() throws Exception {
+
+ int responseCode = deleteOperationPolicy(policyMap.get("customCommonLogPolicyYAML"), null);
+ assertEquals(responseCode, 200);
+ Map<String, String> updatedCommonPolicyMap = restAPIPublisher.getAllCommonOperationPolicies();
+ Assert.assertNull(updatedCommonPolicyMap.get("customCommonLogPolicyYAML"));
+ policyMap.remove("customCommonLogPolicyYAML");
+ }
+
  @AfterClass(alwaysRun = true)
  public void cleanUpArtifacts() throws Exception {
 

modules/integration/tests-integration/tests-backend/src/test/resources/artifacts/AM/operationPolicy/customAPISpecificLogPolicyForYAMLPolicyDefinitionTesting.yaml

@@ -0,0 +1,13 @@
+category: Mediation
+name: customAPISpecificLogPolicyYAML
+version: v1
+displayName: Custom Common Log Policy
+description: Using this policy, you can add a custom log message
+applicableFlows:
+ - request
+ - response
+ - fault
+supportedGateways:
+ - Synapse
+supportedApiTypes:
+ - HTTP

modules/integration/tests-integration/tests-backend/src/test/resources/artifacts/AM/operationPolicy/customCommonLogPolicyForYAMLPolicyDefinitionTesting.yaml

@@ -0,0 +1,13 @@
+category: Mediation
+name: customCommonLogPolicyYAML
+version: v1
+displayName: Custom Common Log Policy
+description: Using this policy, you can add a custom log message
+applicableFlows:
+ - request
+ - response
+ - fault
+supportedGateways:
+ - Synapse
+supportedApiTypes:
+ - HTTP