source: josm/trunk/test/functional/org/openstreetmap/josm/io/MultiFetchServerObjectReaderTest.java@ 6995

Last change on this file since 6995 was 6995, checked in by Don-vip, 10 years ago

fix #9906 - fix reliance on default encoding
File size: 13.6 KB
Line 
1// License: GPL. For details, see LICENSE file.
2package org.openstreetmap.josm.io;
3
4import static org.junit.Assert.assertEquals;
5import static org.junit.Assert.assertFalse;
6import static org.junit.Assert.assertNotNull;
7import static org.junit.Assert.assertTrue;
8import static org.junit.Assert.fail;
9
10import java.io.File;
11import java.io.FileInputStream;
12import java.io.FileOutputStream;
13import java.io.IOException;
14import java.io.InputStream;
15import java.io.OutputStreamWriter;
16import java.io.PrintWriter;
17import java.text.MessageFormat;
18import java.util.ArrayList;
19import java.util.Properties;
20import java.util.logging.Level;
21import java.util.logging.Logger;
22
23import org.junit.Before;
24import org.junit.BeforeClass;
25import org.junit.Test;
26import org.openstreetmap.josm.Main;
27import org.openstreetmap.josm.data.coor.LatLon;
28import org.openstreetmap.josm.data.osm.Changeset;
29import org.openstreetmap.josm.data.osm.DataSet;
30import org.openstreetmap.josm.data.osm.Node;
31import org.openstreetmap.josm.data.osm.OsmPrimitive;
32import org.openstreetmap.josm.data.osm.Relation;
33import org.openstreetmap.josm.data.osm.RelationMember;
34import org.openstreetmap.josm.data.osm.Way;
35import org.openstreetmap.josm.data.projection.Projections;
36import org.openstreetmap.josm.gui.io.UploadStrategy;
37import org.openstreetmap.josm.gui.io.UploadStrategySpecification;
38import org.openstreetmap.josm.gui.progress.NullProgressMonitor;
39import org.openstreetmap.josm.tools.Utils;
40
41public class MultiFetchServerObjectReaderTest {
42 private static Logger logger = Logger.getLogger(MultiFetchServerObjectReader.class.getName());
43
44 /**
45 * builds a large data set to be used later for testing MULTI FETCH on the server
46 *
47 * @return a large data set
48 */
49 protected static DataSet buildTestDataSet() {
50 DataSet ds = new DataSet();
51 ds.setVersion("0.6");
52
53 int numNodes = 1000;
54 int numWays = 1000;
55 int numRelations = 1000;
56
57 ArrayList<Node> nodes = new ArrayList<Node>();
58 ArrayList<Way> ways = new ArrayList<Way>();
59
60 // create a set of nodes
61 //
62 for (int i=0; i< numNodes; i++) {
63 Node n = new Node();
64 n.setCoor(new LatLon(-36.6,47.6));
65 n.put("name", "node-"+i);
66 ds.addPrimitive(n);
67 nodes.add(n);
68 }
69
70 // create a set of ways, each with a random number of
71 // nodes
72 //
73 for (int i=0; i< numWays; i++) {
74 Way w = new Way();
75 int numNodesInWay = 2 + (int)Math.round(Math.random() * 5);
76 int start = (int)Math.round(Math.random() * numNodes);
77 for (int j = 0; j < numNodesInWay;j++) {
78 int idx = (start + j) % numNodes;
79 Node n = nodes.get(idx);
80 w.addNode(n);
81 }
82 w.put("name", "way-"+i);
83 ds.addPrimitive(w);
84 ways.add(w);
85 }
86
87 // create a set of relations each with a random number of nodes,
88 // and ways
89 //
90 for (int i=0; i< numRelations; i++) {
91 Relation r = new Relation();
92 r.put("name", "relation-" +i);
93 int numNodesInRelation = (int)Math.round(Math.random() * 10);
94 int start = (int)Math.round(Math.random() * numNodes);
95 for (int j = 0; j < numNodesInRelation;j++) {
96 int idx = (start + j) % 500;
97 Node n = nodes.get(idx);
98 r.addMember(new RelationMember("role-" + j, n));
99 }
100 int numWaysInRelation = (int)Math.round(Math.random() * 10);
101 start = (int)Math.round(Math.random() * numWays);
102 for (int j = 0; j < numWaysInRelation;j++) {
103 int idx = (start + j) % 500;
104 Way w = ways.get(idx);
105 r.addMember(new RelationMember("role-" + j, w));
106 }
107 ds.addPrimitive(r);
108 }
109
110 return ds;
111 }
112
113 static public DataSet testDataSet;
114 static public Properties testProperties;
115
116 /**
117 * creates the dataset on the server.
118 *
119 * @param ds the data set
120 * @throws OsmTransferException
121 */
122 static public void createDataSetOnServer(DataSet ds) throws OsmTransferException {
123 logger.info("creating data set on the server ...");
124 ArrayList<OsmPrimitive> primitives = new ArrayList<OsmPrimitive>();
125 primitives.addAll(testDataSet.getNodes());
126 primitives.addAll(testDataSet.getWays());
127 primitives.addAll(testDataSet.getRelations());
128
129 OsmServerWriter writer = new OsmServerWriter();
130 Changeset cs = new Changeset();
131 writer.uploadOsm(new UploadStrategySpecification().setStrategy(UploadStrategy.SINGLE_REQUEST_STRATEGY), primitives,cs,NullProgressMonitor.INSTANCE);
132 OsmApi.getOsmApi().closeChangeset(cs, NullProgressMonitor.INSTANCE);
133 }
134
135 @BeforeClass
136 public static void init() throws OsmTransferException {
137 logger.info("initializing ...");
138 testProperties = new Properties();
139
140 // load properties
141 //
142 try {
143 InputStream is = MultiFetchServerObjectReaderTest.class.getResourceAsStream("/test-functional-env.properties");
144 try {
145 testProperties.load(is);
146 } finally {
147 is.close();
148 }
149 } catch(Exception e){
150 logger.log(Level.SEVERE, MessageFormat.format("failed to load property file ''{0}''", "test-functional-env.properties"));
151 fail(MessageFormat.format("failed to load property file ''{0}''", "test-functional-env.properties"));
152 }
153
154 // check josm.home
155 //
156 String josmHome = testProperties.getProperty("josm.home");
157 if (josmHome == null) {
158 fail(MessageFormat.format("property ''{0}'' not set in test environment", "josm.home"));
159 } else {
160 File f = new File(josmHome);
161 if (! f.exists() || ! f.canRead()) {
162 fail(MessageFormat.format("property ''{0}'' points to ''{1}'' which is either not existing or not readable", "josm.home", josmHome));
163 }
164 }
165
166 // check temp output dir
167 //
168 String tempOutputDir = testProperties.getProperty("test.functional.tempdir");
169 if (tempOutputDir == null) {
170 fail(MessageFormat.format("property ''{0}'' not set in test environment", "test.functional.tempdir"));
171 } else {
172 File f = new File(tempOutputDir);
173 if (! f.exists() || ! f.isDirectory() || ! f.canWrite()) {
174 fail(MessageFormat.format("property ''{0}'' points to ''{1}'' which is either not existing, not a directory, or not writeable", "test.functional.tempdir", tempOutputDir));
175 }
176 }
177
178
179 // init preferences
180 //
181 System.setProperty("josm.home", josmHome);
182 Main.pref.init(false);
183 // don't use atomic upload, the test API server can't cope with large diff uploads
184 //
185 Main.pref.put("osm-server.atomic-upload", false);
186 Main.setProjection(Projections.getProjectionByCode("EPSG:3857")); // Mercator
187
188 File dataSetCacheOutputFile = new File(tempOutputDir, MultiFetchServerObjectReaderTest.class.getName() + ".dataset");
189
190 // make sure we don't upload to production
191 //
192 String url = OsmApi.getOsmApi().getBaseUrl().toLowerCase().trim();
193 if (url.startsWith("http://www.openstreetmap.org") || url.startsWith("http://api.openstreetmap.org")
194 || url.startsWith("https://www.openstreetmap.org") || url.startsWith("https://api.openstreetmap.org")) {
195 fail(MessageFormat.format("configured url ''{0}'' seems to be a productive url, aborting.", url));
196 }
197
198 String p = System.getProperties().getProperty("useCachedDataset");
199 if (p != null && Boolean.parseBoolean(p.trim().toLowerCase())) {
200 logger.info(MessageFormat.format("property ''{0}'' set, using cached dataset", "useCachedDataset"));
201 return;
202 }
203
204 logger.info(MessageFormat.format("property ''{0}'' not set to true, creating test dataset on the server. property is ''{1}''", "useCachedDataset", p));
205
206 // build and upload the test data set
207 //
208 logger.info("creating test data set ....");
209 testDataSet = buildTestDataSet();
210 logger.info("uploading test data set ...");
211 createDataSetOnServer(testDataSet);
212
213 try {
214 PrintWriter pw = new PrintWriter(
215 new OutputStreamWriter(new FileOutputStream(dataSetCacheOutputFile), Utils.UTF_8)
216 );
217 logger.info(MessageFormat.format("caching test data set in ''{0}'' ...", dataSetCacheOutputFile.toString()));
218 OsmWriter w = new OsmWriter(pw, false, testDataSet.getVersion());
219 w.header();
220 w.writeDataSources(testDataSet);
221 w.writeContent(testDataSet);
222 w.footer();
223 w.close();
224 pw.close();
225 } catch(IOException e) {
226 fail(MessageFormat.format("failed to open file ''{0}'' for writing", dataSetCacheOutputFile.toString()));
227 }
228 }
229
230 private DataSet ds;
231
232 /**
233 * Setup test.
234 */
235 @Before
236 public void setUp() throws IOException, IllegalDataException {
237 File f = new File(testProperties.getProperty("test.functional.tempdir"), MultiFetchServerObjectReaderTest.class.getName() + ".dataset");
238 logger.info(MessageFormat.format("reading cached dataset ''{0}''", f.toString()));
239 ds = new DataSet();
240 FileInputStream fis = new FileInputStream(f);
241 ds = OsmReader.parseDataSet(fis, NullProgressMonitor.INSTANCE);
242 fis.close();
243 }
244
245 @Test
246 public void testMultiGet10Nodes() throws OsmTransferException {
247 MultiFetchServerObjectReader reader = new MultiFetchServerObjectReader();
248 ArrayList<Node> nodes = new ArrayList<Node>(ds.getNodes());
249 for (int i =0; i< 10; i++) {
250 reader.append(nodes.get(i));
251 }
252 DataSet out = reader.parseOsm(NullProgressMonitor.INSTANCE);
253 assertEquals(10, out.getNodes().size());
254 for (Node n1:out.getNodes()) {
255 Node n2 = (Node)ds.getPrimitiveById(n1);
256 assertNotNull(n2);
257 assertEquals(n2.get("name"),n2.get("name"));
258 }
259 assertTrue(reader.getMissingPrimitives().isEmpty());
260 }
261
262 @Test
263 public void testMultiGet10Ways() throws OsmTransferException {
264 MultiFetchServerObjectReader reader = new MultiFetchServerObjectReader();
265 ArrayList<Way> ways= new ArrayList<Way>(ds.getWays());
266 for (int i =0; i< 10; i++) {
267 reader.append(ways.get(i));
268 }
269 DataSet out = reader.parseOsm(NullProgressMonitor.INSTANCE);
270 assertEquals(10, out.getWays().size());
271 for (Way w1: out.getWays()) {
272 Way w2 = (Way)ds.getPrimitiveById(w1);
273 assertNotNull(w2);
274 assertEquals(w2.getNodesCount(), w1.getNodesCount());
275 assertEquals(w2.get("name"),w1.get("name"));
276 }
277 assertTrue(reader.getMissingPrimitives().isEmpty());
278 }
279
280 @Test
281 public void testMultiGet10Relations() throws OsmTransferException {
282 MultiFetchServerObjectReader reader = new MultiFetchServerObjectReader();
283 ArrayList<Relation> relations= new ArrayList<Relation>(ds.getRelations());
284 for (int i =0; i< 10; i++) {
285 reader.append(relations.get(i));
286 }
287 DataSet out = reader.parseOsm(NullProgressMonitor.INSTANCE);
288 assertEquals(10, out.getRelations().size());
289 for (Relation r1: out.getRelations()) {
290 Relation r2 = (Relation)ds.getPrimitiveById(r1);
291 assertNotNull(r2);
292 assertEquals(r2.getMembersCount(), r1.getMembersCount());
293 assertEquals(r2.get("name"),r2.get("name"));
294 }
295 assertTrue(reader.getMissingPrimitives().isEmpty());
296 }
297
298 @Test
299 public void testMultiGet800Nodes() throws OsmTransferException {
300 MultiFetchServerObjectReader reader = new MultiFetchServerObjectReader();
301 ArrayList<Node> nodes = new ArrayList<Node>(ds.getNodes());
302 for (int i =0; i< 812; i++) {
303 reader.append(nodes.get(i));
304 }
305 DataSet out = reader.parseOsm(NullProgressMonitor.INSTANCE);
306 assertEquals(812, out.getNodes().size());
307 for (Node n1:out.getNodes()) {
308 Node n2 = (Node)ds.getPrimitiveById(n1);
309 assertNotNull(n2);
310 assertEquals(n2.get("name"),n2.get("name"));
311 }
312 assertTrue(reader.getMissingPrimitives().isEmpty());
313 }
314
315 @Test
316 public void multiGetWithNonExistingNode() throws OsmTransferException {
317 MultiFetchServerObjectReader reader = new MultiFetchServerObjectReader();
318 ArrayList<Node> nodes = new ArrayList<Node>(ds.getNodes());
319 for (int i =0; i< 10; i++) {
320 reader.append(nodes.get(i));
321 }
322 Node n = new Node(9999999);
323 reader.append(n); // doesn't exist
324 DataSet out = reader.parseOsm(NullProgressMonitor.INSTANCE);
325 assertEquals(10, out.getNodes().size());
326 for (Node n1:out.getNodes()) {
327 Node n2 = (Node)ds.getPrimitiveById(n1);
328 assertNotNull(n2);
329 assertEquals(n2.get("name"),n2.get("name"));
330 }
331 assertFalse(reader.getMissingPrimitives().isEmpty());
332 assertEquals(1, reader.getMissingPrimitives().size());
333 assertEquals(9999999, reader.getMissingPrimitives().iterator().next().getUniqueId());
334 }
335}
Note: See TracBrowser for help on using the repository browser.