arjuna.lib.state.ArjunaSingleton Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of arjuna-java Show documentation
Arjuna-Java is the client implementation in Java for development of test automation using Arjuna. It uses TestNG as the test engine. With minor tweaks, it can be used with any other test engine or custom test automation implementations. Arjuna is a Python based test automation framework developed by Rahul Verma (www.rahulverma.net)
The newest version!
package arjuna.lib.state;

import java.io.File;
import java.net.SocketException;
import java.util.HashMap;
import java.util.Map;

import org.apache.log4j.ConsoleAppender;
import org.apache.log4j.FileAppender;
import org.apache.log4j.Level;
import org.apache.log4j.Logger;
import org.apache.log4j.PatternLayout;
import org.testng.ITestContext;

import arjuna.client.core.action.SetuActionType;
import arjuna.client.core.connector.SetuService;
import arjuna.client.guiauto.automator.DefaultGuiAutomator;
import arjuna.client.testsession.DefaultTestSession;
import arjuna.client.testsession.TestSession;
import arjuna.lib.config.CliArgsConfig;
import arjuna.lib.config.DefaultTestContext;
import arjuna.lib.console.Console;
import arjuna.lib.sysauto.process.CommandExecutor;
import arjuna.lib.sysauto.process.ProcessOutput;
import arjuna.lib.testng.TestNGSuiteContext;
import arjuna.lib.testng.TestNGTestContext;
import arjuna.tpi.ddauto.DataSourceBuilder;
import arjuna.tpi.enums.ArjunaOption;
import arjuna.tpi.guiauto.GuiAutomator;
import arjuna.tpi.guiauto.GuiDriverExtendedConfig;
import arjuna.tpi.test.TestConfig;
import arjuna.tpi.test.TestContext;

public enum ArjunaSingleton {
	INSTANCE;
	private final String DEFAULT_CONTEXT_NAME = "default_context";
	private String rootDir = null;
	
	private TestSession session;
	private TestConfig centralConfig;
	private CliArgsConfig cliConfig = null;	
	
	private Logger logger = null;

	private Map testContextMap = new HashMap();
	private SetuService setuService;

	public TestContext init(String rootDir) throws Exception {
		this.setuService = new SetuService();
		this.setuService.launch();
		
		this.rootDir = rootDir;
		cliConfig = new CliArgsConfig();
		
		// Finalize logger
		createLogger("arjuna", rootDir + File.separator + "log" + File.separator + "arjuna-java.log");
		logger = Logger.getLogger("arjuna");
		Console.init();
		
		session = new DefaultTestSession();
		centralConfig = session.init(rootDir);
		
		TestContext context = this.createTestContext(DEFAULT_CONTEXT_NAME);
		this.testContextMap.put(DEFAULT_CONTEXT_NAME, context);
		return context;
	}
	
	public CliArgsConfig getCliArgsConfig() {
		return this.cliConfig;
	}
	
	public TestConfig getCentralConfig() throws Exception {
		return centralConfig;
	}
	
	public void registerTestContext(TestContext context) throws Exception {
		this.testContextMap.put(context.getName().toLowerCase(), context);
	}
	
	public TestContext getTestContext(String name) throws Exception {
		if (name == null) {
			throw new Exception("Context name was passed as null.");
		} else {
			try {
				return this.testContextMap.get(name.toLowerCase());
			} catch (Exception e) {
				throw new Exception("No context found with name: " + name);
			}
		}
	}
	
	public TestContext createTestContext(String name) throws Exception {
		return new DefaultTestContext(session, name);
	}
	
	public TestContext createTestNGSuiteContext(TestContext parentContext, ITestContext testngContext) throws Exception {
		TestContext context = new TestNGSuiteContext(session, parentContext, testngContext);
		registerTestContext(context);
		return context;
	}
	
	public TestContext createTestNGTestContext(TestContext parentContext, ITestContext testngContext) throws Exception {
		TestContext context =  new TestNGTestContext(session, parentContext, testngContext);
		registerTestContext(context);
		return context;
	}
	
	public String getRootDir() {
		return this.rootDir;
	}
	
	private CliArgsConfig getCliConfig() {
		return this.cliConfig;
	}
	
	public String normalizeUserOption(String option) {
		return option.trim().toUpperCase().replace(".", "_");
	}
	
	public ArjunaOption normalizeArjunaOption(String option) {
		return ArjunaOption.valueOf(normalizeUserOption(option));
	}
	
	public DataSourceBuilder createDataSourceBuilder() throws Exception {
		return new DataSourceBuilder(session);
	}
	
	public GuiAutomator createGuiAutomator() throws Exception {
		return new DefaultGuiAutomator(this.centralConfig);
	}
	
	public GuiAutomator createGuiAutomator(TestConfig config) throws Exception {
		return new DefaultGuiAutomator(config);
	}
	
	public GuiAutomator createGuiAutomator(TestConfig config, GuiDriverExtendedConfig extendedConfig) throws Exception {
		return new DefaultGuiAutomator(config, extendedConfig);
	}
	
	public static void createLogger(String loggerName, String logFilePath) {
		Logger.getLogger(loggerName).getLoggerRepository().resetConfiguration();
		
		ConsoleAppender console = new ConsoleAppender(); // create appender
		FileAppender fa = new FileAppender();

		String PATTERN = null;
		PATTERN = "(%F:%L)\t%m%n";
		console.setLayout(new PatternLayout(PATTERN));
		console.setThreshold(Level.INFO);
		console.activateOptions();
		Logger.getLogger(loggerName).addAppender(console);

		fa.setName("FileLogger");
		fa.setFile(logFilePath);
		fa.setLayout(new PatternLayout("[%-5p]\t%d{yyyy-MM-dd|HH:mm:ss}\t(%F:%L)\t%m%n"));
		fa.setThreshold(Level.DEBUG);
		fa.setAppend(false);
		fa.activateOptions();
		Logger.getLogger(loggerName).addAppender(fa);
	}

	public Logger getLogger() {
//		if (logger == null) {
//			createLogger("trishanku", "trishanku.log");
//			logger = Logger.getLogger("trishanku");
//		}
		return logger;
	}

	public void exit() throws Exception {
		this.setuService.stop();
	}

}