1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16 package org.yaml.snakeyaml.constructor;
17
18 import java.util.LinkedHashMap;
19 import java.util.Map;
20 import java.util.TreeMap;
21
22 import junit.framework.TestCase;
23
24 import org.yaml.snakeyaml.composer.Composer;
25 import org.yaml.snakeyaml.parser.Parser;
26 import org.yaml.snakeyaml.parser.ParserImpl;
27 import org.yaml.snakeyaml.reader.StreamReader;
28 import org.yaml.snakeyaml.resolver.Resolver;
29
30 public class ConstructorMappingTest extends TestCase {
31
32 @SuppressWarnings("unchecked")
33 public void testGetDefaultMap() {
34 String data = "{ one: 1, two: 2, three: 3 }";
35 Map<Object, Object> map = (Map<Object, Object>) construct(new CustomConstructor(), data);
36 assertNotNull(map);
37 assertTrue(map.getClass().toString(), map instanceof TreeMap);
38 }
39
40 @SuppressWarnings("unchecked")
41 public void testGetArrayList() {
42 String data = "{ one: 1, two: 2, three: 3 }";
43 Map<Object, Object> map = (Map<Object, Object>) construct(data);
44 assertNotNull(map);
45 assertTrue(map.getClass().toString(), map instanceof LinkedHashMap);
46 }
47
48 private Object construct(String data) {
49 return construct(new Constructor(), data);
50 }
51
52 private Object construct(Constructor constructor, String data) {
53 StreamReader reader = new StreamReader(data);
54 Parser parser = new ParserImpl(reader);
55 Resolver resolver = new Resolver();
56 Composer composer = new Composer(parser, resolver);
57 constructor.setComposer(composer);
58 return constructor.getSingleData(Object.class);
59 }
60
61 class CustomConstructor extends Constructor {
62 @Override
63 protected Map<Object, Object> createDefaultMap() {
64 return new TreeMap<Object, Object>();
65 }
66 }
67 }