]> git.aero2k.de Git - tmp/jakarta-migration.git/blob
a0bdc5e98354e99bd6ce49cf000e938bc739897d
[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.JexlContext;\r
7 import org.apache.commons.jexl2.JexlEngine;\r
8 import org.apache.commons.jexl2.MapContext;\r
9 import org.dom4j.*;\r
10 import org.dom4j.io.SAXReader;\r
11 import org.jdom.input.SAXBuilder;\r
12 import org.xml.sax.InputSource;\r
13 \r
14 \r
15 import org.jdom.Document;\r
16 import org.jdom.Element;\r
17 import org.jdom.JDOMException;\r
18 \r
19 import java.io.*;\r
20 import java.util.*;\r
21 \r
22 /**  This class is used to replay a request to the Services layer, by sending the XML payload\r
23  *   in an appropriate Multipart request.\r
24  *   See example usage in calling class XmlReplayTest in services/IntegrationTests, and also in main() in this class.\r
25  *   @author Laramie Crocker\r
26  */\r
27 public class XmlReplay {\r
28 \r
29     public XmlReplay(String basedir){\r
30         this.basedir = basedir;\r
31         this.serviceResultsMap = createResultsMap();\r
32     }\r
33 \r
34     public static final String DEFAULT_CONTROL = "xml-replay-control.xml";\r
35     public static final String DEFAULT_MASTER_CONTROL = "xml-replay-master.xml";\r
36     public static final String DEFAULT_DEV_MASTER_CONTROL = "dev-master.xml";\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 \r
92     public String toString(){\r
93         return "XmlReplay{"+this.basedir+", "+this.defaultAuthsMap+", "+this.dump+'}';\r
94     }\r
95 \r
96     // ============== METHODS ===========================================================\r
97 \r
98     public org.dom4j.Document openMasterConfigFile(String masterFilename) throws FileNotFoundException {\r
99         org.dom4j.Document document = getDocument(Tools.glue(basedir, "/", masterFilename)); //will check full path first, then checks relative to PWD.\r
100         if (document == null){\r
101             throw new FileNotFoundException("XmlReplay master control file ("+masterFilename+") not found in basedir: "+basedir+". Exiting test.");\r
102         }\r
103         return document;\r
104     }\r
105 \r
106     /** specify the master config file, relative to getBaseDir(), but ignore any tests or testGroups in the master.\r
107      *  @return a Document object, which you don't need to use: all options will be stored in XmlReplay instance.\r
108      */\r
109     public org.dom4j.Document readOptionsFromMasterConfigFile(String masterFilename) throws FileNotFoundException {\r
110         org.dom4j.Document document = openMasterConfigFile(masterFilename);\r
111         protoHostPort = document.selectSingleNode("/xmlReplayMaster/protoHostPort").getText().trim();\r
112         AuthsMap authsMap = readAuths(document);\r
113         setDefaultAuthsMap(authsMap);\r
114         Dump dump = XmlReplay.readDumpOptions(document);\r
115         setDump(dump);\r
116         return document;\r
117     }\r
118 \r
119     public List<List<ServiceResult>> runMaster(String masterFilename) throws Exception {\r
120         return runMaster(masterFilename, true);\r
121     }\r
122 \r
123     /** Creates new instances of XmlReplay, one for each controlFile specified in the master,\r
124      *  and setting defaults from this instance, but not sharing ServiceResult objects or maps. */\r
125     public List<List<ServiceResult>> runMaster(String masterFilename, boolean readOptionsFromMaster) throws Exception {\r
126         List<List<ServiceResult>> list = new ArrayList<List<ServiceResult>>();\r
127         org.dom4j.Document document;\r
128         if (readOptionsFromMaster){\r
129             document = readOptionsFromMasterConfigFile(masterFilename);\r
130         } else {\r
131             document = openMasterConfigFile(masterFilename);\r
132         }\r
133         String controlFile, testGroup, test;\r
134         List<Node> runNodes;\r
135         runNodes = document.selectNodes("/xmlReplayMaster/run");\r
136         for (Node runNode : runNodes) {\r
137             controlFile = runNode.valueOf("@controlFile");\r
138             testGroup = runNode.valueOf("@testGroup");\r
139             test = runNode.valueOf("@test"); //may be empty\r
140 \r
141             //Create a new instance and clone only config values, not any results maps.\r
142             XmlReplay replay = new XmlReplay(basedir);\r
143             replay.setControlFileName(controlFile);\r
144             replay.setProtoHostPort(protoHostPort);\r
145             replay.setAutoDeletePOSTS(isAutoDeletePOSTS());\r
146             replay.setDump(dump);\r
147             replay.setDefaultAuthsMap(getDefaultAuthsMap());\r
148 \r
149             //Now run *that* instance.\r
150             List<ServiceResult> results = replay.runTests(testGroup, test);\r
151             list.add(results);\r
152         }\r
153         return list;\r
154     }\r
155 \r
156     /** Use this if you wish to named tests within a testGroup, otherwise call runTestGroup(). */\r
157     public List<ServiceResult>  runTests(String testGroupID, String testID) throws Exception {\r
158         List<ServiceResult> result = runXmlReplayFile(this.basedir,\r
159                                 this.controlFileName,\r
160                                 testGroupID,\r
161                                 testID,\r
162                                 this.serviceResultsMap,\r
163                                 this.autoDeletePOSTS,\r
164                                 dump,\r
165                                 this.protoHostPort,\r
166                                 this.defaultAuthsMap);\r
167         return result;\r
168     }\r
169 \r
170     /** Use this if you wish to specify just ONE test to run within a testGroup, otherwise call runTestGroup(). */\r
171     public ServiceResult  runTest(String testGroupID, String testID) throws Exception {\r
172         List<ServiceResult> result = runXmlReplayFile(this.basedir,\r
173                                 this.controlFileName,\r
174                                 testGroupID,\r
175                                 testID,\r
176                                 this.serviceResultsMap,\r
177                                 this.autoDeletePOSTS,\r
178                                 dump,\r
179                                 this.protoHostPort,\r
180                                 this.defaultAuthsMap);\r
181         if (result.size()>1){\r
182             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
183         }\r
184         return result.get(0);\r
185     }\r
186 \r
187     /** Use this if you wish to run all tests within a testGroup.*/\r
188     public List<ServiceResult> runTestGroup(String testGroupID) throws Exception {\r
189         //NOTE: calling runTest with empty testID runs all tests in a test group, but don't expose this fact.\r
190         // Expose this method (runTestGroup) instead.\r
191         return runTests(testGroupID, "");\r
192     }\r
193 \r
194     public List<ServiceResult>  autoDelete(String logName){\r
195         return autoDelete(this.serviceResultsMap, logName);\r
196     }\r
197 \r
198     /** Use this method to clean up resources created on the server that returned CSIDs, if you have\r
199      *  specified autoDeletePOSTS==false, which means you are managing the cleanup yourself.\r
200      * @param serviceResultsMap a Map of ServiceResult objects, which will contain ServiceResult.deleteURL.\r
201      * @return a List<String> of debug info about which URLs could not be deleted.\r
202      */\r
203     public static List<ServiceResult> autoDelete(Map<String, ServiceResult> serviceResultsMap, String logName){\r
204         List<ServiceResult> results = new ArrayList<ServiceResult>();\r
205         for (ServiceResult pr : serviceResultsMap.values()){\r
206             try {\r
207                 ServiceResult deleteResult = XmlReplayTransport.doDELETE(pr.deleteURL, pr.auth, pr.testID, "[autodelete:"+logName+"]");\r
208                 results.add(deleteResult);\r
209             } catch (Throwable t){\r
210                 String s = (pr!=null) ? "ERROR while cleaning up ServiceResult map: "+pr+" for "+pr.deleteURL+" :: "+t\r
211                                       : "ERROR while cleaning up ServiceResult map (null ServiceResult): "+t;\r
212                 System.err.println(s);\r
213                 ServiceResult errorResult = new ServiceResult();\r
214                 errorResult.fullURL = pr.fullURL;\r
215                 errorResult.testGroupID = pr.testGroupID;\r
216                 errorResult.fromTestID = pr.fromTestID;\r
217                 errorResult.error = s;\r
218                 results.add(errorResult);\r
219             }\r
220         }\r
221         return results;\r
222     }\r
223 \r
224     public static class AuthsMap {\r
225         Map<String,String> map;\r
226         String defaultID="";\r
227         public String getDefaultAuth(){\r
228             return map.get(defaultID);\r
229         }\r
230         public String toString(){\r
231             return "AuthsMap: {default='"+defaultID+"'; "+map.keySet()+'}';\r
232         }\r
233     }\r
234 \r
235     public static AuthsMap readAuths(org.dom4j.Document document){\r
236     Map<String, String> map = new HashMap<String, String>();\r
237         List<Node> authNodes = document.selectNodes("//auths/auth");\r
238         for (Node auth : authNodes) {\r
239             map.put(auth.valueOf("@ID"), auth.getStringValue());\r
240         }\r
241         AuthsMap authsMap = new AuthsMap();\r
242         Node auths = document.selectSingleNode("//auths");\r
243         String defaultID = "";\r
244         if (auths != null){\r
245             defaultID = auths.valueOf("@default");\r
246         }\r
247         authsMap.map = map;\r
248         authsMap.defaultID = defaultID;\r
249         return authsMap;\r
250     }\r
251 \r
252     public static class Dump {\r
253         public boolean payloads = false;\r
254         //public static final ServiceResult.DUMP_OPTIONS dumpServiceResultOptions = ServiceResult.DUMP_OPTIONS;\r
255         public ServiceResult.DUMP_OPTIONS dumpServiceResult = ServiceResult.DUMP_OPTIONS.minimal;\r
256         public String toString(){\r
257             return "payloads: "+payloads+" dumpServiceResult: "+dumpServiceResult;\r
258         }\r
259     }\r
260 \r
261     public static Dump getDumpConfig(){\r
262         return new Dump();\r
263     }\r
264 \r
265     public static Dump readDumpOptions(org.dom4j.Document document){\r
266         Dump dump = getDumpConfig();\r
267         Node dumpNode = document.selectSingleNode("//dump");\r
268         if (dumpNode != null){\r
269             dump.payloads = Tools.isTrue(dumpNode.valueOf("@payloads"));\r
270             String dumpServiceResultStr = dumpNode.valueOf("@dumpServiceResult");\r
271             if (Tools.notEmpty(dumpServiceResultStr)){\r
272                 dump.dumpServiceResult = ServiceResult.DUMP_OPTIONS.valueOf(dumpServiceResultStr);\r
273             }\r
274         }\r
275         return dump;\r
276     }\r
277 \r
278     private static class PartsStruct {\r
279         public List<String> partsList = new ArrayList<String>();\r
280         public List<String> filesList = new ArrayList<String>();\r
281         boolean bDoingSinglePartPayload = false;\r
282         String singlePartPayloadFilename = "";\r
283         String overrideTestID = "";\r
284         public static PartsStruct readParts(Node testNode, final String testID, String xmlReplayBaseDir){\r
285             PartsStruct result = new PartsStruct();\r
286             result.singlePartPayloadFilename = testNode.valueOf("filename");\r
287             String singlePartPayloadFilename = testNode.valueOf("filename");\r
288             if (Tools.notEmpty(singlePartPayloadFilename)){\r
289                 result.bDoingSinglePartPayload = true;\r
290                 result.singlePartPayloadFilename = xmlReplayBaseDir + '/' + singlePartPayloadFilename;\r
291             } else {\r
292                 result.bDoingSinglePartPayload = false;\r
293                 List<Node> parts = testNode.selectNodes("parts/part");\r
294                 if (parts == null || parts.size()==0){\r
295                     //path is just /testGroup/test/part/\r
296                     String commonPartName = testNode.valueOf("part/label");\r
297                     String testfile = testNode.valueOf("part/filename");\r
298                     String fullTestFilename = xmlReplayBaseDir + '/' + testfile;\r
299                     if ( Tools.isEmpty(testID) ){\r
300                         result.overrideTestID = testfile; //It is legal to have a missing ID attribute, and rely on a unique filename.\r
301                     }\r
302                     result.partsList.add(commonPartName);\r
303                     result.filesList.add(fullTestFilename);\r
304                 } else {\r
305                     // path is /testGroup/test/parts/part/\r
306                     for (Node part : parts){\r
307                         String commonPartName = part.valueOf("label");\r
308                         String filename = part.valueOf("filename");\r
309                         String fullTestFilename = xmlReplayBaseDir + '/' + filename;\r
310                         if ( Tools.isEmpty(testID) ){  //if testID is empty, we'll use the *first*  filename as ID.\r
311                             result.overrideTestID = filename; //It is legal to have a missing ID attribute, and rely on a unique filename.\r
312                         }\r
313                         result.partsList.add(commonPartName);\r
314                         result.filesList.add(fullTestFilename);\r
315                     }\r
316                 }\r
317             }\r
318             return result;\r
319         }\r
320     }\r
321 \r
322     private static String fixupFullURL(String fullURL, String protoHostPort, String uri){\r
323         if ( ! uri.startsWith(protoHostPort)){\r
324             fullURL = Tools.glue(protoHostPort, "/", uri);\r
325         } else {\r
326             fullURL = uri;\r
327         }\r
328         return fullURL;\r
329     }\r
330 \r
331     private static String fromTestID(String fullURL, Node testNode, Map<String, ServiceResult> serviceResultsMap){\r
332         String fromTestID = testNode.valueOf("fromTestID");\r
333         if (Tools.notEmpty(fromTestID)){\r
334             ServiceResult getPR = serviceResultsMap.get(fromTestID);\r
335             if (getPR != null){\r
336                 fullURL = Tools.glue(fullURL, "/", getPR.location);\r
337             }\r
338         }\r
339         return fullURL;\r
340     }\r
341 \r
342     private static String CSIDfromTestID(Node testNode, Map<String, ServiceResult> serviceResultsMap){\r
343         String result = "";\r
344         String fromTestID = testNode.valueOf("fromTestID");\r
345         if (Tools.notEmpty(fromTestID)){\r
346             ServiceResult getPR = serviceResultsMap.get(fromTestID);\r
347             if (getPR != null){\r
348                 result = getPR.location;\r
349             }\r
350         }\r
351         return result;\r
352     }\r
353 \r
354     public static org.dom4j.Document getDocument(String xmlFileName) {\r
355         org.dom4j.Document document = null;\r
356         SAXReader reader = new SAXReader();\r
357         try {\r
358             document = reader.read(xmlFileName);\r
359         } catch (DocumentException e) {\r
360             System.out.println("ERROR reading document: "+e);\r
361             e.printStackTrace();\r
362         }\r
363         return document;\r
364     }\r
365 \r
366     protected static String validateResponse(ServiceResult serviceResult,\r
367                                              Map<String, ServiceResult> serviceResultsMap,\r
368                                              PartsStruct expectedResponseParts){\r
369         String OK = "";\r
370         if (expectedResponseParts == null) return OK;\r
371         if (serviceResult == null) return OK;\r
372         if (serviceResult.result.length() == 0) return OK;\r
373         String responseDump = serviceResult.result;\r
374         //System.out.println("responseDump: "+responseDump);\r
375         PayloadLogger.HttpTraffic traffic = PayloadLogger.readPayloads(responseDump, serviceResult.boundary, serviceResult.contentLength);\r
376         try {\r
377             for (int i=0; i<expectedResponseParts.partsList.size(); i++){\r
378                 String fileName = expectedResponseParts.filesList.get(i);\r
379                 String label = expectedResponseParts.partsList.get(i);\r
380                 byte[] b = FileUtils.readFileToByteArray(new File(fileName));\r
381                 String expectedPartContent = new String(b);\r
382                 System.out.println("expected: "+label+ " content ==>\r\n"+expectedPartContent);\r
383                 PayloadLogger.Part partFromServer = traffic.getPart(label);\r
384                 String partFromServerContent = "";\r
385                 if (partFromServer != null){\r
386                     partFromServerContent = partFromServer.getContent();\r
387                 } else {\r
388                     partFromServerContent = "";\r
389                 }\r
390                 //if (partFromServer!=null) {\r
391                     //System.out.println("====part content from server.   label-->"+label+"<-- \r\npart-->"+partFromServerContent+"<--");\r
392 \r
393                     String leftID  = "{from expected part, label:"+label+" filename: "+fileName+"}";\r
394                     String rightID = "{from server, label:"+label\r
395                                         //+" testGroupID: "+serviceResult.testGroupID\r
396                                         +" fromTestID: "+serviceResult.fromTestID\r
397                                         +" URL: "+serviceResult.fullURL\r
398                                         +"}";\r
399                     TreeWalkResults list =\r
400                         XmlCompareJdom.compareParts(expectedPartContent,\r
401                                                     leftID,\r
402                                                     partFromServerContent,\r
403                                                     rightID);\r
404                     //if (list.getMismatchCount()>0){\r
405                         serviceResult.addPartSummary(label, list);\r
406                     //}\r
407                 //}\r
408             }\r
409         } catch (Exception e){\r
410             String err = "ERROR in XmlReplay.validateResponse() : "+e; \r
411             //System.out.println(err);\r
412             return err  ;\r
413         }\r
414         return OK;\r
415     }\r
416 \r
417 \r
418     //================= runXmlReplayFile ======================================================\r
419 \r
420     public static List<ServiceResult> runXmlReplayFile(String xmlReplayBaseDir,\r
421                                           String controlFileName,\r
422                                           String testGroupID,\r
423                                           String oneTestID,\r
424                                           Map<String, ServiceResult> serviceResultsMap,\r
425                                           boolean param_autoDeletePOSTS,\r
426                                           Dump dump,\r
427                                           String protoHostPortParam,\r
428                                           AuthsMap defaultAuths)\r
429                                           throws Exception {\r
430         //Internally, we maintain two collections of ServiceResult:\r
431         //  the first is the return value of this method.\r
432         //  the second is the serviceResultsMap, which is used for keeping track of CSIDs created by POSTs, for later reference by DELETE, etc.\r
433         List<ServiceResult> results = new ArrayList<ServiceResult>();\r
434 \r
435         String controlFile = Tools.glue(xmlReplayBaseDir, "/", controlFileName);\r
436         org.dom4j.Document document;\r
437         document = getDocument(controlFile); //will check full path first, then checks relative to PWD.\r
438         if (document==null){\r
439             throw new FileNotFoundException("XmlReplay control file ("+controlFileName+") not found in basedir: "+xmlReplayBaseDir+" Exiting test.");\r
440         }\r
441         String protoHostPort;\r
442         if (Tools.isEmpty(protoHostPortParam)){\r
443             protoHostPort = document.selectSingleNode("/xmlReplay/protoHostPort").getText().trim();\r
444             System.out.println("DEPRECATED: Using protoHostPort ('"+protoHostPort+"') from xmlReplay file ('"+controlFile+"'), not master.");\r
445         } else {\r
446             protoHostPort = protoHostPortParam;\r
447         }\r
448         if (Tools.isEmpty(protoHostPort)){\r
449             throw new Exception("XmlReplay control file must have a protoHostPort element");\r
450         }\r
451 \r
452         String authsMapINFO;\r
453         AuthsMap authsMap = readAuths(document);\r
454         if (authsMap.map.size()==0){\r
455             authsMap = defaultAuths;\r
456             authsMapINFO = "Using defaultAuths from master file: "+defaultAuths;\r
457         } else {\r
458             authsMapINFO = "Using AuthsMap from control file: "+authsMap;\r
459         }\r
460         System.out.println("XmlReplay running:"\r
461                           +"\r\n   controlFile: "+ (new File(controlFile).getCanonicalPath())\r
462                           +"\r\n   protoHostPort: "+protoHostPort\r
463                           +"\r\n   testGroup: "+testGroupID\r
464                           + (Tools.notEmpty(oneTestID) ? "\r\n   oneTestID: "+oneTestID : "")\r
465                           +"\r\n   AuthsMap: "+authsMapINFO\r
466                           +"\r\n   param_autoDeletePOSTS: "+param_autoDeletePOSTS\r
467                           +"\r\n   Dump info: "+dump\r
468                           +"\r\n");\r
469 \r
470         String autoDeletePOSTS = "";\r
471         List<Node> testgroupNodes;\r
472         if (Tools.notEmpty(testGroupID)){\r
473             testgroupNodes = document.selectNodes("//testGroup[@ID='"+testGroupID+"']");\r
474         } else {\r
475             testgroupNodes = document.selectNodes("//testGroup");\r
476         }\r
477 \r
478         JexlEngine jexl = new JexlEngine();   // Used for expression language expansion from uri field.\r
479         XmlReplayEval evalStruct = new XmlReplayEval();\r
480         evalStruct.serviceResultsMap = serviceResultsMap;\r
481         evalStruct.jexl = jexl;\r
482 \r
483         for (Node testgroup : testgroupNodes) {\r
484             JexlContext jc = new MapContext();  //Get a new JexlContext for each test group.\r
485             evalStruct.jc = jc;\r
486 \r
487             autoDeletePOSTS = testgroup.valueOf("@autoDeletePOSTS");\r
488             List<Node> tests;\r
489             if (Tools.notEmpty(oneTestID)){\r
490                 tests = testgroup.selectNodes("test[@ID='"+oneTestID+"']");\r
491             } else {\r
492                 tests = testgroup.selectNodes("test");\r
493             }\r
494             String authForTest = "";\r
495             int testElementIndex = -1;\r
496 \r
497             for (Node testNode : tests) {\r
498                 try {\r
499                     testElementIndex++;\r
500                     String testID = testNode.valueOf("@ID");\r
501                     String testIDLabel = Tools.notEmpty(testID) ? (testGroupID+'.'+testID) : (testGroupID+'.'+testElementIndex);\r
502                     String method = testNode.valueOf("method");\r
503                     String uri = testNode.valueOf("uri");\r
504                     String fullURL = Tools.glue(protoHostPort, "/", uri);\r
505                     String initURI = uri;\r
506 \r
507                     String authIDForTest = testNode.valueOf("@auth");\r
508                     String currentAuthForTest = authsMap.map.get(authIDForTest);\r
509                     if (Tools.notEmpty(currentAuthForTest)){\r
510                         authForTest = currentAuthForTest; //else just run with current from last loop;\r
511                     }\r
512                     if (Tools.isEmpty(authForTest)){\r
513                         authForTest = defaultAuths.getDefaultAuth();\r
514                     }\r
515 \r
516                     if (uri.indexOf("$")>-1){\r
517                         uri = evalStruct.eval(uri, serviceResultsMap, jexl, jc);\r
518                     }\r
519                     fullURL = fixupFullURL(fullURL, protoHostPort, uri);\r
520 \r
521                     List<Integer> expectedCodes = new ArrayList<Integer>();\r
522                     String expectedCodesStr = testNode.valueOf("expectedCodes");\r
523                     if (Tools.notEmpty(expectedCodesStr)){\r
524                          String[] codesArray = expectedCodesStr.split(",");\r
525                          for (String code : codesArray){\r
526                              expectedCodes.add(new Integer(code.trim()));\r
527                          }\r
528                     }\r
529 \r
530                     Node responseNode = testNode.selectSingleNode("response");\r
531                     PartsStruct expectedResponseParts = null;\r
532                     if (responseNode!=null){\r
533                         expectedResponseParts = PartsStruct.readParts(responseNode, testID, xmlReplayBaseDir);\r
534                         //System.out.println("reponse parts: >>>>>>>>>>>>>>>>>>>>>>>>>>>>>>"+expectedResponseParts);\r
535                     }\r
536 \r
537                     ServiceResult serviceResult;\r
538                     boolean isPOST = method.equalsIgnoreCase("POST");\r
539                     boolean isPUT =  method.equalsIgnoreCase("PUT");\r
540                     if ( isPOST || isPUT ) {\r
541                         PartsStruct parts = PartsStruct.readParts(testNode, testID, xmlReplayBaseDir);\r
542                         if (Tools.notEmpty(parts.overrideTestID)) {\r
543                             testID = parts.overrideTestID;\r
544                         }\r
545                         if (isPOST){\r
546                             String csid = CSIDfromTestID(testNode, serviceResultsMap);\r
547                             if (Tools.notEmpty(csid)) uri = Tools.glue(uri, "/", csid+"/items/");\r
548                         } else if (isPUT) {\r
549                             uri = fromTestID(uri, testNode, serviceResultsMap);\r
550                         }\r
551                         if (parts.bDoingSinglePartPayload){\r
552                             serviceResult = XmlReplayTransport.doPOST_PUTFromXML(parts.singlePartPayloadFilename, protoHostPort, uri, method, XmlReplayTransport.APPLICATION_XML, evalStruct, authForTest, testIDLabel);\r
553                         } else {\r
554                             serviceResult = XmlReplayTransport.doPOST_PUTFromXML_Multipart(parts.filesList, parts.partsList, protoHostPort, uri, method, evalStruct, authForTest, testIDLabel);\r
555                         }\r
556                         results.add(serviceResult);\r
557                         if (isPOST){\r
558                             serviceResultsMap.put(testID, serviceResult);      //PUTs do not return a Location, so don't add PUTs to serviceResultsMap.\r
559                         }\r
560                         fullURL = fixupFullURL(fullURL, protoHostPort, uri);\r
561                     } else if (method.equalsIgnoreCase("DELETE")){\r
562                         String fromTestID = testNode.valueOf("fromTestID");\r
563                         ServiceResult pr = serviceResultsMap.get(fromTestID);\r
564                         if (pr!=null){\r
565                             serviceResult = XmlReplayTransport.doDELETE(pr.deleteURL, authForTest, testIDLabel, fromTestID);\r
566                             serviceResult.fromTestID = fromTestID;\r
567                             if (expectedCodes.size()>0){\r
568                                 serviceResult.expectedCodes = expectedCodes;\r
569                             }\r
570                             results.add(serviceResult);\r
571                             if (serviceResult.gotExpectedResult()){  //gotExpectedResult depends on serviceResult.expectedCodes.\r
572                                 serviceResultsMap.remove(fromTestID);\r
573                             }\r
574                         } else {\r
575                             if (Tools.notEmpty(fromTestID)){\r
576                                 serviceResult = new ServiceResult();\r
577                                 serviceResult.responseCode = 0;\r
578                                 serviceResult.error = "ID not found in element fromTestID: "+fromTestID;\r
579                                 System.err.println("****\r\nServiceResult: "+serviceResult.error+". SKIPPING TEST. Full URL: "+fullURL);\r
580                             } else {\r
581                                 serviceResult = XmlReplayTransport.doDELETE(fullURL, authForTest, testID, fromTestID);\r
582                             }\r
583                             serviceResult.fromTestID = fromTestID;\r
584                             results.add(serviceResult);\r
585                         }\r
586                     } else if (method.equalsIgnoreCase("GET")){\r
587                         fullURL = fromTestID(fullURL, testNode, serviceResultsMap);\r
588                         serviceResult = XmlReplayTransport.doGET(fullURL, authForTest, testIDLabel);\r
589                         results.add(serviceResult);\r
590                     } else if (method.equalsIgnoreCase("LIST")){\r
591                         fullURL = fixupFullURL(fullURL, protoHostPort, uri);\r
592                         String listQueryParams = ""; //TODO: empty for now, later may pick up from XML control file.\r
593                         serviceResult = XmlReplayTransport.doLIST(fullURL, listQueryParams, authForTest, testIDLabel);\r
594                         results.add(serviceResult);\r
595                     } else {\r
596                         throw new Exception("HTTP method not supported by XmlReplay: "+method);\r
597                     }\r
598 \r
599                     serviceResult.testID = testID;\r
600                     serviceResult.fullURL = fullURL;\r
601                     serviceResult.auth = authForTest;\r
602                     serviceResult.method = method;\r
603                     if (expectedCodes.size()>0){\r
604                         serviceResult.expectedCodes = expectedCodes;\r
605                     }\r
606                     if (Tools.isEmpty(serviceResult.testID)) serviceResult.testID = testIDLabel;\r
607                     if (Tools.isEmpty(serviceResult.testGroupID)) serviceResult.testGroupID = testGroupID;\r
608 \r
609                     Node expectedLevel = testNode.selectSingleNode("response/expected");\r
610                     if (expectedLevel!=null){\r
611                         String level = expectedLevel.valueOf("@level");\r
612                         serviceResult.payloadStrictness = level;\r
613                     }\r
614 \r
615                     String vError = validateResponse(serviceResult, serviceResultsMap, expectedResponseParts);\r
616                     if (Tools.notEmpty(vError)){\r
617                         serviceResult.error = vError;\r
618                         serviceResult.failureReason = " : VALIDATION ERROR; ";\r
619                     }\r
620 \r
621                     String serviceResultRow = serviceResult.dump(dump.dumpServiceResult);\r
622                     String leader = (dump.dumpServiceResult == ServiceResult.DUMP_OPTIONS.detailed) ? "XmlReplay:"+testIDLabel+": ": "";\r
623                     System.out.println(leader+serviceResultRow+"\r\n");\r
624                     if (dump.payloads) System.out.println(serviceResult.result);\r
625                 } catch (Throwable t) {\r
626                     System.out.println("ERROR: XmlReplay experienced an error in a test node: "+testNode+" Throwable: "+t);\r
627                 }\r
628             }\r
629             if (Tools.isTrue(autoDeletePOSTS)&&param_autoDeletePOSTS){\r
630                 autoDelete(serviceResultsMap, "default");\r
631             }\r
632         }\r
633         return results;\r
634     }\r
635 \r
636 \r
637     //======================== MAIN ===================================================================\r
638 \r
639     private static Options createOptions() {\r
640         Options options = new Options();\r
641         options.addOption("xmlReplayBaseDir", true, "default/basedir");\r
642         return options;\r
643     }\r
644 \r
645     public static String usage(){\r
646         String result = "org.collectionspace.services.IntegrationTests.xmlreplay.XmlReplay {args}\r\n"\r
647                         +"  -xmlReplayBaseDir <dir> \r\n"\r
648                         +" You may also override these with system args, e.g.: \r\n"\r
649                         +"   -DxmlReplayBaseDir=/path/to/dir \r\n"\r
650                         +" These may also be passed in via the POM.\r\n"\r
651                         +" You can also set these system args, e.g.: \r\n"\r
652                         +"  -DtestGroupID=<oneID> \r\n"\r
653                         +"  -DtestID=<one TestGroup ID>"\r
654                         +"  -DautoDeletePOSTS=<true|false> \r\n"\r
655                         +"    (note: -DautoDeletePOSTS won't force deletion if set to false in control file.";\r
656         return result;\r
657     }\r
658 \r
659     private static String opt(CommandLine line, String option){\r
660         String result;\r
661         String fromProps = System.getProperty(option);\r
662         if (Tools.notEmpty(fromProps)){\r
663             return fromProps;\r
664         }\r
665         result = line.getOptionValue(option);\r
666         if (result == null){\r
667             result = "";\r
668         }\r
669         return result;\r
670     }\r
671 \r
672     public static void main(String[]args) throws Exception {\r
673         Options options = createOptions();\r
674         //System.out.println("System CLASSPATH: "+prop.getProperty("java.class.path", null));\r
675         CommandLineParser parser = new GnuParser();\r
676         try {\r
677             // parse the command line arguments\r
678             CommandLine line = parser.parse(options, args);\r
679 \r
680             String xmlReplayBaseDir = opt(line, "xmlReplayBaseDir");\r
681             String testGroupID      = opt(line, "testGroupID");\r
682             String testID           = opt(line, "testID");\r
683             String autoDeletePOSTS  = opt(line, "autoDeletePOSTS");\r
684             String dumpResults      = opt(line, "dumpResults");\r
685             String controlFilename   = opt(line, "controlFilename");\r
686             String xmlReplayMaster  = opt(line, "xmlReplayMaster");\r
687 \r
688             xmlReplayBaseDir = Tools.fixFilename(xmlReplayBaseDir);\r
689             controlFilename = Tools.fixFilename(controlFilename);\r
690 \r
691             boolean bAutoDeletePOSTS = true;\r
692             if (Tools.notEmpty(autoDeletePOSTS)) {\r
693                 bAutoDeletePOSTS = Tools.isTrue(autoDeletePOSTS);\r
694             }\r
695             boolean bDumpResults = false;\r
696             if (Tools.notEmpty(dumpResults)) {\r
697                 bDumpResults = Tools.isTrue(autoDeletePOSTS);\r
698             }\r
699             if (Tools.isEmpty(xmlReplayBaseDir)){\r
700                 System.err.println("ERROR: xmlReplayBaseDir was not specified.");\r
701                 return;\r
702             }\r
703             File f = new File(Tools.glue(xmlReplayBaseDir, "/", controlFilename));\r
704             if (Tools.isEmpty(xmlReplayMaster) && !f.exists()){\r
705                 System.err.println("Control file not found: "+f.getCanonicalPath());\r
706                 return;\r
707             }\r
708             File fMaster = new File(Tools.glue(xmlReplayBaseDir, "/", xmlReplayMaster));\r
709             if (Tools.notEmpty(xmlReplayMaster)  && !fMaster.exists()){\r
710                 System.err.println("Master file not found: "+fMaster.getCanonicalPath());\r
711                 return;\r
712             }\r
713 \r
714             String xmlReplayBaseDirResolved = (new File(xmlReplayBaseDir)).getCanonicalPath();\r
715             System.out.println("XmlReplay ::"\r
716                             + "\r\n    xmlReplayBaseDir: "+xmlReplayBaseDir\r
717                             + "\r\n    xmlReplayBaseDir(resolved): "+xmlReplayBaseDirResolved\r
718                             + "\r\n    controlFilename: "+controlFilename\r
719                             + "\r\n    xmlReplayMaster: "+xmlReplayMaster\r
720                             + "\r\n    testGroupID: "+testGroupID\r
721                             + "\r\n    testID: "+testID\r
722                             + "\r\n    autoDeletePOSTS: "+bAutoDeletePOSTS\r
723                             + (Tools.notEmpty(xmlReplayMaster)\r
724                                        ? ("\r\n    will use master file: "+fMaster.getCanonicalPath())\r
725                                        : ("\r\n    will use control file: "+f.getCanonicalPath()) )\r
726                              );\r
727             \r
728             if (Tools.notEmpty(xmlReplayMaster)){\r
729                 if (Tools.notEmpty(controlFilename)){\r
730                     System.out.println("WARN: controlFilename: "+controlFilename+" will not be used because master was specified.  Running master: "+xmlReplayMaster);\r
731                 }\r
732                 XmlReplay replay = new XmlReplay(xmlReplayBaseDirResolved);\r
733                 replay.readOptionsFromMasterConfigFile(xmlReplayMaster);\r
734                 replay.setAutoDeletePOSTS(bAutoDeletePOSTS);\r
735                 Dump dumpFromMaster = replay.getDump();\r
736                 dumpFromMaster.payloads = Tools.isTrue(dumpResults);\r
737                 replay.setDump(dumpFromMaster);\r
738                 replay.runMaster(xmlReplayMaster, false); //false, because we already just read the options, and override a few.\r
739             } else {\r
740                 Dump dump = getDumpConfig();\r
741                 dump.payloads = Tools.isTrue(dumpResults);\r
742                 runXmlReplayFile(xmlReplayBaseDirResolved, controlFilename, testGroupID, testID, createResultsMap(), bAutoDeletePOSTS, dump, "", null);\r
743             }\r
744         } catch (ParseException exp) {\r
745             // oops, something went wrong\r
746             System.err.println("Cmd-line parsing failed.  Reason: " + exp.getMessage());\r
747             System.err.println(usage());\r
748         } catch (Exception e) {\r
749             System.out.println("Error : " + e.getMessage());\r
750             e.printStackTrace();\r
751         }\r
752     }\r
753 \r
754 }\r