| /** |
| * 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.Iterator; |
| 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<Enum, Developer> developers <br/> |
| * Developer class must be properly recognised |
| */ |
| public class TypeSafeMap2Test extends TestCase { |
| public void testDumpMap() { |
| MapBean2 bean = new MapBean2(); |
| Map<Developer2, Color> data = new LinkedHashMap<Developer2, Color>(); |
| data.put(new Developer2("Andy", "tester"), Color.BLACK); |
| data.put(new Developer2("Lisa", "owner"), Color.RED); |
| bean.setData(data); |
| Map<Color, Developer2> developers = new LinkedHashMap<Color, Developer2>(); |
| developers.put(Color.WHITE, new Developer2("Fred", "creator")); |
| developers.put(Color.BLACK, 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-12.yaml"); |
| assertEquals(etalon, output); |
| } |
| |
| public void testMap2() { |
| MapBean2 bean = new MapBean2(); |
| Map<Developer2, Color> data = new LinkedHashMap<Developer2, Color>(); |
| data.put(new Developer2("Andy", "tester"), Color.BLACK); |
| data.put(new SuperMan("Bill", "cleaner", false), Color.BLACK); |
| data.put(new Developer2("Lisa", "owner"), Color.RED); |
| bean.setData(data); |
| Map<Color, Developer2> developers = new LinkedHashMap<Color, Developer2>(); |
| developers.put(Color.WHITE, new Developer2("Fred", "creator")); |
| developers.put(Color.RED, new SuperMan("Jason", "contributor", true)); |
| developers.put(Color.BLACK, 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-13.yaml"); |
| assertEquals(etalon, output); |
| // load |
| Yaml beanLoader = new Yaml(); |
| MapBean2 parsed = beanLoader.loadAs(etalon, MapBean2.class); |
| assertNotNull(parsed); |
| Map<Developer2, Color> parsedData = parsed.getData(); |
| assertEquals(3, parsedData.size()); |
| assertTrue(parsedData.containsKey(new SuperMan("Bill", "cleaner", false))); |
| assertEquals(Color.BLACK, parsedData.get(new SuperMan("Bill", "cleaner", false))); |
| // |
| Map<Color, Developer2> parsedDevelopers = parsed.getDevelopers(); |
| assertEquals(3, parsedDevelopers.size()); |
| assertEquals(new SuperMan("Jason", "contributor", true), parsedDevelopers.get(Color.RED)); |
| } |
| |
| public void testLoadMap() { |
| String output = Util.getLocalResource("examples/map-bean-12.yaml"); |
| // System.out.println(output); |
| Yaml beanLoader = new Yaml(); |
| MapBean2 parsed = beanLoader.loadAs(output, MapBean2.class); |
| assertNotNull(parsed); |
| Map<Developer2, Color> data = parsed.getData(); |
| assertEquals(2, data.size()); |
| Iterator<Developer2> iter = data.keySet().iterator(); |
| Developer2 first = iter.next(); |
| assertEquals("Andy", first.getName()); |
| assertEquals("tester", first.getRole()); |
| assertEquals(Color.BLACK, data.get(first)); |
| Developer2 second = iter.next(); |
| assertEquals("Lisa", second.getName()); |
| assertEquals("owner", second.getRole()); |
| assertEquals(Color.RED, data.get(second)); |
| // |
| Map<Color, Developer2> developers = parsed.getDevelopers(); |
| assertEquals(2, developers.size()); |
| Iterator<Color> iter2 = developers.keySet().iterator(); |
| Color firstColor = iter2.next(); |
| assertEquals(Color.WHITE, firstColor); |
| Developer2 dev1 = developers.get(firstColor); |
| assertEquals("Fred", dev1.getName()); |
| assertEquals("creator", dev1.getRole()); |
| Color secondColor = iter2.next(); |
| assertEquals(Color.BLACK, secondColor); |
| Developer2 dev2 = developers.get(secondColor); |
| assertEquals("John", dev2.getName()); |
| assertEquals("committer", dev2.getRole()); |
| } |
| |
| public static enum Color { |
| WHITE, BLACK, RED; |
| } |
| |
| public static class MapBean2 { |
| private Map<Developer2, Color> data; |
| private String name; |
| private Map<Color, Developer2> developers; |
| |
| public MapBean2() { |
| name = "Bean123"; |
| } |
| |
| public String getName() { |
| return name; |
| } |
| |
| public void setName(String name) { |
| this.name = name; |
| } |
| |
| public Map<Color, Developer2> getDevelopers() { |
| return developers; |
| } |
| |
| public void setDevelopers(Map<Color, Developer2> developers) { |
| this.developers = developers; |
| } |
| |
| public Map<Developer2, Color> getData() { |
| return data; |
| } |
| |
| public void setData(Map<Developer2, Color> data) { |
| this.data = data; |
| } |
| |
| } |
| |
| public static class Developer2 implements Comparable<Developer2> { |
| private String name; |
| private String role; |
| |
| public Developer2() { |
| } |
| |
| private 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 int compareTo(Developer2 o) { |
| return name.compareTo(o.name); |
| } |
| |
| @Override |
| public boolean equals(Object obj) { |
| if (obj instanceof Developer2) { |
| return toString().equals(obj.toString()); |
| } else { |
| return false; |
| } |
| } |
| |
| @Override |
| public int hashCode() { |
| return toString().hashCode(); |
| } |
| |
| @Override |
| public String toString() { |
| return "Developer " + name + " " + role; |
| } |
| |
| } |
| |
| public static class SuperMan extends Developer2 { |
| private boolean smart; |
| |
| public SuperMan() { |
| super(); |
| } |
| |
| private SuperMan(String name, String role, boolean smart) { |
| super(name, role); |
| this.smart = smart; |
| } |
| |
| public boolean isSmart() { |
| return smart; |
| } |
| |
| public void setSmart(boolean smart) { |
| this.smart = smart; |
| } |
| |
| @Override |
| public String toString() { |
| return "Super" + super.toString(); |
| } |
| } |
| } |