]> git.aero2k.de Git - tmp/jakarta-migration.git/blob
ebf19c30e36e457a736b8f2602aa9c05f321d6a4
[tmp/jakarta-migration.git] /
1 package org.collectionspace.services.IntegrationTests.xmlreplay;\r
2 \r
3 import org.apache.commons.cli.*;\r
4 \r
5 import org.apache.commons.io.FileUtils;\r
6 import org.apache.commons.jexl2.JexlEngine;\r
7 import org.collectionspace.services.common.api.FileTools;\r
8 import org.collectionspace.services.common.api.Tools;\r
9 import org.collectionspace.services.common.service.ServiceBindingType;\r
10 import org.dom4j.*;\r
11 import org.dom4j.io.SAXReader;\r
12 \r
13 import java.io.*;\r
14 import java.util.*;\r
15 \r
16 /**  This class is used to replay a request to the Services layer, by sending the XML payload\r
17  *   in an appropriate Multipart request.\r
18  *   See example usage in calling class XmlReplayTest in services/IntegrationTests, and also in main() in this class.\r
19  *   @author Laramie Crocker\r
20  */\r
21 public class XmlReplay {\r
22 \r
23     public XmlReplay(String basedir, String reportsDir){\r
24         this.basedir = basedir;\r
25         this.serviceResultsMap = createResultsMap();\r
26         this.reportsList = new ArrayList<String>();\r
27         this.reportsDir = reportsDir;\r
28     }\r
29 \r
30     public static final String DEFAULT_CONTROL = "xml-replay-control.xml";\r
31     public static final String DEFAULT_MASTER_CONTROL = "xml-replay-master.xml";\r
32     public static final String DEFAULT_DEV_MASTER_CONTROL = "dev-master.xml";\r
33 \r
34     private String reportsDir = "";\r
35     public String getReportsDir(){\r
36         return reportsDir;\r
37     }\r
38     private String basedir = ".";  //set from constructor.\r
39     public String getBaseDir(){\r
40         return basedir;\r
41     }\r
42     \r
43     private String controlFileName = DEFAULT_CONTROL;\r
44     public String getControlFileName() {\r
45         return controlFileName;\r
46     }\r
47     public void setControlFileName(String controlFileName) {\r
48         this.controlFileName = controlFileName;\r
49     }\r
50 \r
51     private String protoHostPort = "";\r
52     public String getProtoHostPort() {\r
53         return protoHostPort;\r
54     }\r
55     public void setProtoHostPort(String protoHostPort) {\r
56         this.protoHostPort = protoHostPort;\r
57     }\r
58 \r
59     private boolean autoDeletePOSTS = true;\r
60     public boolean isAutoDeletePOSTS() {\r
61         return autoDeletePOSTS;\r
62     }\r
63     public void setAutoDeletePOSTS(boolean autoDeletePOSTS) {\r
64         this.autoDeletePOSTS = autoDeletePOSTS;\r
65     }\r
66 \r
67     private Dump dump;\r
68     public Dump getDump() {\r
69         return dump;\r
70     }\r
71     public void setDump(Dump dump) {\r
72         this.dump = dump;\r
73     }\r
74 \r
75     AuthsMap defaultAuthsMap;\r
76     public AuthsMap getDefaultAuthsMap(){\r
77         return defaultAuthsMap;\r
78     }\r
79     public void setDefaultAuthsMap(AuthsMap authsMap){\r
80         defaultAuthsMap = authsMap;\r
81     }\r
82 \r
83     private Map<String, ServiceResult> serviceResultsMap;\r
84     public Map<String, ServiceResult> getServiceResultsMap(){\r
85         return serviceResultsMap;\r
86     }\r
87     public static Map<String, ServiceResult> createResultsMap(){\r
88         return new HashMap<String, ServiceResult>();\r
89     }\r
90 \r
91     private List<String> reportsList;\r
92     public  List<String> getReportsList(){\r
93         return reportsList;\r
94     }\r
95 \r
96     public String toString(){\r
97         return "XmlReplay{"+this.basedir+", "+this.defaultAuthsMap+", "+this.dump+", "+this.reportsDir+'}';\r
98     }\r
99 \r
100     // ============== METHODS ===========================================================\r
101 \r
102     /** Optional information method: call this method after instantiating this class using the constructor XmlReplay(String), which sets the basedir.  Then you\r
103      *   pass in your relative masterFilename to that basedir to this method, which will return true if the file is readable, valid xml, etc.\r
104      *   Do this in preference to  just seeing if File.exists(), because there are rules to finding the file relative to the maven test dir, yada, yada.\r
105      *   This method makes it easy to have a development test file that you don't check in, so that dev tests can be missing gracefully, etc.\r
106      */\r
107     public boolean masterConfigFileExists(String masterFilename){\r
108         try {\r
109             org.dom4j.Document doc = openMasterConfigFile(masterFilename);\r
110             if (doc == null){\r
111                 return false;\r
112             }\r
113             return true;\r
114         } catch (Throwable t){\r
115             return false;\r
116         }\r
117     }\r
118 \r
119     public org.dom4j.Document openMasterConfigFile(String masterFilename) throws FileNotFoundException {\r
120         String fullPath = Tools.glue(basedir, "/", masterFilename);\r
121         File f = new File(fullPath);\r
122         if (!f.exists()){\r
123             return null;\r
124         }\r
125         org.dom4j.Document document = getDocument(fullPath); //will check full path first, then checks relative to PWD.\r
126         if (document == null){\r
127             throw new FileNotFoundException("XmlReplay master control file ("+masterFilename+") not found in basedir: "+basedir+". Exiting test.");\r
128         }\r
129         return document;\r
130     }\r
131 \r
132     /** specify the master config file, relative to getBaseDir(), but ignore any tests or testGroups in the master.\r
133      *  @return a Document object, which you don't need to use: all options will be stored in XmlReplay instance.\r
134      */\r
135     public org.dom4j.Document readOptionsFromMasterConfigFile(String masterFilename) throws FileNotFoundException {\r
136         org.dom4j.Document document = openMasterConfigFile(masterFilename);\r
137         if (document == null){\r
138             throw new FileNotFoundException(masterFilename);\r
139         }\r
140         protoHostPort = document.selectSingleNode("/xmlReplayMaster/protoHostPort").getText().trim();\r
141         AuthsMap authsMap = readAuths(document);\r
142         setDefaultAuthsMap(authsMap);\r
143         Dump dump = XmlReplay.readDumpOptions(document);\r
144         setDump(dump);\r
145         return document;\r
146     }\r
147 \r
148     public List<List<ServiceResult>> runMaster(String masterFilename) throws Exception {\r
149         return runMaster(masterFilename, true);\r
150     }\r
151 \r
152     /** Creates new instances of XmlReplay, one for each controlFile specified in the master,\r
153      *  and setting defaults from this instance, but not sharing ServiceResult objects or maps. */\r
154     public List<List<ServiceResult>> runMaster(String masterFilename, boolean readOptionsFromMaster) throws Exception {\r
155         List<List<ServiceResult>> list = new ArrayList<List<ServiceResult>>();\r
156         org.dom4j.Document document;\r
157         if (readOptionsFromMaster){\r
158             document = readOptionsFromMasterConfigFile(masterFilename);\r
159         } else {\r
160             document = openMasterConfigFile(masterFilename);\r
161         }\r
162         if (document==null){\r
163             throw new FileNotFoundException(masterFilename);\r
164         }\r
165         String controlFile, testGroup, test;\r
166         List<Node> runNodes;\r
167         runNodes = document.selectNodes("/xmlReplayMaster/run");\r
168         for (Node runNode : runNodes) {\r
169             controlFile = runNode.valueOf("@controlFile");\r
170             testGroup = runNode.valueOf("@testGroup");\r
171             test = runNode.valueOf("@test"); //may be empty\r
172 \r
173             //Create a new instance and clone only config values, not any results maps.\r
174             XmlReplay replay = new XmlReplay(basedir, this.reportsDir);\r
175             replay.setControlFileName(controlFile);\r
176             replay.setProtoHostPort(protoHostPort);\r
177             replay.setAutoDeletePOSTS(isAutoDeletePOSTS());\r
178             replay.setDump(dump);\r
179             replay.setDefaultAuthsMap(getDefaultAuthsMap());\r
180 \r
181             //Now run *that* instance.\r
182             List<ServiceResult> results = replay.runTests(testGroup, test);\r
183             list.add(results);\r
184             this.reportsList.addAll(replay.getReportsList());   //Add all the reports from the inner replay, to our master replay's reportsList, to generate the index.html file.\r
185         }\r
186         XmlReplayReport.saveIndexForMaster(basedir, reportsDir, masterFilename, this.reportsList);\r
187         return list;\r
188     }\r
189 \r
190     /** Use this if you wish to run named tests within a testGroup, otherwise call runTestGroup(). */\r
191     public List<ServiceResult>  runTests(String testGroupID, String testID) throws Exception {\r
192         List<ServiceResult> result = runXmlReplayFile(this.basedir,\r
193                                 this.controlFileName,\r
194                                 testGroupID,\r
195                                 testID,\r
196                                 this.serviceResultsMap,\r
197                                 this.autoDeletePOSTS,\r
198                                 dump,\r
199                                 this.protoHostPort,\r
200                                 this.defaultAuthsMap,\r
201                                 this.reportsList,\r
202                                 this.reportsDir);\r
203         return result;\r
204     }\r
205 \r
206     /** Use this if you wish to specify just ONE test to run within a testGroup, otherwise call runTestGroup(). */\r
207     public ServiceResult  runTest(String testGroupID, String testID) throws Exception {\r
208         List<ServiceResult> result = runXmlReplayFile(this.basedir,\r
209                                 this.controlFileName,\r
210                                 testGroupID,\r
211                                 testID,\r
212                                 this.serviceResultsMap,\r
213                                 this.autoDeletePOSTS,\r
214                                 dump,\r
215                                 this.protoHostPort,\r
216                                 this.defaultAuthsMap,\r
217                                 this.reportsList,\r
218                                 this.reportsDir);\r
219         if (result.size()>1){\r
220             throw new IndexOutOfBoundsException("Multiple ("+result.size()+") tests with ID='"+testID+"' were found within test group '"+testGroupID+"', but there should only be one test per ID attribute.");\r
221         }\r
222         return result.get(0);\r
223     }\r
224 \r
225     /** Use this if you wish to run all tests within a testGroup.*/\r
226     public List<ServiceResult> runTestGroup(String testGroupID) throws Exception {\r
227         //NOTE: calling runTest with empty testID runs all tests in a test group, but don't expose this fact.\r
228         // Expose this method (runTestGroup) instead.\r
229         return runTests(testGroupID, "");\r
230     }\r
231 \r
232     public List<ServiceResult>  autoDelete(String logName){\r
233         return autoDelete(this.serviceResultsMap, logName);\r
234     }\r
235 \r
236     /** Use this method to clean up resources created on the server that returned CSIDs, if you have\r
237      *  specified autoDeletePOSTS==false, which means you are managing the cleanup yourself.\r
238      * @param serviceResultsMap a Map of ServiceResult objects, which will contain ServiceResult.deleteURL.\r
239      * @return a List<String> of debug info about which URLs could not be deleted.\r
240      */\r
241     public static List<ServiceResult> autoDelete(Map<String, ServiceResult> serviceResultsMap, String logName){\r
242         List<ServiceResult> results = new ArrayList<ServiceResult>();\r
243         for (ServiceResult pr : serviceResultsMap.values()){\r
244             try {\r
245                 if (Tools.notEmpty(pr.deleteURL)){\r
246                     ServiceResult deleteResult = XmlReplayTransport.doDELETE(pr.deleteURL, pr.auth, pr.testID, "[autodelete:"+logName+"]");\r
247                     results.add(deleteResult);\r
248                 } else {\r
249                     ServiceResult errorResult = new ServiceResult();\r
250                     errorResult.fullURL = pr.fullURL;\r
251                     errorResult.testGroupID = pr.testGroupID;\r
252                     errorResult.fromTestID = pr.fromTestID;\r
253                     errorResult.overrideGotExpectedResult();\r
254                     results.add(errorResult);\r
255                 }\r
256             } catch (Throwable t){\r
257                 String s = (pr!=null) ? "ERROR while cleaning up ServiceResult map: "+pr+" for "+pr.deleteURL+" :: "+t\r
258                                       : "ERROR while cleaning up ServiceResult map (null ServiceResult): "+t;\r
259                 System.err.println(s);\r
260                 ServiceResult errorResult = new ServiceResult();\r
261                 errorResult.fullURL = pr.fullURL;\r
262                 errorResult.testGroupID = pr.testGroupID;\r
263                 errorResult.fromTestID = pr.fromTestID;\r
264                 errorResult.error = s;\r
265                 results.add(errorResult);\r
266             }\r
267         }\r
268         return results;\r
269     }\r
270 \r
271     public static class AuthsMap {\r
272         Map<String,String> map;\r
273         String defaultID="";\r
274         public String getDefaultAuth(){\r
275             return map.get(defaultID);\r
276         }\r
277         public String toString(){\r
278             return "AuthsMap: {default='"+defaultID+"'; "+map.keySet()+'}';\r
279         }\r
280     }\r
281 \r
282     public static AuthsMap readAuths(org.dom4j.Document document){\r
283     Map<String, String> map = new HashMap<String, String>();\r
284         List<Node> authNodes = document.selectNodes("//auths/auth");\r
285         for (Node auth : authNodes) {\r
286             map.put(auth.valueOf("@ID"), auth.getStringValue());\r
287         }\r
288         AuthsMap authsMap = new AuthsMap();\r
289         Node auths = document.selectSingleNode("//auths");\r
290         String defaultID = "";\r
291         if (auths != null){\r
292             defaultID = auths.valueOf("@default");\r
293         }\r
294         authsMap.map = map;\r
295         authsMap.defaultID = defaultID;\r
296         return authsMap;\r
297     }\r
298 \r
299     public static class Dump {\r
300         public boolean payloads = false;\r
301         //public static final ServiceResult.DUMP_OPTIONS dumpServiceResultOptions = ServiceResult.DUMP_OPTIONS;\r
302         public ServiceResult.DUMP_OPTIONS dumpServiceResult = ServiceResult.DUMP_OPTIONS.minimal;\r
303         public String toString(){\r
304             return "payloads: "+payloads+" dumpServiceResult: "+dumpServiceResult;\r
305         }\r
306     }\r
307 \r
308     public static Dump getDumpConfig(){\r
309         return new Dump();\r
310     }\r
311 \r
312     public static Dump readDumpOptions(org.dom4j.Document document){\r
313         Dump dump = getDumpConfig();\r
314         Node dumpNode = document.selectSingleNode("//dump");\r
315         if (dumpNode != null){\r
316             dump.payloads = Tools.isTrue(dumpNode.valueOf("@payloads"));\r
317             String dumpServiceResultStr = dumpNode.valueOf("@dumpServiceResult");\r
318             if (Tools.notEmpty(dumpServiceResultStr)){\r
319                 dump.dumpServiceResult = ServiceResult.DUMP_OPTIONS.valueOf(dumpServiceResultStr);\r
320             }\r
321         }\r
322         return dump;\r
323     }\r
324 \r
325     private static class PartsStruct {\r
326         public List<Map<String,String>> varsList = new ArrayList<Map<String,String>>();\r
327         String responseFilename = "";\r
328         String overrideTestID = "";\r
329         String startElement = "";\r
330         String label = "";\r
331 \r
332         public static PartsStruct readParts(Node testNode, final String testID, String xmlReplayBaseDir){\r
333             PartsStruct resultPartsStruct = new PartsStruct();\r
334             resultPartsStruct.responseFilename = testNode.valueOf("filename");\r
335             resultPartsStruct.startElement = testNode.valueOf("startElement");\r
336             resultPartsStruct.label = testNode.valueOf("label");\r
337             String responseFilename = testNode.valueOf("filename");\r
338             if (Tools.notEmpty(responseFilename)){\r
339                 resultPartsStruct.responseFilename = xmlReplayBaseDir + '/' + responseFilename;\r
340                 List<Node> varNodes = testNode.selectNodes("vars/var");\r
341                 readVars(testNode, varNodes, resultPartsStruct);\r
342             }\r
343             return resultPartsStruct;\r
344         }\r
345 \r
346         private static void readVars(Node testNode, List<Node> varNodes, PartsStruct resultPartsStruct){\r
347             Map<String,String> vars = new HashMap<String,String>();\r
348             resultPartsStruct.varsList.add(vars);\r
349             //System.out.println("### vars: "+vars.size()+" ########");\r
350             for (Node var: varNodes){\r
351                 String ID = var.valueOf("@ID");\r
352                 String value = var.getText();\r
353                 //System.out.println("ID: "+ID+" value: "+value);\r
354                 vars.put(ID, value); //vars is already part of resultPartsStruct.varsList\r
355             }\r
356             //System.out.println("### end-vars ########");\r
357         }\r
358     }\r
359 \r
360 \r
361 \r
362     private static String fixupFullURL(String fullURL, String protoHostPort, String uri){\r
363         if ( ! uri.startsWith(protoHostPort)){\r
364             fullURL = Tools.glue(protoHostPort, "/", uri);\r
365         } else {\r
366             fullURL = uri;\r
367         }\r
368         return fullURL;\r
369     }\r
370 \r
371     private static String fromTestID(String fullURL, Node testNode, Map<String, ServiceResult> serviceResultsMap){\r
372         String fromTestID = testNode.valueOf("fromTestID");\r
373         if (Tools.notEmpty(fromTestID)){\r
374             ServiceResult getPR = serviceResultsMap.get(fromTestID);\r
375             if (getPR != null){\r
376                 fullURL = Tools.glue(fullURL, "/", getPR.location);\r
377             }\r
378         }\r
379         return fullURL;\r
380     }\r
381 \r
382     private static String CSIDfromTestID(Node testNode, Map<String, ServiceResult> serviceResultsMap){\r
383         String result = "";\r
384         String fromTestID = testNode.valueOf("fromTestID");\r
385         if (Tools.notEmpty(fromTestID)){\r
386             ServiceResult getPR = serviceResultsMap.get(fromTestID);\r
387             if (getPR != null){\r
388                 result = getPR.location;\r
389             }\r
390         }\r
391         return result;\r
392     }\r
393 \r
394     public static org.dom4j.Document getDocument(String xmlFileName) {\r
395         org.dom4j.Document document = null;\r
396         SAXReader reader = new SAXReader();\r
397         try {\r
398             document = reader.read(xmlFileName);\r
399         } catch (DocumentException e) {\r
400             System.out.println("ERROR reading document: "+e);\r
401             //e.printStackTrace();\r
402         }\r
403         return document;\r
404     }\r
405 \r
406     protected static String validateResponseSinglePayload(ServiceResult serviceResult,\r
407                                                  Map<String, ServiceResult> serviceResultsMap,\r
408                                                  PartsStruct expectedResponseParts,\r
409                                                  XmlReplayEval evalStruct)\r
410     throws Exception {\r
411         String OK = "";\r
412         byte[] b = FileUtils.readFileToByteArray(new File(expectedResponseParts.responseFilename));\r
413         String expectedPartContent = new String(b);\r
414         Map<String,String> vars = expectedResponseParts.varsList.get(0);  //just one part, so just one varsList.  Must be there, even if empty.\r
415         expectedPartContent = evalStruct.eval(expectedPartContent, serviceResultsMap, vars, evalStruct.jexl, evalStruct.jc);\r
416         String label = "NOLABEL";\r
417         String leftID  = "{from expected part, label:"+label+" filename: "+expectedResponseParts.responseFilename+"}";\r
418         String rightID = "{from server, label:"+label\r
419                             +" fromTestID: "+serviceResult.fromTestID\r
420                             +" URL: "+serviceResult.fullURL\r
421                             +"}";\r
422         String startElement = expectedResponseParts.startElement;\r
423         String partLabel = expectedResponseParts.label;\r
424         if (Tools.isBlank(startElement)){\r
425             if (Tools.notBlank(partLabel))\r
426             startElement = "/document/*[local-name()='"+partLabel+"']";\r
427         }\r
428         TreeWalkResults.MatchSpec matchSpec = TreeWalkResults.MatchSpec.createDefault();\r
429         TreeWalkResults list =\r
430             XmlCompareJdom.compareParts(expectedPartContent,\r
431                                         leftID,\r
432                                         serviceResult.result,\r
433                                         rightID,\r
434                                         startElement,\r
435                                         matchSpec);\r
436         serviceResult.addPartSummary(label, list);\r
437         return OK;\r
438     }\r
439 \r
440     protected static String validateResponse(ServiceResult serviceResult,\r
441                                              Map<String, ServiceResult> serviceResultsMap,\r
442                                              PartsStruct expectedResponseParts,\r
443                                              XmlReplayEval evalStruct){\r
444         String OK = "";\r
445         if (expectedResponseParts == null) return OK;\r
446         if (serviceResult == null) return OK;\r
447         if (serviceResult.result.length() == 0) return OK;\r
448         try {\r
449             return validateResponseSinglePayload(serviceResult, serviceResultsMap, expectedResponseParts, evalStruct);\r
450         } catch (Exception e){\r
451             String err = "ERROR in XmlReplay.validateResponse() : "+e;\r
452             return err  ;\r
453         }\r
454     }\r
455 \r
456     //================= runXmlReplayFile ======================================================\r
457 \r
458     public static List<ServiceResult> runXmlReplayFile(String xmlReplayBaseDir,\r
459                                           String controlFileName,\r
460                                           String testGroupID,\r
461                                           String oneTestID,\r
462                                           Map<String, ServiceResult> serviceResultsMap,\r
463                                           boolean param_autoDeletePOSTS,\r
464                                           Dump dump,\r
465                                           String protoHostPortParam,\r
466                                           AuthsMap defaultAuths,\r
467                                           List<String> reportsList,\r
468                                           String reportsDir)\r
469                                           throws Exception {\r
470         //Internally, we maintain two collections of ServiceResult:\r
471         //  the first is the return value of this method.\r
472         //  the second is the serviceResultsMap, which is used for keeping track of CSIDs created by POSTs, for later reference by DELETE, etc.\r
473         List<ServiceResult> results = new ArrayList<ServiceResult>();\r
474 \r
475         XmlReplayReport report = new XmlReplayReport(reportsDir);\r
476 \r
477         String controlFile = Tools.glue(xmlReplayBaseDir, "/", controlFileName);\r
478         org.dom4j.Document document;\r
479         document = getDocument(controlFile); //will check full path first, then checks relative to PWD.\r
480         if (document==null){\r
481             throw new FileNotFoundException("XmlReplay control file ("+controlFileName+") not found in basedir: "+xmlReplayBaseDir+" Exiting test.");\r
482         }\r
483         String protoHostPort;\r
484         if (Tools.isEmpty(protoHostPortParam)){\r
485             protoHostPort = document.selectSingleNode("/xmlReplay/protoHostPort").getText().trim();\r
486             System.out.println("DEPRECATED: Using protoHostPort ('"+protoHostPort+"') from xmlReplay file ('"+controlFile+"'), not master.");\r
487         } else {\r
488             protoHostPort = protoHostPortParam;\r
489         }\r
490         if (Tools.isEmpty(protoHostPort)){\r
491             throw new Exception("XmlReplay control file must have a protoHostPort element");\r
492         }\r
493 \r
494         String authsMapINFO;\r
495         AuthsMap authsMap = readAuths(document);\r
496         if (authsMap.map.size()==0){\r
497             authsMap = defaultAuths;\r
498             authsMapINFO = "Using defaultAuths from master file: "+defaultAuths;\r
499         } else {\r
500             authsMapINFO = "Using AuthsMap from control file: "+authsMap;\r
501         }\r
502 \r
503         report.addTestGroup(testGroupID, controlFileName);   //controlFileName is just the short name, without the full path.\r
504         String xmlReplayHeader = "========================================================================"\r
505                           +"\r\nXmlReplay running:"\r
506                           +"\r\n   controlFile: "+ (new File(controlFile).getCanonicalPath())\r
507                           +"\r\n   protoHostPort: "+protoHostPort\r
508                           +"\r\n   testGroup: "+testGroupID\r
509                           + (Tools.notEmpty(oneTestID) ? "\r\n   oneTestID: "+oneTestID : "")\r
510                           +"\r\n   AuthsMap: "+authsMapINFO\r
511                           +"\r\n   param_autoDeletePOSTS: "+param_autoDeletePOSTS\r
512                           +"\r\n   Dump info: "+dump\r
513                           +"\r\n========================================================================"\r
514                           +"\r\n";\r
515         report.addRunInfo(xmlReplayHeader);\r
516 \r
517         System.out.println(xmlReplayHeader);\r
518 \r
519         String autoDeletePOSTS = "";\r
520         List<Node> testgroupNodes;\r
521         if (Tools.notEmpty(testGroupID)){\r
522             testgroupNodes = document.selectNodes("//testGroup[@ID='"+testGroupID+"']");\r
523         } else {\r
524             testgroupNodes = document.selectNodes("//testGroup");\r
525         }\r
526 \r
527         JexlEngine jexl = new JexlEngine();   // Used for expression language expansion from uri field.\r
528         XmlReplayEval evalStruct = new XmlReplayEval();\r
529         evalStruct.serviceResultsMap = serviceResultsMap;\r
530         evalStruct.jexl = jexl;\r
531 \r
532         for (Node testgroup : testgroupNodes) {\r
533 \r
534             XmlReplayEval.MapContextWKeys jc = new XmlReplayEval.MapContextWKeys();//MapContext();  //Get a new JexlContext for each test group.\r
535             evalStruct.jc = jc;\r
536 \r
537             autoDeletePOSTS = testgroup.valueOf("@autoDeletePOSTS");\r
538             List<Node> tests;\r
539             if (Tools.notEmpty(oneTestID)){\r
540                 tests = testgroup.selectNodes("test[@ID='"+oneTestID+"']");\r
541             } else {\r
542                 tests = testgroup.selectNodes("test");\r
543             }\r
544             String authForTest = "";\r
545             int testElementIndex = -1;\r
546 \r
547             for (Node testNode : tests) {\r
548                 long startTime = System.currentTimeMillis();\r
549                 try {\r
550                     testElementIndex++;\r
551                     String testID = testNode.valueOf("@ID");\r
552                     String testIDLabel = Tools.notEmpty(testID) ? (testGroupID+'.'+testID) : (testGroupID+'.'+testElementIndex);\r
553                     String method = testNode.valueOf("method");\r
554                     String uri = testNode.valueOf("uri");\r
555                     String fullURL = Tools.glue(protoHostPort, "/", uri);\r
556 \r
557                     String authIDForTest = testNode.valueOf("@auth");\r
558                     String currentAuthForTest = authsMap.map.get(authIDForTest);\r
559                     if (Tools.notEmpty(currentAuthForTest)){\r
560                         authForTest = currentAuthForTest; //else just run with current from last loop;\r
561                     }\r
562                     if (Tools.isEmpty(authForTest)){\r
563                         authForTest = defaultAuths.getDefaultAuth();\r
564                     }\r
565 \r
566                     if (uri.indexOf("$")>-1){\r
567                         uri = evalStruct.eval(uri, serviceResultsMap, null, jexl, jc);\r
568                     }\r
569                     fullURL = fixupFullURL(fullURL, protoHostPort, uri);\r
570 \r
571                     List<Integer> expectedCodes = new ArrayList<Integer>();\r
572                     String expectedCodesStr = testNode.valueOf("expectedCodes");\r
573                     if (Tools.notEmpty(expectedCodesStr)){\r
574                          String[] codesArray = expectedCodesStr.split(",");\r
575                          for (String code : codesArray){\r
576                              expectedCodes.add(new Integer(code.trim()));\r
577                          }\r
578                     }\r
579 \r
580                     Node responseNode = testNode.selectSingleNode("response");\r
581                     PartsStruct expectedResponseParts = null;\r
582                     if (responseNode!=null){\r
583                         expectedResponseParts = PartsStruct.readParts(responseNode, testID, xmlReplayBaseDir);\r
584                         //System.out.println("reponse parts: >>>>>>>>>>>>>>>>>>>>>>>>>>>>>>"+expectedResponseParts);\r
585                     }\r
586 \r
587                     ServiceResult serviceResult;\r
588                     boolean isPOST = method.equalsIgnoreCase("POST");\r
589                     boolean isPUT =  method.equalsIgnoreCase("PUT");\r
590                     if ( isPOST || isPUT ) {\r
591                         PartsStruct parts = PartsStruct.readParts(testNode, testID, xmlReplayBaseDir);\r
592                         if (Tools.notEmpty(parts.overrideTestID)) {\r
593                             testID = parts.overrideTestID;\r
594                         }\r
595                         if (isPOST){\r
596                             String csid = CSIDfromTestID(testNode, serviceResultsMap);\r
597                             if (Tools.notEmpty(csid)) uri = Tools.glue(uri, "/", csid+"/items/");\r
598                         } else if (isPUT) {\r
599                             uri = fromTestID(uri, testNode, serviceResultsMap);\r
600                         }\r
601                         //vars only make sense in two contexts: POST/PUT, because you are submitting another file with internal expressions,\r
602                         // and in <response> nodes. For GET, DELETE, there is no payload, so all the URLs with potential expressions are right there in the testNode.\r
603                         Map<String,String> vars = null;\r
604                         if (parts.varsList.size()>0){\r
605                             vars = parts.varsList.get(0);\r
606                         }\r
607                         serviceResult = XmlReplayTransport.doPOST_PUTFromXML(parts.responseFilename, vars, protoHostPort, uri, method, XmlReplayTransport.APPLICATION_XML, evalStruct, authForTest, testIDLabel);\r
608                         if (vars!=null) {\r
609                             serviceResult.addVars(vars);\r
610                         }\r
611                         results.add(serviceResult);\r
612                         //if (isPOST){\r
613                             serviceResultsMap.put(testID, serviceResult);      //PUTs do not return a Location, so don't add PUTs to serviceResultsMap.\r
614                         //}\r
615                         fullURL = fixupFullURL(fullURL, protoHostPort, uri);\r
616                     } else if (method.equalsIgnoreCase("DELETE")){\r
617                         String fromTestID = testNode.valueOf("fromTestID");\r
618                         ServiceResult pr = serviceResultsMap.get(fromTestID);\r
619                         if (pr!=null){\r
620                             serviceResult = XmlReplayTransport.doDELETE(pr.deleteURL, authForTest, testIDLabel, fromTestID);\r
621                             serviceResult.fromTestID = fromTestID;\r
622                             if (expectedCodes.size()>0){\r
623                                 serviceResult.expectedCodes = expectedCodes;\r
624                             }\r
625                             results.add(serviceResult);\r
626                             if (serviceResult.codeInSuccessRange(serviceResult.responseCode)){  //gotExpectedResult depends on serviceResult.expectedCodes.\r
627                                 serviceResultsMap.remove(fromTestID);\r
628                             }\r
629                         } else {\r
630                             if (Tools.notEmpty(fromTestID)){\r
631                                 serviceResult = new ServiceResult();\r
632                                 serviceResult.responseCode = 0;\r
633                                 serviceResult.error = "ID not found in element fromTestID: "+fromTestID;\r
634                                 System.err.println("****\r\nServiceResult: "+serviceResult.error+". SKIPPING TEST. Full URL: "+fullURL);\r
635                             } else {\r
636                                 serviceResult = XmlReplayTransport.doDELETE(fullURL, authForTest, testID, fromTestID);\r
637                             }\r
638                             serviceResult.fromTestID = fromTestID;\r
639                             results.add(serviceResult);\r
640                         }\r
641                     } else if (method.equalsIgnoreCase("GET")){\r
642                         fullURL = fromTestID(fullURL, testNode, serviceResultsMap);\r
643                         serviceResult = XmlReplayTransport.doGET(fullURL, authForTest, testIDLabel);\r
644                         results.add(serviceResult);\r
645                         serviceResultsMap.put(testID, serviceResult);\r
646                     } else if (method.equalsIgnoreCase("LIST")){\r
647                         fullURL = fixupFullURL(fullURL, protoHostPort, uri);\r
648                         String listQueryParams = ""; //TODO: empty for now, later may pick up from XML control file.\r
649                         serviceResult = XmlReplayTransport.doLIST(fullURL, listQueryParams, authForTest, testIDLabel);\r
650                         results.add(serviceResult);\r
651                         serviceResultsMap.put(testID, serviceResult);\r
652                     } else {\r
653                         throw new Exception("HTTP method not supported by XmlReplay: "+method);\r
654                     }\r
655 \r
656                     serviceResult.testID = testID;\r
657                     serviceResult.fullURL = fullURL;\r
658                     serviceResult.auth = authForTest;\r
659                     serviceResult.method = method;\r
660                     if (expectedCodes.size()>0){\r
661                         serviceResult.expectedCodes = expectedCodes;\r
662                     }\r
663                     if (Tools.isEmpty(serviceResult.testID)) serviceResult.testID = testIDLabel;\r
664                     if (Tools.isEmpty(serviceResult.testGroupID)) serviceResult.testGroupID = testGroupID;\r
665 \r
666                     Node expectedLevel = testNode.selectSingleNode("response/expected");\r
667                     if (expectedLevel!=null){\r
668                         String level = expectedLevel.valueOf("@level");\r
669                         serviceResult.payloadStrictness = level;\r
670                     }\r
671                     //=====================================================\r
672                     //  ALL VALIDATION FOR ALL REQUESTS IS DONE HERE:\r
673                     //=====================================================\r
674                     boolean hasError = false;\r
675                     String vError = validateResponse(serviceResult, serviceResultsMap, expectedResponseParts, evalStruct);\r
676                     if (Tools.notEmpty(vError)){\r
677                         serviceResult.error = vError;\r
678                         serviceResult.failureReason = " : VALIDATION ERROR; ";\r
679                         hasError = true;\r
680                     }\r
681                     if (hasError == false){\r
682                         hasError = ! serviceResult.gotExpectedResult();\r
683                     }\r
684 \r
685                     boolean doingAuto = (dump.dumpServiceResult == ServiceResult.DUMP_OPTIONS.auto);\r
686                     String serviceResultRow = serviceResult.dump(dump.dumpServiceResult, hasError)+"; time:"+(System.currentTimeMillis()-startTime);\r
687                     String leader = (dump.dumpServiceResult == ServiceResult.DUMP_OPTIONS.detailed) ? "XmlReplay:"+testIDLabel+": ": "";\r
688 \r
689                     report.addTestResult(serviceResult);\r
690 \r
691                     if (   (dump.dumpServiceResult == ServiceResult.DUMP_OPTIONS.detailed)\r
692                         || (dump.dumpServiceResult == ServiceResult.DUMP_OPTIONS.full)         ){\r
693                         System.out.println("\r\n#---------------------#");\r
694                     }\r
695                     System.out.println(leader+serviceResultRow+"\r\n");\r
696                     if (dump.payloads || (doingAuto&&hasError) ) {\r
697                         if (Tools.notBlank(serviceResult.requestPayload)){\r
698                             System.out.println("\r\n========== request payload ===============");\r
699                             System.out.println(serviceResult.requestPayload);\r
700                             System.out.println("==========================================\r\n");\r
701                         }\r
702                     }\r
703                     if (dump.payloads || (doingAuto&&hasError)) {\r
704                         if (Tools.notBlank(serviceResult.result)){\r
705                             System.out.println("\r\n========== response payload ==============");\r
706                             System.out.println(serviceResult.result);\r
707                             System.out.println("==========================================\r\n");\r
708                         }\r
709                     }\r
710                 } catch (Throwable t) {\r
711                     String msg = "ERROR: XmlReplay experienced an error in a test node: "+testNode+" Throwable: "+t;\r
712                     System.out.println(msg);\r
713                     System.out.println(Tools.getStackTrace(t));\r
714                     ServiceResult serviceResult = new ServiceResult();\r
715                     serviceResult.error = msg;\r
716                     serviceResult.failureReason = " : SYSTEM ERROR; ";\r
717                     results.add(serviceResult);\r
718                 }\r
719             }\r
720             if (Tools.isTrue(autoDeletePOSTS)&&param_autoDeletePOSTS){\r
721                 autoDelete(serviceResultsMap, "default");\r
722             }\r
723         }\r
724 \r
725         //=== Now spit out the HTML report file ===\r
726         File m = new File(controlFileName);\r
727         String localName = m.getName();//don't instantiate, just use File to extract file name without directory.\r
728         String reportName = localName+'-'+testGroupID+".html";\r
729 \r
730         File resultFile = report.saveReport(xmlReplayBaseDir, reportsDir, reportName);\r
731         if (resultFile!=null) {\r
732             String toc = report.getTOC(reportName);\r
733             reportsList.add(toc);\r
734         }\r
735         //================================\r
736 \r
737         return results;\r
738     }\r
739 \r
740 \r
741 \r
742     //======================== MAIN ===================================================================\r
743 \r
744     private static Options createOptions() {\r
745         Options options = new Options();\r
746         options.addOption("xmlReplayBaseDir", true, "default/basedir");\r
747         return options;\r
748     }\r
749 \r
750     public static String usage(){\r
751         String result = "org.collectionspace.services.IntegrationTests.xmlreplay.XmlReplay {args}\r\n"\r
752                         +"  -xmlReplayBaseDir <dir> \r\n"\r
753                         +" You may also override these with system args, e.g.: \r\n"\r
754                         +"   -DxmlReplayBaseDir=/path/to/dir \r\n"\r
755                         +" These may also be passed in via the POM.\r\n"\r
756                         +" You can also set these system args, e.g.: \r\n"\r
757                         +"  -DtestGroupID=<oneID> \r\n"\r
758                         +"  -DtestID=<one TestGroup ID>"\r
759                         +"  -DautoDeletePOSTS=<true|false> \r\n"\r
760                         +"    (note: -DautoDeletePOSTS won't force deletion if set to false in control file.";\r
761         return result;\r
762     }\r
763 \r
764     private static String opt(CommandLine line, String option){\r
765         String result;\r
766         String fromProps = System.getProperty(option);\r
767         if (Tools.notEmpty(fromProps)){\r
768             return fromProps;\r
769         }\r
770         if (line==null){\r
771             return "";\r
772         }\r
773         result = line.getOptionValue(option);\r
774         if (result == null){\r
775             result = "";\r
776         }\r
777         return result;\r
778     }\r
779 \r
780     public static void main(String[]args) throws Exception {\r
781         Options options = createOptions();\r
782         //System.out.println("System CLASSPATH: "+prop.getProperty("java.class.path", null));\r
783         CommandLineParser parser = new GnuParser();\r
784         try {\r
785             // parse the command line arguments\r
786             CommandLine line = parser.parse(options, args);\r
787 \r
788             String xmlReplayBaseDir = opt(line, "xmlReplayBaseDir");\r
789             String reportsDir = opt(line, "reportsDir");\r
790             String testGroupID      = opt(line, "testGroupID");\r
791             String testID           = opt(line, "testID");\r
792             String autoDeletePOSTS  = opt(line, "autoDeletePOSTS");\r
793             String dumpResults      = opt(line, "dumpResults");\r
794             String controlFilename   = opt(line, "controlFilename");\r
795             String xmlReplayMaster  = opt(line, "xmlReplayMaster");\r
796 \r
797             if (Tools.isBlank(reportsDir)){\r
798                 reportsDir = xmlReplayBaseDir + XmlReplayTest.REPORTS_DIRNAME;\r
799             }\r
800             reportsDir = Tools.fixFilename(reportsDir);\r
801             xmlReplayBaseDir = Tools.fixFilename(xmlReplayBaseDir);\r
802             controlFilename = Tools.fixFilename(controlFilename);\r
803 \r
804             boolean bAutoDeletePOSTS = true;\r
805             if (Tools.notEmpty(autoDeletePOSTS)) {\r
806                 bAutoDeletePOSTS = Tools.isTrue(autoDeletePOSTS);\r
807             }\r
808             boolean bDumpResults = false;\r
809             if (Tools.notEmpty(dumpResults)) {\r
810                 bDumpResults = Tools.isTrue(autoDeletePOSTS);\r
811             }\r
812             if (Tools.isEmpty(xmlReplayBaseDir)){\r
813                 System.err.println("ERROR: xmlReplayBaseDir was not specified.");\r
814                 return;\r
815             }\r
816             File f = new File(Tools.glue(xmlReplayBaseDir, "/", controlFilename));\r
817             if (Tools.isEmpty(xmlReplayMaster) && !f.exists()){\r
818                 System.err.println("Control file not found: "+f.getCanonicalPath());\r
819                 return;\r
820             }\r
821             File fMaster = new File(Tools.glue(xmlReplayBaseDir, "/", xmlReplayMaster));\r
822             if (Tools.notEmpty(xmlReplayMaster)  && !fMaster.exists()){\r
823                 System.err.println("Master file not found: "+fMaster.getCanonicalPath());\r
824                 return;\r
825             }\r
826 \r
827             String xmlReplayBaseDirResolved = (new File(xmlReplayBaseDir)).getCanonicalPath();\r
828             System.out.println("XmlReplay ::"\r
829                             + "\r\n    xmlReplayBaseDir: "+xmlReplayBaseDir\r
830                             + "\r\n    xmlReplayBaseDir(resolved): "+xmlReplayBaseDirResolved\r
831                             + "\r\n    controlFilename: "+controlFilename\r
832                             + "\r\n    xmlReplayMaster: "+xmlReplayMaster\r
833                             + "\r\n    testGroupID: "+testGroupID\r
834                             + "\r\n    testID: "+testID\r
835                             + "\r\n    autoDeletePOSTS: "+bAutoDeletePOSTS\r
836                             + (Tools.notEmpty(xmlReplayMaster)\r
837                                        ? ("\r\n    will use master file: "+fMaster.getCanonicalPath())\r
838                                        : ("\r\n    will use control file: "+f.getCanonicalPath()) )\r
839                              );\r
840             \r
841             if (Tools.notEmpty(xmlReplayMaster)){\r
842                 if (Tools.notEmpty(controlFilename)){\r
843                     System.out.println("WARN: controlFilename: "+controlFilename+" will not be used because master was specified.  Running master: "+xmlReplayMaster);\r
844                 }\r
845                 XmlReplay replay = new XmlReplay(xmlReplayBaseDirResolved, reportsDir);\r
846                 replay.readOptionsFromMasterConfigFile(xmlReplayMaster);\r
847                 replay.setAutoDeletePOSTS(bAutoDeletePOSTS);\r
848                 Dump dumpFromMaster = replay.getDump();\r
849                 dumpFromMaster.payloads = Tools.isTrue(dumpResults);\r
850                 replay.setDump(dumpFromMaster);\r
851                 replay.runMaster(xmlReplayMaster, false); //false, because we already just read the options, and override a few.\r
852             } else {\r
853                 Dump dump = getDumpConfig();\r
854                 dump.payloads = Tools.isTrue(dumpResults);\r
855                 List<String> reportsList = new ArrayList<String>();\r
856                 runXmlReplayFile(xmlReplayBaseDirResolved, controlFilename, testGroupID, testID, createResultsMap(), bAutoDeletePOSTS, dump, "", null, reportsList, reportsDir);\r
857                 System.out.println("DEPRECATED: reportsList is generated, but not dumped: "+reportsList.toString());\r
858             }\r
859         } catch (ParseException exp) {\r
860             // oops, something went wrong\r
861             System.err.println("Cmd-line parsing failed.  Reason: " + exp.getMessage());\r
862             System.err.println(usage());\r
863         } catch (Exception e) {\r
864             System.out.println("Error : " + e.getMessage());\r
865             e.printStackTrace();\r
866         }\r
867     }\r
868 \r
869 }\r