9567
|
1 /** Copyright (C) 2017 by Bundesanstalt für Gewässerkunde |
|
2 * Software engineering by |
|
3 * Björnsen Beratende Ingenieure GmbH |
|
4 * Dr. Schumacher Ingenieurbüro für Wasser und Umwelt |
|
5 * |
|
6 * This file is Free Software under the GNU AGPL (>=v3) |
|
7 * and comes with ABSOLUTELY NO WARRANTY! Check out the |
|
8 * documentation coming with Dive4Elements River for details. |
|
9 */ |
|
10 package test; |
|
11 |
|
12 import java.io.File; |
|
13 import java.io.FileNotFoundException; |
|
14 import java.io.IOException; |
|
15 import java.io.InputStream; |
|
16 import java.net.URL; |
|
17 import java.util.HashMap; |
|
18 import java.util.Map; |
|
19 |
|
20 import org.apache.commons.io.FileUtils; |
|
21 import org.apache.commons.io.IOUtils; |
|
22 import org.dive4elements.artifacts.common.utils.ClientProtocolUtils; |
|
23 import org.dive4elements.artifacts.httpclient.http.HttpClient; |
|
24 import org.dive4elements.river.client.server.ChartServiceHelper; |
|
25 import org.junit.Assert; |
|
26 import org.w3c.dom.Document; |
|
27 |
|
28 /** |
|
29 * @author Domenico Nardi Tironi |
|
30 * |
|
31 */ |
|
32 |
|
33 public class ModuleRunnerOutput { |
|
34 |
|
35 private static final String IGNORE_ERSTELLDATUM = "# Datum der Erstellung"; |
|
36 private static final String IGNORE_FLYS_VERSION = "# FLYS-Version:"; |
|
37 private static final String IGNORE_BEARBEITER = "# Bearbeiter:"; |
|
38 |
|
39 private static final String LINE_SEPARATOR = System.getProperty("line.separator"); |
|
40 |
|
41 final FilenameSupplier filename; |
|
42 final String outputModeName; |
|
43 |
|
44 // private final String collectionIdentifier; |
|
45 |
|
46 private final Map<String, String> params; |
|
47 |
|
48 public final String getOutputModename() { |
|
49 return this.outputModeName; |
|
50 } |
|
51 |
|
52 public ModuleRunnerOutput(final FilenameSupplier filename, final String outputModeName, final String currentKm) { |
|
53 this.filename = filename; |
|
54 this.outputModeName = outputModeName; |
|
55 |
|
56 this.params = getChartExportAttributeMap(currentKm, outputModeName); |
|
57 } |
|
58 |
9570
|
59 protected final void assertAndWriteToFile(final HttpClient client, final String collectionId, final File testFolder) throws IOException { |
9567
|
60 |
|
61 final String type = "csv"; |
|
62 |
|
63 final String enc = "UTF-8";// req.getParameter("encoding"); |
|
64 |
|
65 final URL expectedResource = getClass().getResource(this.filename.getFilename()); |
|
66 final Document attr = ChartServiceHelper.getChartAttributes(this.params); |
|
67 |
|
68 final Document request = ClientProtocolUtils.newOutCollectionDocument(collectionId, this.outputModeName, type, attr); |
|
69 |
|
70 final InputStream response = client.collectionOut(request, collectionId, this.outputModeName); |
|
71 |
|
72 final String actual = deleteErstelldatum(IOUtils.toString(response, "UTF-8")); |
|
73 |
|
74 final String expected = deleteErstelldatum(FileUtils.readFileToString(new File(expectedResource.getFile()), enc)); |
|
75 |
9570
|
76 final File fileTmp = doGetWriteToDisk(actual, testFolder); |
|
77 Assert.assertEquals(expected, actual); |
9567
|
78 |
9570
|
79 fileTmp.delete(); // if assert fails, file won't be deleted |
9567
|
80 |
|
81 } |
|
82 |
9570
|
83 public final File doGetWriteToDisk(final String actual_text, final File testFolder) throws FileNotFoundException, IOException { |
|
84 |
|
85 final File file = new File(testFolder, this.filename.getFilename()); |
|
86 new File(file.getParent()).mkdir(); |
9567
|
87 |
|
88 // final String name = mode; |
9570
|
89 // final String type = "csv"; |
9567
|
90 |
9570
|
91 FileUtils.write(file, actual_text, "UTF-8"); |
|
92 return file; |
9567
|
93 } |
|
94 |
|
95 private final String deleteErstelldatum(final String input) { |
|
96 String result = ""; |
|
97 final String[] lines = input.split(LINE_SEPARATOR); |
|
98 for (final String line : lines) { |
|
99 if (!line.contains(IGNORE_ERSTELLDATUM) && !line.contains(IGNORE_FLYS_VERSION) && !line.contains(IGNORE_BEARBEITER)) { |
|
100 result = result + line + LINE_SEPARATOR; |
|
101 } |
|
102 } |
|
103 return result; |
|
104 } |
|
105 |
|
106 private final Map<String, String> getChartExportAttributeMap(final String currentKm, final String outputmodeName) { |
|
107 |
|
108 // From ChartOutputServiceImpl |
|
109 final Map<String, String> attr = new HashMap<String, String>(); |
|
110 |
|
111 attr.put("width", "-1"); // req.getParameter("width")); |
|
112 attr.put("height", "-1"); // req.getParameter("height")); |
|
113 // attr.put("minx", null);// req.getParameter("minx")); |
|
114 // attr.put("maxx", req.getParameter("maxx")); |
|
115 // attr.put("miny", req.getParameter("miny")); |
|
116 // attr.put("maxy", req.getParameter("maxy")); |
|
117 attr.put("format", "csv"); // req.getParameter("format")); |
|
118 attr.put("export", "true");// req.getParameter("export")); |
|
119 |
|
120 if (currentKm != null) |
|
121 attr.put("km", currentKm);// req.getParameter("currentKm")); |
|
122 // encoding? |
|
123 |
|
124 // Weiteres (nicht mehr von CHartOutputServiceImpl, sondern von URL-Parametern): |
|
125 attr.put("type", outputmodeName); |
|
126 attr.put("locale", "de"); |
|
127 attr.put("timestamp", String.valueOf(System.currentTimeMillis())); |
|
128 attr.put("encoding", "windows-1252"); |
|
129 |
|
130 return attr; |
|
131 } |
|
132 |
|
133 } |