selenium-remote-control-0.8.0.zip_Selenium web 测试_remote_seleniu

/* * Created on Mar 12, 2006 * */ package org.openqa.selenium.server.testgenerator; import java.io.*; import java.util.*; import javax.xml.parsers.*; import org.w3c.dom.*; /** * Given an HTML file containing a Selenese test case, generate equivalent Java code w/ calls * to the Selenium object to execute that same test case. * * @author nsproul * */ public class XlateHtmlSeleneseToJava { static Set generatedJavaClassNames = new HashSet(); private static Map funcTypes = null; static final String BEGIN_SELENESE = ">>>>>"; static final String END_SELENESE = "<<<<<"; static final String SELENESE_TOKEN_DIVIDER = "//////"; static HashMap declaredVariables = new HashMap(); private static int varNameSeed = 1; private static String EOL = "\n\t\t"; private static int timeOut = 30000; private static String domain; private static boolean silentMode = false; private static boolean dontThrowOnTranslationDifficulties = false; public static void main(String[] args) throws IOException { boolean generateSuite = false; if (args.length < 2) { Usage("too few args"); return; } HashMap skipList = new HashMap(); String javaSeleneseFileDirectoryName = args[0]; for (int j = 1; j < args.length; j++) { if (args[j].equals("-silent")) { silentMode = true; } else if (args[j].equals("-skip")) { skipList.put(args[++j], Boolean.TRUE); } else if (args[j].equals("-dontThrowOnTranslationDifficulties")) { dontThrowOnTranslationDifficulties = true; } else if (args[j].equals("-suite")) { generateSuite = true; } else if (args[j].equals("-dir")) { String dirName = args[++j]; File dir = new File(dirName); if (!dir.isDirectory()) { Usage("-dir is not a directory: " + dirName); } String children[] = dir.list(); for (int k = 0; k < children.length; k++) { String fileName = children[k]; if (!skipList.containsKey(fileName) && fileName.indexOf(".htm")!=-1 && fileName.indexOf("Suite")==-1) { generateJavaClassFromSeleneseHtml(dirName + "/" + fileName, javaSeleneseFileDirectoryName); } } } else { String htmlSeleneseFileName = args[j]; generateJavaClassFromSeleneseHtml(htmlSeleneseFileName, javaSeleneseFileDirectoryName); } } if (generateSuite) { generateSuite(javaSeleneseFileDirectoryName); } } private static void initializeFuncTypes() { if (funcTypes != null) return; funcTypes = new HashMap(); InputStream stream = XlateHtmlSeleneseToJava.class.getResourceAsStream("/core/iedoc.xml"); if (stream==null) { throw new RuntimeException("could not find /core/iedoc.xml on the class path"); } try { Document d = DocumentBuilderFactory.newInstance().newDocumentBuilder().parse(stream); NodeList functions = d.getElementsByTagName("function"); for (int i = 0; i < functions.getLength(); i++) { Element function = (Element) functions.item(i); String funcName = function.getAttribute("name"); NodeList returnElements = function.getElementsByTagName("return"); if (returnElements.getLength() == 0) { funcTypes.put(funcName, void.class); } else { Element ret = (Element) returnElements.item(0); String retType = ret.getAttribute("type"); if ("boolean".equals(retType)) { funcTypes.put(funcName, boolean.class); } else if ("string".equals(retType)) { funcTypes.put(funcName, String.class); } } } } catch (Exception e) { throw new RuntimeException(e); } } private static boolean isBoolean(String op) { return (boolean.class.equals(getOpType(op))); } private static Class getOpType(String op) { initializeFuncTypes(); if (funcTypes.get(op) != null) { return (Class) funcTypes.get(op); } String swappedOp = op.replaceFirst("store", "get"); if (funcTypes.get(swappedOp) != null) { return (Class) funcTypes.get(swappedOp); } swappedOp = op.replaceFirst("store", "is"); if (funcTypes.get(swappedOp) != null) { return (Class) funcTypes.get(swappedOp); } // if we get here, apparently op has no direct analog in Selenium. So just look at the name and guess: if (op.matches(".*(Present|Visible|Editable)$")) { return boolean.class; } return String.class; } private static void generateSuite(String javaSeleneseFileDirectoryName) throws IOException { if (generatedJavaClassNames.size()==1) { return; // this is a test run focusing on a single file, so a suite wouldn't be useful } String beginning = "package com.thoughtworks.selenium.corebased;\n" + "\n" + "import junit.framework.Test;\n" + "import junit.framework.TestSuite;\n" + "\n" + "public class SeleneseSuite {\n" + " static public Test suite() {\n" + " TestSuite suite = new TestSuite();\n"; String ending = " return suite;\n" + " }\n" + "}\n"; StringBuffer middle = new StringBuffer(); Iterator i = generatedJavaClassNames.iterator(); while (i.hasNext()) { String generatedJavaClassName = (String) i.next(); if (!generatedJavaClassName.equals("TestJavascriptParameters")){ middle.append(" suite.addTestSuite(") .append(generatedJavaClassName) .append(".class);\n"); } } WriteFileContents(beginning + middle + ending, openFile(javaSeleneseFileDirectoryName + "/SeleneseSuite.java")); } private static void generateJavaClassFromSeleneseHtml(String htmlSeleneseFileName, String javaSeleneseFileDirectoryName) throws IOException { String base = htmlSeleneseFileName; base = base.replaceAll(".*/", ""); base = base.replaceAll("\\.html?$", ""); String javaSeleneseFileName = javaSeleneseFileDirectoryName + "/" + base + ".java"; generatedJavaClassNames.add(base); System.out.println("Generating test class " + base + " from\t" + htmlSeleneseFileName + "..."); File htmlSeleneseFile = openFile(htmlSeleneseFileName); File javaSeleneseFile = openFile(javaSeleneseFileName); try { javaSeleneseFile.createNewFile(); } catch (IOException e) { Usage(e.toString()); } if (!htmlSeleneseFile.canRead()) { Usage("can't read " + htmlSeleneseFileName); } if (!javaSeleneseFile.canWrite()) { Usage("can't write " + javaSeleneseFileName); } String htmlSelenese = ReadFileContents(htmlSeleneseFile); String javaSelenese = XlateString(base, htmlSeleneseFileName, htmlSelenese); WriteFileContents(javaSelenese, javaSeleneseFile); }