/** | |
* Copyright (c) 2008-2011, 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.JavaBeanDumper; | |
import org.yaml.snakeyaml.JavaBeanLoader; | |
import org.yaml.snakeyaml.Util; | |
/** | |
* 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); | |
JavaBeanDumper dumper = new JavaBeanDumper(false); | |
String output = dumper.dump(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); | |
JavaBeanDumper dumper = new JavaBeanDumper(false); | |
String output = dumper.dump(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); | |
JavaBeanLoader<MapBean> beanLoader = new JavaBeanLoader<MapBean>(MapBean.class); | |
MapBean parsed = beanLoader.load(output); | |
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); | |
JavaBeanLoader<MapBeanNoGenerics> beanLoader = new JavaBeanLoader<MapBeanNoGenerics>( | |
MapBeanNoGenerics.class); | |
MapBeanNoGenerics parsed = beanLoader.load(output); | |
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; | |
} | |
} | |
} |