org.wso2.appserver.integration.tests.carbontools.CleanRegistryCommandTestCase.java Source code

Introduction

Here is the source code for org.wso2.appserver.integration.tests.carbontools.CleanRegistryCommandTestCase.java

Source

/*
 * Copyright (c) 2015, WSO2 Inc. (http://www.wso2.org) All Rights Reserved.
 *
 * WSO2 Inc. 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.appserver.integration.tests.carbontools;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.testng.annotations.AfterClass;
import org.testng.annotations.BeforeClass;
import org.testng.annotations.Test;
import org.wso2.appserver.integration.common.clients.ResourceAdminServiceClient;
import org.wso2.appserver.integration.common.utils.ASIntegrationConstants;
import org.wso2.appserver.integration.common.utils.ASIntegrationTest;
import org.wso2.appserver.integration.common.utils.CarbonCommandToolsUtil;
import org.wso2.appserver.integration.tests.carbontools.test.servers.CarbonTestServerManager;
import org.wso2.carbon.authenticator.stub.LoginAuthenticationExceptionException;
import org.wso2.carbon.automation.engine.context.AutomationContext;
import org.wso2.carbon.automation.engine.context.ContextXpathConstants;
import org.wso2.carbon.automation.engine.exceptions.AutomationFrameworkException;
import org.wso2.carbon.automation.engine.frameworkutils.FrameworkPathUtil;
import org.wso2.carbon.integration.common.utils.LoginLogoutClient;
import org.wso2.carbon.integration.common.utils.exceptions.AutomationUtilException;
import org.xml.sax.SAXException;

import javax.activation.DataHandler;
import javax.xml.stream.XMLStreamException;
import javax.xml.xpath.XPathExpressionException;
import java.io.File;
import java.io.IOException;
import java.net.URISyntaxException;
import java.net.URL;
import java.util.HashMap;

import static org.testng.Assert.assertFalse;

/**
 * This class to test --cleanRegistry by adding a resource and clean it by this command
 */
public class CleanRegistryCommandTestCase extends ASIntegrationTest {

    private ResourceAdminServiceClient resourceAdminServiceClient;
    private static final String configRegistryRepoPath = "/_system/config/repository/";

    private static final Log log = LogFactory.getLog(CleanRegistryCommandTestCase.class);
    private HashMap<String, String> serverPropertyMap = new HashMap<String, String>();
    private String carbonHome;
    private AutomationContext context;
    private String sessionCookieForInstance002;
    private String backendURLForInstance002;
    private int portOffset = 1;
    private Process process;

    @BeforeClass(alwaysRun = true)
    public void init() throws Exception {
        context = new AutomationContext(ASIntegrationConstants.AS_PRODUCT_GROUP,
                ASIntegrationConstants.AS_INSTANCE_0002, ContextXpathConstants.SUPER_TENANT,
                ContextXpathConstants.ADMIN);

        String offsetString = context.getInstance().getProperty("portOffset");
        portOffset = Integer.parseInt(offsetString);
        serverPropertyMap.put("-DportOffset", offsetString);

        if (CarbonTestServerManager.isServerRunning()) {
            carbonHome = CarbonTestServerManager.getCarbonHome();

        } else {
            CarbonTestServerManager.start(portOffset);
            carbonHome = CarbonTestServerManager.getCarbonHome();

        }

        initEnvironment();

        resourceAdminServiceClient = new ResourceAdminServiceClient(backendURLForInstance002,
                sessionCookieForInstance002);

    }

    @Test(groups = "wso2.as", description = "Add resource and test --cleanRegistry startup argument")
    public void testCleanResource() throws Exception {
        boolean isResourceFound;

        String resourcePath = FrameworkPathUtil.getSystemResourceLocation() + "artifacts" + File.separator + "AS"
                + File.separator + "carbontools" + File.separator + "resource.txt";

        DataHandler dh = new DataHandler(new URL("file:///" + resourcePath));
        resourceAdminServiceClient.addResource(configRegistryRepoPath + "resource.txt", "txt", "testDesc", dh);
        isResourceFound = true;
        CarbonTestServerManager.stop();
        try {
            // start with -Dsetup command
            serverPropertyMap.put("--cleanRegistry", "");

            CarbonTestServerManager.start(serverPropertyMap);

            boolean startupStatus = CarbonCommandToolsUtil.isServerStartedUp(context, portOffset);
            log.info("Server startup status : " + startupStatus);
            initEnvironment();
            resourceAdminServiceClient = new ResourceAdminServiceClient(backendURLForInstance002,
                    sessionCookieForInstance002);
            resourceAdminServiceClient.getResource(configRegistryRepoPath + "resource.txt");
        } catch (Exception ex) {
            if (ex.getMessage().contains("Resource does not exist")) {
                isResourceFound = false;
            }
        } finally {
            if (process != null) {
                process.destroy();
            }
        }
        assertFalse(isResourceFound, "Resource not deleted successfully");
    }

    private void initEnvironment() throws IOException, XPathExpressionException, URISyntaxException, SAXException,
            XMLStreamException, LoginAuthenticationExceptionException, AutomationUtilException {
        LoginLogoutClient loginLogoutClientForInstance002 = new LoginLogoutClient(context);
        sessionCookieForInstance002 = loginLogoutClientForInstance002.login();
        backendURLForInstance002 = context.getContextUrls().getBackEndUrl();
    }

    @AfterClass(alwaysRun = true)
    public void serverShutDown() throws AutomationFrameworkException {
        CarbonTestServerManager.stop();
    }

}