package org.wso2.am.integration.tests.other;

import org.json.JSONObject;
import org.json.simple.parser.JSONParser;
import org.testng.annotations.AfterClass;
import org.testng.annotations.BeforeClass;
import org.testng.annotations.Test;
import org.wso2.am.admin.clients.registry.ResourceAdminServiceClient;
import org.wso2.am.integration.test.utils.APIManagerIntegrationTestException;
import org.wso2.am.integration.test.utils.bean.*;
import org.wso2.am.integration.test.utils.clients.APIPublisherRestClient;
import org.wso2.am.integration.test.utils.clients.APIStoreRestClient;
import org.wso2.am.integration.tests.api.lifecycle.APIManagerLifecycleBaseTest;
import org.wso2.carbon.apimgt.api.WorkflowResponse;
import org.wso2.carbon.apimgt.api.model.APIIdentifier;
import org.wso2.carbon.automation.engine.annotations.ExecutionEnvironment;
import org.wso2.carbon.automation.engine.annotations.SetEnvironment;
import org.wso2.carbon.automation.test.utils.http.client.HttpResponse;
import org.wso2.carbon.integration.common.utils.exceptions.AutomationUtilException;
import org.wso2.carbon.integration.common.utils.mgt.ServerConfigurationManager;
import org.wso2.carbon.registry.resource.stub.ResourceAdminServiceExceptionException;

import javax.xml.xpath.XPathExpressionException;
import java.io.File;
import java.io.IOException;
import java.net.URISyntaxException;
import java.net.URL;

import static org.testng.Assert.assertEquals;
import static org.testng.Assert.assertFalse;

/**
 * The test class that tests the custom Subscription creation workflow which has a HTTP redirect.
 */
@SetEnvironment(executionEnvironments = {ExecutionEnvironment.STANDALONE})
public class SubscriptionWFHTTPRedirectTest  extends APIManagerLifecycleBaseTest{


    private APIPublisherRestClient apiPublisher;
    private String publisherURLHttp;
    private String storeURLHttp;
    private String userName;
    private String originalWFExtentionsXML;
    private String newWFExtentionsXML;
    private ResourceAdminServiceClient resourceAdminServiceClient;
    private APIIdentifier apiIdentifier;
    private final String DEFAULT_WF_EXTENTIONS_XML_REG_CONFIG_LOCATION = "/_system/governance/apimgt/applicationdata/workflow-extensions.xml";
    private String appName = "sample-application-workflow2";
    private static JSONParser parser = new JSONParser();

    @BeforeClass(alwaysRun = true)
    public void setEnvironment() throws AutomationUtilException, XPathExpressionException, IOException,
            APIManagerIntegrationTestException, URISyntaxException, ResourceAdminServiceExceptionException {
        super.init();

        ServerConfigurationManager serverConfigurationManager;
        serverConfigurationManager = new ServerConfigurationManager(gatewayContextMgt);
        //Copies the custom workflow jar to the lib.
        serverConfigurationManager.copyToComponentLib(new File(getAMResourceLocation()
                + File.separator + "configFiles" + File.separator + "workflowHTTPRedirect" + File.separator +
                "SubscriptionCreationCustomWorkflowExecutor-1.0.0.jar"));
        serverConfigurationManager.restartGracefully();

        publisherURLHttp = getPublisherURLHttp();
        storeURLHttp = getStoreURLHttp();
        userName = user.getUserName();
        resourceAdminServiceClient =
                new ResourceAdminServiceClient(gatewayContextMgt.getContextUrls().getBackEndUrl(),
                        createSession(gatewayContextMgt));
        //Gets the original workflow-extentions.xml file's content from the registry.
        originalWFExtentionsXML = resourceAdminServiceClient.getTextContent(DEFAULT_WF_EXTENTIONS_XML_REG_CONFIG_LOCATION);
        //Gets the new configuration of the workflow-extentions.xml
        newWFExtentionsXML = readFile(getAMResourceLocation()
                + File.separator + "configFiles" + File.separator + "workflowHTTPRedirect" + File.separator + "workflow-extentions.xml");
    }


    @Test(groups = "wso2.am", description = "Test if the custom subscription workflow response returns the redirect URI")
    public void testHTTPredirectInSubscriptionWorkflow() throws Exception {

        APIStoreRestClient apiStore;
        //Updates the content of the workflow-extentions.xml of the registry file, to have the new configurations.
        resourceAdminServiceClient.updateTextContent(DEFAULT_WF_EXTENTIONS_XML_REG_CONFIG_LOCATION, newWFExtentionsXML);
        String apiName = "HTTPRedirectTestAPI2";
        //Just a dummy URL
        String url = "http://localhost:9443/carbon";
        String description = "This is test API create by API manager integration test";
        String apiVersion = "1.0.0";

        apiPublisher = new APIPublisherRestClient(publisherURLHttp);
        apiStore = new APIStoreRestClient(storeURLHttp);

        apiPublisher.login(userName, userName);

        APIRequest apiRequest = new APIRequest(apiName, apiVersion, new URL(url));
        apiRequest.setDescription(description);
        apiRequest.setVersion(apiVersion);
        apiRequest.setSandbox(url);
        apiRequest.setProvider(userName);
        apiPublisher.addAPI(apiRequest);

        apiIdentifier = new APIIdentifier(userName, apiName, apiVersion);
        APILifeCycleStateRequest updateRequest = new APILifeCycleStateRequest(apiName, userName,
                APILifeCycleState.PUBLISHED);
        apiPublisher.changeAPILifeCycleStatus(updateRequest);

        apiStore.login(userName, user.getPassword());

        apiStore.addApplication(appName, "Gold", "", "this-is-test");

        SubscriptionRequest subscriptionRequest = new SubscriptionRequest(apiName, userName);
        subscriptionRequest.setApplicationName(appName);
       //Subscribed to the API.
        HttpResponse httpresponse= apiStore.subscribe(subscriptionRequest);
        String httpresponseData = httpresponse.getData();
        JSONObject jsondata = new JSONObject(httpresponseData);
        JSONObject status = (JSONObject)jsondata.get("status");
        if(status != null){
            JSONObject workflowResponse = null;
            org.json.simple.JSONObject jsonpayloadObject = null;
            workflowResponse = (JSONObject)status.get("workflowResponse");
            if(workflowResponse != null) {
                jsonpayloadObject = (org.json.simple.JSONObject) parser.parse((String) (workflowResponse.get("jsonPayload")));
                if(jsonpayloadObject != null) {
                    assertEquals(jsonpayloadObject.get("redirectUrl").toString(), "http://google.lk");
                }
            }
        }else{
            assertFalse(true);
        }
    }

    @AfterClass(alwaysRun = true)
    public void cleanUpArtifacts() throws Exception {
        super.cleanUp();
        //restore the original workflow-extentions.xml content.
        resourceAdminServiceClient.updateTextContent(DEFAULT_WF_EXTENTIONS_XML_REG_CONFIG_LOCATION, originalWFExtentionsXML);
    }

}

@@ -370,32 +370,105 @@
 nTestException;%0A
+import org.wso2.am.integration.test.utils.base.APIMIntegrationConstants;%0A
 import org.wso2.
@@ -1339,32 +1339,33 @@
 ssionException;%0A
+%0A
 import java.io.F
@@ -5605,14 +5605,55 @@
 me, 
-%22Gold%22
+APIMIntegrationConstants.APPLICATION_TIER.LARGE
 , %22%22

modify tier in the test case