View Javadoc

1   /**
2    * Copyright (c) 2008-2012, http://www.snakeyaml.org
3    *
4    * Licensed under the Apache License, Version 2.0 (the "License");
5    * you may not use this file except in compliance with the License.
6    * You may obtain a copy of the License at
7    *
8    *     http://www.apache.org/licenses/LICENSE-2.0
9    *
10   * Unless required by applicable law or agreed to in writing, software
11   * distributed under the License is distributed on an "AS IS" BASIS,
12   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13   * See the License for the specific language governing permissions and
14   * limitations under the License.
15   */
16  package org.yaml.snakeyaml.issues.issue100;
17  
18  import java.util.HashSet;
19  import java.util.List;
20  import java.util.Map;
21  import java.util.Set;
22  
23  import junit.framework.TestCase;
24  
25  import org.yaml.snakeyaml.Util;
26  import org.yaml.snakeyaml.Yaml;
27  import org.yaml.snakeyaml.constructor.Constructor;
28  
29  public class MergeJavaBeanTest extends TestCase {
30  
31      @SuppressWarnings("unchecked")
32      public void testNoMerge() {
33          String input = "- &id001 !!org.yaml.snakeyaml.issues.issue100.Data {age: 11, id: id123}\n- *id001";
34          // System.out.println(input);
35          Yaml yaml = new Yaml();
36          List<Data> list = (List<Data>) yaml.load(input);
37          for (Data data : list) {
38              // System.out.println(data);
39              assertEquals("id123", data.getId());
40              assertEquals(11, data.getAge());
41          }
42      }
43  
44      public void testMergeWithTags() {
45          String input = Util.getLocalResource("issues/issue100-1.yaml");
46          // System.out.println(input);
47          Yaml yaml = new Yaml();
48  
49          List<?> list = (List<?>) yaml.load(input);
50          // First object: Data ( 11, "id123" )
51          assertEquals(list.get(0).getClass(), Data.class);
52          assertEquals(11, ((Data) list.get(0)).getAge());
53          assertEquals("id123", ((Data) list.get(0)).getId());
54  
55          // Second object: Data ( 13, "id456" )
56          assertEquals(list.get(1).getClass(), Data.class);
57          assertEquals(13, ((Data) list.get(1)).getAge());
58          assertEquals("id456", ((Data) list.get(1)).getId());
59  
60          // Third object: Data ( 11, "id789" )
61          assertEquals(list.get(2).getClass(), Data.class);
62          assertEquals(11, ((Data) list.get(2)).getAge());
63          assertEquals("id789", ((Data) list.get(2)).getId());
64  
65          // 4th object: DataMore ( 30, "id123" )
66          assertEquals(list.get(3).getClass(), DataMore.class);
67          assertEquals(30, ((DataMore) list.get(3)).getAge());
68          assertEquals("id123", ((DataMore) list.get(3)).getId());
69          assertTrue(((DataMore) list.get(3)).isComplete());
70  
71          // 5th object: Map ( age: 100 )
72          assertTrue(list.get(4) instanceof Map);
73          assertEquals(1, ((Map<?, ?>) list.get(4)).size());
74          assertTrue(((Map<?, ?>) list.get(4)).containsKey("age"));
75          assertEquals(100, ((Map<?, ?>) list.get(4)).get("age"));
76  
77          // 6th object: DataMore ( 100, "id789" )
78          assertEquals(list.get(5).getClass(), DataMore.class);
79          assertEquals(100, ((DataMore) list.get(5)).getAge());
80          assertEquals("id789", ((DataMore) list.get(5)).getId());
81          assertFalse(((DataMore) list.get(5)).isComplete());
82          // All instances except the last Map must be different Data instances
83          Set<Data> dataSet = new HashSet<Data>();
84          for (Object data : list) {
85              if (data instanceof Data) {
86                  dataSet.add((Data) data);
87              }
88          }
89          assertEquals("Must be all but one Data instances.", list.size() - 1, dataSet.size());
90      }
91  
92      /**
93       * Since to explicit tag is provided JavaBean properties are used to create
94       * a map
95       */
96      @SuppressWarnings("unchecked")
97      public void testMergeBeanToMap() {
98          String input = "- &id001 !!org.yaml.snakeyaml.issues.issue100.Data {age: 11, id: id123}\n- << : *id001";
99          // System.out.println(input);
100         Yaml yaml = new Yaml();
101         List<Object> list = (List<Object>) yaml.load(input);
102         // First object: Data ( 11, "id123" )
103         Data first = (Data) list.get(0);
104         assertEquals(11, first.getAge());
105         assertEquals("id123", first.getId());
106         // Second object is map
107         Map<?, ?> second = (Map<?, ?>) list.get(1);
108         assertEquals(11, second.get("age"));
109         assertEquals("id123", second.get("id"));
110     }
111 
112     @SuppressWarnings("unchecked")
113     public void testMergeAndDeviate() {
114         String input = "- &id001 !!org.yaml.snakeyaml.issues.issue100.Data {age: 11, id: id123}\n- <<: *id001\n  id: id456";
115         // System.out.println(input);
116         Yaml yaml = new Yaml();
117         List<Object> list = (List<Object>) yaml.load(input);
118         // First object: Data ( 11, "id123" )
119         Data first = (Data) list.get(0);
120         assertEquals(11, first.getAge());
121         assertEquals("id123", first.getId());
122         // Second object is map with a diffrent id
123         Map<?, ?> second = (Map<?, ?>) list.get(1);
124         assertEquals(11, second.get("age"));
125         assertEquals("id456", second.get("id"));
126     }
127 
128     /**
129      * <pre>
130      * Test that the merge-override works correctly in the case of custom classes / data types.
131      * Two base objects are specified:
132      *    id001 refers to Data ( 11, "id123" )
133      *    id002 refers to Data ( 37, null )  with a literal null value
134      * The third object is specified as having the properties of id001, with the properties of
135      * id002 overriding where they conflict.
136      *    Data ( 37, "id123" )
137      * </pre>
138      */
139     @SuppressWarnings("unchecked")
140     public void testMergeAndDeviateOverride() {
141         String input = "- &id001 !!org.yaml.snakeyaml.issues.issue100.Data {age: 11, id: id123}\n- &id002 !!org.yaml.snakeyaml.issues.issue100.Data {age: 37}\n- <<: [ *id002, *id001 ]";
142         // System.out.println(input);
143         Yaml yaml = new Yaml();
144         List<Data> list = (List<Data>) yaml.load(input);
145 
146         // First object: Data ( 11, "id123" )
147         assertEquals(11, list.get(0).getAge());
148         assertEquals("id123", list.get(0).getId());
149 
150         // Second object: Data ( 37, null )
151         assertEquals(37, list.get(1).getAge());
152         assertEquals(null, list.get(1).getId());
153 
154         // Third object: map
155         Map<?, ?> third = (Map<?, ?>) list.get(2);
156         assertEquals(37, third.get("age"));
157         assertEquals("id123", third.get("id"));
158     }
159 
160     /**
161      * When the merged JavaBean is itself a JavaBean property then explicit tag
162      * is not required
163      */
164     public void testMergeBeanProperty() {
165         String input = Util.getLocalResource("issues/issue100-3.yaml");
166         // System.out.println(input);
167         Yaml yaml = new Yaml(new Constructor(DataBean.class));
168         DataBean bean = (DataBean) yaml.load(input);
169         assertEquals("id001", bean.getId());
170         assertEquals("id002", bean.getData().getId());
171         assertEquals(17, bean.getData().getAge());
172         assertEquals(17, bean.getMore().getAge());
173         assertEquals("more003", bean.getMore().getId());
174         assertTrue(bean.getMore().isComplete());
175     }
176 
177     /**
178      * Merge map to JavaBean
179      */
180     @SuppressWarnings("unchecked")
181     public void testMergeMapToJavaBean() {
182         String input = "- &id001 { age: 11, id: id123 }\n- !!org.yaml.snakeyaml.issues.issue100.Data\n  <<: *id001\n  id: id456";
183         // System.out.println(input);
184         Yaml yaml = new Yaml(new Constructor());
185         List<Object> objects = (List<Object>) yaml.load(input);
186         assertEquals(2, objects.size());
187         // Check first type
188         Object first = objects.get(0);
189         Map<Object, Object> firstMap = (Map<Object, Object>) first;
190         // Check first contents
191         assertEquals(11, firstMap.get("age"));
192         assertEquals("id123", firstMap.get("id"));
193         // Check second contents
194         Data secondData = (Data) objects.get(1);
195         assertEquals(11, secondData.getAge());
196         assertEquals("id456", secondData.getId());
197     }
198 }