/** | |
* Copyright (c) 2008-2010, http://code.google.com/p/snakeyaml/ | |
* | |
* 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.JavaBeanDumper; | |
import org.yaml.snakeyaml.JavaBeanLoader; | |
import org.yaml.snakeyaml.Util; | |
/** | |
* 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); | |
JavaBeanDumper dumper = new JavaBeanDumper(false); | |
String output = dumper.dump(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); | |
JavaBeanDumper dumper = new JavaBeanDumper(false); | |
String output = dumper.dump(bean); | |
// System.out.println(output); | |
String etalon = Util.getLocalResource("examples/map-bean-13.yaml"); | |
assertEquals(etalon, output); | |
// load | |
JavaBeanLoader<MapBean2> beanLoader = new JavaBeanLoader<MapBean2>(MapBean2.class); | |
MapBean2 parsed = beanLoader.load(etalon); | |
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); | |
JavaBeanLoader<MapBean2> beanLoader = new JavaBeanLoader<MapBean2>(MapBean2.class); | |
MapBean2 parsed = beanLoader.load(output); | |
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(); | |
} | |
} | |
} |