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