pieShareAppITs.helper.ITUtil.java Source code

Introduction

Here is the source code for pieShareAppITs.helper.ITUtil.java

Source

/*
 * To change this license header, choose License Headers in Project Properties.
 * To change this template file, choose Tools | Templates
 * and open the template in the editor.
 */
package pieShareAppITs.helper;

import java.io.BufferedReader;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import org.apache.commons.io.FileUtils;
import org.pieShare.pieShareApp.model.command.LoginCommand;
import org.pieShare.pieShareApp.service.PieShareService;
import org.pieShare.pieShareAppFx.springConfiguration.PiePlateConfiguration;
import org.pieShare.pieShareAppFx.springConfiguration.PieShareApp.PieShareAppTasks;
import org.pieShare.pieShareApp.task.commandTasks.loginTask.LoginTask;
import org.pieShare.pieShareApp.task.commandTasks.loginTask.api.ILoginFinished;
import org.pieShare.pieShareApp.task.commandTasks.loginTask.exceptions.WrongPasswordException;
import org.pieShare.pieShareAppFx.springConfiguration.ProviderConfiguration;
import org.pieShare.pieTools.pieUtilities.model.PlainTextPassword;
import org.pieShare.pieTools.pieUtilities.service.pieLogger.PieLogger;
import org.springframework.context.annotation.AnnotationConfigApplicationContext;
import org.testng.Assert;
import pieShareAppITs.helper.config.PieShareAppModelITConfig;
import pieShareAppITs.helper.config.PieShareAppServiceConfig;
import pieShareAppITs.helper.config.PieShareUtilitiesITConfig;

/**
 *
 * @author Svetoslav
 */
public class ITUtil {

    public static String getMainWorkingDir() {
        return "workingDirTestMain";
    }

    public static String getMainTmpDir() {
        return "pieTempTestMain";
    }

    public static String getBotWorkingDir() {
        return "workingDirTestBot";
    }

    public static String getBotTmpDir() {
        return "pieTempTestBot";
    }

    public static String getMainKey() {
        return "testMainKey";
    }

    public static String getBotKey() {
        return "testBotKey";
    }

    public static String getMainDbDir() {
        return "mainDb";
    }

    public static String getBotDbDir() {
        return "botDb";
    }

    public static void setUpEnviroment(boolean main) {
        System.setProperty("java.net.preferIPv4Stack", "true");
        System.setProperty("jgroups.logging.log_factory_class",
                "org.pieShare.pieTools.piePlate.service.cluster.jgroupsCluster.JGroupsLoggerFactory");
        PieShareAppServiceConfig.main = main;
        PieLogger.info(ITUtil.class, "MainInstance: {}", main);
    }

    public static void performTearDownDelete() throws Exception {
        FileUtils.deleteDirectory(new File(getMainWorkingDir()));
        FileUtils.deleteDirectory(new File(getMainTmpDir()));
        FileUtils.deleteDirectory(new File(getBotWorkingDir()));
        FileUtils.deleteDirectory(new File(getBotTmpDir()));
        FileUtils.deleteDirectory(new File(getMainDbDir()));
        FileUtils.deleteDirectory(new File(getBotDbDir()));
        (new File(getMainKey())).delete();
        (new File(getBotKey())).delete();
    }

    public static void performTearDown(AnnotationConfigApplicationContext context) throws Exception {
        //shutdown application
        PieShareService service = context.getBean(PieShareService.class);
        context.close();
        service.stop();

        //get dirs to delete
        /*PieShareConfiguration config = context.getBean("pieUser", PieUser.class).getPieShareConfiguration();
        File mainWorkingDir = config.getWorkingDir();//config.getWorkingDirectory();
        File mainTmpDir = config.getTmpDir();
        File configMain = config.getPwdFile();
        config = context.getBean("botPieUser", PieUser.class).getPieShareConfiguration();
        File botWorkingDir = config.getWorkingDir();
        File botTmpDir = config.getTmpDir();
        File configBot = config.getPwdFile();*/

        //stop context
        //      context.close();
        context = null;
    }

    public static Process startProcess(Class mainClazz, String cloudName, String password) throws IOException {
        String separator = System.getProperty("file.separator");
        String classpath = System.getProperty("java.class.path");
        String path = System.getProperty("java.home") + separator + "bin" + separator + "java";
        ProcessBuilder processBuilder = new ProcessBuilder(path, "-Xdebug",
                "-Xrunjdwp:transport=dt_socket,server=y,suspend=n,address=5005", "-cp", classpath,
                mainClazz.getName(), cloudName, password);
        return processBuilder.start();
    }

    public static void waitForProcessToStartup(Process process) throws Exception {
        InputStream stream = process.getInputStream();
        InputStreamReader in = new InputStreamReader(stream);
        BufferedReader reader = new BufferedReader(in);
        String line = reader.readLine();
        while (!line.equals("!loggedIn")) {
            line = reader.readLine();
        }
    }

    public static boolean waitForFileToBeFreed(File file, int sec) {
        boolean done = false;
        int time = 0;

        //todo: this has to move to utils: this is a check if the access to the file has been restored
        //after torrent work
        while (!done || time >= sec) {
            try {
                Thread.sleep(1000);
                FileInputStream st = new FileInputStream(file);
                done = true;
                st.close();
                return true;
            } catch (FileNotFoundException ex) {
                //nothing needed to do here
            } catch (IOException ex) {
                //nothing needed to do here
            } catch (InterruptedException ex) {
                //nothing needed to do here
            }
        }

        return false;
    }

    public static void executeLoginToTestCloud(AnnotationConfigApplicationContext context, String cloudName,
            String password) throws Exception {
        LoginCommand command = new LoginCommand();
        PlainTextPassword pwd = new PlainTextPassword();
        pwd.password = password.getBytes();
        command.setPlainTextPassword(pwd);
        command.setUserName(cloudName);

        command.setCallback(new ILoginFinished() {

            @Override
            public void error(Exception ex) {
                ex.printStackTrace();
                Assert.fail(ex.getLocalizedMessage());
            }

            @Override
            public void wrongPassword(WrongPasswordException ex) {
                Assert.fail(ex.getMessage());
            }

            @Override
            public void OK() {
            }
        });

        LoginTask task = context.getBean(LoginTask.class);
        task.setEvent(command);
        task.run();
    }

    public static AnnotationConfigApplicationContext getContext() {
        AnnotationConfigApplicationContext context = new AnnotationConfigApplicationContext();
        context.register(PieShareUtilitiesITConfig.class);
        context.register(PiePlateConfiguration.class);
        context.register(PieShareAppModelITConfig.class);
        context.register(PieShareAppServiceConfig.class);
        context.register(PieShareAppTasks.class);
        context.register(ProviderConfiguration.class);
        context.refresh();
        return context;
    }

    public static AnnotationConfigApplicationContext getContextWithSpecialPiePlateContext(Class piePlateContext) {
        AnnotationConfigApplicationContext context = new AnnotationConfigApplicationContext();
        context.register(PieShareUtilitiesITConfig.class);
        context.register(piePlateContext);
        context.register(PieShareAppModelITConfig.class);
        context.register(PieShareAppServiceConfig.class);
        context.register(PieShareAppTasks.class);
        context.register(ProviderConfiguration.class);
        context.refresh();
        return context;
    }
}