| /** |
| * Copyright (c) 2008-2014, http://www.snakeyaml.org |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| package examples.collections; |
| |
| import java.util.LinkedHashMap; |
| import java.util.Map; |
| |
| import junit.framework.TestCase; |
| |
| import org.yaml.snakeyaml.Util; |
| import org.yaml.snakeyaml.Yaml; |
| |
| /** |
| * Test MapBean->Map<String, Developer> developers <br/> |
| * Developer class must be properly recognised |
| */ |
| public class TypeSafeMapTest extends TestCase { |
| public void testDumpMap() { |
| MapBean bean = new MapBean(); |
| Map<String, Integer> data = new LinkedHashMap<String, Integer>(); |
| data.put("aaa", 1); |
| data.put("bbb", 2); |
| data.put("zzz", 3); |
| bean.setData(data); |
| Map<String, Developer2> developers = new LinkedHashMap<String, Developer2>(); |
| developers.put("team1", new Developer2("Fred", "creator")); |
| developers.put("team2", new Developer2("John", "committer")); |
| bean.setDevelopers(developers); |
| Yaml yaml = new Yaml(); |
| String output = yaml.dumpAsMap(bean); |
| // System.out.println(output); |
| String etalon = Util.getLocalResource("examples/map-bean-10.yaml"); |
| assertEquals(etalon, output); |
| } |
| |
| public void testDumpMap2() { |
| MapBean bean = new MapBean(); |
| Map<String, Integer> data = new LinkedHashMap<String, Integer>(); |
| data.put("aaa", 1); |
| data.put("bbb", 2); |
| bean.setData(data); |
| Map<String, Developer2> developers = new LinkedHashMap<String, Developer2>(); |
| developers.put("team1", new Developer2("Fred", "creator")); |
| developers.put("team2", new Developer2("John", "committer")); |
| developers.put("team3", new Developer222("Bill", "head")); |
| bean.setDevelopers(developers); |
| Yaml yaml = new Yaml(); |
| String output = yaml.dumpAsMap(bean); |
| // System.out.println(output); |
| String etalon = Util.getLocalResource("examples/map-bean-11.yaml"); |
| assertEquals(etalon, output); |
| } |
| |
| public void testLoadMap() { |
| String output = Util.getLocalResource("examples/map-bean-10.yaml"); |
| // System.out.println(output); |
| Yaml beanLoader = new Yaml(); |
| MapBean parsed = beanLoader.loadAs(output, MapBean.class); |
| assertNotNull(parsed); |
| Map<String, Integer> data = parsed.getData(); |
| assertEquals(3, data.size()); |
| assertEquals(new Integer(1), data.get("aaa")); |
| assertEquals(new Integer(2), data.get("bbb")); |
| assertEquals(new Integer(3), data.get("zzz")); |
| Map<String, Developer2> developers = parsed.getDevelopers(); |
| assertEquals(2, developers.size()); |
| assertEquals("Developer must be recognised.", Developer2.class, developers.get("team1") |
| .getClass()); |
| Developer2 fred = developers.get("team1"); |
| assertEquals("Fred", fred.getName()); |
| assertEquals("creator", fred.getRole()); |
| } |
| |
| public static class MapBean { |
| private Map<String, Integer> data; |
| private String name; |
| private Map<String, Developer2> developers; |
| |
| public MapBean() { |
| name = "Bean123"; |
| } |
| |
| public Map<String, Integer> getData() { |
| return data; |
| } |
| |
| public void setData(Map<String, Integer> data) { |
| this.data = data; |
| } |
| |
| public String getName() { |
| return name; |
| } |
| |
| public void setName(String name) { |
| this.name = name; |
| } |
| |
| public Map<String, Developer2> getDevelopers() { |
| return developers; |
| } |
| |
| public void setDevelopers(Map<String, Developer2> developers) { |
| this.developers = developers; |
| } |
| } |
| |
| public static class Developer2 { |
| private String name; |
| private String role; |
| |
| public Developer2() { |
| } |
| |
| public Developer2(String name, String role) { |
| this.name = name; |
| this.role = role; |
| } |
| |
| public String getName() { |
| return name; |
| } |
| |
| public void setName(String name) { |
| this.name = name; |
| } |
| |
| public String getRole() { |
| return role; |
| } |
| |
| public void setRole(String role) { |
| this.role = role; |
| } |
| } |
| |
| public static class Developer222 extends Developer2 { |
| public Developer222() { |
| super(); |
| } |
| |
| public Developer222(String name, String role) { |
| super(name, role); |
| } |
| } |
| |
| /* |
| * No generic collection |
| */ |
| @SuppressWarnings({ "unchecked", "rawtypes" }) |
| public void testLoadMapWithObject() { |
| String output = Util.getLocalResource("examples/map-bean-10.yaml"); |
| // System.out.println(output); |
| Yaml beanLoader = new Yaml(); |
| MapBeanNoGenerics parsed = beanLoader.loadAs(output, MapBeanNoGenerics.class); |
| assertNotNull(parsed); |
| Map<String, Integer> data = parsed.getData(); |
| assertEquals(3, data.size()); |
| assertEquals(new Integer(1), data.get("aaa")); |
| assertEquals(new Integer(2), data.get("bbb")); |
| assertEquals(new Integer(3), data.get("zzz")); |
| Map developers = parsed.getDevelopers(); |
| assertNotNull(developers); |
| assertEquals(2, developers.size()); |
| Object o1 = developers.get("team1"); |
| // because of erasure we get simply Map |
| Map<String, String> developer = (Map<String, String>) o1; |
| assertEquals("Fred", developer.get("name")); |
| assertEquals("creator", developer.get("role")); |
| } |
| |
| @SuppressWarnings("rawtypes") |
| public static class MapBeanNoGenerics { |
| private Map data; |
| private String name; |
| private Map developers; |
| |
| public MapBeanNoGenerics() { |
| name = "Bean123"; |
| } |
| |
| public Map getData() { |
| return data; |
| } |
| |
| public void setData(Map data) { |
| this.data = data; |
| } |
| |
| public String getName() { |
| return name; |
| } |
| |
| public void setName(String name) { |
| this.name = name; |
| } |
| |
| public Map getDevelopers() { |
| return developers; |
| } |
| |
| public void setDevelopers(Map developers) { |
| this.developers = developers; |
| } |
| } |
| } |