maxq0.98测试工具

package com.bitmechanic.maxq.generator; import java.util.regex.Pattern; import java.util.regex.Matcher; import com.bitmechanic.maxq.*; import java.util.HashMap; import java.util.LinkedList; import java.util.StringTokenizer; import java.util.List; import java.util.Iterator; import org.apache.commons.logging.Log; import org.apache.commons.logging.LogFactory; /** * Abstract class providing general useful utility methods for code generation. * @author Robert Dawson robert@rojotek.com * @author Wayne Hearn wayne_hearn@dell.com */ public abstract class AbstractCodeGenerator implements IScriptGenerator, ProxyServer.Observer, Runnable { private static Log log = LogFactory.getLog(AbstractCodeGenerator.class); private static String cpRspTo = System.getProperty("http.cpRspTo"); private static boolean cpRspToStdout; private static boolean cpRspToFile = false; private static final String MIME_DEFAULTS[] = { "text/plain", "text/html", "text/comma-separated-values" }; private static HashMap mimeTypes; private IScriptAdapter scriptAdapter; private String testName; private String defaultTestName; private String testPath; private boolean ignoreNextResponse; private Pattern[] namePatterns; private LinkedList outstandingInserts; private Thread insertThread; protected long assertNumber = 0; protected static final String END_STATEMENT = ";" + EOL; /* * Routines for our descendents. */ /** * Implementer must add script code for the parameters provided. * This will only be called if there is at least one parameter. * * @param params One or more parameters. */ public abstract void doParameterList(Param[] params) throws Utils.UserException; // Not yet documented. public abstract void doTestUrlMessage(String url) throws Utils.UserException; public abstract void doSetData(String data) throws Utils.UserException; public abstract void doCallUrl(String url, String method, String data, String contentLength) throws Utils.UserException; public abstract void doAssertResponse(String respCode) throws Utils.UserException; public abstract void doTidyCode(String url) throws Utils.UserException; public abstract void doResponseForStdOut(String url) throws Utils.UserException; public abstract void doResponseForFile() throws Utils.UserException; /** * Called after all other calls for a particular transaction (request + * response). */ public abstract void doEndTransaction() throws Utils.UserException; /** jtidyConfigFile variable serves two purpose: * one as a flag indicating we need spit out jtidy integration code, * and as a variable indicating the location of jtidy.cfg file. */ protected static String jtidyConfigFile = System.getProperty("jtidy.config"); static { if (cpRspTo != null) { if (cpRspTo.toLowerCase().compareTo("stdout") == 0) cpRspToStdout = true; else if (cpRspTo.toLowerCase().compareTo("file") == 0) cpRspToFile = true; } mimeTypes = new HashMap(); for (int i = 0; i < MIME_DEFAULTS.length; i++) { mimeTypes.put(MIME_DEFAULTS[i], MIME_DEFAULTS[i]); } } public static boolean isCpRspToStdout() { return cpRspToStdout; } public static boolean isCpRspToFile() { return cpRspToFile; } /** * @param nameRegExps A list of regular expressions. Each finds an * instance of the script's name embedded in the * script. This class will use the regular expressions * to keep these names up-to-date when the user saves. */ public AbstractCodeGenerator(IScriptAdapter adapter, String[] nameRegExps) { this.scriptAdapter = adapter; int i; namePatterns = new Pattern[nameRegExps.length]; for (i = 0; i < nameRegExps.length; i ++) namePatterns[i] = Pattern.compile(nameRegExps[i], Pattern.MULTILINE); testName = Config.getProperty("test.default_testname", "MaxQTest"); defaultTestName = testName; // Create a thread to allow multiple inserts to be collated and // inserted together. insertThread = new Thread(this); insertThread.start(); } /** * Necessary because the thread started in the constructor will otherwise * never exit. */ public void close() { insertThread.interrupt(); } public IScriptAdapter getScriptAdapter() { return scriptAdapter; } /** * Returns the name of the test. * Currently the same as the name of the file the test is stored in * @return testName */ public String getTestName() { return testName; } /** * Sets the name of the test class. * Defaults to value set in maxq.properties under test.default_testname * @param testName */ public void setTestName(String testName) { this.testName = testName; } public String getDefaultTestName() { return defaultTestName; } public void setDefaultTestName(String defaultTestName) { this.defaultTestName = defaultTestName; } /** * Returns the file path of the test. Null if test is not previously saved. * @return */ public String getTestPath() { return testPath; } /** * Sets the file path of the test * @param testPath */ public void setTestPath(String testPath) { this.testPath = testPath; } /** * Returns the full script from the ScriptAdapter. */ protected String getScript() { return this.scriptAdapter.getText(); } /** * Processes the response and generates test events for the code generator implementation class. * @param header * @param response */ public void processResponse(HttpRequestHeader header, byte[] response) throws Exception { if (ignoreNextResponse) { log.debug("Ignoring response"); ignoreNextResponse = false; return; } String respStr = new String(response).toLowerCase(); String respCode = parseResponseCode(respStr); String contentType = parseContentType(respStr); if(contentType != null && mimeTypes.get(contentType) != null && respCode != null ) { doAssertResponse(respCode); if (respCode.startsWith("200")) { if (contentType.toLowerCase().compareTo("text/html") == 0) { doTidyCode(HttpTestCase.staticUrlDecode(header.url)); if (cpRspToStdout) { doResponseForStdOut(HttpTestCase.staticUrlDecode(header.url).trim()); } else if (cpRspToFile) { doResponseForFile(); } } //text/html }// respCode.startsWith("200") doEndTransaction(); // add in an empty line for readability } else { log.debug("Ignoring response because content type is not known: " + contentType); } } private String parseContentType(String response){ String contentType = null; int pos = response.indexOf("content-type:"); if (pos > -1) { pos += 14; // okay we found 'content-type:', advance 14 char positions int end = response.indexOf("\r\n", pos); int end2 = response.indexOf(";", pos); if (end2 > -1 && end2 < end) end = end2; if (end > -1){ contentType = response.substring(pos, end).trim();