1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17 package examples;
18
19 import java.io.IOException;
20 import java.util.Map;
21 import java.util.TreeMap;
22
23 import junit.framework.TestCase;
24
25 import org.yaml.snakeyaml.Yaml;
26 import org.yaml.snakeyaml.constructor.Constructor;
27
28 public class CustomMapExampleTest extends TestCase {
29 public void testMap() throws IOException {
30 Yaml yaml = new Yaml(new CustomConstructor());
31 @SuppressWarnings("unchecked")
32 Map<Integer, String> data = (Map<Integer, String>) yaml
33 .load("{2: '222', 1: '111', 3: '333'}");
34 assertTrue(data instanceof TreeMap);
35 Object[] keys = data.keySet().toArray();
36
37 assertEquals(new Integer(1), keys[0]);
38 assertEquals(new Integer(2), keys[1]);
39 assertEquals(new Integer(3), keys[2]);
40 }
41
42 class CustomConstructor extends Constructor {
43 @Override
44 protected Map<Object, Object> createDefaultMap() {
45 return new TreeMap<Object, Object>();
46 }
47 }
48 }